Merge pull request #17 from Nerox9/suspend

Add suspend and wakeup commands

api/admin/admin.proto

@@ -73,6 +73,8 @@ service AdminService {
     rpc StopApplication(ApplicationRequest) returns (ApplicationResponse) {}
     rpc Poweroff(Empty) returns (Empty) {}
     rpc Reboot(Empty) returns (Empty) {}
+    rpc Suspend(Empty) returns (Empty) {}
+    rpc Wakeup(Empty) returns (Empty) {}
 
     rpc QueryList(Empty) returns (QueryListResponse) {}
     rpc Watch(Empty) returns (stream WatchItem) {}

client/src/client.rs

@@ -110,6 +110,16 @@ impl AdminClient {
         let _response = self.connect_to().await?.poweroff(request).await?;
         Ok(())
     }
+    pub async fn suspend(&self) -> anyhow::Result<()> {
+        let request = pb::admin::Empty {};
+        let _response = self.connect_to().await?.suspend(request).await?;
+        Ok(())
+    }
+    pub async fn wakeup(&self) -> anyhow::Result<()> {
+        let request = pb::admin::Empty {};
+        let _response = self.connect_to().await?.wakeup(request).await?;
+        Ok(())
+    }
 
     pub async fn query(
         &self,

nixos/modules/host.nix

@@ -59,6 +59,8 @@ in
       default = [
         "reboot.target"
         "poweroff.target"
+        "sleep.target"
+        "suspend.target"
       ];
       example = "[ 'my-service.service' ]";
     };

nixos/tests/admin.nix

@@ -75,6 +75,8 @@ in
                   "[email protected]"
                   "poweroff.target"
                   "reboot.target"
+                  "sleep.target"
+                  "suspend.target"
                 ];
                 tls = mkTls "ghaf-host";
               };
@@ -154,6 +156,8 @@ in
                   services = [
                     "poweroff.target"
                     "reboot.target"
+                    "sleep.target"
+                    "suspend.target"
                   ];
                 };
 

src/admin/server.rs

@@ -385,6 +385,30 @@ impl pb::admin_service_server::AdminService for AdminService {
         })
         .await
     }
+    async fn suspend(
+        &self,
+        request: tonic::Request<Empty>,
+    ) -> std::result::Result<tonic::Response<Empty>, tonic::Status> {
+        escalate(request, |_| async {
+            self.inner
+                .send_system_command(String::from("suspend.target"))
+                .await?;
+            Ok(Empty {})
+        })
+        .await
+    }
+    async fn wakeup(
+        &self,
+        request: tonic::Request<Empty>,
+    ) -> std::result::Result<tonic::Response<Empty>, tonic::Status> {
+        escalate(request, |_| async {
+            self.inner
+                .send_system_command(String::from("sleep.target"))
+                .await?;
+            Ok(Empty {})
+        })
+        .await
+    }
 
     async fn query_list(
         &self,

src/bin/givc-cli.rs

@@ -59,6 +59,8 @@ enum Commands {
     },
     Reboot {},
     Poweroff {},
+    Suspend {},
+    Wakeup {},
     Query {
         #[arg(long, default_value_t = false)]
         as_json: bool, // Would it useful for scripts?
@@ -140,6 +142,8 @@ async fn main() -> std::result::Result<(), Box<dyn std::error::Error>> {
         Commands::Resume { app } => admin.resume(app).await?,
         Commands::Reboot {} => admin.reboot().await?,
         Commands::Poweroff {} => admin.poweroff().await?,
+        Commands::Suspend {} => admin.suspend().await?,
+        Commands::Wakeup {} => admin.wakeup().await?,
 
         Commands::Query {
             by_type,