package fic

import (
	"log"
	"math"
	"time"
)

// UnlockedChallengeDepth is the number of challenges to unlock ahead (0: only the next one, -1: all)
var UnlockedChallengeDepth int

// WchoiceCoefficient is the current coefficient applied on the cost of changing flag into choices
var WChoiceCoefficient = 1.0

// Team represents a group of players, come to solve our challenges.
type Team struct {
	Id         int64  `json:"id"`
	Name       string `json:"name"`
	Color      uint32 `json:"color"`
	Active     bool   `json:"active"`
	ExternalId string `json:"external_id"`
}

func getTeams(filter string) ([]Team, error) {
	if rows, err := DBQuery("SELECT id_team, name, color, active, external_id FROM teams " + filter); 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, &t.Active, &t.ExternalId); err != nil {
				return nil, err
			}
			teams = append(teams, t)
		}
		if err := rows.Err(); err != nil {
			return nil, err
		}

		return teams, nil
	}
}

// GetTeams returns a list of registered Team from the database.
func GetTeams() ([]Team, error) {
	return getTeams("")
}

// GetActiveTeams returns a list of registered Team from the database, limited to team to generate.
func GetActiveTeams() ([]Team, error) {
	return getTeams("WHERE active = 1")
}

// GetTeam retrieves a Team from its identifier.
func GetTeam(id int64) (Team, error) {
	var t Team
	if err := DBQueryRow("SELECT id_team, name, color, active, external_id FROM teams WHERE id_team = ?", id).Scan(&t.Id, &t.Name, &t.Color, &t.Active, &t.ExternalId); err != nil {
		return t, err
	}

	return t, nil
}

// GetTeamBySerial retrieves a Team from one of its associated certificates.
func GetTeamBySerial(serial int64) (Team, error) {
	var t Team
	if err := DBQueryRow("SELECT T.id_team, T.name, T.color, T.active, T.external_id FROM certificates C INNER JOIN teams T ON T.id_team = C.id_team WHERE id_cert = ?", serial).Scan(&t.Id, &t.Name, &t.Color, &t.Active, &t.ExternalId); err != nil {
		return t, err
	}

	return t, nil
}

// CreateTeam creates and fills a new struct Team and registers it into the database.
func CreateTeam(name string, color uint32, externalId string) (Team, error) {
	if res, err := DBExec("INSERT INTO teams (name, color, external_id) VALUES (?, ?, ?)", name, color, externalId); err != nil {
		return Team{}, err
	} else if tid, err := res.LastInsertId(); err != nil {
		return Team{}, err
	} else {
		return Team{tid, name, color, true, ""}, nil
	}
}

// Update applies modifications back to the database.
func (t Team) Update() (int64, error) {
	if res, err := DBExec("UPDATE teams SET name = ?, color = ?, active = ?, external_id = ? WHERE id_team = ?", t.Name, t.Color, t.Active, t.ExternalId, t.Id); err != nil {
		return 0, err
	} else if nb, err := res.RowsAffected(); err != nil {
		return 0, err
	} else {
		return nb, err
	}
}

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

// HasAccess checks if the Team has access to the given challenge.
func (t Team) HasAccess(e Exercice) bool {
	if UnlockedChallengeDepth < 0 {
		return true
	}

	for i := UnlockedChallengeDepth; i >= 0; i-- {
		// An exercice without dependency is accessible
		if e.Depend == nil {
			return true
		}

		ed := Exercice{}
		ed.Id = *e.Depend
		s, _ := t.HasSolved(ed)
		if s {
			return s
		}

		// Prepare next iteration
		var err error
		e, err = GetExercice(ed.Id)
		if err != nil {
			return false
		}
	}
	return false
}

// CanDownload checks if the Team has access to the given file.
func (t Team) CanDownload(f EFile) bool {
	if deps, err := f.GetDepends(); err != nil {
		log.Printf("Unable to retrieve file dependencies: %s\n", err)
		return false
	} else {
		res := true

		for _, dep := range deps {
			if t.HasPartiallySolved(dep) == nil {
				res = false
				break
			}
		}

		return res
	}
}

// CanSeeHint checks if the Team has access to the given hint.
func (t Team) CanSeeHint(h EHint) bool {
	if deps, err := h.GetDepends(); err != nil {
		log.Printf("Unable to retrieve flag dependencies: %s\n", err)
		return false
	} else {
		res := true

		for _, dep := range deps {
			if t.HasPartiallySolved(dep) == nil {
				res = false
				break
			}
		}

		return res
	}
}

// CanSeeFlag checks if the Team has access to the given flag.
func (t Team) CanSeeFlag(k Flag) bool {
	if deps, err := k.GetDepends(); err != nil {
		log.Printf("Unable to retrieve flag dependencies: %s\n", err)
		return false
	} else {
		res := true

		for _, dep := range deps {
			if t.HasPartiallySolved(dep) == nil {
				res = false
				break
			}
		}

		return res
	}
}

