Beta program option in CLI

1 files changed, 54 insertions, 0 deletions

diff --git a/mullvad-cli/src/cmds/beta_program.rs b/mullvad-cli/src/cmds/beta_program.rs
new file mode 100644
index 0000000000..c63027ccfd
--- /dev/null
+++ b/mullvad-cli/src/cmds/beta_program.rs
@@ -0,0 +1,54 @@
+use crate::{new_rpc_client, Command, Result};
+use clap::value_t_or_exit;
+
+pub struct BetaProgram;
+
+impl Command for BetaProgram {
+    fn name(&self) -> &'static str {
+        "beta-program"
+    }
+
+    fn clap_subcommand(&self) -> clap::App<'static, 'static> {
+        clap::SubCommand::with_name(self.name())
+            .about("Receive notifications about beta updates")
+            .setting(clap::AppSettings::SubcommandRequiredElseHelp)
+            .subcommand(
+                clap::SubCommand::with_name("set")
+                    .about("Change beta notifications setting")
+                    .arg(
+                        clap::Arg::with_name("policy")
+                            .required(true)
+                            .possible_values(&["on", "off"]),
+                    ),
+            )
+            .subcommand(clap::SubCommand::with_name("get").about("Get beta notifications setting"))
+    }
+
+    fn run(&self, matches: &clap::ArgMatches<'_>) -> Result<()> {
+        match matches.subcommand() {
+            ("get", Some(_)) => {
+                let mut rpc = new_rpc_client()?;
+                let settings = rpc.get_settings()?;
+                let enabled_str = if settings.get_show_beta_releases().unwrap_or(false) {
+                    "on"
+                } else {
+                    "off"
+                };
+                println!("Beta program: {}", enabled_str);
+                Ok(())
+            }
+            ("set", Some(matches)) => {
+                let enabled_str = value_t_or_exit!(matches.value_of("policy"), String);
+
+                let mut rpc = new_rpc_client()?;
+                rpc.set_show_beta_releases(enabled_str == "on")?;
+
+                println!("Beta program: {}", enabled_str);
+                Ok(())
+            }
+            _ => {
+                unreachable!("unhandled comand");
+            }
+        }
+    }
+}