package fic

import (
	"time"
)

type Team struct {
	Id    int64  `json:"id"`
	Name  string `json:"name"`
	Color int64  `json:"color"`
}

func GetTeams() ([]Team, error) {
	if rows, err := DBQuery("SELECT id_team, name, color FROM teams"); err != nil {
		return nil, err
	} else {
		defer rows.Close()

		var teams = make([]Team, 0)
		for rows.Next() {
			var t Team
			if err := rows.Scan(&t.Id, &t.Name, &t.Color); err != nil {
				return nil, err
			}
			teams = append(teams, t)
		}
		if err := rows.Err(); err != nil {
			return nil, err
		}

		return teams, nil
	}
}

func GetTeam(id int) (Team, error) {
	var t Team
	if err := DBQueryRow("SELECT id_team, name, color FROM teams WHERE id_team = ?", id).Scan(&t.Id, &t.Name, &t.Color); err != nil {
		return t, err
	}

	return t, nil
}

func CreateTeam(name string, color int64) (Team, error) {
	if res, err := DBExec("INSERT INTO teams (name, color) VALUES (?, ?)", name, color); err != nil {
		return Team{}, err
	} else if tid, err := res.LastInsertId(); err != nil {
		return Team{}, err
	} else {
		return Team{tid, name, color}, nil
	}
}

func (t Team) Update() (int64, error) {
	if res, err := DBExec("UPDATE teams SET name = ?, color = ? WHERE id_team = ?", t.Name, t.Color, t.Id); err != nil {
		return 0, err
	} else if nb, err := res.RowsAffected(); err != nil {
		return 0, err
	} else {
		return nb, err
	}
}

func (t Team) Delete() (int64, error) {
	if res, err := DBExec("DELETE FROM team_members WHERE id_team = ?; DELETE FROM teams WHERE id_team = ?", t.Id, t.Id); err != nil {
		return 0, err
	} else if nb, err := res.RowsAffected(); err != nil {
		return 0, err
	} else {
		return nb, err
	}
}

func (t Team) HasAccess(e Exercice) bool {
	if e.Depend == nil {
		return true
	} else {
		ed := Exercice{}
		ed.Id = *e.Depend
		s, _, _ := t.HasSolved(ed)
		return s
	}
}

func (t Team) HasSolved(e Exercice) (bool, time.Time, int64) {
	var nb int64
	var tm time.Time
	if err := DBQueryRow("SELECT time FROM exercice_solved WHERE id_team = ? AND id_exercice = ?", t.Id, e.Id).Scan(&tm); err != nil {
		return false, time.Now(), 0
	} else if err := DBQueryRow("SELECT COUNT(id_exercice) FROM exercice_solved WHERE id_exercice = ? AND time < ?", e.Id, tm).Scan(&nb); err != nil {
		return true, tm, 0
	} else {
		return true, tm, nb
	}
}