From e42f0db4cc81549e25f9e663e307606eff75ce53 Mon Sep 17 00:00:00 2001
From: GeckoEidechse <gecko.eidechse+git@pm.me>
Date: Sun, 23 Jul 2023 17:23:51 +0200
Subject: feat: Add button to DevView to calculate checksums of files in game
 install location

---
 src-tauri/Cargo.lock                | 44 +++++++++++++++++++--
 src-tauri/Cargo.toml                |  4 ++
 src-tauri/src/development/mod.rs    | 78 +++++++++++++++++++++++++++++++++++++
 src-tauri/src/main.rs               |  1 +
 src-vue/src/views/DeveloperView.vue | 59 ++++++++++++++++++++++++++++
 5 files changed, 183 insertions(+), 3 deletions(-)

diff --git a/src-tauri/Cargo.lock b/src-tauri/Cargo.lock
index e8ac30cc..dca74714 100644
--- a/src-tauri/Cargo.lock
+++ b/src-tauri/Cargo.lock
@@ -87,6 +87,7 @@ dependencies = [
  "async-recursion",
  "chrono",
  "const_format",
+ "crypto-hash",
  "dirs",
  "glob",
  "json5",
@@ -108,6 +109,7 @@ dependencies = [
  "tauri-plugin-store",
  "tokio",
  "ts-rs",
+ "walkdir",
  "winapi",
  "winreg 0.11.0",
  "zip",
@@ -595,6 +597,24 @@ dependencies = [
  "memchr",
 ]
 
+[[package]]
+name = "commoncrypto"
+version = "0.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d056a8586ba25a1e4d61cb090900e495952c7886786fc55f909ab2f819b69007"
+dependencies = [
+ "commoncrypto-sys",
+]
+
+[[package]]
+name = "commoncrypto-sys"
+version = "0.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1fed34f46747aa73dfaa578069fd8279d2818ade2b55f38f22a9401c7f4083e2"
+dependencies = [
+ "libc",
+]
+
 [[package]]
 name = "concurrent-queue"
 version = "2.2.0"
@@ -747,6 +767,18 @@ dependencies = [
  "typenum",
 ]
 
+[[package]]
+name = "crypto-hash"
+version = "0.3.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8a77162240fd97248d19a564a565eb563a3f592b386e4136fb300909e67dddca"
+dependencies = [
+ "commoncrypto",
+ "hex 0.3.2",
+ "openssl",
+ "winapi",
+]
+
 [[package]]
 name = "cssparser"
 version = "0.27.2"
@@ -1596,6 +1628,12 @@ version = "0.3.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "443144c8cdadd93ebf52ddb4056d257f5b52c04d3c804e657d19eb73fc33668b"
 
+[[package]]
+name = "hex"
+version = "0.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "805026a5d0141ffc30abb3be3173848ad46a1b1664fe632428479619a3644d77"
+
 [[package]]
 name = "hex"
 version = "0.4.3"
@@ -3465,7 +3503,7 @@ checksum = "360ee3270f7a4a1eee6c667f7d38360b995431598a73b740dfe420da548d9cc9"
 dependencies = [
  "debugid",
  "getrandom 0.2.10",
- "hex",
+ "hex 0.4.3",
  "serde",
  "serde_json",
  "thiserror",
@@ -3545,7 +3583,7 @@ checksum = "21e47d95bc83ed33b2ecf84f4187ad1ab9685d18ff28db000c99deac8ce180e3"
 dependencies = [
  "base64 0.21.2",
  "chrono",
- "hex",
+ "hex 0.4.3",
  "indexmap 1.9.3",
  "serde",
  "serde_json",
@@ -5331,7 +5369,7 @@ dependencies = [
  "futures-core",
  "futures-sink",
  "futures-util",
- "hex",
+ "hex 0.4.3",
  "nix",
  "once_cell",
  "ordered-stream",
diff --git a/src-tauri/Cargo.toml b/src-tauri/Cargo.toml
index d2f1e87e..50e75987 100644
--- a/src-tauri/Cargo.toml
+++ b/src-tauri/Cargo.toml
@@ -65,6 +65,10 @@ semver = "1.0"
 # simplified filesystem access
 glob = "0.3.1"
 dirs = "5"
+# Walk directories
+walkdir = "2.3"
+# Crypto stuff like checksum calculation
+crypto-hash = "0.3"
 
 [target.'cfg(windows)'.dependencies]
 # Windows API stuff
diff --git a/src-tauri/src/development/mod.rs b/src-tauri/src/development/mod.rs
index be02966d..cb2aeec9 100644
--- a/src-tauri/src/development/mod.rs
+++ b/src-tauri/src/development/mod.rs
@@ -3,6 +3,9 @@ use crate::github::{
     pull_requests::{check_github_api, download_zip_into_memory, get_launcher_download_link},
     CommitInfo,
 };
+use crate::GameInstall;
+use serde::{Deserialize, Serialize};
+use std::io::Read;
 
 #[tauri::command]
 pub async fn install_git_main(game_install_path: &str) -> Result<String, String> {
@@ -82,3 +85,78 @@ pub async fn install_git_main(game_install_path: &str) -> Result<String, String>
     );
     Ok(latest_commit_sha)
 }
+
+#[derive(Serialize, Deserialize, Debug)]
+pub struct Checksum {
+    path: String,
+    checksum: String,
+}
+
+impl ToString for Checksum {
+    fn to_string(&self) -> String {
+        format!("{} {}", self.checksum, self.path)
+    }
+}
+
+/// Computes the checksum of a given file
+fn compute_checksum<P: AsRef<std::path::Path>>(
+    path: P,
+) -> Result<Checksum, Box<dyn std::error::Error>> {
+    dbg!(path.as_ref().to_string_lossy().into_owned());
+    let mut file = std::fs::File::open(&path)?;
+    let mut buffer = Vec::new();
+    file.read_to_end(&mut buffer)?;
+
+    let checksum = crypto_hash::hex_digest(crypto_hash::Algorithm::SHA256, &buffer);
+    Ok(Checksum {
+        path: path.as_ref().to_string_lossy().into_owned(),
+        checksum,
+    })
+}
+
+fn convert_to_string(checksums: Vec<Checksum>) -> String {
+    let mut result = String::new();
+
+    for entry in checksums {
+        result.push_str(&entry.to_string());
+        result.push('\n');
+    }
+    result
+}
+
+#[tauri::command]
+/// Calculates checksums over the passed game_install folder and returns results
+pub async fn calculate_checksums_gameinstall(game_install: GameInstall) -> Result<String, String> {
+    log::info!("Computing checksums");
+
+    let path = game_install.game_path;
+    let mut checksums = Vec::new();
+
+    // Iterate over folder
+    for entry in walkdir::WalkDir::new(path.clone()) {
+        let entry = entry.unwrap();
+        if !entry.file_type().is_file() {
+            continue;
+        }
+
+        match compute_checksum(entry.path()) {
+            Ok(mut checksum) => {
+                checksum.path = checksum
+                    .path
+                    .strip_prefix(&path.clone())
+                    .unwrap()
+                    .to_string();
+                checksums.push(checksum)
+            }
+            Err(err) => log::warn!("Failed to compute checksum for {:?}: {:?}", entry, err),
+        }
+    }
+
+    for checksum in &checksums {
+        println!("{:?}", checksum);
+    }
+
+    log::info!("Done calculating");
+    let s = convert_to_string(checksums);
+    Ok(s)
+}
diff --git a/src-tauri/src/main.rs b/src-tauri/src/main.rs
index 1067f5d3..db0fd3bd 100644
--- a/src-tauri/src/main.rs
+++ b/src-tauri/src/main.rs
@@ -156,6 +156,7 @@ fn main() {
             github::pull_requests::get_launcher_download_link,
             close_application,
             development::install_git_main,
+            development::calculate_checksums_gameinstall,
             get_available_northstar_versions,
         ])
         .run(tauri::generate_context!())
diff --git a/src-vue/src/views/DeveloperView.vue b/src-vue/src/views/DeveloperView.vue
index 28ab3892..ce91af8c 100644
--- a/src-vue/src/views/DeveloperView.vue
+++ b/src-vue/src/views/DeveloperView.vue
@@ -50,6 +50,24 @@
             <br />
             <br />
 
+            <el-button type="primary" @click="computeChecksumsGameInstallFolder">
+                Compute checksums
+            </el-button>
+            <el-button type="primary" @click="copyToClipboard">
+                Copy to clipboard
+            </el-button>
+
+
+            <el-input
+                v-model="checksum_results"
+                type="textarea"
+                :rows="1"
+                placeholder="Output"
+            />
+
+            <br />
+            <br />
+
             <el-button type="primary" @click="getAvailableNorthstarVersions">
                 Get available versions
             </el-button>
@@ -130,6 +148,7 @@
 </template>
 
 <script lang="ts">
+import { Checksum } from "../../../src-tauri/bindings/Checksum";
 import { defineComponent } from "vue";
 import { invoke } from "@tauri-apps/api";
 import { GameInstall } from "../utils/GameInstall";
@@ -148,6 +167,7 @@ export default defineComponent({
         return {
             mod_to_install_field_string: "",
             release_notes_text: "",
+            checksum_results: "",
             first_tag: { label: '', value: { name: '' } },
             second_tag: { label: '', value: { name: '' } },
             ns_release_tags: [] as TagWrapper[],
@@ -336,6 +356,45 @@ export default defineComponent({
                 .then((message) => { showNotification(`NSProton Version`, message as string); })
                 .catch((error) => { showNotification(`Error`, error, "error"); })
         },
+        async computeChecksumsGameInstallFolder() {
+            let game_install = {
+                game_path: this.$store.state.game_path,
+                install_type: this.$store.state.install_type
+            } as GameInstall;
+
+            // Send notification telling the user to wait for the process to finish
+            const notification = showNotification(
+                "Calculating checksums",
+                "Please wait",
+                'info',
+                0
+            );
+
+            let checksum_calc_result = invoke<string>("calculate_checksums_gameinstall", { gameInstall: game_install });
+            await checksum_calc_result
+                .then((message) => {
+                    // Send notification
+                    this.checksum_results = message;
+                    showNotification("Done calculating checksums");
+                })
+                .catch((error) => {
+                    showErrorNotification(error);
+                    console.error(error);
+                })
+                .finally(() => {
+                    // Clear old notification
+                    notification.close();
+                });
+        },
+        async copyToClipboard() {
+            navigator.clipboard.writeText(this.checksum_results)
+                .then(() => {
+                    showNotification("Copied to clipboard");
+                })
+                .catch(() => {
+                    showErrorNotification("Failed copying to clipboard");
+                });
+        }
     }
 });
 </script>
-- 
cgit v1.2.3