info.go

package codeforces_client

import (
	"errors"
	"fmt"
	"path/filepath"
	"strings"
)

var ProblemTypes = [...]string{
	"contest",
	"gym",
	"group",
	"acmsguru",
}

type Info struct {
	ProblemType  string `json:"problem_type"`
	ContestID    string `json:"contest_id"`
	GroupID      string `json:"group_id"`
	ProblemID    string `json:"problem_id"`
	SubmissionID string `json:"submission_id"`
	RootPath     string
}

const ErrorNeedProblemID = "you have to specify the Problem ID"
const ErrorNeedContestID = "you have to specify the Contest ID"
const ErrorNeedGymID = "you have to specify the Gym ID"
const ErrorNeedGroupID = "you have to specify the Group ID"
const ErrorNeedSubmissionID = "you have to specify the Submission ID"
const ErrorUnknownType = "unknown type"
const ErrorNotSupportAcmsguru = "not support acmsguru"

func (info *Info) errorContest() (string, error) {
	if info.ProblemType == "gym" {
		return "", errors.New(ErrorNeedGymID)
	}
	return "", errors.New(ErrorNeedContestID)
}

func (info *Info) Hint() string {
	text := strings.ToUpper(info.ProblemType)
	if info.GroupID != "" {
		text = text + " " + info.GroupID
	}
	if info.ProblemType != "acmsguru" && info.ContestID != "" {
		if info.ProblemType != "group" {
			text = text + " " + info.ContestID
		} else {
			text = text + ", contest " + info.ContestID
		}
	}
	if info.ProblemID != "" {
		text = text + ", problem " + info.ProblemID
	}
	if info.SubmissionID != "" {
		text = text + ", submission " + info.SubmissionID
	}
	return text
}

func (info *Info) Path() string {
	path := info.RootPath
	if info.GroupID != "" {
		path = filepath.Join(path, info.GroupID)
	}
	if info.ProblemType != "acmsguru" && info.ContestID != "" {
		path = filepath.Join(path, info.ContestID)
	}
	if info.ProblemID != "" {
		path = filepath.Join(path, strings.ToLower(info.ProblemID))
	}
	return path
}

func (info *Info) PackagePath() (string, error) {
	if info.GroupID == "" {
		return "", errors.New(ErrorNeedGroupID)
	}
	if info.ProblemType == "acmsguru" || info.ContestID == "" {
		return "", errors.New(ErrorNeedContestID)
	}
	if info.ProblemID == "" {
		return "", errors.New(ErrorNeedProblemID)
	}
	path := info.RootPath
	path = filepath.Join(path, info.GroupID)
	path = filepath.Join(path, info.ContestID)
	path = filepath.Join(path, strings.ToLower(info.ProblemID))
	return path, nil
}

func (info *Info) ProblemSetURL(host string) (string, error) {
	if info.ContestID == "" {
		return info.errorContest()
	}
	switch info.ProblemType {
	case "contest":
		return fmt.Sprintf(host+"/contest/%v", info.ContestID), nil
	case "gym":
		return fmt.Sprintf(host+"/gym/%v", info.ContestID), nil
	case "group":
		if info.GroupID == "" {
			return "", errors.New(ErrorNeedGroupID)
		}
		return fmt.Sprintf(host+"/group/%v/contest/%v", info.GroupID, info.ContestID), nil
	case "acmsguru":
		return host + "/problemsets/acmsguru", nil
	}
	return "", errors.New(ErrorUnknownType)
}

func (info *Info) ProblemURL(host string) (string, error) {
	if info.ProblemID == "" {
		return "", errors.New(ErrorNeedProblemID)
	}
	if info.ContestID == "" {
		return info.errorContest()
	}
	switch info.ProblemType {
	case "contest":
		return fmt.Sprintf(host+"/contest/%v/problem/%v", info.ContestID, info.ProblemID), nil
	case "gym":
		return fmt.Sprintf(host+"/gym/%v/problem/%v", info.ContestID, info.ProblemID), nil
	case "group":
		if info.GroupID == "" {
			return "", errors.New(ErrorNeedGroupID)
		}
		return fmt.Sprintf(host+"/group/%v/contest/%v/problem/%v", info.GroupID, info.ContestID, info.ProblemID), nil
	case "acmsguru":
		return fmt.Sprintf(host+"/problemsets/acmsguru/problem/%v/%v", info.ContestID, info.ProblemID), nil
	}
	return "", errors.New(ErrorUnknownType)
}

