274 lines
5.4 KiB
Go
Raw Normal View History

2017-10-27 00:00:54 +00:00
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)
}