//! Creating repositories and their settings pages (hub spec §6.3).

use super::settings::{csrf, require};

use super::{filters, lookup_role, Error, Page, RepoShell, Shell};

use crate::auth::{roles, Access, Role, Viewer};

use crate::db::{self, accounts};

use crate::repos::RepoName;

use crate::Shared;

use askama::Template;

use axum::extract::{Form, Path, State};

use axum::response::{IntoResponse, Redirect, Response};

use hub_api::Visibility;

use serde::Deserialize;

#[derive(Template)]

#[template(path = "repo_new.html")]

pub struct NewRepoPage {

    shell: Shell,

    owners: Vec<String>,

    error: Option<String>,

}

pub(crate) async fn owners_for(app: &Shared, me: &accounts::Account) -> Vec<String> {

    let mut owners = vec![me.username.clone()];

    if let Ok(orgs) = accounts::orgs_of(&app.db, me.id).await {

        owners.extend(orgs.into_iter().map(|(o, _)| o.name));

    }

    owners

}

pub async fn new_form(State(app): State<Shared>, viewer: Viewer) -> Result<Page<NewRepoPage>, Error> {

    let shell = Shell::new(&app, "/new", &viewer);

    let me = require(&viewer, &shell)?;

    let owners = owners_for(&app, &me.account).await;

    Ok(Page(NewRepoPage { shell, owners, error: None }))

}

#[derive(Deserialize)]

pub struct NewRepoForm {

    #[serde(rename = "_csrf")]

    csrf_token: String,

    owner: String,

    name: String,

    #[serde(default)]

    description: String,

    #[serde(default = "default_visibility")]

    visibility: String,

    /// `hub` (this server's disk, the default) or `bucket` (oxygenhub#96).

    #[serde(default)]

    storage: String,

    #[serde(flatten)]

    bucket: BucketForm,

}

/// A bucket and the credentials for it, as the New repository form and

/// Settings → Storage give them; only the kind the URL names is read.

#[derive(Deserialize, Default)]

pub struct BucketForm {

    #[serde(default)]

    store_url: String,

    #[serde(default)]

    access_key_id: String,

    #[serde(default)]

    secret_access_key: String,

    #[serde(default)]

    service_account_key: String,

    #[serde(default)]

    tenant_id: String,

    #[serde(default)]

    client_id: String,

    #[serde(default)]

    client_secret: String,

}

impl BucketForm {

    fn credentials(&self, scheme: &str) -> Result<oxy_strata::Credentials, String> {

        crate::storage::credentials_from(

            scheme,

            &self.access_key_id,

            &self.secret_access_key,

            &self.service_account_key,

            &self.tenant_id,

            &self.client_id,

            &self.client_secret,

        )

    }

}

fn default_visibility() -> String {

    "public".into()

}

pub async fn create(State(app): State<Shared>, viewer: Viewer, Form(f): Form<NewRepoForm>) -> Result<Response, Error> {

    let shell = Shell::new(&app, "/new", &viewer);

    let me = require(&viewer, &shell)?;

    csrf(&viewer, &f.csrf_token, &shell)?;

    let owners = owners_for(&app, &me.account).await;

    let fail = |msg: &str| Page(NewRepoPage { shell: shell.clone(), owners: owners.clone(), error: Some(msg.into()) }).into_response();

    let Some(name) = RepoName::new(f.owner.trim(), f.name.trim()) else {

        return Ok(fail("A repository name is letters, digits, dots, dashes and underscores."));

    };

    if name.name.ends_with(crate::wiki::SUFFIX) {

        return Ok(fail("A name cannot end in .wiki: that is the address of a repository's wiki."));

    }

    if !roles::may_create_under(&app.db, &me.account, &name.owner).await.map_err(|e| Error::from_db(shell.clone(), e))? {

        return Ok(fail("You cannot create repositories under that owner."));

    }

    let Some(visibility) = Visibility::parse(&f.visibility) else { return Ok(fail("Visibility is public, internal or private.")) };

    if db::repos::get(&app.db, &name.owner, &name.name).await.map_err(|e| Error::from_db(shell.clone(), e))?.is_some()

        || app.repos.exists(&name)

    {

        return Ok(fail("That repository exists."));

    }

    // An organisation's storage policy (oxygenhub#96).

    if let Some(org) = accounts::org_by_name(&app.db, &name.owner).await.map_err(|e| Error::from_db(shell.clone(), e))? {

        let policy = crate::storage::policy(&app.db, org.id).await.map_err(|e| Error::from_db(shell.clone(), e))?;

        let store = (f.storage == "bucket").then_some(f.bucket.store_url.as_str());

        if let Err(e) = policy.permits(&org.name, store) {

            return Ok(fail(&format!("{e}.")));

        }

    }

    if f.storage == "bucket" {

        // Kept in the owner's bucket (oxygenhub#96): checked before

        // anything is made, and nothing is left behind when it fails.

        let url = f.bucket.store_url.trim();

        let parsed = match crate::storage::check_url(url) {

            Ok(p) => p,

            Err(e) => return Ok(fail(&format!("Store: {e}."))),

        };

        let credentials = match f.bucket.credentials(&parsed.scheme) {

            Ok(c) => c,

            Err(e) => return Ok(fail(&format!("Credentials: {e}."))),

        };

        let row = db::repos::create(&app.db, &name.owner, &name.name, f.description.trim(), visibility)

            .await

            .map_err(|e| Error::from_db(shell.clone(), e))?;

        if let Err(e) = crate::storage::create(&app, &name, row.id, url, credentials).await {

            db::repos::delete(&app.db, row.id).await.ok();

            return Ok(fail(&format!("The bucket did not pass the check, so nothing was created: {e}")));

        }

        accounts::audit(

            &app.db,

            Some(&me.account),

            "repo.created",

            &name.to_string(),

            &format!("{} in {}", visibility.as_str(), parsed.plain()),

        )

        .await

        .ok();

        return Ok(Redirect::to(&name.url()).into_response());

    }

    app.repos.create(&name, None).map_err(|e| Error::from_repo(shell.clone(), e))?;

    db::repos::create(&app.db, &name.owner, &name.name, f.description.trim(), visibility)

        .await

        .map_err(|e| Error::from_db(shell.clone(), e))?;

    accounts::audit(&app.db, Some(&me.account), "repo.created", &name.to_string(), visibility.as_str()).await.ok();

    Ok(Redirect::to(&name.url()).into_response())

}

