//! Proposal pages (hub spec §7): list, new, detail with conversation,

//! changes and files, comments anchored to atoms, reviews, landing.

use super::history::{render_files, FileRows};

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

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

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

use crate::db::{self, accounts, proposals as pdb};

use crate::render::{self, diff};

use crate::repos::proposals::{self as pl, Projection, Source};

use crate::repos::view::{self, ChangeView, ConflictView};

use crate::repos::RepoName;

use crate::Shared;

use askama::Template;

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

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

use hub_api::{Approval, ProposalSource, ProposalState};

use oxy_core::hash::ChangeHash;

use serde::Deserialize;

use std::collections::{BTreeMap, HashMap};

// ------------------------------------------------------------------ list

#[derive(Template)]

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

pub struct ProposalsPage {

    shell: Shell,

    repo: RepoShell,

    rows: Vec<pdb::ProposalRow>,

    filter: String,

    counts: (usize, usize, usize),

    bookmarks: Vec<String>,

    q: String,

    sort: String,

    pager: super::Pager,

    labels: std::collections::HashMap<i64, Vec<db::issues::Label>>,

}

impl ProposalsPage {

    fn labels_of(&self, id: &i64) -> &[db::issues::Label] {

        self.labels.get(id).map(Vec::as_slice).unwrap_or(&[])

    }

}

#[derive(Deserialize)]

pub struct ListQuery {

    state: Option<String>,

    q: Option<String>,

    sort: Option<String>,

    page: Option<usize>,

}

/// The list's sort orders.

pub const SORTS: &[(&str, &str)] = &[("updated", "recently updated"), ("newest", "newest"), ("oldest", "oldest")];

pub async fn list(

    State(app): State<Shared>,

    viewer: Viewer,

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

    Query(q): Query<ListQuery>,

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

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

    let (repo, row, role) = lookup(&app, &shell, &viewer, &owner, &name).await?;

    let all = pdb::list(&app.db, row.id, &[]).await.map_err(|e| Error::from_db(shell.clone(), e))?;

    let counts = (

        all.iter().filter(|p| p.is_open()).count(),

        all.iter().filter(|p| p.state == "landed").count(),

        all.iter().filter(|p| p.state == "closed").count(),

    );

    let filter = q.state.unwrap_or_else(|| "open".into());

    let mut labels: std::collections::HashMap<i64, Vec<db::issues::Label>> = std::collections::HashMap::new();

    for (id, l) in db::issues::labels_by_proposal(&app.db, row.id).await.map_err(|e| Error::from_db(shell.clone(), e))? {

        labels.entry(id).or_default().push(l);

    }

    // `author:bob label:ui target:main words…`

    let qtext = q.q.clone().unwrap_or_default();

    let (mut authors, mut wanted, mut targets, mut words) = (vec![], vec![], vec![], vec![]);

    for tok in qtext.split_whitespace() {

        match tok.split_once(':') {

            Some(("author", v)) => authors.push(v.to_lowercase()),

            Some(("label", v)) => wanted.push(v.to_lowercase()),

            Some(("target", v)) => targets.push(v.to_string()),

            _ => words.push(tok.to_lowercase()),

        }

    }

    let mut rows: Vec<pdb::ProposalRow> = all

        .into_iter()

        .filter(|p| match filter.as_str() {

            "open" => p.is_open(),

            "landed" => p.state == "landed",

            "closed" => p.state == "closed",

            _ => true,

        })

        .filter(|p| authors.is_empty() || authors.contains(&p.author.to_lowercase()))

        .filter(|p| targets.is_empty() || targets.contains(&p.target))

        .filter(|p| wanted.iter().all(|w| labels.get(&p.id).is_some_and(|ls| ls.iter().any(|l| l.name.to_lowercase() == *w))))

        .filter(|p| words.iter().all(|w| p.title.to_lowercase().contains(w) || p.body.to_lowercase().contains(w)))

        .collect();

    let sort = q.sort.filter(|s| SORTS.iter().any(|(k, _)| k == s)).unwrap_or_else(|| "updated".into());

    match sort.as_str() {

        "newest" => rows.sort_by_key(|p| std::cmp::Reverse(p.number)),

        "oldest" => rows.sort_by_key(|p| p.number),

        _ => rows.sort_by_key(|p| std::cmp::Reverse(p.updated_ms)),

    }

    let pager = super::Pager::new(rows.len(), q.page, &[("state", &filter), ("q", &qtext), ("sort", &sort)]);

    let rows = pager.slice(rows);

    let bookmarks = app

        .repos

        .read(&repo, |r, _| Ok(r.bookmarks()?.into_iter().map(|b| b.name).collect::<Vec<_>>()))

        .await

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

    Ok(Page(ProposalsPage {

        shell,

        repo: RepoShell::new(&app, &row, "proposals").with_role(role),

        rows,

        filter,

        counts,

        bookmarks,

        q: qtext,

        sort,

        pager,

        labels,

    }))

}