// NbTry retrieves the number of attempts made by the Team to the given challenge.
func NbTry(t *Team, e Exercice) int {
	tries_table := "exercice_tries"
	if SubmissionUniqueness {
		tries_table = "exercice_distinct_tries"
	}

	var cnt *int

	if t != nil {
		DBQueryRow("SELECT COUNT(*) FROM "+tries_table+" WHERE id_team = ? AND id_exercice = ?", t.Id, e.Id).Scan(&cnt)
	} else {
		DBQueryRow("SELECT COUNT(*) FROM "+tries_table+" WHERE id_exercice = ?", e.Id).Scan(&cnt)
	}

	if cnt == nil {
		return 0
	} else {
		return *cnt
	}
}

// HasHint checks if the Team has revealed the given Hint.
func (t Team) HasHint(h EHint) bool {
	var tm *time.Time
	DBQueryRow("SELECT MIN(time) FROM team_hints WHERE id_team = ? AND id_hint = ?", t.Id, h.Id).Scan(&tm)
	return tm != nil
}

// OpenHint registers to the database that the Team has now revealed.
func (t Team) OpenHint(h EHint) error {
	_, err := DBExec("INSERT INTO team_hints (id_team, id_hint, time, coefficient) VALUES (?, ?, ?, ?)", t.Id, h.Id, time.Now(), math.Max(HintCoefficient, 0.0))
	return err
}

// SeeChoices checks if the Team has revealed the given choices.
func (t Team) SeeChoices(k FlagKey) bool {
	var tm *time.Time
	DBQueryRow("SELECT MIN(time) FROM team_wchoices WHERE id_team = ? AND id_flag = ?", t.Id, k.Id).Scan(&tm)
	return tm != nil
}

// DisplayChoices registers to the database that the Team has now revealed.
func (t Team) DisplayChoices(k FlagKey) error {
	_, err := DBExec("INSERT INTO team_wchoices (id_team, id_flag, time, coefficient) VALUES (?, ?, ?, ?)", t.Id, k.Id, time.Now(), math.Max(WChoiceCoefficient, 0.0))
	return err
}

// CountTries gets the amount of attempts made by the Team and retrieves the time of the latest attempt.
func (t Team) CountTries(e Exercice) (int64, time.Time) {
	var nb *int64
	var tm *time.Time
	if DBQueryRow("SELECT COUNT(id_exercice), MAX(time) FROM exercice_tries WHERE id_team = ? AND id_exercice = ?", t.Id, e.Id).Scan(&nb, &tm); tm == nil {
		return 0, time.Unix(0, 0)
	} else if nb == nil {
		return 0, *tm
	} else {
		return *nb, *tm
	}
}

// LastTryDist retrieves the distance to the correct answers, for the given challenge.
// The distance is the number of bad responses given in differents MCQs.
func (t Team) LastTryDist(e Exercice) int64 {
	var nb *int64
	if DBQueryRow("SELECT nbdiff FROM exercice_tries WHERE id_team = ? AND id_exercice = ? ORDER BY time DESC LIMIT 1", t.Id, e.Id).Scan(&nb); nb == nil {
		return 0
	} else {
		return *nb
	}
}

// HasSolved checks if the Team already has validated the given challenge.
// Note that the function also returns the effective validation timestamp.
func (t Team) HasSolved(e Exercice) (bool, time.Time) {
	var tm *time.Time
	if DBQueryRow("SELECT MIN(time) FROM exercice_solved WHERE id_team = ? AND id_exercice = ?", t.Id, e.Id).Scan(&tm); tm == nil {
		return false, time.Unix(0, 0)
	} else {
		return true, *tm
	}
}

// GetSolvedRank returns the number of teams that solved the challenge before the Team.
func (t Team) GetSolvedRank(e Exercice) (nb int64, err error) {
	if rows, errr := DBQuery("SELECT id_team FROM exercice_solved WHERE id_exercice = ? ORDER BY time ASC", e.Id); errr != nil {
		return nb, errr
	} else {
		defer rows.Close()

		for rows.Next() {
			var tid int64
			if err = rows.Scan(&tid); err != nil {
				return
			}

			nb += 1
			if t.Id == tid {
				break
			}
		}
		return
	}
}

// HasPartiallySolved checks if the Team already has unlocked the given flag and returns the validation's timestamp.
func (t Team) HasPartiallySolved(f Flag) (tm *time.Time) {
	if k, ok := f.(FlagKey); ok {
		DBQueryRow("SELECT MIN(time) FROM flag_found WHERE id_team = ? AND id_flag = ?", t.Id, k.Id).Scan(&tm)
	} else if m, ok := f.(MCQ); ok {
		DBQueryRow("SELECT MIN(time) FROM mcq_found WHERE id_team = ? AND id_mcq = ?", t.Id, m.Id).Scan(&tm)
	} else {
		log.Fatal("Unknown flag type")
	}
	return
}