module main

import time

enum CiStatusEnum {
    pending   = 0
    running   = 1
    success   = 2
    failure   = 3
    cancelled = 4
}

fn (s CiStatusEnum) str() string {
    return match s {
        .pending { 'pending' }
        .running { 'running' }
        .success { 'success' }
        .failure { 'failure' }
        .cancelled { 'cancelled' }
    }
}

fn (s CiStatusEnum) css_class() string {
    return match s {
        .pending { 'ci-pending' }
        .running { 'ci-running' }
        .success { 'ci-success' }
        .failure { 'ci-failure' }
        .cancelled { 'ci-cancelled' }
    }
}

fn (s CiStatusEnum) icon() string {
    return match s {
        .pending { '⏳' }
        .running { '🔄' }
        .success { '✓' }
        .failure { '✗' }
        .cancelled { '⊘' }
    }
}

struct CiStatus {
    id          int @[primary; sql: serial]
    repo_id     int
    commit_hash string
    branch      string
    status      CiStatusEnum
    ci_run_id   int
    created_at  int
    updated_at  int
}

fn ci_status_from_string(s string) CiStatusEnum {
    return match s {
        'pending' { CiStatusEnum.pending }
        'running' { CiStatusEnum.running }
        'success' { CiStatusEnum.success }
        'failure' { CiStatusEnum.failure }
        'cancelled' { CiStatusEnum.cancelled }
        else { CiStatusEnum.pending }
    }
}

fn (mut app App) find_ci_status_for_commit(repo_id int, commit_hash string) ?CiStatus {
    results := sql app.db {
        select from CiStatus where repo_id == repo_id && commit_hash == commit_hash order by id desc limit 1
    } or { return none }
    if results.len == 0 {
        return none
    }
    return results[0]
}

fn (mut app App) find_ci_status_for_branch(repo_id int, branch string) ?CiStatus {
    results := sql app.db {
        select from CiStatus where repo_id == repo_id && branch == branch order by id desc limit 1
    } or { return none }
    if results.len == 0 {
        return none
    }
    return results[0]
}

fn (mut app App) find_ci_runs_for_repo(repo_id int) []CiStatus {
    return sql app.db {
        select from CiStatus where repo_id == repo_id order by id desc
    } or { []CiStatus{} }
}

fn (mut app App) add_ci_status(ci CiStatus) ! {
    sql app.db {
        insert ci into CiStatus
    }!
}

fn (mut app App) update_ci_status(repo_id int, commit_hash string, status CiStatusEnum) ! {
    updated := int(time.now().unix())
    sql app.db {
        update CiStatus set status = status, updated_at = updated where repo_id == repo_id
        && commit_hash == commit_hash
    }!
}

fn (mut app App) upsert_ci_status(repo_id int, commit_hash string, branch string, status CiStatusEnum, ci_run_id int) ! {
    existing := app.find_ci_status_for_commit(repo_id, commit_hash) or {
        // Insert new
        app.add_ci_status(CiStatus{
            repo_id:     repo_id
            commit_hash: commit_hash
            branch:      branch
            status:      status
            ci_run_id:   ci_run_id
            created_at:  int(time.now().unix())
            updated_at:  int(time.now().unix())
        })!
        return
    }
    // Update existing
    id := existing.id
    updated := int(time.now().unix())
    sql app.db {
        update CiStatus set status = status, ci_run_id = ci_run_id, updated_at = updated
        where id == id
    }!
}

fn (mut app App) delete_repo_ci_statuses(repo_id int) ! {
    sql app.db {
        delete from CiStatus where repo_id == repo_id
    }!
}

fn (ci &CiStatus) relative_time() string {
    if ci.updated_at == 0 && ci.created_at == 0 {
        return ''
    }
    t := if ci.updated_at > 0 { ci.updated_at } else { ci.created_at }
    return time.unix(t).relative()
}