Method execute

crates/chat-cli/src/cli/user.rs:78–182

(self, os: &mut Os)

impl LoginArgs {
    pub async fn execute(self, os: &mut Os) -> Result<ExitCode> {
        if crate::auth::is_logged_in(&mut os.database).await {
            eyre::bail!(
                "Already logged in, please logout with {} first",
                StyledText::command(&format!("{CLI_BINARY_NAME} logout"))
            );
        }

        let login_method = match self.license {
            Some(LicenseType::Free) => AuthMethod::BuilderId,
            Some(LicenseType::Pro) => AuthMethod::IdentityCenter,
            None => {
                if self.identity_provider.is_some() && self.region.is_some() {
                    // If license is specified and --identity-provider and --region are specified,
                    // the license is determined to be pro
                    AuthMethod::IdentityCenter
                } else {
                    // --license is not specified, prompt the user to choose
                    let options = [AuthMethod::BuilderId, AuthMethod::IdentityCenter];
                    let i = match choose("Select login method", &options)? {
                        Some(i) => i,
                        None => bail!("No login method selected"),
                    };
                    options[i]
                }
            },
        };

        match login_method {
            AuthMethod::BuilderId | AuthMethod::IdentityCenter => {
                let (start_url, region) = match login_method {
                    AuthMethod::BuilderId => (None, None),
                    AuthMethod::IdentityCenter => {
                        let default_start_url = match self.identity_provider {
                            Some(start_url) => Some(start_url),
                            None => os.database.get_start_url()?,
                        };
                        let default_region = match self.region {
                            Some(region) => Some(region),
                            None => os.database.get_idc_region()?,
                        };

                        let start_url = input("Enter Start URL", default_start_url.as_deref())?;
                        let region = input("Enter Region", default_region.as_deref())?.trim().to_string();

                        let _ = os.database.set_start_url(start_url.clone());
                        let _ = os.database.set_idc_region(region.clone());

                        (Some(start_url), Some(region))
                    },
                };

                // Remote machine won't be able to handle browser opening and redirects,
                // hence always use device code flow.
                if is_remote() || self.use_device_flow {
                    try_device_authorization(os, start_url.clone(), region.clone()).await?;
                } else {
                    let (client, registration) = start_pkce_authorization(start_url.clone(), region.clone()).await?;