// ------------------------------------------------------------------- new

#[derive(Template)]

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

pub struct NewProposalPage {

    shell: Shell,

    repo: RepoShell,

    bookmarks: Vec<String>,

    source: String,

    target: String,

    error: Option<String>,

    /// Preview: the changes the proposal would add.

    preview: Vec<ChangeView>,

    conflicted: bool,

    /// What the form holds so far (after a failed open, or a draft).

    title: String,

    body: String,

    /// AI assistants that may draft the description (hub spec §13.4).

    assistants: Vec<String>,

}

#[derive(Deserialize)]

pub struct NewQuery {

    source: Option<String>,

    target: Option<String>,

}

/// `owner/name:bookmark` for a fork, else a bookmark of this repository.

/// A proposal's source: a bookmark of this repository by name (which may

/// hold `/`, as `feature/16-hud` does), or `owner/name:bookmark` of a fork;

/// `owner/name:bookmark` naming this repository is its own bookmark.

fn parse_source(app: &Shared, repo: &RepoName, source: &str) -> Option<(Option<RepoName>, String)> {

    let plain = |b: &str| !b.is_empty() && !b.chars().any(|c| c.is_control() || c == ':');

    if let Some((slug, bookmark)) = source.split_once(':') {

        let fork = RepoName::parse(slug)?;

        if !plain(bookmark) {

            return None;

        }

        if fork == *repo {

            return Some((None, bookmark.to_string()));

        }

        if !app.repos.exists(&fork) {

            return None;

        }

        return Some((Some(fork), bookmark.to_string()));

    }

    plain(source).then(|| (None, source.to_string()))

}

async fn preview_for(

    app: &Shared,

    repo: &RepoName,

    target: &str,

    source: &str,

) -> Result<(Vec<ChangeView>, bool), crate::repos::RepoError> {

    let Some((fork, bookmark)) = parse_source(app, repo, source) else { return Ok((vec![], false)) };

    let target = target.to_string();

    let fork_dir = fork.as_ref().map(|f| app.repos.dir(f));

    app.repos

        .read(repo, move |r, cache| {

            let projection = match &fork_dir {

                Some(dir) => {

                    let fork_repo = oxy_core::repo::Repo::open(dir)?;

                    let heads = fork_repo.bookmarks()?.into_iter().find(|b| b.name == bookmark).map(|b| b.heads).unwrap_or_default();

                    pl::project(r, cache, &target, Source::Fork { fork: &fork_repo, heads })?

                }

                None => {

                    let heads = r.bookmarks()?.into_iter().find(|b| b.name == bookmark).map(|b| b.heads).unwrap_or_default();

                    pl::project(r, cache, &target, Source::Bookmark(heads))?

                }

            };

            Ok((projection.changes, projection.conflicted))

        })

        .await

}

pub async fn new_form(

    State(app): State<Shared>,

    viewer: Viewer,

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

    Query(q): Query<NewQuery>,

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

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

    require(&viewer, &shell)?;

    let (repo, row, role) = lookup(&app, &shell, &viewer, &owner, &name).await?;

    let bookmarks = app

        .repos

        .read(&repo, |r, _| Ok(r.bookmarks()?.into_iter().map(|b| b.name).collect::<Vec<_>>()))

        .await

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

    let target = q.target.unwrap_or_else(|| row.default_bookmark.clone());

    let source = q.source.unwrap_or_default();

    let (preview, conflicted) =

        if source.is_empty() { (vec![], false) } else { preview_for(&app, &repo, &target, &source).await.unwrap_or((vec![], false)) };

    let assistants = if preview.is_empty() { vec![] } else { assistant_names(&app, &row).await };

    Ok(Page(NewProposalPage {

        shell,

        repo: RepoShell::new(&app, &row, "proposals").with_role(role),

        bookmarks,

        source,

        target,

        error: None,

        preview,

        conflicted,

        title: String::new(),

        body: String::new(),

        assistants,

    }))

}

