//! The `repositories` and `pushes` tables.

use super::{now_ms, Pool};

use hub_api::Visibility;

use sqlx::FromRow;

#[derive(Clone, Debug, FromRow)]

pub struct RepoRow {

    pub id: i64,

    pub owner: String,

    pub name: String,

    pub description: String,

    pub default_bookmark: String,

    pub visibility: String,

    pub created_ms: i64,

    pub updated_ms: i64,

    #[sqlx(default)]

    pub git_mirror: bool,

    #[sqlx(default)]

    pub git_mirror_state: String,

    /// The repository this one was forked from (migration 0009).

    #[sqlx(default)]

    pub forked_from: Option<i64>,

}

impl RepoRow {

    pub fn visibility(&self) -> Visibility {

        Visibility::parse(&self.visibility).unwrap_or(Visibility::Private)

    }

    pub fn slug(&self) -> String {

        format!("{}/{}", self.owner, self.name)

    }

}

#[derive(Clone, Debug, FromRow)]

pub struct PushRow {

    pub id: i64,

    pub at_ms: i64,

    pub reference: String,

    pub ok: bool,

    pub reason: String,

    pub old_state: String,

    pub new_state: String,

    pub heads: i64,

    pub changes: i64,

    pub pusher: String,

}

/// One reference's outcome in a push, as the protocol backend reports it.

#[derive(Clone, Debug)]

pub struct PushOutcome {

    pub reference: String,

    pub ok: bool,

    pub reason: String,

    pub old_state: String,

    pub new_state: String,

    pub heads: i64,

    pub changes: i64,

}

pub async fn list(pool: &Pool) -> sqlx::Result<Vec<RepoRow>> {

    sqlx::query_as("SELECT * FROM repositories ORDER BY updated_ms DESC, owner, name").fetch_all(pool).await

}

pub async fn get_by_id(pool: &Pool, id: i64) -> sqlx::Result<Option<RepoRow>> {

    sqlx::query_as("SELECT * FROM repositories WHERE id = ?").bind(id).fetch_optional(pool).await

}

pub async fn list_by_owner(pool: &Pool, owner: &str) -> sqlx::Result<Vec<RepoRow>> {

    sqlx::query_as("SELECT * FROM repositories WHERE owner = ? ORDER BY updated_ms DESC, name").bind(owner).fetch_all(pool).await

}

pub async fn get(pool: &Pool, owner: &str, name: &str) -> sqlx::Result<Option<RepoRow>> {

    sqlx::query_as("SELECT * FROM repositories WHERE owner = ? AND name = ?").bind(owner).bind(name).fetch_optional(pool).await

}

pub async fn create(pool: &Pool, owner: &str, name: &str, description: &str, visibility: Visibility) -> sqlx::Result<RepoRow> {

    let now = now_ms();

    sqlx::query_as(

        "INSERT INTO repositories (owner, name, description, visibility, created_ms, updated_ms) VALUES (?, ?, ?, ?, ?, ?) RETURNING *",

    )

    .bind(owner)

    .bind(name)

    .bind(description)

    .bind(visibility.as_str())

    .bind(now)

    .bind(now)

    .fetch_one(pool)

    .await

}

pub async fn delete(pool: &Pool, id: i64) -> sqlx::Result<()> {

    sqlx::query("DELETE FROM repositories WHERE id = ?").bind(id).execute(pool).await?;

    Ok(())

}

pub async fn set_description(pool: &Pool, id: i64, description: &str) -> sqlx::Result<()> {

    sqlx::query("UPDATE repositories SET description = ? WHERE id = ?").bind(description).bind(id).execute(pool).await?;

    Ok(())

}

pub async fn set_default_bookmark(pool: &Pool, id: i64, bookmark: &str) -> sqlx::Result<()> {

    sqlx::query("UPDATE repositories SET default_bookmark = ? WHERE id = ?").bind(bookmark).bind(id).execute(pool).await?;

    Ok(())

}

pub async fn set_visibility(pool: &Pool, id: i64, visibility: Visibility) -> sqlx::Result<()> {

    sqlx::query("UPDATE repositories SET visibility = ? WHERE id = ?").bind(visibility.as_str()).bind(id).execute(pool).await?;

    Ok(())

}

/// Record a push's per-reference results and bump the repository's activity.

pub async fn record_push(pool: &Pool, id: i64, pusher: Option<(i64, &str)>, results: &[PushOutcome]) -> sqlx::Result<()> {

    let now = now_ms();

    let mut tx = pool.begin().await?;

    for r in results {

        sqlx::query("INSERT INTO pushes (repo_id, at_ms, reference, ok, reason, old_state, new_state, heads, changes, account_id, pusher) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)")

            .bind(id)

            .bind(now)

            .bind(&r.reference)

            .bind(r.ok)

            .bind(&r.reason)

            .bind(&r.old_state)

            .bind(&r.new_state)

            .bind(r.heads)

            .bind(r.changes)

            .bind(pusher.map(|p| p.0).filter(|id| *id > 0))

            .bind(pusher.map(|p| p.1).unwrap_or(""))

            .execute(&mut *tx)

            .await?;

    }

    if results.iter().any(|r| r.ok) {

        sqlx::query("UPDATE repositories SET updated_ms = ? WHERE id = ?").bind(now).bind(id).execute(&mut *tx).await?;

    }

    tx.commit().await

}

pub async fn pushes(pool: &Pool, id: i64, limit: i64) -> sqlx::Result<Vec<PushRow>> {

    sqlx::query_as("SELECT id, at_ms, reference, ok, reason, old_state, new_state, heads, changes, pusher FROM pushes WHERE repo_id = ? ORDER BY at_ms DESC, id DESC LIMIT ?")

        .bind(id)

        .bind(limit)

        .fetch_all(pool)

        .await

}

/// Record that `id` is a fork of `from`.

pub async fn set_forked_from(pool: &Pool, id: i64, from: i64) -> sqlx::Result<()> {

    sqlx::query("UPDATE repositories SET forked_from = ? WHERE id = ?").bind(from).bind(id).execute(pool).await?;

    Ok(())

}

/// The forks of `id`, newest activity first.

pub async fn forks_of(pool: &Pool, id: i64) -> sqlx::Result<Vec<RepoRow>> {

    sqlx::query_as("SELECT * FROM repositories WHERE forked_from = ? ORDER BY updated_ms DESC").bind(id).fetch_all(pool).await

}

pub async fn set_git_mirror(pool: &Pool, id: i64, on: bool) -> sqlx::Result<()> {

    sqlx::query("UPDATE repositories SET git_mirror = ? WHERE id = ?").bind(on).bind(id).execute(pool).await?;

    Ok(())

}

pub async fn set_git_mirror_state(pool: &Pool, id: i64, state: &str) -> sqlx::Result<()> {

    sqlx::query("UPDATE repositories SET git_mirror_state = ? WHERE id = ?").bind(state).bind(id).execute(pool).await?;

    Ok(())

}