2019-06-07 10:50:38 +09:00

89 lines
2.3 KiB
Go

package arn
import (
"fmt"
"github.com/aerogo/nano"
)
// IgnoreAnimeDifferenceEditorScore represents how many points you get for a diff ignore.
const IgnoreAnimeDifferenceEditorScore = 2
// IgnoreAnimeDifference saves which differences between anime databases can be ignored.
type IgnoreAnimeDifference struct {
// The ID is built like this: arn:323|mal:356|JapaneseTitle
ID string `json:"id"`
ValueHash uint64 `json:"valueHash"`
hasCreator
}
// GetIgnoreAnimeDifference ...
func GetIgnoreAnimeDifference(id string) (*IgnoreAnimeDifference, error) {
obj, err := DB.Get("IgnoreAnimeDifference", id)
if err != nil {
return nil, err
}
return obj.(*IgnoreAnimeDifference), nil
}
// CreateDifferenceID ...
func CreateDifferenceID(animeID AnimeID, dataProvider string, malAnimeID string, typeName string) string {
return fmt.Sprintf("arn:%s|%s:%s|%s", animeID, dataProvider, malAnimeID, typeName)
}
// IsAnimeDifferenceIgnored tells you whether the given difference is being ignored.
func IsAnimeDifferenceIgnored(animeID AnimeID, dataProvider string, malAnimeID string, typeName string, hash uint64) bool {
key := CreateDifferenceID(animeID, dataProvider, malAnimeID, typeName)
ignore, err := GetIgnoreAnimeDifference(key)
if err != nil {
return false
}
return ignore.ValueHash == hash
}
// StreamIgnoreAnimeDifferences returns a stream of all ignored differences.
func StreamIgnoreAnimeDifferences() <-chan *IgnoreAnimeDifference {
channel := make(chan *IgnoreAnimeDifference, nano.ChannelBufferSize)
go func() {
for obj := range DB.All("IgnoreAnimeDifference") {
channel <- obj.(*IgnoreAnimeDifference)
}
close(channel)
}()
return channel
}
// AllIgnoreAnimeDifferences returns a slice of all ignored differences.
func AllIgnoreAnimeDifferences() []*IgnoreAnimeDifference {
all := make([]*IgnoreAnimeDifference, 0, DB.Collection("IgnoreAnimeDifference").Count())
stream := StreamIgnoreAnimeDifferences()
for obj := range stream {
all = append(all, obj)
}
return all
}
// FilterIgnoreAnimeDifferences filters all ignored differences by a custom function.
func FilterIgnoreAnimeDifferences(filter func(*IgnoreAnimeDifference) bool) []*IgnoreAnimeDifference {
var filtered []*IgnoreAnimeDifference
for obj := range StreamIgnoreAnimeDifferences() {
if filter(obj) {
filtered = append(filtered, obj)
}
}
return filtered
}