#[derive(Template)]

#[template(path = "repo_settings.html")]

pub struct RepoSettingsPage {

    shell: Shell,

    repo: RepoShell,

    row: db::repos::RepoRow,

    collaborators: Vec<accounts::Collaborator>,

    teams: Vec<accounts::Team>,

    role: Role,

    requirements: db::proposals::Requirements,

    hooks: Vec<super::webhooks::HookRow>,

    required_checks: String,

    /// Whether hub.toml alone would run Flow here, and whether it runs.

    flow_default: bool,

    /// hub.toml's `[diffusion] follow` (#148).

    diffusion_site: String,

    flow_enabled: bool,

    site_admin: bool,

    /// Whether the repository or its owner has a runner of its own (#225).

    own_runner: bool,

    notice: Option<String>,

    error: Option<String>,

    /// Runners that may take this repository's jobs: (runner, what it

    /// serves, whether it is this repository's own).

    runners: Vec<(db::runners::Runner, String, bool)>,

    /// A registration token just made, shown once.

    runner_token: Option<String>,

    /// The repository's AI assistants and prompt (#97).

    assistants: super::assistants::Panel,

    /// The repository's Flow variables (#102).

    variables: super::variables::Panel,

    /// The repository's Flow secrets (#107).

    secrets: super::secrets::Panel,

    /// Where its data lives, when in a bucket (oxygenhub#96).

    storage: Option<crate::storage::Storage>,

    /// The coverage check's rules (#113).

    coverage: crate::coverage::Rules,

}

async fn settings_page(

    app: &Shared,

    shell: Shell,

    row: db::repos::RepoRow,

    role: Role,

    notice: Option<String>,

    error: Option<String>,

) -> Result<Page<RepoSettingsPage>, Error> {

    let collaborators = accounts::collaborators(&app.db, row.id).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let teams = match accounts::org_by_name(&app.db, &row.owner).await.map_err(|e| Error::from_db(shell.clone(), e))? {

        Some(org) => accounts::teams(&app.db, org.id).await.map_err(|e| Error::from_db(shell.clone(), e))?,

        None => vec![],

    };

    let repo = RepoShell::new(app, &row, "settings").with_role(role);

    let requirements = db::proposals::requirements(&app.db, row.id).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let hooks = db::hooks::hooks_for_repo(&app.db, row.id).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let hooks = super::webhooks::rows(app, hooks).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let required_checks = db::hooks::required_checks(&app.db, row.id).await.map_err(|e| Error::from_db(shell.clone(), e))?.join(", ");

    let flow_default = app.config.flow.allows(&row.owner, &row.name);

    let diffusion_site = app.config.diffusion.follow.clone();

    let flow_enabled = crate::flow::enabled(app, &row);

    let site_admin = shell.is_admin;

    let own_runner = has_own_runner(app, &row).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let runners = db::runners::serving(&app.db, &row.owner, row.id)

        .await

        .map_err(|e| Error::from_db(shell.clone(), e))?

        .into_iter()

        .map(|r| {

            let own = r.scope_repo_id == Some(row.id);

            let serves = if own {

                "this repository".to_string()

            } else if r.scope_owner.is_empty() {

                "the whole site".to_string()

            } else {

                format!("{}'s repositories", r.scope_owner)

            };

            (r, serves, own)

        })

        .collect();

    let assistants = super::assistants::panel(app, &repo_assistants(&row)).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let variables = super::variables::panel(app, &repo_assistants(&row)).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let secrets = super::secrets::panel(app, &repo_assistants(&row)).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let storage = crate::storage::of(&app.db, row.id).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let coverage = crate::coverage::rules(&app.db, row.id).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    Ok(Page(RepoSettingsPage {

        coverage,

        storage,

        assistants,

        variables,

        secrets,

        runners,

        runner_token: None,

        shell,

        repo,

        row,

        collaborators,

        teams,

        role,

        requirements,

        hooks,

        required_checks,

        flow_default,

        diffusion_site,

        flow_enabled,

        site_admin,

        own_runner,

        notice,

        error,

    }))

}

