From 010198a44f743530c39330c2d788f0c9cef474f7 Mon Sep 17 00:00:00 2001
From: Fankai Liu <liufankai137@outlook.com>
Date: Wed, 15 Nov 2023 23:02:38 +0800
Subject: [PATCH] add static routers

---
 {assets => src/template/assets}/favicon.ico   | Bin
 .../template/assets}/js/alpinejs.js           |   0
 .../template/assets}/js/sweetalert2.js        |   0
 .../template/assets}/js/tailwindcss.js        |   0
 src/template/src/routers/mod.hbs              |   5 +++
 src/template/src/routers/static_routers.hbs   |  23 ++++++++++++
 src/template/templates/404.hbs                |   4 +--
 src/template/templates/login.hbs              |   6 ++--
 src/template/templates/user_list_page.hbs     |   6 ++--
 src/utils/create_project.rs                   |  34 +++++++++++++++++-
 10 files changed, 69 insertions(+), 9 deletions(-)
 rename {assets => src/template/assets}/favicon.ico (100%)
 rename {assets => src/template/assets}/js/alpinejs.js (100%)
 rename {assets => src/template/assets}/js/sweetalert2.js (100%)
 rename {assets => src/template/assets}/js/tailwindcss.js (100%)
 create mode 100644 src/template/src/routers/static_routers.hbs

diff --git a/assets/favicon.ico b/src/template/assets/favicon.ico
similarity index 100%
rename from assets/favicon.ico
rename to src/template/assets/favicon.ico
diff --git a/assets/js/alpinejs.js b/src/template/assets/js/alpinejs.js
similarity index 100%
rename from assets/js/alpinejs.js
rename to src/template/assets/js/alpinejs.js
diff --git a/assets/js/sweetalert2.js b/src/template/assets/js/sweetalert2.js
similarity index 100%
rename from assets/js/sweetalert2.js
rename to src/template/assets/js/sweetalert2.js
diff --git a/assets/js/tailwindcss.js b/src/template/assets/js/tailwindcss.js
similarity index 100%
rename from assets/js/tailwindcss.js
rename to src/template/assets/js/tailwindcss.js
diff --git a/src/template/src/routers/mod.hbs b/src/template/src/routers/mod.hbs
index 39b1a1c..137ef3b 100644
--- a/src/template/src/routers/mod.hbs
+++ b/src/template/src/routers/mod.hbs
@@ -15,6 +15,7 @@ use self::{
 };
 pub mod demo;
 pub mod user;
+mod static_routers;
 
 pub fn router() -> Router {
     let mut no_auth_routers = vec![
@@ -33,6 +34,8 @@ pub fn router() -> Router {
                 .delete(delete_user),
         ),
     ];
+    let static_routers = static_routers::create_static_routers();
+    no_auth_routers.extend(static_routers);
     let router = Router::new()
         .hoop(Logger::new())
         .hoop(CatchPanic::new())
@@ -78,6 +81,8 @@ pub fn router() -> Router {
                 .delete(delete_user),
         ),
     ];
+    let static_routers = static_routers::create_static_routers();
+    no_auth_routers.extend(static_routers);
     let router = Router::new()
         .hoop(Logger::new())
         .hoop(CatchPanic::new())
