model list impl

src-tauri/src/main.rs

@@ -3,7 +3,6 @@
 use mouse_position::mouse_position::Mouse;
 use std::path::PathBuf;
 use tauri::Manager;
-use window_shadows::set_shadow;
 
 #[derive(serde::Serialize, Clone)]
 struct MousePosition {
@@ -18,6 +17,10 @@ struct CursorGrab {
 struct ModelViewParam {
     vrm: String,
 }
+#[derive(serde::Serialize, Clone)]
+struct ModelList {
+    models: Vec<String>,
+}
 
 fn main() {
     //simple_logger::init_with_env().unwrap();
@@ -75,15 +78,40 @@ fn main() {
 
             let app_handle = app.app_handle();
             tauri::async_runtime::spawn(async move {
+                use axum::{
+                    http::StatusCode,
+                    routing::{get, post},
+                    Json,
+                };
                 use tower_http::cors::{AllowMethods, Any, CorsLayer};
 
                 let data_dir = app_handle.path_resolver().app_data_dir().unwrap();
                 let data_dir2 = data_dir.clone();
+                let data_dir3 = data_dir.clone();
 
                 let app = axum::Router::new()
+                    .route(
+                        "/vrm",
+                        get(|| async move {
+                            let vrm_dir = data_dir3.join("vrm");
+                            if let Ok(mut files) = tokio::fs::read_dir(&vrm_dir).await {
+                                let mut models = Vec::new();
+                                while let Ok(Some(file)) = files.next_entry().await {
+                                    let filename = file.file_name().to_string_lossy().to_string();
+                                    if filename.ends_with(".vrm") {
+                                        models.push(filename);
+                                    }
+                                }
+                                (StatusCode::OK, Json(ModelList { models }))
+                            } else {
+                                let _ = tokio::fs::create_dir_all(&vrm_dir).await;
+                                (StatusCode::OK, Json(ModelList { models: Vec::new() }))
+                            }
+                        }),
+                    )
                     .route(
                         "/vrm/{vrm}",
-                        axum::routing::get(|p: axum::extract::Path<String>| async move {
+                        get(|p: axum::extract::Path<String>| async move {
                             if let Ok(data) = tokio::fs::read(
                                 data_dir
                                     .join("vrm")
@@ -99,7 +127,7 @@ fn main() {
                     )
                     .route(
                         "/vrm/{vrm}",
-                        axum::routing::post(
+                        post(
                             |p: axum::extract::Path<String>, body: axum::body::Bytes| async move {
                                 if let Ok(_) = tokio::fs::write(
                                     data_dir2

src/components/VrmLibrary.tsx

@@ -1,7 +1,5 @@
 import { useState, useEffect, ChangeEventHandler } from "react";
-import { readDir, createDir, exists } from "@tauri-apps/api/fs";
-import { vrmDir } from "@/util/path";
-import { writeModel } from "@/util/fetch";
+import { listModel, writeModel } from "@/util/fetch";
 
 type Props = {
   onSelect: (path: string) => any;
@@ -11,14 +9,8 @@ export default ({ onSelect }: Props) => {
   const [models, setModels] = useState<string[]>([]);
   const [processing, setProcessing] = useState(false);
   const readModels = async () => {
-    if (!(await exists(vrmDir))) {
-      await createDir(vrmDir, { recursive: true });
-    }
-    const read = await readDir(vrmDir);
-    const vrmFiles = read
-      .filter((p) => p.path.endsWith(".vrm") && typeof p.name === "string")
-      .map((p) => p.name!);
-    setModels(vrmFiles);
+    const models = await listModel();
+    setModels(models);
   };
   useEffect(() => {
     readModels();

src/util/fetch.ts

@@ -1,3 +1,12 @@
+export const listModel = async (): Promise<string[]> => {
+  const resp = await fetch(new URL(`http://localhost:8108/vrm`));
+  if (resp.ok) {
+    return (await resp.json()).models;
+  } else {
+    return [];
+  }
+};
+
 export const readModel = async (name: string): Promise<Uint8Array | null> => {
   const resp = await fetch(new URL(`http://localhost:8108/vrm/${name}`));
   if (resp.ok) {