#[derive(Deserialize)]

pub struct NewForm {

    #[serde(rename = "_csrf")]

    csrf_token: String,

    title: String,

    #[serde(default)]

    body: String,

    target: String,

    source: String,

    #[serde(default)]

    draft: Option<String>,

    /// For `/proposals/new/draft`: which assistant writes the description.

    #[serde(default)]

    assistant: String,

}

/// Open a proposal from a form or the API: validate, create, index, notify.

#[allow(clippy::too_many_arguments)]

pub async fn open_proposal(

    app: &Shared,

    repo: &RepoName,

    row: &db::repos::RepoRow,

    me: &accounts::Account,

    title: &str,

    body: &str,

    target: &str,

    source: &str,

    draft: bool,

) -> Result<pdb::ProposalRow, String> {

    if title.trim().is_empty() {

        return Err("Give the proposal a title.".into());

    }

    let bookmarks =

        app.repos.read(repo, |r, _| Ok(r.bookmarks()?.into_iter().map(|b| b.name).collect::<Vec<_>>())).await.map_err(|e| e.to_string())?;

    if !bookmarks.iter().any(|b| b == target) {

        return Err("Pick a target bookmark of this repository.".into());

    }

    let Some((fork, bookmark)) = parse_source(app, repo, source.trim()) else {

        return Err("The source is a bookmark of this repository, or owner/name:bookmark of a fork.".into());

    };

    if fork.is_none() && !bookmarks.contains(&bookmark) {

        return Err(format!("There is no bookmark {bookmark} in this repository."));

    }

    if fork.is_none() && bookmark == target {

        return Err("The source and the target are the same bookmark.".into());

    }

    let (preview, _) = preview_for(app, repo, target, source.trim()).await.map_err(|e| e.to_string())?;

    if preview.is_empty() {

        return Err("The source adds nothing to the target.".into());

    }

    let source_repo_id = match &fork {

        Some(fk) => db::repos::get(&app.db, &fk.owner, &fk.name).await.map_err(|e| e.to_string())?.map(|r| r.id),

        None => None,

    };

    let created = pdb::create(

        &app.db,

        pdb::NewProposal {

            repo_id: row.id,

            title: title.trim(),

            body: body.trim(),

            author_id: Some(me.id),

            author: &me.username,

            draft,

            target,

            source_kind: if fork.is_some() { "fork" } else { "bookmark" },

            source_repo_id,

            source_bookmark: &bookmark,

            source_changes: "",

        },

    )

    .await

    .map_err(|e| e.to_string())?;

    pdb::add_event(&app.db, created.id, &me.username, "opened", "").await.ok();

    accounts::audit(&app.db, Some(me), "proposal.opened", &format!("{}#{}", row.slug(), created.number), title).await.ok();

    db::notify::index(&app.db, "proposal", row.id, created.id, created.number, &created.title, &created.body).await.ok();

    let url = format!("/{}/proposal/{}", row.slug(), created.number);

    crate::notify::deliver(

        app,

        crate::notify::Event {

            kind: "proposal.opened",

            title: format!("{}#{}: {}", row.slug(), created.number, created.title),

            url,

            actor: me,

            repo_id: row.id,

            thread: Some(("proposal", created.id)),

            mention_text: body,

            direct: vec![],

        },

    )

    .await;

    crate::hooks::emit(

        app,

        row,

        "proposal.opened",

        &me.username,

        serde_json::json!({ "number": created.number, "title": created.title, "target": target, "source": source }),

    )

    .await;

    Ok(created)

}

pub async fn create(

    State(app): State<Shared>,

    viewer: Viewer,

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

    Form(f): Form<NewForm>,

) -> Result<Response, Error> {

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

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

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

    let (repo, row, role) = lookup(&app, &shell, &viewer, &owner, &name).await?;

    super::writable(&row, &shell)?;

    match open_proposal(&app, &repo, &row, &me.account, &f.title, &f.body, &f.target, &f.source, f.draft.is_some()).await {

        Ok(created) => Ok(Redirect::to(&format!("/{owner}/{name}/proposal/{}", created.number)).into_response()),

        Err(msg) => {

            let bookmarks =

                app.repos.read(&repo, |r, _| Ok(r.bookmarks()?.into_iter().map(|b| b.name).collect::<Vec<_>>())).await.unwrap_or_default();

            Ok(Page(NewProposalPage {

                shell,

                repo: RepoShell::new(&app, &row, "proposals").with_role(role),

                bookmarks,

                source: f.source.clone(),

                target: f.target.clone(),

                error: Some(msg),

                preview: vec![],

                conflicted: false,

                title: f.title.clone(),

                body: f.body.clone(),

                assistants: vec![],

            })

            .into_response())

        }

    }

}

