server/admin/api_team.go

package main

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

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 nginxGenTeam() (string, error) {
	if teams, err := 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) == 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 := GetTeam(tid); err != nil {
			return nil, err
		} else {
			return team.GetMembers()
		}
	} else if len(args) == 0 {
		// List all teams
		return GetTeams()
	} else {
		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 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 := 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 := 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 := GetTeam(tid); err != nil {
			return nil, err
		} else {
			return team.Delete()
		}
	} else {
		return nil, nil
	}
}