use --browser option

This commit is contained in:
Kevin Yue 2024-08-14 22:42:35 +08:00
parent a1c63f8498
commit f92b04e99a
No known key found for this signature in database
GPG Key ID: 4D3A6EE977B15AC4
3 changed files with 13 additions and 13 deletions

View File

@ -41,7 +41,7 @@ struct Cli {
#[arg(long)] #[arg(long)]
default_browser: bool, default_browser: bool,
#[arg(long)] #[arg(long)]
external_browser: Option<String>, browser: Option<String>,
} }
impl Cli { impl Cli {
@ -61,8 +61,8 @@ impl Cli {
None => portal_prelogin(&self.server, &gp_params).await?, None => portal_prelogin(&self.server, &gp_params).await?,
}; };
let browser_auth = if let Some(external_browser) = &self.external_browser { let browser_auth = if let Some(browser) = &self.browser {
Some(BrowserAuthenticator::new_with_browser(&saml_request, external_browser)) Some(BrowserAuthenticator::new_with_browser(&saml_request, browser))
} else if self.default_browser { } else if self.default_browser {
Some(BrowserAuthenticator::new(&saml_request)) Some(BrowserAuthenticator::new(&saml_request))
} else { } else {

View File

@ -102,7 +102,7 @@ pub(crate) struct ConnectArgs {
long, long,
help = "Use the specified browser to authenticate, e.g., firefox, chromium, chrome, or the path to the browser" help = "Use the specified browser to authenticate, e.g., firefox, chromium, chrome, or the path to the browser"
)] )]
external_browser: Option<String>, browser: Option<String>,
} }
impl ConnectArgs { impl ConnectArgs {
@ -332,8 +332,8 @@ impl<'a> ConnectHandler<'a> {
match prelogin { match prelogin {
Prelogin::Saml(prelogin) => { Prelogin::Saml(prelogin) => {
let use_default_browser = prelogin.support_default_browser() && self.args.default_browser; let use_default_browser = prelogin.support_default_browser() && self.args.default_browser;
let external_browser = if prelogin.support_default_browser() { let browser = if prelogin.support_default_browser() {
self.args.external_browser.as_deref() self.args.browser.as_deref()
} else { } else {
None None
}; };
@ -349,7 +349,7 @@ impl<'a> ConnectHandler<'a> {
.ignore_tls_errors(self.shared_args.ignore_tls_errors) .ignore_tls_errors(self.shared_args.ignore_tls_errors)
.clean(self.args.clean) .clean(self.args.clean)
.default_browser(use_default_browser) .default_browser(use_default_browser)
.external_browser(external_browser) .browser(browser)
.launch() .launch()
.await?; .await?;

View File

@ -19,7 +19,7 @@ pub struct SamlAuthLauncher<'a> {
ignore_tls_errors: bool, ignore_tls_errors: bool,
clean: bool, clean: bool,
default_browser: bool, default_browser: bool,
external_browser: Option<&'a str>, browser: Option<&'a str>,
} }
impl<'a> SamlAuthLauncher<'a> { impl<'a> SamlAuthLauncher<'a> {
@ -36,7 +36,7 @@ impl<'a> SamlAuthLauncher<'a> {
ignore_tls_errors: false, ignore_tls_errors: false,
clean: false, clean: false,
default_browser: false, default_browser: false,
external_browser: None, browser: None,
} }
} }
@ -90,8 +90,8 @@ impl<'a> SamlAuthLauncher<'a> {
self self
} }
pub fn external_browser(mut self, external_browser: Option<&'a str>) -> Self { pub fn browser(mut self, browser: Option<&'a str>) -> Self {
self.external_browser = external_browser; self.browser = browser;
self self
} }
@ -140,8 +140,8 @@ impl<'a> SamlAuthLauncher<'a> {
auth_cmd.arg("--default-browser"); auth_cmd.arg("--default-browser");
} }
if let Some(external_browser) = self.external_browser { if let Some(browser) = self.browser {
auth_cmd.arg("--external-browser").arg(external_browser); auth_cmd.arg("--browser").arg(browser);
} }
let mut non_root_cmd = auth_cmd.into_non_root()?; let mut non_root_cmd = auth_cmd.into_non_root()?;