Added export script for Aero DB

patches/export-aero-db/export-aero-db.go

@@ -0,0 +1,273 @@
+package main
+
+import (
+	"time"
+
+	"github.com/aerogo/database"
+	"github.com/animenotifier/arn"
+	"github.com/fatih/color"
+)
+
+func main() {
+	arn.DB.SetScanPriority("high")
+
+	aeroDB := database.New("db", arn.DBTypes)
+	defer aeroDB.Close()
+
+	for typeName := range arn.DB.Types() {
+		count := 0
+
+		switch typeName {
+		case "Anime":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.Anime) {
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "AnimeEpisodes":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.AnimeEpisodes) {
+				aeroDB.Set(typeName, obj.AnimeID, obj)
+				count++
+			}
+
+		case "AnimeList":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.AnimeList) {
+				aeroDB.Set(typeName, obj.UserID, obj)
+				count++
+			}
+
+		case "AnimeCharacters":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.AnimeCharacters) {
+				aeroDB.Set(typeName, obj.AnimeID, obj)
+				count++
+			}
+
+		case "AnimeRelations":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.AnimeRelations) {
+				aeroDB.Set(typeName, obj.AnimeID, obj)
+				count++
+			}
+
+		case "Character":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.Character) {
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "Purchase":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.Purchase) {
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "PushSubscriptions":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.PushSubscriptions) {
+				aeroDB.Set(typeName, obj.UserID, obj)
+				count++
+			}
+
+		case "User":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.User) {
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "Post":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.Post) {
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "Thread":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.Thread) {
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "Analytics":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.Analytics) {
+				aeroDB.Set(typeName, obj.UserID, obj)
+				count++
+			}
+
+		case "SoundTrack":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.SoundTrack) {
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "Item":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.Item) {
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "Inventory":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.Inventory) {
+				aeroDB.Set(typeName, obj.UserID, obj)
+				count++
+			}
+
+		case "Settings":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.Settings) {
+				aeroDB.Set(typeName, obj.UserID, obj)
+				count++
+			}
+
+		case "UserFollows":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.UserFollows) {
+				aeroDB.Set(typeName, obj.UserID, obj)
+				count++
+			}
+
+		case "PayPalPayment":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.PayPalPayment) {
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "AniListToAnime":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.AniListToAnime) {
+				aeroDB.Set(typeName, obj.ServiceID, obj)
+				count++
+			}
+
+		case "MyAnimeListToAnime":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.MyAnimeListToAnime) {
+				aeroDB.Set(typeName, obj.ServiceID, obj)
+				count++
+			}
+
+		case "SearchIndex":
+			anime, _ := arn.DB.Get(typeName, "Anime")
+			aeroDB.Set(typeName, "Anime", anime)
+
+			users, _ := arn.DB.Get(typeName, "User")
+			aeroDB.Set(typeName, "User", users)
+
+			posts, _ := arn.DB.Get(typeName, "Post")
+			aeroDB.Set(typeName, "Post", posts)
+
+			threads, _ := arn.DB.Get(typeName, "Thread")
+			aeroDB.Set(typeName, "Thread", threads)
+
+			count += 4
+
+		case "DraftIndex":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.DraftIndex) {
+				aeroDB.Set(typeName, obj.UserID, obj)
+				count++
+			}
+
+		case "EmailToUser":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.EmailToUser) {
+				if obj.Email == "" {
+					continue
+				}
+
+				aeroDB.Set(typeName, obj.Email, obj)
+				count++
+			}
+
+		case "FacebookToUser":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.FacebookToUser) {
+				if obj.ID == "" {
+					continue
+				}
+
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "GoogleToUser":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.GoogleToUser) {
+				if obj.ID == "" {
+					continue
+				}
+
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "TwitterToUser":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.TwitterToUser) {
+				if obj.ID == "" {
+					continue
+				}
+
+				aeroDB.Set(typeName, obj.ID, obj)
+				count++
+			}
+
+		case "NickToUser":
+			channel, _ := arn.DB.All(typeName)
+
+			for obj := range channel.(chan *arn.NickToUser) {
+				if obj.Nick == "" {
+					continue
+				}
+
+				aeroDB.Set(typeName, obj.Nick, obj)
+				count++
+			}
+
+		default:
+			color.Yellow("Skipping %s", typeName)
+			continue
+		}
+
+		color.Green("Export %d %s", count, typeName)
+	}
+
+	time.Sleep(1 * time.Second)
+}