//! Runner updates (oxygenhub#322): the policy at each level and the test
//! groups runners are put in.

use super::{now_ms, Pool};
use sqlx::FromRow;

/// One level's settings; `None` inherits from the next level out.
#[derive(Clone, Debug, Default, PartialEq, Eq, FromRow)]
pub struct Policy {
    pub scope: String,
    pub key: String,
    pub updates: Option<bool>,
    pub channel: Option<String>,
    pub window_from: Option<i64>,
    pub window_to: Option<i64>,
    pub dry_run: Option<bool>,
}

impl Policy {
    /// Nothing set: the level inherits everything.
    pub fn is_empty(&self) -> bool {
        self.updates.is_none() && self.channel.is_none() && self.window_from.is_none() && self.dry_run.is_none()
    }
}

#[derive(Clone, Debug, PartialEq, Eq, FromRow)]
pub struct Group {
    pub id: i64,
    /// Empty: a group of the site's runners.
    pub owner: String,
    pub name: String,
    /// A share of the runners in reach, by a stable hash of their ids;
    /// `None`: only the chosen ones.
    pub percent: Option<i64>,
    pub created_ms: i64,
}

pub async fn policies(pool: &Pool) -> sqlx::Result<Vec<Policy>> {
    sqlx::query_as("SELECT * FROM runner_update_policies").fetch_all(pool).await
}

pub async fn policy(pool: &Pool, scope: &str, key: &str) -> sqlx::Result<Policy> {
    let p: Option<Policy> = sqlx::query_as("SELECT * FROM runner_update_policies WHERE scope = $1 AND key = $2")
        .bind(scope)
        .bind(key)
        .fetch_optional(pool)
        .await?;
    Ok(p.unwrap_or(Policy { scope: scope.into(), key: key.into(), ..Policy::default() }))
}

/// Set a level's settings; one with nothing set is removed.
pub async fn set_policy(pool: &Pool, p: &Policy) -> sqlx::Result<()> {
    sqlx::query("DELETE FROM runner_update_policies WHERE scope = $1 AND key = $2").bind(&p.scope).bind(&p.key).execute(pool).await?;
    if p.is_empty() {
        return Ok(());
    }
    sqlx::query(
        "INSERT INTO runner_update_policies (scope, key, updates, channel, window_from, window_to, dry_run)
         VALUES ($1, $2, $3, $4, $5, $6, $7)",
    )
    .bind(&p.scope)
    .bind(&p.key)
    .bind(p.updates)
    .bind(&p.channel)
    .bind(p.window_from)
    .bind(p.window_to)
    .bind(p.dry_run)
    .execute(pool)
    .await?;
    Ok(())
}

pub async fn groups(pool: &Pool) -> sqlx::Result<Vec<Group>> {
    sqlx::query_as("SELECT * FROM runner_groups ORDER BY owner, name").fetch_all(pool).await
}

pub async fn group(pool: &Pool, id: i64) -> sqlx::Result<Option<Group>> {
    sqlx::query_as("SELECT * FROM runner_groups WHERE id = $1").bind(id).fetch_optional(pool).await
}

pub async fn create_group(pool: &Pool, owner: &str, name: &str, percent: Option<i64>) -> sqlx::Result<Group> {
    sqlx::query_as("INSERT INTO runner_groups (owner, name, percent, created_ms) VALUES ($1, $2, $3, $4) RETURNING *")
        .bind(owner)
        .bind(name)
        .bind(percent)
        .bind(now_ms())
        .fetch_one(pool)
        .await
}

pub async fn set_percent(pool: &Pool, id: i64, percent: Option<i64>) -> sqlx::Result<()> {
    sqlx::query("UPDATE runner_groups SET percent = $1 WHERE id = $2").bind(percent).bind(id).execute(pool).await?;
    Ok(())
}

/// Remove a group, its members and its policy.
pub async fn delete_group(pool: &Pool, id: i64) -> sqlx::Result<()> {
    sqlx::query("DELETE FROM runner_group_members WHERE group_id = $1").bind(id).execute(pool).await?;
    sqlx::query("DELETE FROM runner_update_policies WHERE scope = 'group' AND key = $1").bind(id.to_string()).execute(pool).await?;
    sqlx::query("DELETE FROM runner_groups WHERE id = $1").bind(id).execute(pool).await?;
    Ok(())
}

/// Every chosen member, as (group, runner).
pub async fn members(pool: &Pool) -> sqlx::Result<Vec<(i64, i64)>> {
    sqlx::query_as("SELECT group_id, runner_id FROM runner_group_members").fetch_all(pool).await
}

/// Make `runners` the group's chosen members.
pub async fn set_members(pool: &Pool, group: i64, runners: &[i64]) -> sqlx::Result<()> {
    sqlx::query("DELETE FROM runner_group_members WHERE group_id = $1").bind(group).execute(pool).await?;
    for r in runners {
        sqlx::query("INSERT INTO runner_group_members (group_id, runner_id) VALUES ($1, $2)").bind(group).bind(r).execute(pool).await?;
    }
    Ok(())
}

/// What a runner said about its update.
pub async fn report(pool: &Pool, runner: i64, note: &str) -> sqlx::Result<()> {
    sqlx::query("UPDATE flow_runners SET update_note = $1, update_ms = $2 WHERE id = $3")
        .bind(note)
        .bind(now_ms())
        .bind(runner)
        .execute(pool)
        .await?;
    Ok(())
}