pub async fn settings(

    State(app): State<Shared>,

    viewer: Viewer,

    Path((owner, name)): Path<(String, String)>,

) -> Result<Page<RepoSettingsPage>, Error> {

    let shell = Shell::new(&app, &format!("/{owner}/{name}/settings"), &viewer);

    let (_, row, role) = lookup_role(&app, &shell, &viewer, &owner, &name, Access::Maintain).await?;

    settings_page(&app, shell, row, role, None, None).await

}

#[derive(Deserialize)]

pub struct GeneralForm {

    #[serde(rename = "_csrf")]

    csrf_token: String,

    description: String,

    default_bookmark: String,

    visibility: String,

    #[serde(default)]

    topics: String,

}

pub async fn update(

    State(app): State<Shared>,

    viewer: Viewer,

    Path((owner, name)): Path<(String, String)>,

    Form(f): Form<GeneralForm>,

) -> Result<Page<RepoSettingsPage>, Error> {

    let shell = Shell::new(&app, &format!("/{owner}/{name}/settings"), &viewer);

    let (_, row, role) = lookup_role(&app, &shell, &viewer, &owner, &name, Access::Maintain).await?;

    csrf(&viewer, &f.csrf_token, &shell)?;

    let e = |e| Error::from_db(shell.clone(), e);

    let topics = match db::repos::parse_topics(&f.topics) {

        Ok(t) => t,

        Err(msg) => return settings_page(&app, shell, row, role, None, Some(msg)).await,

    };

    db::repos::set_description(&app.db, row.id, f.description.trim()).await.map_err(e)?;

    db::repos::set_topics(&app.db, row.id, &topics).await.map_err(e)?;

    if crate::repos::names::valid_segment(f.default_bookmark.trim()) {

        db::repos::set_default_bookmark(&app.db, row.id, f.default_bookmark.trim()).await.map_err(e)?;

    }

    if role >= Role::Admin {

        if let Some(v) = Visibility::parse(&f.visibility) {

            db::repos::set_visibility(&app.db, row.id, v).await.map_err(e)?;

        }

    }

    accounts::audit(&app.db, viewer.account(), "repo.updated", &format!("{owner}/{name}"), "").await.ok();

    let row = db::repos::get_by_id(&app.db, row.id).await.map_err(e)?.unwrap_or(row);

    settings_page(&app, shell, row, role, Some("Saved.".into()), None).await

}

#[derive(Deserialize)]

pub struct CollaboratorForm {

    #[serde(rename = "_csrf")]

    csrf_token: String,

    #[serde(default)]

    username: String,

    #[serde(default)]

    team: String,

    #[serde(default)]

    role: String,

    #[serde(default)]

    remove: Option<String>,

}

