package models

import "time"

type Expense struct {
    ExpID int
    ExpDesc string
    ExpDate time.Time
    ExpPayer int
    ExpCat int
    ExpValue string
}

func GetExpenses() ([]*Expense, error) {
    rows, err := db.Query("SELECT * FROM expenses")
    if err != nil {
        return nil, err
    }
    defer rows.Close()

    exps := make([]*Expense, 0)
    for rows.Next() {
        exp := new(Expense)
        err := rows.Scan(&exp.ExpID, &exp.ExpDesc, &exp.ExpDate, &exp.ExpPayer, &exp.ExpCat, &exp.ExpValue)
        if err != nil {
            return nil, err
        }
        exps = append(exps, exp)
    }
    if err = rows.Err(); err != nil {
        return nil, err
    }
    return exps, nil
}

func AddExpense(desc, date, catdesc, pplname, value string) error {
    var cat int
    sqlStat := "SELECT catid FROM categories WHERE catdesc = $1"
    err := db.QueryRow(sqlStat, catdesc).Scan(&cat)
    if err != nil {
        return err
    }

    var payer int
    sqlStat = "SELECT peopid FROM people WHERE peopname = $1"
    err = db.QueryRow(sqlStat, pplname).Scan(&payer)
    if err != nil {
        return err
    }

    var expid int
    sqlStat = "INSERT INTO expenses (expdesc, expdate, exppayer, expcat, expvalue) VALUES ($1, $2, $3, $4, $5) RETURNING expid"
    err = db.QueryRow(sqlStat, desc, date, payer, cat, value).Scan(&expid)
    if err != nil {
        panic(err)
    }
    return nil
}

func DeleteExpense(id int) error {
    var expid int
    sqlStat := "DELETE FROM expenses WHERE expid = $1 RETURNING expid"
    err := db.QueryRow(sqlStat, id).Scan(&expid)
    if err != nil {
        return err
    }
    return nil
}