3 files changed, 76 insertions, 15 deletions

diff --git a/server/src/ui/admin/import.rs b/server/src/ui/admin/import.rs
new file mode 100644
index 0000000..b6bb858
--- /dev/null
+++ b/server/src/ui/admin/import.rs
@@ -0,0 +1,70 @@
+/*
+    This file is part of jellything (https://codeberg.org/metamuffin/jellything)
+    which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
+    Copyright (C) 2025 metamuffin <metamuffin.org>
+*/
+
+use std::time::Duration;
+
+use crate::{
+    helper::{A, RequestInfo},
+    ui::error::MyResult,
+};
+use jellycommon::routes::u_admin_import;
+use jellyimport::{
+    is_importing,
+    reporting::{IMPORT_ERRORS, IMPORT_PROGRESS},
+};
+use jellylogic::{admin::do_import, session::Session};
+use jellyui::{admin::import::AdminImportPage, locale::tr, render_page};
+use rocket::{
+    get, post,
+    request::FlashMessage,
+    response::{Flash, Redirect, content::RawHtml},
+};
+use rocket_ws::{Message, Stream, WebSocket};
+use tokio::time::sleep;
+
+#[get("/admin/import", rank = 2)]
+pub async fn r_admin_import(
+    ri: RequestInfo,
+    flash: Option<FlashMessage<'_>>,
+) -> MyResult<RawHtml<String>> {
+    ri.session.assert_admin()?;
+
+    let last_import_err = IMPORT_ERRORS.read().await.clone();
+    Ok(RawHtml(render_page(
+        &AdminImportPage {
+            busy: is_importing(),
+            flash: &flash.map(FlashMessage::into_inner),
+            lang: &ri.lang,
+            last_import_err: &last_import_err,
+        },
+        ri.render_info(),
+    )))
+}
+
+#[post("/admin/import?<incremental>")]
+pub async fn r_admin_import_post(ri: RequestInfo, incremental: bool) -> MyResult<Flash<Redirect>> {
+    ri.session.assert_admin()?;
+    do_import(&ri.session, incremental).await?;
+    Ok(Flash::success(
+        Redirect::to(u_admin_import()),
+        tr(ri.lang, "admin.import_success"),
+    ))
+}
+
+#[get("/admin/import", rank = 1)]
+pub fn r_admin_import_stream(_session: A<Session>, ws: WebSocket) -> Stream!['static] {
+    Stream! { ws =>
+        loop {
+            let Some(p) = IMPORT_PROGRESS.read().await.clone() else {
+                break;
+            };
+            yield Message::Text(serde_json::to_string(&p).unwrap());
+            sleep(Duration::from_secs_f32(0.05)).await;
+        }
+        yield Message::Text("done".to_string());
+        let _ = ws;
+    }
+}
diff --git a/server/src/ui/admin/log.rs b/server/src/ui/admin/log.rs
index 24671bb..0a0e5ca 100644
--- a/server/src/ui/admin/log.rs
+++ b/server/src/ui/admin/log.rs
@@ -4,7 +4,7 @@
     Copyright (C) 2025 metamuffin <metamuffin.org>
 */
 use crate::{
-    helper::{RequestInfo, A},
+    helper::{A, RequestInfo},
     ui::error::MyResult,
 };
 use jellylogic::{
@@ -12,7 +12,7 @@ use jellylogic::{
     session::Session,
 };
 use jellyui::{
-    admin::log::{render_log_line, ServerLogPage},
+    admin::log::{ServerLogPage, render_log_line},
     render_page,
 };
 use rocket::{get, response::content::RawHtml};
diff --git a/server/src/ui/admin/mod.rs b/server/src/ui/admin/mod.rs
index 02a7605..3bd4771 100644
--- a/server/src/ui/admin/mod.rs
+++ b/server/src/ui/admin/mod.rs
@@ -3,6 +3,7 @@
     which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
     Copyright (C) 2025 metamuffin <metamuffin.org>
 */
+pub mod import;
 pub mod log;
 pub mod user;
 
@@ -11,15 +12,15 @@ use crate::helper::RequestInfo;
 use jellycommon::routes::u_admin_dashboard;
 use jellyimport::is_importing;
 use jellylogic::admin::{
-    create_invite, delete_invite, do_import, get_import_errors, list_invites, update_search_index,
+    create_invite, delete_invite, get_import_errors, list_invites, update_search_index,
 };
 use jellyui::{admin::AdminDashboardPage, locale::tr, render_page};
 use rocket::{
+    FromForm,
     form::Form,
     get, post,
     request::FlashMessage,
-    response::{content::RawHtml, Flash, Redirect},
-    FromForm,
+    response::{Flash, Redirect, content::RawHtml},
 };
 
 #[get("/admin/dashboard")]
@@ -76,16 +77,6 @@ pub async fn r_admin_remove_invite(
     ))
 }
 
-#[post("/admin/import?<incremental>")]
-pub async fn r_admin_import(ri: RequestInfo, incremental: bool) -> MyResult<Flash<Redirect>> {
-    ri.session.assert_admin()?;
-    do_import(&ri.session, incremental).await?.1?;
-    Ok(Flash::success(
-        Redirect::to(u_admin_dashboard()),
-        tr(ri.lang, "admin.import_success"),
-    ))
-}
-
 #[post("/admin/update_search")]
 pub async fn r_admin_update_search(ri: RequestInfo) -> MyResult<Flash<Redirect>> {
     ri.session.assert_admin()?;