pub async fn collaborator(

    State(app): State<Shared>,

    viewer: Viewer,

    Path((owner, name)): Path<(String, String)>,

    Form(f): Form<CollaboratorForm>,

) -> Result<Page<RepoSettingsPage>, Error> {

    let shell = Shell::new(&app, &format!("/{owner}/{name}/settings"), &viewer);

    let (_, row, role) = lookup_role(&app, &shell, &viewer, &owner, &name, Access::Admin).await?;

    csrf(&viewer, &f.csrf_token, &shell)?;

    let e = |e| Error::from_db(shell.clone(), e);

    let (account_id, team_id, label) = if !f.username.trim().is_empty() {

        match accounts::account_by_name(&app.db, f.username.trim()).await.map_err(e)? {

            Some(a) => (Some(a.id), None, a.username),

            None => return settings_page(&app, shell, row, role, None, Some("No such account.".into())).await,

        }

    } else if !f.team.trim().is_empty() {

        let org = accounts::org_by_name(&app.db, &row.owner).await.map_err(e)?;

        let team = match org {

            Some(o) => accounts::team_by_name(&app.db, o.id, f.team.trim()).await.map_err(e)?,

            None => None,

        };

        match team {

            Some(t) => (None, Some(t.id), format!("team {}", t.name)),

            None => return settings_page(&app, shell, row, role, None, Some("No such team in this organisation.".into())).await,

        }

    } else {

        return settings_page(&app, shell, row, role, None, Some("Name an account or a team.".into())).await;

    };

    if f.remove.is_some() {

        accounts::remove_collaborator(&app.db, row.id, account_id, team_id).await.map_err(e)?;

        accounts::audit(&app.db, viewer.account(), "repo.collaborator.removed", &format!("{owner}/{name}"), &label).await.ok();

        return settings_page(&app, shell, row, role, Some(format!("Removed {label}.")), None).await;

    }

    let Some(new_role) = Role::parse(&f.role) else {

        return settings_page(&app, shell, row, role, None, Some("Pick a role.".into())).await;

    };

    accounts::set_collaborator(&app.db, row.id, account_id, team_id, new_role.as_str()).await.map_err(e)?;

    accounts::audit(

        &app.db,

        viewer.account(),

        "repo.collaborator.set",

        &format!("{owner}/{name}"),

        &format!("{label} {}", new_role.as_str()),

    )

    .await

    .ok();

    settings_page(&app, shell, row, role, Some(format!("{label} now has {} access.", new_role.as_str())), None).await

}

#[derive(Deserialize)]

pub struct TransferForm {

    #[serde(rename = "_csrf")]

    csrf_token: String,

    owner: String,

    confirm: String,

}

/// `POST /{owner}/{name}/settings/transfer`: move the repository to a

/// user or organisation the admin may create repositories under (#108).

/// The old URL redirects and oxy remotes keep working.

pub async fn transfer(

    State(app): State<Shared>,

    viewer: Viewer,

    Path((owner, name)): Path<(String, String)>,

    Form(f): Form<TransferForm>,

) -> Result<Response, Error> {

    let shell = Shell::new(&app, &format!("/{owner}/{name}/settings"), &viewer);

    let (repo, row, role) = lookup_role(&app, &shell, &viewer, &owner, &name, Access::Admin).await?;

    csrf(&viewer, &f.csrf_token, &shell)?;

    let me = viewer.account().cloned().ok_or_else(|| Error::not_found(shell.clone()))?;

    let fail = |msg: &str| settings_page(&app, shell.clone(), row.clone(), role, None, Some(msg.into()));

    if f.confirm.trim() != repo.to_string() {

        return Ok(fail("Type the repository's full name to confirm.").await?.into_response());

    }

    let to = f.owner.trim().trim_start_matches('@').to_string();

    let Some(dest) = RepoName::new(&to, &row.name) else {

        return Ok(fail("That is not an owner name.").await?.into_response());

    };

    if dest.owner == row.owner {

        return Ok(fail("The repository is already there.").await?.into_response());

    }

    let e = |e| Error::from_db(shell.clone(), e);

    let known = accounts::account_by_name(&app.db, &dest.owner).await.map_err(e)?.is_some()

        || accounts::org_by_name(&app.db, &dest.owner).await.map_err(e)?.is_some();

    if !known {

        return Ok(fail("There is no user or organisation of that name.").await?.into_response());

    }

    if !roles::may_create_under(&app.db, &me, &dest.owner).await.map_err(e)? {

        return Ok(fail("You cannot create repositories under that owner, so you cannot move one there.").await?.into_response());

    }

    if db::repos::get(&app.db, &dest.owner, &dest.name).await.map_err(e)?.is_some() || app.repos.exists(&dest) {

        return Ok(fail("That owner already has a repository of this name.").await?.into_response());

    }

    {

        let _guard = app.repos.write_guard(&repo).await;

        app.repos.rename(&repo, &dest).map_err(|e| Error::from_repo(shell.clone(), e))?;

        crate::storage::renamed(&app, &repo, &dest);

        if app.wikis.exists(&repo) {

            app.wikis.rename(&repo, &dest).ok();

        }

        db::repos::transfer(&app.db, row.id, &row.owner, &row.name, &dest.owner).await.map_err(e)?;

        let mirror = crate::mirror::mirror_dir(&app, &repo);

        if mirror.is_dir() {

            let to = crate::mirror::mirror_dir(&app, &dest);

            if let Some(parent) = to.parent() {