// ---------------------------------------------------------------- detail

/// A comment placed on a diff row, or general.

pub struct CommentView {

    pub row: pdb::CommentRow,

    pub body_html: String,

    pub line: Option<usize>,

}

/// One entry of a proposal's conversation, in time order: what kind it

/// is and its index in `comments`, `reviews` or `events`.

pub struct Entry {

    pub kind: &'static str,

    pub index: usize,

}

pub struct ReviewView {

    pub row: pdb::ReviewRow,

    pub body_html: String,

    pub sealed: bool,

}

/// Everything the detail page and the API need, computed once under the lock.

pub struct Loaded {

    pub row: pdb::ProposalRow,

    pub changes: Vec<ChangeView>,

    pub hashes: Vec<ChangeHash>,

    pub heads: Vec<ChangeHash>,

    pub target_heads: Vec<ChangeHash>,

    pub state: String,

    pub target_state: String,

    pub conflicted: bool,

    pub conflicts: Vec<ConflictView>,

    pub files: Vec<FileRows>,

    /// The per-file hunks and bases behind `files`, to render again with other options.

    pub diffs: Vec<oxy_core::repo::FileDiff>,

    pub bases: std::collections::BTreeMap<String, Vec<u8>>,

    /// Atom per (path, new line number) of the projected state.

    pub atoms: HashMap<(String, usize), (String, u32)>,

    pub sealing_keys: Vec<String>,

    pub touched: Vec<String>,

    pub owners: Vec<String>,

    pub landed_already: bool,

}

/// Compute the projection and everything derived from it.

pub async fn load(app: &Shared, repo: &RepoName, row: pdb::ProposalRow) -> Result<Loaded, crate::repos::RepoError> {

    let fork_dir = match (row.source_kind.as_str(), row.source_repo_id) {

        ("fork", Some(id)) => {

            db::repos::get_by_id(&app.db, id).await.ok().flatten().and_then(|r| RepoName::new(&r.owner, &r.name)).map(|n| app.repos.dir(&n))

        }

        _ => None,

    };

    let r2 = row.clone();

    let loaded = app

        .repos

        .read(repo, move |r, cache| {

            let source = match (r2.source_kind.as_str(), &fork_dir) {

                ("changes", _) => Source::Changes(r2.source_changes().iter().filter_map(|h| ChangeHash::from_hex(h)).collect()),

                ("fork", Some(dir)) => {

                    let fork_repo = oxy_core::repo::Repo::open(dir)?;

                    let heads =

                        fork_repo.bookmarks()?.into_iter().find(|b| b.name == r2.source_bookmark).map(|b| b.heads).unwrap_or_default();

                    // Bring the fork's changes over now, so the projection below can read them.

                    let fork_state = fork_repo.load_state(&heads)?;

                    let missing: Vec<ChangeHash> = fork_state.changes().map(|(h, _)| *h).filter(|h| !r.store().has(h)).collect();

                    if !missing.is_empty() {

                        r.import_bundle(&fork_repo.bundle_for(&missing)?, r.current_policy()?.is_some())?;

                    }

                    Source::Changes(heads)

                }

                _ => {

                    let heads = r.bookmarks()?.into_iter().find(|b| b.name == r2.source_bookmark).map(|b| b.heads).unwrap_or_default();

                    Source::Bookmark(heads)

                }

            };

            let p: Projection = pl::project(r, cache, &r2.target, source)?;

            let target_loaded = cache.load(r, &p.target_heads)?;

            let files = render_files(&p.files, &p.bases);

            let mut atoms = HashMap::new();

            for f in &files {

                for l in &f.lines {

                    if let Some(n) = l.new_no {

                        if let Some((h, i)) = pl::atom_at(&p.loaded, &f.path, n as usize) {

                            atoms.insert((f.path.clone(), n as usize), (h.to_hex(), i));