Added arn to the main repository

arn/IgnoreAnimeDifference.go

@@ -0,0 +1,88 @@
+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 string, 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 string, 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
+}