fix packaging into a crate

Signed-off-by: Teo Koon Peng <teokoonpeng@gmail.com>

Author: koonpeng

.gitignore

@@ -8,7 +8,6 @@ node_modules
 
 # diagram-editor
 /diagram-editor/dist
-/diagram-editor/dist.tar.gz
 
 # storybook
 *storybook.log

diagram-editor/.gitignore

@@ -8,17 +8,16 @@
 mod frontend {
     use flate2::{write::GzEncoder, Compression};
     use std::fs::File;
-    use std::path::PathBuf;
+    use std::io::{BufReader, Read};
+    use std::path::{Path, PathBuf};
     use std::process::Command;
     use tar::Builder;
 
-    pub fn build_frontend() {
-        println!("cargo:rerun-if-changed=build.rs");
+    fn build_pnpm<P: AsRef<Path>>(out_dir: P) {
         println!("cargo:rerun-if-changed=package.json");
         println!("cargo:rerun-if-changed=../pnpm-lock.yaml");
         println!("cargo:rerun-if-changed=rsbuild.config.ts");
         println!("cargo:rerun-if-changed=frontend");
-        println!("cargo:rerun-if-changed=dist.tar.gz");
 
         let status = Command::new("pnpm")
             .arg("install")
@@ -40,17 +39,14 @@ mod frontend {
         }
 
         let dist_dir_path = "dist";
-        // We put the output in `CARGO_MANIFEST_DIR` instead of `OUT_DIR` because we want to include
-        // it in the crate.
-        let out_dir =
-            PathBuf::from(std::env::var("CARGO_MANIFEST_DIR").expect("CARGO_MANIFEST_DIR not set"));
-        let output_tar_gz_path = out_dir.join("dist.tar.gz");
+        let output_tar_gz_path = out_dir.as_ref().join("dist.tar.gz");
 
-        if std::path::Path::new(dist_dir_path).exists() {
-            let tar_gz_file = File::create(&output_tar_gz_path)
-                .expect("Failed to create output tar.gz file in CARGO_MANIFEST_DIR");
+        if Path::new(dist_dir_path).exists() {
+            let tar_gz_file =
+                File::create(&output_tar_gz_path).expect("Failed to create output tar.gz file");
             let enc = GzEncoder::new(tar_gz_file, Compression::default());
             let mut tar_builder = Builder::new(enc);
+            tar_builder.mode(tar::HeaderMode::Deterministic);
 
             // Add the entire "dist" directory to the archive, preserving its name within the archive.
             tar_builder
@@ -68,9 +64,58 @@ mod frontend {
             );
         }
     }
+
+    pub fn build_frontend() {
+        // We put the output in `CARGO_MANIFEST_DIR` instead of `OUT_DIR` because we want to include
+        // it in the crate.
+        build_pnpm(PathBuf::from(
+            std::env::var("CARGO_MANIFEST_DIR").expect("CARGO_MANIFEST_DIR not set"),
+        ));
+    }
+
+    pub fn verify_frontend() {
+        let manifest_dir = std::env::var("CARGO_MANIFEST_DIR").expect("CARGO_MANIFEST_DIR not set");
+        let current_dist_path = Path::new(&manifest_dir).join("dist.tar.gz");
+        let current_dist = File::open(&current_dist_path).expect(&format!(
+            "Failed to open {}",
+            current_dist_path.to_str().unwrap()
+        ));
+        let current_dist = BufReader::new(current_dist);
+
+        let out_dir = std::env::var("OUT_DIR").expect("OUT_DIR not set");
+        build_pnpm(Path::new(&out_dir));
+
+        let new_dist_path = Path::new(&out_dir).join("dist.tar.gz");
+        let new_dist = File::open(&new_dist_path).expect(&format!(
+            "Failed to open {}",
+            new_dist_path.to_str().unwrap()
+        ));
+        let new_dist = BufReader::new(new_dist);
+
+        let mut current_bytes = current_dist.bytes();
+        let mut new_bytes = new_dist.bytes();
+        loop {
+            let a = current_bytes.next();
+            let b = new_bytes.next();
+            if a.is_none() && b.is_none() {
+                break;
+            }
+            if a.is_none() || b.is_none() || a.unwrap().unwrap() != b.unwrap().unwrap() {
+                println!("cargo::error=Frontend has changed, run `BUILD_FRONTEND=1 cargo build` to update it");
+                return;
+            }
+        }
+    }
 }
 
 fn main() {
     #[cfg(feature = "frontend")]
-    frontend::build_frontend();
+    {
+        println!("cargo::rerun-if-env-changed=BUILD_FRONTEND");
+        if std::env::var("BUILD_FRONTEND").is_ok() {
+            frontend::build_frontend();
+        } else {
+            frontend::verify_frontend();
+        }
+    }
 }

diagram-editor/build.rs

@@ -13,7 +13,7 @@
     "check": "biome check --write",
     "check:ts": "tsc --noEmit",
     "dev": "rsbuild dev --open --environment rest-backend",
-    "dev:backend": "cargo run --manifest-path ../examples/diagram/calculator/Cargo.toml -- serve --port 3001",
+    "dev:backend": "BUILD_FRONTEND=1 cargo run --manifest-path ../examples/diagram/calculator/Cargo.toml -- serve --port 3001",
     "dev:wasm": "scripts/dev-wasm.sh",
     "format": "biome format --write",
     "preview": "rsbuild preview",