135 lines
3.1 KiB
Go
Raw Normal View History

2019-06-03 09:32:43 +00:00
package arn
import (
"errors"
"sync"
"github.com/aerogo/nano"
"github.com/akyoto/color"
)
// AnimeCharacters is a list of characters for an anime.
type AnimeCharacters struct {
AnimeID string `json:"animeId" mainID:"true"`
Items []*AnimeCharacter `json:"items" editable:"true"`
sync.Mutex
}
// Anime returns the anime the characters refer to.
func (characters *AnimeCharacters) Anime() *Anime {
anime, _ := GetAnime(characters.AnimeID)
return anime
}
// Add adds an anime character to the list.
func (characters *AnimeCharacters) Add(animeCharacter *AnimeCharacter) error {
if animeCharacter.CharacterID == "" || animeCharacter.Role == "" {
return errors.New("Empty ID or role")
}
characters.Lock()
characters.Items = append(characters.Items, animeCharacter)
characters.Unlock()
return nil
}
// FindByMapping finds an anime character by the given mapping.
func (characters *AnimeCharacters) FindByMapping(service string, serviceID string) *AnimeCharacter {
characters.Lock()
defer characters.Unlock()
for _, animeCharacter := range characters.Items {
character := animeCharacter.Character()
if character == nil {
color.Red("Anime %s has an incorrect Character ID inside AnimeCharacter: %s", characters.AnimeID, animeCharacter.CharacterID)
continue
}
if character.GetMapping(service) == serviceID {
return animeCharacter
}
}
return nil
}
// Link returns the link for that object.
func (characters *AnimeCharacters) Link() string {
return "/anime/" + characters.AnimeID + "/characters"
}
// String implements the default string serialization.
func (characters *AnimeCharacters) String() string {
return characters.Anime().String()
}
// GetID returns the anime ID.
func (characters *AnimeCharacters) GetID() string {
return characters.AnimeID
}
// TypeName returns the type name.
func (characters *AnimeCharacters) TypeName() string {
return "AnimeCharacters"
}
// Self returns the object itself.
func (characters *AnimeCharacters) Self() Loggable {
return characters
}
// Contains tells you whether the given character ID exists.
func (characters *AnimeCharacters) Contains(characterID string) bool {
characters.Lock()
defer characters.Unlock()
for _, item := range characters.Items {
if item.CharacterID == characterID {
return true
}
}
return false
}
// First gives you the first "count" anime characters.
func (characters *AnimeCharacters) First(count int) []*AnimeCharacter {
characters.Lock()
defer characters.Unlock()
if count > len(characters.Items) {
count = len(characters.Items)
}
return characters.Items[:count]
}
// GetAnimeCharacters ...
func GetAnimeCharacters(animeID string) (*AnimeCharacters, error) {
obj, err := DB.Get("AnimeCharacters", animeID)
if err != nil {
return nil, err
}
return obj.(*AnimeCharacters), nil
}
// StreamAnimeCharacters returns a stream of all anime characters.
func StreamAnimeCharacters() <-chan *AnimeCharacters {
channel := make(chan *AnimeCharacters, nano.ChannelBufferSize)
go func() {
for obj := range DB.All("AnimeCharacters") {
channel <- obj.(*AnimeCharacters)
}
close(channel)
}()
return channel
}