diff --git a/src/template/src/routers/static_routers.hbs b/src/template/src/routers/static_routers.hbs
new file mode 100644
index 0000000..8800d53
--- /dev/null
+++ b/src/template/src/routers/static_routers.hbs
@@ -0,0 +1,23 @@
+use rust_embed::RustEmbed;
+use salvo::{Router, serve_static::static_embed, endpoint, Response, http::ResBody, hyper::body::Bytes};
+
+#[derive(RustEmbed)]
+#[folder = "assets"]
+struct Assets;
+
+pub fn create_static_routers() -> Vec<Router> {
+    {{#if need_db_conn}}
+    let static_router = Router::with_path("assets/<**path>").get(static_embed::<Assets>());
+    let icon_router = Router::with_path("favicon.ico").get(get_icon);
+    vec![static_router,icon_router]
+    {{else}}
+    let icon_router = Router::with_path("favicon.ico").get(get_icon);
+    vec![icon_router]
+    {{/if}}
+}
+
+#[endpoint(tags("comm"))]
+pub async fn get_icon(res: &mut Response) {
+    let icon = Assets::get("favicon.ico").unwrap();
+    res.body(ResBody::Once(Bytes::from(icon.data.to_vec())));
+}
diff --git a/src/template/templates/404.hbs b/src/template/templates/404.hbs
index dd97fe6..0164311 100644
--- a/src/template/templates/404.hbs
+++ b/src/template/templates/404.hbs
@@ -4,8 +4,8 @@
     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>404 {{page_not_found}}</title>
-    <script src="https://cdn.tailwindcss.com"></script>
-</head>
+  <script src="assets/js/tailwindcss.js"></script>
+  </head>
   <body
     class="grid min-h-full place-items-center bg-white px-6 py-24 sm:py-32 lg:px-8"
   >
diff --git a/src/template/templates/login.hbs b/src/template/templates/login.hbs
index cfe249d..bc124fa 100644
--- a/src/template/templates/login.hbs
+++ b/src/template/templates/login.hbs
@@ -80,9 +80,9 @@
       </template>
     </div>
   </body>
-  <script src="https://cdn.tailwindcss.com"></script>
-  <script src="https://cdn.jsdelivr.net/npm/sweetalert2@11"></script>
-  <script src="//unpkg.com/alpinejs" defer></script>
+  <script src="assets/js/tailwindcss.js"></script>
+  <script src="assets/js/sweetalert2.js"></script>
+  <script src="assets/js/alpinejs.js" defer></script>
   <script>
     function loginForm() {
       return {
diff --git a/src/template/templates/user_list_page.hbs b/src/template/templates/user_list_page.hbs
index df98780..328ab53 100644
--- a/src/template/templates/user_list_page.hbs
+++ b/src/template/templates/user_list_page.hbs
@@ -7,9 +7,9 @@
     <title>salvo</title>
   </head>
   {% include "user_list.html" %}
-  <script src="https://cdn.tailwindcss.com"></script>
-  <script src="https://cdn.jsdelivr.net/npm/sweetalert2@11"></script>
-  <script src="//unpkg.com/alpinejs" defer></script>
+  <script src="assets/js/tailwindcss.js"></script>
+  <script src="assets/js/sweetalert2.js"></script>
+  <script src="assets/js/alpinejs.js" defer></script>
   <script>
     function userForm() {
       return {
diff --git a/src/utils/create_project.rs b/src/utils/create_project.rs
index 743ac3b..887107b 100644
--- a/src/utils/create_project.rs
+++ b/src/utils/create_project.rs
@@ -169,6 +169,7 @@ fn write_project_file(
     let mut dependencies = data["dependencies"].clone();
     handle_dependencies(
         &mut dependencies,
+        is_web_site,
         need_db_conn,
         user_selected.db_type,
         user_selected.db_conn_type,
@@ -193,8 +194,28 @@ fn write_project_file(
             handlebars.render_template(handle_404_template, &data)?;
         let mut handle_404_file = File::create(template_path.join("handle_404.html"))?;
         handle_404_file.write_all(handle_404_template_rendered.as_bytes())?;
-
+        //assets
+        let assets_path = project_path.join("assets");
+        //assets/favicon.ico
+        let favicon_bytes = include_bytes!("../template/assets/favicon.ico");
+        let mut favicon_file = File::create(assets_path.join("favicon.ico"))?;
+        favicon_file.write_all(favicon_bytes)?;
         if need_db_conn {
+            //assets/js
+            let js_path = assets_path.join("js");
+            std::fs::create_dir_all(&js_path)?;
+            //assets/js/alpinejs.js
+            let alpinejs_bytes = include_bytes!("../template/assets/js/alpinejs.js");
+            let mut alpinejs_file = File::create(js_path.join("alpinejs.js"))?;
+            alpinejs_file.write_all(alpinejs_bytes)?;
+            //assets/js/sweetalert2.js
+            let sweetalert2_bytes = include_bytes!("../template/assets/js/sweetalert2.js");
+            let mut sweetalert2_file = File::create(js_path.join("sweetalert2.js"))?;
+            sweetalert2_file.write_all(sweetalert2_bytes)?;
+            //assets/js/tailwindcss.js
+            let tailwindcss_bytes = include_bytes!("../template/assets/js/tailwindcss.js");
+            let mut tailwindcss_file = File::create(js_path.join("tailwindcss.js"))?;
+            tailwindcss_file.write_all(tailwindcss_bytes)?;
             //template/login.html
             let login_template = include_str!("../template/templates/login.hbs");
             let login_template_rendered = handlebars.render_template(login_template, &data)?;
@@ -627,10 +648,21 @@ fn create_basic_file(
 
 fn handle_dependencies(
     dependencies: &mut serde_json::Value,
+    is_website: bool,
     need_db_conn: bool,
     db_type: DbType,
     conn_type: DbConnectionType,
 ) {
+    if is_website {
+        dependencies["rust-embed"] = json!({
+            "version": "8.0.0",
+        });
+        if let Some(salvo) = dependencies["salvo"].as_object_mut() {
+            if let Some(features) = salvo["features"].as_array_mut() {
+                features.push(serde_json::json!("serve-static"));
+            }
+        }
+    }
     if need_db_conn {
         match (conn_type, db_type) {
             (DbConnectionType::Sqlx, DbType::Mysql) => {