781 lines
18 KiB
Go
Raw Normal View History

2019-06-03 09:32:43 +00:00
package arn
import (
"errors"
"fmt"
"sort"
"strconv"
"strings"
"time"
"github.com/aerogo/nano"
"github.com/akyoto/color"
"github.com/animenotifier/kitsu"
2019-10-19 14:57:58 +00:00
"github.com/animenotifier/notify.moe/arn/validate"
2019-06-03 09:32:43 +00:00
"github.com/animenotifier/shoboi"
2019-10-19 14:57:58 +00:00
"github.com/animenotifier/twist"
2019-06-03 09:32:43 +00:00
)
2019-06-07 01:50:38 +00:00
// AnimeID represents an anime ID.
2019-11-18 05:01:13 +00:00
type AnimeID = ID
2019-06-07 01:50:38 +00:00
2019-06-03 09:32:43 +00:00
// Anime represents an anime.
type Anime struct {
ID AnimeID `json:"id" primary:"true"`
2019-06-03 09:32:43 +00:00
Type string `json:"type" editable:"true" datalist:"anime-types"`
2019-10-16 05:44:53 +00:00
Title MediaTitle `json:"title" editable:"true"`
2019-06-03 09:32:43 +00:00
Summary string `json:"summary" editable:"true" type:"textarea"`
Status string `json:"status" editable:"true" datalist:"anime-status"`
Genres []string `json:"genres" editable:"true"`
StartDate string `json:"startDate" editable:"true"`
EndDate string `json:"endDate" editable:"true"`
EpisodeCount int `json:"episodeCount" editable:"true"`
EpisodeLength int `json:"episodeLength" editable:"true"`
Source string `json:"source" editable:"true" datalist:"anime-sources"`
2019-06-04 03:16:27 +00:00
Image Image `json:"image"`
2019-06-03 09:32:43 +00:00
FirstChannel string `json:"firstChannel"`
2019-10-16 05:44:53 +00:00
Rating AnimeRating `json:"rating"`
Popularity AnimePopularity `json:"popularity"`
2019-06-03 09:32:43 +00:00
Trailers []*ExternalMedia `json:"trailers" editable:"true"`
2019-10-16 05:44:53 +00:00
EpisodeIDs []string `json:"episodes"`
2019-06-03 09:32:43 +00:00
// Mixins
hasMappings
hasPosts
hasLikes
hasCreator
hasEditor
hasDraft
// Company IDs
2019-11-18 05:01:13 +00:00
StudioIDs []CompanyID `json:"studios" editable:"true"`
ProducerIDs []CompanyID `json:"producers" editable:"true"`
LicensorIDs []CompanyID `json:"licensors" editable:"true"`
2019-06-03 09:32:43 +00:00
// Links to external websites
Links []*Link `json:"links" editable:"true"`
}
// NewAnime creates a new anime.
func NewAnime() *Anime {
2019-11-18 06:13:51 +00:00
anime := &Anime{}
2019-10-16 05:44:53 +00:00
return anime.init()
}
// init is the constructor for Anime.
func (anime *Anime) init() *Anime {
anime.ID = GenerateID("Anime")
anime.Type = "tv"
anime.Status = "upcoming"
anime.Trailers = []*ExternalMedia{}
anime.Mappings = []*Mapping{}
anime.Created = DateTimeUTC()
return anime
2019-06-03 09:32:43 +00:00
}
// GetAnime gets the anime with the given ID.
2019-06-07 01:50:38 +00:00
func GetAnime(id AnimeID) (*Anime, error) {
2019-06-03 09:32:43 +00:00
obj, err := DB.Get("Anime", id)
if err != nil {
return nil, err
}
return obj.(*Anime), nil
}
// TitleByUser returns the preferred title for the given user.
func (anime *Anime) TitleByUser(user *User) string {
return anime.Title.ByUser(user)
}
2019-10-16 05:44:53 +00:00
// Publish publishes the anime draft.
func (anime *Anime) Publish() error {
// No type
if anime.Type == "" {
return errors.New("No type")
}
// No name
if anime.Title.Canonical == "" {
return errors.New("No canonical anime name")
}
// No status
if anime.Status == "" {
return errors.New("No status")
}
// No genres
if len(anime.Genres) == 0 {
return errors.New("No genres")
}
// No image
if !anime.HasImage() {
return errors.New("No anime image")
}
return publish(anime)
}
// Unpublish turns the anime into a draft.
func (anime *Anime) Unpublish() error {
return unpublish(anime)
}
2019-06-03 09:32:43 +00:00
// AddStudio adds the company ID to the studio ID list if it doesn't exist already.
func (anime *Anime) AddStudio(companyID string) {
// Is the ID valid?
if companyID == "" {
return
}
// If it already exists we don't need to add it
for _, id := range anime.StudioIDs {
if id == companyID {
return
}
}
anime.StudioIDs = append(anime.StudioIDs, companyID)
}
// AddProducer adds the company ID to the producer ID list if it doesn't exist already.
func (anime *Anime) AddProducer(companyID string) {
// Is the ID valid?
if companyID == "" {
return
}
// If it already exists we don't need to add it
for _, id := range anime.ProducerIDs {
if id == companyID {
return
}
}
anime.ProducerIDs = append(anime.ProducerIDs, companyID)
}
// AddLicensor adds the company ID to the licensor ID list if it doesn't exist already.
func (anime *Anime) AddLicensor(companyID string) {
// Is the ID valid?
if companyID == "" {
return
}
// If it already exists we don't need to add it
for _, id := range anime.LicensorIDs {
if id == companyID {
return
}
}
anime.LicensorIDs = append(anime.LicensorIDs, companyID)
}
// Studios returns the list of studios for this anime.
func (anime *Anime) Studios() []*Company {
companies := []*Company{}
for _, obj := range DB.GetMany("Company", anime.StudioIDs) {
if obj == nil {
continue
}
companies = append(companies, obj.(*Company))
}
return companies
}
// Producers returns the list of producers for this anime.
func (anime *Anime) Producers() []*Company {
companies := []*Company{}
for _, obj := range DB.GetMany("Company", anime.ProducerIDs) {
if obj == nil {
continue
}
companies = append(companies, obj.(*Company))
}
return companies
}
// Licensors returns the list of licensors for this anime.
func (anime *Anime) Licensors() []*Company {
companies := []*Company{}
for _, obj := range DB.GetMany("Company", anime.LicensorIDs) {
if obj == nil {
continue
}
companies = append(companies, obj.(*Company))
}
return companies
}
// Prequels returns the list of prequels for that anime.
func (anime *Anime) Prequels() []*Anime {
prequels := []*Anime{}
relations := anime.Relations()
relations.Lock()
defer relations.Unlock()
for _, relation := range relations.Items {
if relation.Type != "prequel" {
continue
}
prequel := relation.Anime()
if prequel == nil {
color.Red("Anime %s has invalid anime relation ID %s", anime.ID, relation.AnimeID)
continue
}
prequels = append(prequels, prequel)
}
return prequels
}
2019-09-10 00:49:51 +00:00
// ImageLink requires a size parameter and returns a link to the image in the given size.
2019-06-03 09:32:43 +00:00
func (anime *Anime) ImageLink(size string) string {
extension := ".jpg"
if size == "original" {
extension = anime.Image.Extension
}
return fmt.Sprintf("//%s/images/anime/%s/%s%s?%v", MediaHost, size, anime.ID, extension, anime.Image.LastModified)
}
// HasImage returns whether the anime has an image or not.
func (anime *Anime) HasImage() bool {
return anime.Image.Extension != "" && anime.Image.Width > 0
}
// AverageColor returns the average color of the image.
func (anime *Anime) AverageColor() string {
color := anime.Image.AverageColor
if color.Hue == 0 && color.Saturation == 0 && color.Lightness == 0 {
return ""
}
return color.String()
}
// Season returns the season the anime started airing in.
func (anime *Anime) Season() string {
if !validate.Date(anime.StartDate) && !validate.YearMonth(anime.StartDate) {
2019-06-03 09:32:43 +00:00
return ""
}
return DateToSeason(anime.StartDateTime())
}
2019-09-10 00:49:51 +00:00
// Characters returns the anime characters for this anime.
2019-06-03 09:32:43 +00:00
func (anime *Anime) Characters() *AnimeCharacters {
characters, _ := GetAnimeCharacters(anime.ID)
return characters
}
// Relations ...
func (anime *Anime) Relations() *AnimeRelations {
relations, _ := GetAnimeRelations(anime.ID)
return relations
}
// Link returns the URI to the anime page.
func (anime *Anime) Link() string {
return "/anime/" + anime.ID
}
// StartDateTime returns the start date as a time object.
func (anime *Anime) StartDateTime() time.Time {
2019-11-18 06:13:51 +00:00
format := validate.DateFormat
2019-06-03 09:32:43 +00:00
switch {
2019-11-18 06:13:51 +00:00
case len(anime.StartDate) >= len(validate.DateFormat):
2019-06-03 09:32:43 +00:00
// ...
case len(anime.StartDate) >= len("2006-01"):
format = "2006-01"
case len(anime.StartDate) >= len("2006"):
format = "2006"
}
t, _ := time.Parse(format, anime.StartDate)
return t
}
// EndDateTime returns the end date as a time object.
func (anime *Anime) EndDateTime() time.Time {
2019-11-18 06:13:51 +00:00
format := validate.DateFormat
2019-06-03 09:32:43 +00:00
switch {
2019-11-18 06:13:51 +00:00
case len(anime.EndDate) >= len(validate.DateFormat):
2019-06-03 09:32:43 +00:00
// ...
case len(anime.EndDate) >= len("2006-01"):
format = "2006-01"
case len(anime.EndDate) >= len("2006"):
format = "2006"
}
t, _ := time.Parse(format, anime.EndDate)
return t
}
2019-08-28 08:06:42 +00:00
// Episodes returns the anime episodes.
func (anime *Anime) Episodes() EpisodeList {
objects := DB.GetMany("Episode", anime.EpisodeIDs)
episodes := make([]*Episode, 0, len(anime.EpisodeIDs))
2019-06-03 09:32:43 +00:00
2019-08-28 08:06:42 +00:00
for _, obj := range objects {
if obj == nil {
continue
}
episodes = append(episodes, obj.(*Episode))
2019-06-03 09:32:43 +00:00
}
2019-08-28 08:06:42 +00:00
return episodes
2019-06-03 09:32:43 +00:00
}
// UsersWatchingOrPlanned returns a list of users who are watching the anime right now.
func (anime *Anime) UsersWatchingOrPlanned() []*User {
users := FilterUsers(func(user *User) bool {
item := user.AnimeList().Find(anime.ID)
if item == nil {
return false
}
return item.Status == AnimeListStatusWatching || item.Status == AnimeListStatusPlanned
})
return users
}
// RefreshEpisodes will refresh the episode data.
func (anime *Anime) RefreshEpisodes() error {
// Fetch episodes
episodes := anime.Episodes()
// Save number of available episodes for comparison later
2019-11-18 06:19:04 +00:00
oldAvailableCount := episodes.AvailableCount()
2019-06-03 09:32:43 +00:00
// Shoboi
shoboiEpisodes, err := anime.ShoboiEpisodes()
if err != nil {
2019-06-05 06:45:54 +00:00
return err
2019-06-03 09:32:43 +00:00
}
2019-08-28 09:37:09 +00:00
episodes = episodes.Merge(shoboiEpisodes)
2019-06-03 09:32:43 +00:00
2019-11-28 01:01:51 +00:00
// // AnimeTwist
// twistEpisodes, err := anime.TwistEpisodes()
2019-06-03 09:32:43 +00:00
2019-11-28 01:01:51 +00:00
// if err != nil {
// return err
// }
2019-06-03 09:32:43 +00:00
2019-11-28 01:01:51 +00:00
// episodes = episodes.Merge(twistEpisodes)
2019-06-03 09:32:43 +00:00
// Count number of available episodes
2019-11-18 06:19:04 +00:00
newAvailableCount := episodes.AvailableCount()
if anime.Status != "finished" && newAvailableCount > oldAvailableCount {
// New episodes have been released.
// Notify all users who are watching the anime.
go func() {
for _, user := range anime.UsersWatchingOrPlanned() {
if !user.Settings().Notification.AnimeEpisodeReleases {
continue
}
user.SendNotification(&PushNotification{
Title: anime.Title.ByUser(user),
Message: "Episode " + strconv.Itoa(newAvailableCount) + " has been released!",
Icon: anime.ImageLink("medium"),
Link: "https://notify.moe" + anime.Link(),
Type: NotificationTypeAnimeEpisode,
})
}
}()
}
2019-06-03 09:32:43 +00:00
// Number remaining episodes
startNumber := 0
2019-08-28 08:06:42 +00:00
for _, episode := range episodes {
2019-06-03 09:32:43 +00:00
if episode.Number != -1 {
startNumber = episode.Number
continue
}
startNumber++
episode.Number = startNumber
}
// Guess airing dates
oneWeek := 7 * 24 * time.Hour
lastAiringDate := ""
timeDifference := oneWeek
2019-08-28 08:06:42 +00:00
for _, episode := range episodes {
2019-06-03 09:32:43 +00:00
if validate.DateTime(episode.AiringDate.Start) {
if lastAiringDate != "" {
a, _ := time.Parse(time.RFC3339, lastAiringDate)
b, _ := time.Parse(time.RFC3339, episode.AiringDate.Start)
timeDifference = b.Sub(a)
// Cap time difference at one week
if timeDifference > oneWeek {
timeDifference = oneWeek
}
}
lastAiringDate = episode.AiringDate.Start
continue
}
// Add 1 week to the last known airing date
nextAiringDate, _ := time.Parse(time.RFC3339, lastAiringDate)
nextAiringDate = nextAiringDate.Add(timeDifference)
// Guess start and end time
episode.AiringDate.Start = nextAiringDate.Format(time.RFC3339)
episode.AiringDate.End = nextAiringDate.Add(30 * time.Minute).Format(time.RFC3339)
// Set this date as the new last known airing date
lastAiringDate = episode.AiringDate.Start
}
2019-08-28 08:06:42 +00:00
// Save new episode ID list
episodeIDs := make([]string, len(episodes))
2019-08-28 09:37:09 +00:00
for index, episode := range episodes {
episodeIDs[index] = episode.ID
episode.AnimeID = anime.ID
episode.Save()
2019-08-28 08:06:42 +00:00
}
anime.EpisodeIDs = episodeIDs
anime.Save()
2019-06-03 09:32:43 +00:00
return nil
}
// ShoboiEpisodes returns a slice of episode info from cal.syoboi.jp.
2019-08-28 08:06:42 +00:00
func (anime *Anime) ShoboiEpisodes() (EpisodeList, error) {
2019-06-03 09:32:43 +00:00
shoboiID := anime.GetMapping("shoboi/anime")
if shoboiID == "" {
return nil, errors.New("Missing shoboi/anime mapping")
}
shoboiAnime, err := shoboi.GetAnime(shoboiID)
if err != nil {
return nil, err
}
2019-08-28 08:06:42 +00:00
arnEpisodes := []*Episode{}
2019-06-03 09:32:43 +00:00
shoboiEpisodes := shoboiAnime.Episodes()
for _, shoboiEpisode := range shoboiEpisodes {
episode := NewAnimeEpisode()
episode.Number = shoboiEpisode.Number
episode.Title.Japanese = shoboiEpisode.TitleJapanese
// Try to get airing date
airingDate := shoboiEpisode.AiringDate
if airingDate != nil {
episode.AiringDate.Start = airingDate.Start
episode.AiringDate.End = airingDate.End
} else {
episode.AiringDate.Start = ""
episode.AiringDate.End = ""
}
arnEpisodes = append(arnEpisodes, episode)
}
return arnEpisodes, nil
}
// TwistEpisodes returns a slice of episode info from twist.moe.
2019-08-28 08:06:42 +00:00
func (anime *Anime) TwistEpisodes() (EpisodeList, error) {
2019-06-03 09:32:43 +00:00
idList, err := GetIDList("animetwist index")
if err != nil {
return nil, err
}
// Does the index contain the ID?
kitsuID := anime.GetMapping("kitsu/anime")
found := false
for _, id := range idList {
if id == kitsuID {
found = true
break
}
}
// If the ID is not the index we don't need to query the feed
if !found {
return nil, errors.New("Not available in twist.moe anime index")
}
// Get twist.moe feed
feed, err := twist.GetFeedByKitsuID(kitsuID)
if err != nil {
return nil, err
}
episodes := feed.Episodes
// Sort by episode number
sort.Slice(episodes, func(a, b int) bool {
return episodes[a].Number < episodes[b].Number
})
2019-08-28 08:06:42 +00:00
arnEpisodes := []*Episode{}
2019-06-03 09:32:43 +00:00
for _, episode := range episodes {
arnEpisode := NewAnimeEpisode()
arnEpisode.Number = episode.Number
arnEpisode.Links = map[string]string{
"twist.moe": strings.Replace(episode.Link, "https://test.twist.moe/", "https://twist.moe/", 1),
}
arnEpisodes = append(arnEpisodes, arnEpisode)
}
return arnEpisodes, nil
}
// UpcomingEpisodes ...
func (anime *Anime) UpcomingEpisodes() []*UpcomingEpisode {
var upcomingEpisodes []*UpcomingEpisode
now := time.Now().UTC().Format(time.RFC3339)
2019-08-28 08:06:42 +00:00
for _, episode := range anime.Episodes() {
2019-06-03 09:32:43 +00:00
if episode.AiringDate.Start > now && validate.DateTime(episode.AiringDate.Start) {
upcomingEpisodes = append(upcomingEpisodes, &UpcomingEpisode{
Anime: anime,
Episode: episode,
})
}
}
return upcomingEpisodes
}
// UpcomingEpisode ...
func (anime *Anime) UpcomingEpisode() *UpcomingEpisode {
now := time.Now().UTC().Format(time.RFC3339)
2019-08-28 08:06:42 +00:00
for _, episode := range anime.Episodes() {
2019-06-03 09:32:43 +00:00
if episode.AiringDate.Start > now && validate.DateTime(episode.AiringDate.Start) {
return &UpcomingEpisode{
Anime: anime,
Episode: episode,
}
}
}
return nil
}
2019-09-10 00:49:51 +00:00
// EpisodeCountString formats the episode count and displays
// a question mark when the number of episodes is unknown.
2019-06-03 09:32:43 +00:00
func (anime *Anime) EpisodeCountString() string {
if anime.EpisodeCount == 0 {
return "?"
}
return strconv.Itoa(anime.EpisodeCount)
}
// ImportKitsuMapping imports the given Kitsu mapping.
func (anime *Anime) ImportKitsuMapping(mapping *kitsu.Mapping) {
switch mapping.Attributes.ExternalSite {
case "myanimelist/anime":
anime.SetMapping("myanimelist/anime", mapping.Attributes.ExternalID)
case "anidb":
anime.SetMapping("anidb/anime", mapping.Attributes.ExternalID)
case "trakt":
anime.SetMapping("trakt/anime", mapping.Attributes.ExternalID)
// case "hulu":
// anime.SetMapping("hulu/anime", mapping.Attributes.ExternalID)
case "anilist":
externalID := mapping.Attributes.ExternalID
externalID = strings.TrimPrefix(externalID, "anime/")
anime.SetMapping("anilist/anime", externalID)
case "thetvdb", "thetvdb/series":
externalID := mapping.Attributes.ExternalID
slashPos := strings.Index(externalID, "/")
if slashPos != -1 {
externalID = externalID[:slashPos]
}
anime.SetMapping("thetvdb/anime", externalID)
case "thetvdb/season":
// Ignore
default:
color.Yellow("Unknown mapping: %s %s", mapping.Attributes.ExternalSite, mapping.Attributes.ExternalID)
}
}
// TypeHumanReadable ...
func (anime *Anime) TypeHumanReadable() string {
switch anime.Type {
case "tv":
return "TV"
case "movie":
return "Movie"
case "ova":
return "OVA"
case "ona":
return "ONA"
case "special":
return "Special"
case "music":
return "Music"
default:
return anime.Type
}
}
// StatusHumanReadable ...
func (anime *Anime) StatusHumanReadable() string {
switch anime.Status {
case "finished":
return "Finished"
case "current":
return "Airing"
case "upcoming":
return "Upcoming"
case "tba":
return "To be announced"
default:
return anime.Status
}
}
// CalculatedStatus returns the status of the anime inferred by the start and end date.
func (anime *Anime) CalculatedStatus() string {
// If we are past the end date, the anime is finished.
if validate.Date(anime.EndDate) {
end := anime.EndDateTime()
if time.Since(end) > 0 {
return "finished"
}
}
// If we have a start date and we didn't reach the end date, it's either current or upcoming.
if validate.Date(anime.StartDate) {
start := anime.StartDateTime()
if time.Since(start) > 0 {
return "current"
}
return "upcoming"
}
// If we have no date information it's to be announced.
return "tba"
}
// EpisodeByNumber returns the episode with the given number.
2019-08-28 08:06:42 +00:00
func (anime *Anime) EpisodeByNumber(number int) *Episode {
for _, episode := range anime.Episodes() {
2019-06-03 09:32:43 +00:00
if number == episode.Number {
return episode
}
}
return nil
}
// String implements the default string serialization.
func (anime *Anime) String() string {
return anime.Title.Canonical
}
2019-06-07 01:50:38 +00:00
// GetID returns the ID.
func (anime *Anime) GetID() string {
return anime.ID
}
2019-06-03 09:32:43 +00:00
// TypeName returns the type name.
func (anime *Anime) TypeName() string {
return "Anime"
}
// Self returns the object itself.
func (anime *Anime) Self() Loggable {
return anime
}
// StreamAnime returns a stream of all anime.
func StreamAnime() <-chan *Anime {
channel := make(chan *Anime, nano.ChannelBufferSize)
go func() {
for obj := range DB.All("Anime") {
channel <- obj.(*Anime)
}
close(channel)
}()
return channel
}
// AllAnime returns a slice of all anime.
func AllAnime() []*Anime {
all := make([]*Anime, 0, DB.Collection("Anime").Count())
stream := StreamAnime()
for obj := range stream {
all = append(all, obj)
}
return all
}
// FilterAnime filters all anime by a custom function.
func FilterAnime(filter func(*Anime) bool) []*Anime {
var filtered []*Anime
channel := DB.All("Anime")
for obj := range channel {
realObject := obj.(*Anime)
if filter(realObject) {
filtered = append(filtered, realObject)
}
}
return filtered
}