package arn

// Register a list of supported anime relation types.
func init() {
	DataLists["anime-relation-types"] = []*Option{
		{"prequel", HumanReadableAnimeRelation("prequel")},
		{"sequel", HumanReadableAnimeRelation("sequel")},
		{"alternative version", "Alternative version"},
		{"alternative setting", "Alternative setting"},
		{"side story", HumanReadableAnimeRelation("side story")},
		{"parent story", HumanReadableAnimeRelation("parent story")},
		{"full story", HumanReadableAnimeRelation("full story")},
		{"spinoff", HumanReadableAnimeRelation("spinoff")},
		{"summary", HumanReadableAnimeRelation("summary")},
		{"other", HumanReadableAnimeRelation("other")},
	}
}

// AnimeRelation ...
type AnimeRelation struct {
	AnimeID AnimeID `json:"animeId" editable:"true"`
	Type    string  `json:"type" editable:"true" datalist:"anime-relation-types"`
}

// Anime ...
func (relation *AnimeRelation) Anime() *Anime {
	anime, _ := GetAnime(relation.AnimeID)
	return anime
}

// HumanReadableType ...
func (relation *AnimeRelation) HumanReadableType() string {
	return HumanReadableAnimeRelation(relation.Type)
}

// HumanReadableAnimeRelation ...
func HumanReadableAnimeRelation(relationName string) string {
	switch relationName {
	case "prequel":
		return "Prequel"
	case "sequel":
		return "Sequel"
	case "alternative version":
		return "Alternative"
	case "alternative setting":
		return "Alternative"
	case "side story":
		return "Side story"
	case "parent story":
		return "Parent story"
	case "full story":
		return "Full story"
	case "spinoff":
		return "Spin-off"
	case "summary":
		return "Summary"
	case "other":
		return "Other"
	}

	return relationName
}