improve oauth2 config deserialization and configuration

src/config/mod.rs

@@ -1,6 +1,6 @@
 pub mod args;
 pub mod config;
 pub mod prelude;
-mod wizard;
+pub(crate) mod wizard;
 
 pub use config::*;

src/config/prelude.rs

@@ -1,7 +1,7 @@
 use pimalaya_email::{
     folder::sync::Strategy as SyncFoldersStrategy, EmailHooks, EmailSender, EmailTextPlainFormat,
-    ImapAuthConfig, MaildirConfig, OAuth2Config, OAuth2Method, OAuth2Scopes, SendmailConfig,
-    SmtpConfig,
+    ImapAuthConfig, MaildirConfig, OAuth2ClientSecret, OAuth2Config, OAuth2Method, OAuth2Scopes,
+    SendmailConfig, SmtpConfig,
 };
 use serde::{Deserialize, Serialize};
 use std::{collections::HashSet, path::PathBuf};
@@ -47,7 +47,7 @@ pub struct ImapConfigDef {
     pub insecure: Option<bool>,
     #[serde(rename = "imap-login")]
     pub login: String,
-    #[serde(flatten, with = "ImapAuthConfigDef")]
+    #[serde(rename = "imap-auth", with = "ImapAuthConfigDef")]
     pub auth: ImapAuthConfig,
     #[serde(rename = "imap-notify-cmd")]
     pub notify_cmd: Option<String>,
@@ -57,16 +57,14 @@ pub struct ImapConfigDef {
     pub watch_cmds: Option<Vec<String>>,
 }
 
-#[derive(Clone, Debug, Default, Eq, PartialEq, Deserialize, Serialize)]
+#[derive(Clone, Debug, Eq, PartialEq, Deserialize, Serialize)]
 #[serde(remote = "ImapAuthConfig", rename_all = "kebab-case")]
 pub enum ImapAuthConfigDef {
-    #[default]
+    #[serde(skip)]
     None,
-    #[serde(rename = "imap-passwd")]
-    Passwd(String),
-    #[serde(rename = "imap-passwd-cmd")]
+    RawPasswd(String),
     PasswdCmd(String),
-    #[serde(rename = "imap-oauth2", with = "OAuth2ConfigDef")]
+    #[serde(with = "OAuth2ConfigDef", rename = "oauth2")]
     OAuth2(OAuth2Config),
 }
 
@@ -76,7 +74,8 @@ pub struct OAuth2ConfigDef {
     #[serde(with = "OAuth2MethodDef")]
     pub method: OAuth2Method,
     pub client_id: String,
-    pub client_secret: String,
+    #[serde(with = "OAuth2ClientSecretDef")]
+    pub client_secret: OAuth2ClientSecret,
     pub auth_url: String,
     pub token_url: String,
     #[serde(flatten, with = "OAuth2ScopesDef")]
@@ -85,6 +84,14 @@ pub struct OAuth2ConfigDef {
     pub pkce: bool,
 }
 
+#[derive(Clone, Debug, Eq, PartialEq, Deserialize, Serialize)]
+#[serde(remote = "OAuth2ClientSecret", rename_all = "kebab-case")]
+pub enum OAuth2ClientSecretDef {
+    Raw(String),
+    Cmd(String),
+    Keyring,
+}
+
 #[derive(Clone, Debug, Eq, PartialEq, Deserialize, Serialize)]
 #[serde(remote = "OAuth2Method", rename_all = "lowercase")]
 pub enum OAuth2MethodDef {

src/config/wizard/imap.rs

@@ -1,10 +1,13 @@
-use super::{SECURITY_PROTOCOLS, THEME};
-use crate::account::{
-    DeserializedAccountConfig, DeserializedBaseAccountConfig, DeserializedImapAccountConfig,
-};
 use anyhow::Result;
 use dialoguer::{Input, Select};
 use pimalaya_email::ImapConfig;
+use std::io;
+
+use crate::account::{
+    DeserializedAccountConfig, DeserializedBaseAccountConfig, DeserializedImapAccountConfig,
+};
+
+use super::{SECURITY_PROTOCOLS, THEME};
 
 #[cfg(feature = "imap-backend")]
 pub(crate) fn configure(base: DeserializedBaseAccountConfig) -> Result<DeserializedAccountConfig> {
@@ -56,3 +59,11 @@ pub(crate) fn configure(base: DeserializedBaseAccountConfig) -> Result<Deseriali
         DeserializedImapAccountConfig { base, backend },
     ))
 }
+
+#[cfg(feature = "imap-backend")]
+pub(crate) fn configure_oauth2_client_secret() -> io::Result<String> {
+    Input::with_theme(&*THEME)
+        .with_prompt("Enter your OAuth 2.0 client secret:")
+        .report(false)
+        .interact()
+}

src/config/wizard/mod.rs

@@ -1,5 +1,5 @@
 #[cfg(feature = "imap-backend")]
-mod imap;
+pub(crate) mod imap;
 mod maildir;
 #[cfg(feature = "notmuch-backend")]
 mod notmuch;

src/domain/account/args.rs

@@ -10,12 +10,14 @@ use crate::{folder, ui::table};
 
 const ARG_ACCOUNT: &str = "account";
 const ARG_DRY_RUN: &str = "dry-run";
+const ARG_RESET: &str = "reset";
 const CMD_ACCOUNTS: &str = "accounts";
 const CMD_CONFIGURE: &str = "configure";
 const CMD_LIST: &str = "list";
 const CMD_SYNC: &str = "sync";
 
 type DryRun = bool;
+type Reset = bool;
 
 /// Represents the account commands.
 #[derive(Debug, PartialEq, Eq)]
@@ -25,7 +27,7 @@ pub enum Cmd {
     /// Represents the sync account command.
     Sync(Option<SyncFoldersStrategy>, DryRun),
     /// Configure the current selected account.
-    Configure,
+    Configure(Reset),
 }
 
 /// Represents the account command matcher.
@@ -54,9 +56,10 @@ pub fn matches(m: &ArgMatches) -> Result<Option<Cmd>> {
             info!("list accounts subcommand matched");
             let max_table_width = table::args::parse_max_width(m);
             Some(Cmd::List(max_table_width))
-        } else if let Some(_) = m.subcommand_matches(CMD_CONFIGURE) {
+        } else if let Some(m) = m.subcommand_matches(CMD_CONFIGURE) {
             info!("configure account subcommand matched");
-            Some(Cmd::Configure)
+            let reset = parse_reset_flag(m);
+            Some(Cmd::Configure(reset))
         } else {
             info!("no account subcommand matched, falling back to subcommand list");
             Some(Cmd::List(None))
@@ -88,7 +91,8 @@ pub fn subcmd() -> Command {
                 .arg(dry_run()),
             Command::new(CMD_CONFIGURE)
                 .about("Configure the current selected account")
-                .aliases(["config", "conf", "cfg"]),
+                .aliases(["config", "conf", "cfg"])
+                .arg(reset_flag()),
         ])
 }
 
