AnimeListItemAPI.go

package arn

import (
	"errors"
	"fmt"
	"reflect"
	"time"

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

// Force interface implementations
var (
	_ api.CustomEditable = (*AnimeListItem)(nil)
	_ api.AfterEditable  = (*AnimeListItem)(nil)
)

// Edit ...
func (item *AnimeListItem) Edit(ctx aero.Context, key string, value reflect.Value, newValue reflect.Value) (bool, error) {
	user := GetUserFromContext(ctx)

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

	switch key {
	case "Episodes":
		if !newValue.IsValid() {
			return true, errors.New("Invalid episode number")
		}

		oldEpisodes := item.Episodes
		newEpisodes := int(newValue.Float())

		// Fetch last activity
		lastActivity := user.LastActivityConsumeAnime(item.AnimeID)

		if lastActivity == nil || time.Since(lastActivity.GetCreatedTime()) > 1*time.Hour {
			// If there is no last activity for the given anime,
			// or if the last activity happened more than an hour ago,
			// create a new activity.
			if newEpisodes > oldEpisodes {
				activity := NewActivityConsumeAnime(item.AnimeID, newEpisodes, newEpisodes, user.ID)
				activity.Save()

				// Broadcast event to all users so they can reload the activity page if needed.
				for receiver := range StreamUsers() {
					receiverIsFollowing := Contains(receiver.Follows().Items, user.ID)

					receiver.BroadcastEvent(&aero.Event{
						Name: "activity",
						Data: receiverIsFollowing,
					})
				}
			}
		} else if newEpisodes >= lastActivity.FromEpisode {
			// Otherwise, update the last activity.
			lastActivity.ToEpisode = newEpisodes
			lastActivity.Created = DateTimeUTC()
			lastActivity.Save()
		}

		item.Episodes = newEpisodes

		if item.Episodes < 0 {
			item.Episodes = 0
		}

		item.OnEpisodesChange()
		return true, nil

	case "Status":
		newStatus := newValue.String()

		switch newStatus {
		case AnimeListStatusWatching, AnimeListStatusCompleted, AnimeListStatusPlanned, AnimeListStatusHold, AnimeListStatusDropped:
			item.Status = newStatus
			item.OnStatusChange()
			return true, nil

		default:
			return true, fmt.Errorf("Invalid anime list item status: %s", newStatus)
		}
	}

	return false, nil
}

// AfterEdit is called after the item is edited.
func (item *AnimeListItem) AfterEdit(ctx aero.Context) error {
	item.Rating.Clamp()
	item.Edited = DateTimeUTC()
	return nil
}