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

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

use super::{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>,

}

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,

}

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 !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."));

    }

    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,

    notice: Option<String>,

    error: Option<String>,

}

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");

    Ok(Page(RepoSettingsPage { shell, repo, row, collaborators, teams, role, 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,

}

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);

    db::repos::set_description(&app.db, row.id, f.description.trim()).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 DeleteForm {

    #[serde(rename = "_csrf")]

    csrf_token: String,

    confirm: String,

}

pub async fn delete(

    State(app): State<Shared>,

    viewer: Viewer,

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

    Form(f): Form<DeleteForm>,

) -> 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)?;

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

        return Ok(settings_page(&app, shell, row, role, None, Some("Type the repository's full name to delete it.".into()))

            .await?

            .into_response());

    }

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

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

    db::repos::delete(&app.db, row.id).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    accounts::audit(&app.db, viewer.account(), "repo.deleted", &repo.to_string(), "moved to trash").await.ok();

    Ok(Redirect::to(&format!("/{owner}")).into_response())

}