func (info *Info) MySubmissionURL(host string) (string, error) {
	if info.ContestID == "" {
		return info.errorContest()
	}
	switch info.ProblemType {
	case "contest":
		return fmt.Sprintf(host+"/contest/%v/my", info.ContestID), nil
	case "gym":
		return fmt.Sprintf(host+"/gym/%v/my", info.ContestID), nil
	case "group":
		if info.GroupID == "" {
			return "", errors.New(ErrorNeedGroupID)
		}
		return fmt.Sprintf(host+"/group/%v/contest/%v/my", info.GroupID, info.ContestID), nil
	case "acmsguru":
		return "", errors.New(ErrorNotSupportAcmsguru)
	}
	return "", errors.New(ErrorUnknownType)
}

func (info *Info) SubmissionURL(host string) (string, error) {
	if info.SubmissionID == "" {
		return "", errors.New(ErrorNeedSubmissionID)
	}
	if info.ContestID == "" {
		return info.errorContest()
	}
	switch info.ProblemType {
	case "contest":
		return fmt.Sprintf(host+"/contest/%v/submission/%v", info.ContestID, info.SubmissionID), nil
	case "gym":
		return fmt.Sprintf(host+"/gym/%v/submission/%v", info.ContestID, info.SubmissionID), nil
	case "group":
		if info.GroupID == "" {
			return "", errors.New(ErrorNeedGroupID)
		}
		return fmt.Sprintf(host+"/group/%v/contest/%v/submission/%v", info.GroupID, info.ContestID, info.SubmissionID), nil
	case "acmsguru":
		return fmt.Sprintf(host+"/problemsets/acmsguru/submission/%v/%v", info.ContestID, info.SubmissionID), nil
	}
	return "", errors.New(ErrorUnknownType)
}

func (info *Info) StandingsURL(host string) (string, error) {
	if info.ContestID == "" {
		return info.errorContest()
	}
	switch info.ProblemType {
	case "contest":
		return fmt.Sprintf(host+"/contest/%v/standings", info.ContestID), nil
	case "gym":
		return fmt.Sprintf(host+"/gym/%v/standings", info.ContestID), nil
	case "group":
		if info.GroupID == "" {
			return "", errors.New(ErrorNeedGroupID)
		}
		return fmt.Sprintf(host+"/group/%v/contest/%v/standings", info.GroupID, info.ContestID), nil
	case "acmsguru":
		return host + "/problemsets/acmsguru/standings", nil
	}
	return "", errors.New(ErrorUnknownType)
}

func (info *Info) SubmitURL(host string) (string, error) {
	URL, err := info.ProblemSetURL(host)
	if err != nil {
		return "", err
	}
	return URL + "/submit", nil
}

func (info *Info) OpenURL(host string) (string, error) {
	switch info.ProblemType {
	case "contest":
		if info.ContestID == "" {
			return host + "/contests", nil
		} else if info.ProblemID == "" {
			return fmt.Sprintf(host+"/contest/%v", info.ContestID), nil
		}
		return fmt.Sprintf(host+"/contest/%v/problem/%v", info.ContestID, info.ProblemID), nil
	case "gym":
		if info.ContestID == "" {
			return host + "/gyms", nil
		} else if info.ProblemID == "" {
			return fmt.Sprintf(host+"/gym/%v", info.ContestID), nil
		}
		return fmt.Sprintf(host+"/gym/%v/problem/%v", info.ContestID, info.ProblemID), nil
	case "group":
		if info.GroupID == "" {
			return host + "/groups", nil
		} else if info.ContestID == "" {
			return fmt.Sprintf(host+"/group/%v", info.GroupID), nil
		} else if info.ProblemID == "" {
			return fmt.Sprintf(host+"/group/%v/contest/%v", info.GroupID, info.ContestID), nil
		}
		return fmt.Sprintf(host+"/group/%v/contest/%v/problem/%v", info.GroupID, info.ContestID, info.ProblemID), nil
	case "acmsguru":
		if info.ProblemID == "" {
			return host + "/problemsets/acmsguru/", nil
		}
		return fmt.Sprintf(host+"/problemsets/acmsguru/problem/%v/%v", info.ContestID, info.ProblemID), nil
	}
	return "", errors.New("hmmm I don't know what you want to do~")
}