server/admin/api_team.go

package main

import (
	"encoding/json"
	"fmt"
	"strconv"

	"srs.epita.fr/fic-server/libfic"
)

var ApiTeamsRouting = map[string]DispatchFunction{
	"GET":    listTeam,
	"PUT":    creationTeamMembers,
	"POST":   creationTeam,
	"DELETE": deletionTeam,
}

type uploadedTeam struct {
	Name string
}

type uploadedMember struct {
	Firstname string
	Lastname  string
	Nickname  string
	Company   string
}

func myJSONTeam(t fic.Team) (interface{}, error) {
	ret := map[string]interface{}{}

	ret["team_id"] = t.Id

	exercices := map[string]interface{}{}

	if exos, err := fic.GetExercices(); err != nil {
		return ret, err
	} else {
		for _, e := range exos {
			if t.HasAccess(e) {
				exercice := map[string]interface{}{}
				exercice["statement"] = e.Statement
				exercice["hint"] = e.Hint
				exercice["solved"] = t.HasSolved(e)

				exercices[fmt.Sprintf("%d", e.Id)] = exercice
			}
		}
	}

	ret["exercices"] = exercices

	return ret, nil
}

func nginxGenTeam() (string, error) {
	if teams, err := fic.GetTeams(); err != nil {
		return "", err
	} else {
		ret := ""
		for _, team := range teams {
			ret += fmt.Sprintf("        if ($ssl_client_s_dn ~ \"/C=FR/ST=France/O=Epita/OU=SRS/CN=%s\") { set $team %d; }\n", team.Name, team.Id)
		}

		return ret, nil
	}
}

func listTeam(args []string, body []byte) (interface{}, error) {
	if len(args) == 2 {
		if tid, err := strconv.Atoi(string(args[0])); err != nil {
			return nil, err
		} else if team, err := fic.GetTeam(tid); err != nil {
			return nil, err
		} else if args[1] == "my.json" {
			return myJSONTeam(team)
		}
	} else if len(args) == 1 {
		if args[0] == "nginx" {
			return nginxGenTeam()
		} else if tid, err := strconv.Atoi(string(args[0])); err != nil {
			return nil, err
		} else if team, err := fic.GetTeam(tid); err != nil {
			return nil, err
		} else {
			return team.GetMembers()
		}
	} else if len(args) == 0 {
		// List all teams
		return fic.GetTeams()
	}
	return nil, nil
}

func creationTeam(args []string, body []byte) (interface{}, error) {
	if len(args) == 0 {
		// Create a new team
		var ut uploadedTeam
		if err := json.Unmarshal(body, &ut); err != nil {
			return nil, err
		}

		return fic.CreateTeam(ut.Name)
	} else {
		return nil, nil
	}
}

func creationTeamMembers(args []string, body []byte) (interface{}, error) {
	if len(args) == 1 {
		// List given team
		if tid, err := strconv.Atoi(string(args[0])); err != nil {
			return nil, err
		} else if team, err := fic.GetTeam(tid); err != nil {
			return nil, err
		} else {
			var member uploadedMember
			if err := json.Unmarshal(body, &member); err != nil {
				return nil, err
			}

			team.AddMember(member.Firstname, member.Lastname, member.Nickname, member.Company)

			return team.GetMembers()
		}
	} else if len(args) == 0 {
		// Create a new team
		var members []uploadedMember
		if err := json.Unmarshal(body, &members); err != nil {
			return nil, err
		}

		if team, err := fic.CreateTeam(""); err != nil {
			return nil, err
		} else {
			for _, member := range members {
				if _, err := team.AddMember(member.Firstname, member.Lastname, member.Nickname, member.Company); err != nil {
					return nil, err
				}
			}

			return team, nil
		}
	} else {
		return nil, nil
	}
}

func deletionTeam(args []string, body []byte) (interface{}, error) {
	if len(args) == 1 {
		if tid, err := strconv.Atoi(string(args[0])); err != nil {
			return nil, err
		} else if team, err := fic.GetTeam(tid); err != nil {
			return nil, err
		} else {
			return team.Delete()
		}
	} else {
		return nil, nil
	}
}