From 592de08bbcb615971776f65b0dbfba25813aa690 Mon Sep 17 00:00:00 2001
From: "Frederik P." <frepbh@gmail.com>
Date: Sun, 3 Dec 2023 03:47:31 +0100
Subject: [PATCH] fixed stupid mistake

---
 Cargo.lock                   |  6 +++---
 Cargo.toml                   |  2 +-
 build.rs                     |  3 +++
 src/main.rs                  | 10 ++++++++--
 src/notification_provider.rs |  1 -
 variables.ps1                |  2 +-
 6 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index a42af78..2e87b89 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -4,7 +4,7 @@ version = 3
 
 [[package]]
 name = "autopower"
-version = "2.0.3"
+version = "2.0.4"
 dependencies = [
  "autopower_shared",
  "bincode",
@@ -13,7 +13,7 @@ dependencies = [
 
 [[package]]
 name = "autopower_notification_provider"
-version = "2.0.3"
+version = "2.0.4"
 dependencies = [
  "autopower_shared",
  "bincode",
@@ -23,7 +23,7 @@ dependencies = [
 
 [[package]]
 name = "autopower_shared"
-version = "2.0.3"
+version = "2.0.4"
 dependencies = [
  "bincode",
  "once_cell",
diff --git a/Cargo.toml b/Cargo.toml
index 790b896..36f3e2e 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -2,7 +2,7 @@
 members = ["shared", "notification_provider"]
 
 [workspace.package]
-version = "2.0.3"
+version = "2.0.4"
 edition = "2021"
 publish = false
 repository = "https://github.com/F0903/AutoPower"
diff --git a/build.rs b/build.rs
index 9ea1680..363a5f9 100644
--- a/build.rs
+++ b/build.rs
@@ -15,6 +15,9 @@ fn copy_powershell_scripts() -> Result<()> {
             continue;
         }
 
+        #[cfg(debug_assertions)]
+        std::fs::copy(file.path(), format!("./target/debug/{}", file_name_str))?;
+        #[cfg(not(debug_assertions))]
         std::fs::copy(file.path(), format!("./target/release/{}", file_name_str))?;
     }
     Ok(())
diff --git a/src/main.rs b/src/main.rs
index a754dc6..24e3bfc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -218,11 +218,14 @@ unsafe extern "system" fn service_main(_arg_num: u32, _args: *mut PWSTR) {
 
 fn service_setup() -> Result<()> {
     std::panic::set_hook(Box::new(|info| {
-        LOGGER.error(info);
+        LOGGER.error(format!("Fatal panic!\n    {}", info));
     }));
 
     LOGGER.debug("Starting setup...");
     let mut service_name = to_win32_wstr(SERVICE_NAME);
+    LOGGER.debug(format!("Service name is: {}", unsafe {
+        service_name.get_const().display()
+    }));
     let service_entry = SERVICE_TABLE_ENTRYW {
         lpServiceName: service_name.get_mut(),
         lpServiceProc: Some(service_main),
@@ -244,5 +247,8 @@ fn main() -> Result<()> {
         return Ok(());
     }
 
-    service_setup()
+    if let Err(e) = service_setup() {
+        LOGGER.error(format!("Fatal error!\n  {}", e))
+    }
+    Ok(())
 }
diff --git a/src/notification_provider.rs b/src/notification_provider.rs
index 60a0217..3049d65 100644
--- a/src/notification_provider.rs
+++ b/src/notification_provider.rs
@@ -28,7 +28,6 @@ impl NotificationProvider {
             name: "display".to_owned(),
             content: format!("{}\n{}", title, description),
         };
-        let command = bincode::serialize(&command)?;
         self.pipe.write_as(&command)?;
         Ok(())
     }
diff --git a/variables.ps1 b/variables.ps1
index 8474073..66ad789 100644
--- a/variables.ps1
+++ b/variables.ps1
@@ -1,4 +1,4 @@
-$Global:ServiceName = 'AutoPower'
+$Global:ServiceName = "AutoPower"
 $Global:Dir = "$($PSScriptRoot)\autopower.exe"
 $Global:NotifierPath = "$($PSScriptRoot)\autopower_notification_provider.exe"
 $Global:NotifierName = "AutoPower Notification Provider"