106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
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) GetPoints() (int64, error) {
|
|
var nb *int64
|
|
err := DBQueryRow("SELECT SUM(E.gain) FROM exercice_solved S INNER JOIN exercices E ON E.id_exercice = S.id_exercice WHERE id_team = ?", t.Id).Scan(&nb)
|
|
if nb != nil {
|
|
return *nb, err
|
|
} else {
|
|
return 0, 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.Unix()).Scan(&nb); err != nil {
|
|
return true, tm, 0
|
|
} else {
|
|
return true, tm, nb + 1
|
|
}
|
|
}
|