notify.moe/arn/AnimeAPI.go

package arn

import (
	"errors"
	"fmt"
	"os"
	"path"
	"reflect"
	"strings"

	"github.com/aerogo/aero"
	"github.com/aerogo/api"
	"github.com/akyoto/color"
)

// Force interface implementations
var (
	_ fmt.Stringer           = (*Anime)(nil)
	_ Likeable               = (*Anime)(nil)
	_ PostParent             = (*Anime)(nil)
	_ Publishable            = (*Anime)(nil)
	_ api.Creatable          = (*Anime)(nil)
	_ api.Deletable          = (*Anime)(nil)
	_ api.Editable           = (*Anime)(nil)
	_ api.CustomEditable     = (*Anime)(nil)
	_ api.ArrayEventListener = (*Anime)(nil)
)

// Actions
func init() {
	API.RegisterActions("Anime", []*api.Action{
		// Like anime
		LikeAction(),

		// Unlike anime
		UnlikeAction(),
	})
}

// Create sets the data for a new anime.
func (anime *Anime) Create(ctx aero.Context) error {
	user := GetUserFromContext(ctx)

	if user == nil {
		return errors.New("Not logged in")
	}

	anime.init()
	anime.CreatedBy = user.ID

	// Characters
	characters := anime.Characters()

	if characters == nil {
		characters = &AnimeCharacters{
			AnimeID: anime.ID,
			Items:   []*AnimeCharacter{},
		}

		characters.Save()
	}

	// Relations
	relations := anime.Relations()

	if relations == nil {
		relations = &AnimeRelations{
			AnimeID: anime.ID,
			Items:   []*AnimeRelation{},
		}

		relations.Save()
	}

	// Write log entry
	logEntry := NewEditLogEntry(user.ID, "create", "Anime", anime.ID, "", "", "")
	logEntry.Save()

	return anime.Unpublish()
}

// Edit creates an edit log entry.
func (anime *Anime) Edit(ctx aero.Context, key string, value reflect.Value, newValue reflect.Value) (consumed bool, err error) {
	user := GetUserFromContext(ctx)

	if key == "Status" {
		oldStatus := value.String()
		newStatus := newValue.String()

		// Notify people who want to know about finished series
		if oldStatus == "current" && newStatus == "finished" {
			go func() {
				for _, user := range anime.UsersWatchingOrPlanned() {
					if !user.Settings().Notification.AnimeFinished {
						continue
					}

					user.SendNotification(&PushNotification{
						Title:   anime.Title.ByUser(user),
						Message: anime.Title.ByUser(user) + " has finished airing!",
						Icon:    anime.ImageLink("medium"),
						Link:    "https://notify.moe" + anime.Link(),
						Type:    NotificationTypeAnimeFinished,
					})
				}
			}()
		}
	}

	// Write log entry
	logEntry := NewEditLogEntry(user.ID, "edit", "Anime", anime.ID, key, fmt.Sprint(value.Interface()), fmt.Sprint(newValue.Interface()))
	logEntry.Save()

	return false, nil
}

// OnAppend saves a log entry.
func (anime *Anime) OnAppend(ctx aero.Context, key string, index int, obj interface{}) {
	onAppend(anime, ctx, key, index, obj)
}

// OnRemove saves a log entry.
func (anime *Anime) OnRemove(ctx aero.Context, key string, index int, obj interface{}) {
	onRemove(anime, ctx, key, index, obj)
}

// Authorize returns an error if the given API POST request is not authorized.
func (anime *Anime) Authorize(ctx aero.Context, action string) error {
	user := GetUserFromContext(ctx)

	if user == nil || (user.Role != "editor" && user.Role != "admin") {
		return errors.New("Not logged in or not authorized to edit this anime")
	}

	return nil
}

// DeleteInContext deletes the anime in the given context.
func (anime *Anime) DeleteInContext(ctx aero.Context) error {
	user := GetUserFromContext(ctx)

	// Write log entry
	logEntry := NewEditLogEntry(user.ID, "delete", "Anime", anime.ID, "", fmt.Sprint(anime), "")
	logEntry.Save()

	return anime.Delete()
}

// Delete deletes the anime from the database.
func (anime *Anime) Delete() error {
	if anime.IsDraft {
		draftIndex := anime.Creator().DraftIndex()
		draftIndex.AnimeID = ""
		draftIndex.Save()
	}

	// Delete anime characters
	DB.Delete("AnimeCharacters", anime.ID)

	// Delete anime relations
	DB.Delete("AnimeRelations", anime.ID)

	// Delete anime episodes
	for _, episode := range anime.Episodes() {
		err := episode.Delete()

		if err != nil {
			return err
		}
	}

	// Delete anime list items
	for animeList := range StreamAnimeLists() {
		removed := animeList.Remove(anime.ID)

		if removed {
			animeList.Save()
		}
	}

	// Delete anime ID from existing relations
	for relations := range StreamAnimeRelations() {
		removed := relations.Remove(anime.ID)

		if removed {
			relations.Save()
		}
	}

	// Delete anime ID from quotes
	for quote := range StreamQuotes() {
		if quote.AnimeID == anime.ID {
			quote.AnimeID = ""
			quote.Save()
		}
	}

	// Remove posts
	for _, post := range anime.Posts() {
		err := post.Delete()

		if err != nil {
			return err
		}
	}

	// Delete soundtrack tags
	for track := range StreamSoundTracks() {
		newTags := []string{}

		for _, tag := range track.Tags {
			if strings.HasPrefix(tag, "anime:") {
				parts := strings.Split(tag, ":")
				id := parts[1]

				if id == anime.ID {
					continue
				}
			}

			newTags = append(newTags, tag)
		}

		if len(track.Tags) != len(newTags) {
			track.Tags = newTags
			track.Save()
		}
	}

	// Delete images on file system
	if anime.HasImage() {
		err := os.Remove(path.Join(Root, "images/anime/original/", anime.ID+anime.Image.Extension))

		if err != nil {
			// Don't return the error.
			// It's too late to stop the process at this point.
			// Instead, log the error.
			color.Red(err.Error())
		}

		os.Remove(path.Join(Root, "images/anime/large/", anime.ID+".jpg"))
		os.Remove(path.Join(Root, "images/anime/large/", anime.ID+"@2.jpg"))
		os.Remove(path.Join(Root, "images/anime/large/", anime.ID+".webp"))
		os.Remove(path.Join(Root, "images/anime/large/", anime.ID+"@2.webp"))

		os.Remove(path.Join(Root, "images/anime/medium/", anime.ID+".jpg"))
		os.Remove(path.Join(Root, "images/anime/medium/", anime.ID+"@2.jpg"))
		os.Remove(path.Join(Root, "images/anime/medium/", anime.ID+".webp"))
		os.Remove(path.Join(Root, "images/anime/medium/", anime.ID+"@2.webp"))

		os.Remove(path.Join(Root, "images/anime/small/", anime.ID+".jpg"))
		os.Remove(path.Join(Root, "images/anime/small/", anime.ID+"@2.jpg"))
		os.Remove(path.Join(Root, "images/anime/small/", anime.ID+".webp"))
		os.Remove(path.Join(Root, "images/anime/small/", anime.ID+"@2.webp"))
	}

	// Delete the actual anime
	DB.Delete("Anime", anime.ID)

	return nil
}

// Save saves the anime in the database.
func (anime *Anime) Save() {
	DB.Set("Anime", anime.ID, anime)
}