fix: change vue to MPA mode and primarily add upload logic in web page

Author: debugdoctor

server/src/db.rs

@@ -1,4 +1,4 @@
-use std::{sync::Arc, u32};
+use std::{sync::Arc};
 use crate::utils::memdb::MemDB;
 
 use axum::body::Bytes;
@@ -27,7 +27,6 @@ impl AccessCode {
 #[derive(Clone)]
 pub struct FileBlock {
     pub data: Bytes,
-    pub size: u64,
     pub is_final: bool,
     pub filename: String,
     pub start: u64,
@@ -40,10 +39,9 @@ impl FileBlock {
         FILE_BLOCK_DB.clone()
     }
 
-    pub fn new(data: &Bytes, size: u64, is_final: bool, filename: String, start: u64, end: u64, total: u64) -> Self {
+    pub fn new(data: &Bytes, is_final: bool, filename: String, start: u64, end: u64, total: u64) -> Self {
         FileBlock {
             data: data.clone(),
-            size,
             is_final,
             filename,
             start,

server/src/router.rs

@@ -2,19 +2,21 @@ use axum::{routing::{get, post}, serve, Router};
 use tokio::net::{TcpListener};
 use tracing::{event, instrument};
 
-use crate::service::handler::{get_file, get_id, get_status, root, upload_file};
+use crate::service::handler::{download, get_file, get_id, get_status, upload, upload_file};
 
 fn api_router() -> Router {
     Router::new()
+        .route("/hello", get(|| async { "Hi!" }))
         .route("/get_id", get(get_id))
         .route("/{id}/status", get(get_status))
         .route("/{id}/upload", post(upload_file))
+        .route("/{id}/file", get(get_file))
 }
 
 fn view_router() -> Router {
     Router::new()
-        .route("/", get(root))
-        .route("/{id}/file", get(get_file))
+        .route("/", get(upload))
+        .route("/{id}/file", get(download))
 }
 
 #[instrument(skip_all)]

server/src/service/handler.rs

@@ -29,8 +29,18 @@ struct FileInfo {
 }
 
 #[instrument]
-pub async fn root() -> impl IntoResponse {
-    match StaticFiles::get("index.html") {
+pub async fn upload() -> impl IntoResponse {
+    match StaticFiles::get("./src/pages/upload/index.html") {
+        Some(content) => {
+            let html = String::from_utf8(content.data.to_vec()).unwrap();
+            Html(html).into_response()
+        }
+        None => (StatusCode::NOT_FOUND, "Page not found").into_response(),
+    }
+}
+
+pub async fn download() -> impl IntoResponse {
+    match StaticFiles::get("./src/pages/download/index.html") {
         Some(content) => {
             let html = String::from_utf8(content.data.to_vec()).unwrap();
             Html(html).into_response()
@@ -144,8 +154,8 @@ pub async fn get_file(
 
     // Get file size and filename
     let file_block_db = FileBlock::get_db();
-    let mut total_size = 0u64;
-    let mut filename = String::new();
+    let total_size: u64 ;
+    let filename: String;
 
     // Calculate file size and get filename
     let mut retry = 0;
@@ -470,7 +480,6 @@ pub async fn upload_file(Path(id): Path<String>, multipart: Multipart) -> impl I
         }
         let file_block = FileBlock::new(
             &data,
-            data.len() as u64,
             is_final > 0,
             filename,
             start,

web/components.d.ts

@@ -0,0 +1,15 @@
+/* eslint-disable */
+// @ts-nocheck
+// Generated by unplugin-vue-components
+// Read more: https://github.com/vuejs/core/pull/3399
+// biome-ignore lint: disable
+export {}
+
+/* prettier-ignore */
+declare module 'vue' {
+  export interface GlobalComponents {
+    AButton: typeof import('ant-design-vue/es')['Button']
+    AProgress: typeof import('ant-design-vue/es')['Progress']
+    AUpload: typeof import('ant-design-vue/es')['Upload']
+  }
+}

web/package.json

@@ -16,6 +16,9 @@
     "format": "prettier --write src/"
   },
   "dependencies": {
+    "@ant-design/icons-vue": "^7.0.1",
+    "ant-design-vue": "~4.2.6",
+    "lucide-vue-next": "^0.544.0",
     "vue": "^3.5.18"
   },
   "devDependencies": {
@@ -31,6 +34,7 @@
     "npm-run-all2": "^8.0.4",
     "prettier": "3.6.2",
     "typescript": "~5.8.0",
+    "unplugin-vue-components": "^29.0.0",
     "vite": "^7.0.6",
     "vite-plugin-vue-devtools": "^8.0.0",
     "vue-tsc": "^3.0.4"