89 lines
2.3 KiB
Go
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 {
|
|
ValueHash uint64 `json:"valueHash"`
|
|
|
|
// The ID is built like this: arn:323|mal:356|JapaneseTitle
|
|
hasID
|
|
hasCreator
|
|
}
|
|
|
|
// GetIgnoreAnimeDifference ...
|
|
func GetIgnoreAnimeDifference(id ID) (*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
|
|
}
|