@@ -126,3 +130,15 @@ Changes can be visualized with the RUST_LOG=trace environment variable.",
 pub fn parse_dry_run_arg(m: &ArgMatches) -> bool {
     m.get_flag(ARG_DRY_RUN)
 }
+
+pub fn reset_flag() -> Arg {
+    Arg::new(ARG_RESET)
+        .help("Reset the configuration")
+        .short('r')
+        .long("reset")
+        .action(ArgAction::SetTrue)
+}
+
+pub fn parse_reset_flag(m: &ArgMatches) -> bool {
+    m.get_flag(ARG_RESET)
+}

src/domain/account/handlers.rs

@@ -11,15 +11,32 @@ use pimalaya_email::{
 };
 
 use crate::{
-    config::DeserializedConfig,
+    config::{wizard::imap::configure_oauth2_client_secret, DeserializedConfig},
     printer::{PrintTableOpts, Printer},
     Accounts,
 };
 
 /// Configure the current selected account
-pub fn configure(account_config: &AccountConfig, backend_config: &BackendConfig) -> Result<()> {
+pub fn configure(
+    account_config: &AccountConfig,
+    backend_config: &BackendConfig,
+    reset: bool,
+) -> Result<()> {
     info!("entering the configure account handler");
-    backend_config.configure(&account_config.name)?;
+    match backend_config {
+        BackendConfig::None => (),
+        BackendConfig::Maildir(_) => (),
+        #[cfg(feature = "imap-backend")]
+        BackendConfig::Imap(imap_config) => {
+            imap_config.auth.configure(
+                &account_config.name,
+                reset,
+                configure_oauth2_client_secret,
+            )?;
+        }
+        #[cfg(feature = "notmuch-backend")]
+        BackendConfig::Notmuch(config) => (),
+    };
     println!("Account {} configured!", account_config.name);
     Ok(())
 }

src/main.rs

@@ -137,8 +137,8 @@ fn main() -> Result<()> {
             backend.close()?;
             return Ok(());
         }
-        Some(account::args::Cmd::Configure) => {
-            return account::handlers::configure(&account_config, &backend_config);
+        Some(account::args::Cmd::Configure(reset)) => {
+            return account::handlers::configure(&account_config, &backend_config, reset);
         }
         _ => (),
     }