trust level 100 w ts one

wakey-control-plane/src/cli.rs

@@ -3,8 +3,8 @@ use std::path::PathBuf;
 
 use clap::{ArgAction, Args, Parser, Subcommand};
 
-pub const DEFAULT_STATE_FILE: &str = "/var/lib/wakey-control-plane/state.db";
-pub const DEFAULT_PID_FILE: &str = "/var/run/wakey-control-plane.pid";
+pub const DEFAULT_PID_FILE: &str = "/var/lib/wakey-control-plane/wakey-control-plane.pid";
+pub const DEFAULT_DATA_DIR: &str = "/var/lib/wakey-control-plane";
 pub const DEFAULT_CONFIG_FILE: &str = "/etc/wakey-control-plane/config.toml";
 
 #[derive(Parser)]
@@ -32,6 +32,9 @@ pub enum Command {
 
 #[derive(Args, Clone)]
 pub struct ServeArgs {
+    #[arg(long)]
+    pub data_dir: Option<PathBuf>,
+
     #[arg(long)]
     pub bind: Option<SocketAddr>,
 
@@ -47,6 +50,9 @@ pub struct ServeArgs {
     #[arg(long)]
     pub command_timeout_ms: Option<u64>,
 
+    #[arg(long)]
+    pub enroll_token_ttl_seconds: Option<u64>,
+
     #[arg(long)]
     pub pid_file: Option<PathBuf>,
 
@@ -59,6 +65,9 @@ pub struct InitConfigArgs {
     #[arg(long, default_value = DEFAULT_CONFIG_FILE)]
     pub config_file: PathBuf,
 
+    #[arg(long)]
+    pub data_dir: Option<PathBuf>,
+
     #[arg(long)]
     pub bind: Option<SocketAddr>,
 
@@ -74,6 +83,9 @@ pub struct InitConfigArgs {
     #[arg(long)]
     pub command_timeout_ms: Option<u64>,
 
+    #[arg(long)]
+    pub enroll_token_ttl_seconds: Option<u64>,
+
     #[arg(long = "enroll-token")]
     pub enroll_tokens: Vec<String>,
 
@@ -92,14 +104,23 @@ pub struct InitConfigArgs {
 
 #[derive(Args)]
 pub struct IssueEnrollTokenArgs {
-    #[arg(long, default_value = DEFAULT_STATE_FILE)]
-    pub state_file: PathBuf,
+    #[arg(long, default_value = DEFAULT_CONFIG_FILE)]
+    pub config_file: PathBuf,
+
+    #[arg(long)]
+    pub state_file: Option<PathBuf>,
+
+    #[arg(long)]
+    pub data_dir: Option<PathBuf>,
 
     #[arg(long = "enroll-token")]
     pub enroll_tokens: Vec<String>,
 
     #[arg(long)]
     pub public_url: Option<String>,
+
+    #[arg(long)]
+    pub ttl_seconds: Option<u64>,
 }
 
 #[derive(Args)]