// Copyright (c) 2019-2026 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by a GPL license that can be found in the LICENSE file.
module main

import time

struct Org {
    id            int    @[primary; sql: serial]
    name          string @[unique]
    contact_email string
    kind          string
    created_at    time.Time
    created_by    int
}

struct OrgMember {
    id      int @[primary; sql: serial]
    org_id  int @[unique: 'org_member']
    user_id int @[unique: 'org_member']
    role    string
}

pub fn (mut app App) add_org(name string, contact_email string, kind string, created_by int) !int {
    new_org := Org{
        name:          name
        contact_email: contact_email
        kind:          kind
        created_at:    time.now()
        created_by:    created_by
    }
    sql app.db {
        insert new_org into Org
    }!
    row := app.get_org_by_name(name) or { return error('failed to load newly created org') }
    return row.id
}

pub fn (app App) get_org_by_name(name string) ?Org {
    rows := sql app.db {
        select from Org where name == name limit 1
    } or { [] }
    if rows.len == 0 {
        return none
    }
    return rows.first()
}

pub fn (app App) get_org_by_id(id int) ?Org {
    rows := sql app.db {
        select from Org where id == id limit 1
    } or { [] }
    if rows.len == 0 {
        return none
    }
    return rows.first()
}

pub fn (mut app App) add_org_member(org_id int, user_id int, role string) ! {
    member := OrgMember{
        org_id:  org_id
        user_id: user_id
        role:    role
    }
    sql app.db {
        insert member into OrgMember
    }!
}

pub fn (app App) find_orgs_for_user(user_id int) []Org {
    members := sql app.db {
        select from OrgMember where user_id == user_id
    } or { [] }
    mut orgs := []Org{cap: members.len}
    for m in members {
        org := app.get_org_by_id(m.org_id) or { continue }
        orgs << org
    }
    return orgs
}

pub fn (app App) is_org_member(org_id int, user_id int) bool {
    count := sql app.db {
        select count from OrgMember where org_id == org_id && user_id == user_id
    } or { 0 }
    return count > 0
}