mycelium-cli 9.0.0-rc.6

use crate::functions::try_to_resolve_database_url;

use clap::Parser;
use myc_core::{
    domain::{
        dtos::{instance_settings::STAFF_BOOTSTRAP_KEY, written_by::WrittenBy},
        entities::{
            AccountRegistration, InstanceSettingsRegistration,
            ResourceAuditLogRegistration, UserRegistration,
        },
    },
    use_cases::super_users::staff::account::create_seed_staff_account,
};
use myc_diesel::repositories::{
    DieselDbPoolProvider, DieselDbPoolProviderParameters, SqlAppModule,
};
use mycelium_base::dtos::Children;
use mycelium_base::entities::GetOrCreateResponseKind;
use shaku::HasComponent;
use std::sync::Arc;

#[derive(Parser, Debug)]
pub(crate) struct Arguments {
    #[clap(subcommand)]
    pub create_seed_account: Commands,
}

#[derive(Parser, Debug)]
pub(crate) enum Commands {
    CreateSeedAccount(CreateSeedAccountArguments),
}

#[derive(Parser, Debug)]
pub(crate) struct CreateSeedAccountArguments {
    email: String,
    account_name: String,
    first_name: String,
    last_name: String,
}

pub(crate) async fn create_seed_staff_account_cmd(
    args: CreateSeedAccountArguments,
) {
    //
    // Ask for the database url
    //
    let database_url = try_to_resolve_database_url();

    //
    // Ask for the password
    //
    let password =
        rpassword::prompt_password("Your password: ".to_string()).unwrap();

    //
    // Initialize the dependency
    //
    let module = Arc::new(
        SqlAppModule::builder()
            .with_component_parameters::<DieselDbPoolProvider>(
                DieselDbPoolProviderParameters {
                    pool: DieselDbPoolProvider::new(&database_url.as_str()),
                },
            )
            .build(),
    );

    let user_repo: &dyn UserRegistration = module.resolve_ref();
    let account_repo: &dyn AccountRegistration = module.resolve_ref();
    let audit_repo: &dyn ResourceAuditLogRegistration = module.resolve_ref();

    //
    // Create the seed staff account
    //
    match create_seed_staff_account(
        args.email.to_owned(),
        args.account_name.to_owned(),
        args.first_name.to_owned(),
        args.last_name.to_owned(),
        password,
        Box::new(user_repo),
        Box::new(account_repo),
        Box::new(audit_repo),
    )
    .await
    {
        Err(err) => tracing::error!("{err}"),
        Ok(res) => match res {
            GetOrCreateResponseKind::NotCreated(_, msg) => {
                tracing::info!(
                    "Seed staff account already exists created: {:?}",
                    msg
                )
            }
            GetOrCreateResponseKind::Created(account) => {
                tracing::info!(
                    "\n
    Seed staff account successfully created:
    - Email: {}
    - First Name: {}
    - Last Name: {}
    - Account Name: {}
        ",
                    args.email,
                    args.first_name,
                    args.last_name,
                    account.name,
                );

                close_staff_bootstrap_best_effort(&account, &module).await;
            }
        },
    };
}

/// Best-effort: close the web staff-bootstrap flow (feature staff-bootstrap,
/// spec SB-R9) when the manual CLI path is used instead. Never fails the CLI
/// command -- `instance_settings` bookkeeping is secondary to the seed
/// account itself, which is already the source of truth at this point.
async fn close_staff_bootstrap_best_effort(
    account: &myc_core::domain::dtos::account::Account,
    module: &Arc<SqlAppModule>,
) {
    let Children::Records(owners) = &account.owners else {
        tracing::warn!(
            "Could not close staff bootstrap: account owners were not \
             returned as records"
        );
        return;
    };

    let Some(owner) = owners.first() else {
        tracing::warn!(
            "Could not close staff bootstrap: no owner user available"
        );
        return;
    };

    let Some(user_id) = owner.id else {
        tracing::warn!(
            "Could not close staff bootstrap: no owner user id available"
        );
        return;
    };

    let instance_settings_repo: &dyn InstanceSettingsRegistration =
        module.resolve_ref();

    let created_by =
        WrittenBy::new_from_user_with_email(user_id, &owner.email.email());

    match instance_settings_repo
        .get_or_create(
            STAFF_BOOTSTRAP_KEY.to_string(),
            serde_json::json!({}),
            Some(created_by),
        )
        .await
    {
        Ok(GetOrCreateResponseKind::Created(_)) => {
            tracing::info!(
                "Staff bootstrap closed (claimed via manual CLI path)"
            );
        }
        Ok(GetOrCreateResponseKind::NotCreated(..)) => {
            // Already claimed (e.g. the web flow got there first) -- fine.
        }
        Err(err) => {
            tracing::warn!(
                "Could not close staff bootstrap (non-fatal): {err}"
            );
        }
    }
}