main.go

package main

import (
	"flag"
	"io"
	"log"

	"github.com/aksiksi/gelatin/emby"
	"github.com/aksiksi/gelatin/jellyfin"
	gelatin "github.com/aksiksi/gelatin/lib"
)

var (
	jellyfinAdminUser string
	jellyfinAdminPass string
	embyAdminUser     string
	embyAdminPass     string
)

func verifyJellyfin() {
	client := jellyfin.NewJellyfinApiClient("http://192.168.0.99:8097", nil)

	if err := client.System().Ping(); err != nil {
		log.Fatalf("failed to ping: %s", err)
	}

	adminKey, err := client.User().Authenticate(jellyfinAdminUser, jellyfinAdminPass)
	if err != nil {
		log.Fatal("failed to authenticate")
	}

	client.SetApiKey(adminKey)

	systemInfo, err := client.System().Info(true)
	if err != nil {
		log.Fatalf("failed to get system info: %s", err)
	}

	log.Printf("Jellyfin info: %+v", systemInfo)

	logsInfo, err := client.System().GetLogs()
	if err != nil {
		log.Fatalf("failed to get system logs: %s", err)
	}

	log.Printf("Jellyfin logs: %+v", logsInfo)

	logName, logSize := logsInfo[0].Name, logsInfo[0].Size
	data, err := client.System().GetLogFile(logName)
	if err != nil {
		log.Fatalf("failed to get system log %s: %s", logName, err)
	}

	logData, _ := io.ReadAll(data)

	log.Printf("got log %s, size = %d, expected = %d", logName, len(logData), logSize)

	// Query public users
	_, err = client.User().GetUsers(true)
	if err != nil {
		log.Fatalf("failed to query users: %s", err)
	}

	// Query available users
	users, err := client.User().GetUsers(false)
	if err != nil {
		log.Fatalf("failed to query users: %s", err)
	}

	log.Printf("Users count: %d", len(users))

	// Create a new user
	user, err := client.User().CreateUser("test123")
	if err != nil {
		log.Fatalf("failed to create new user: %s", err)
	}

	log.Printf("User: %v", user)

	// Set user password
	err = client.User().UpdatePassword(user.Id, "", "abcd1234", false)
	log.Printf("%v", err)

	// Make the user an admin
	user.Policy.IsAdministrator = true
	err = client.User().UpdatePolicy(user.Id, &user.Policy)
	if err != nil {
		log.Fatal(err)
	}

	items, err := client.Library().GetItemsByUser(user.Id, map[string]string{
		"IncludeItemTypes": "Movie,Series",
	})
	if err != nil {
		log.Fatal(err)
	}

	log.Printf("Num items: %d", len(items))
	log.Printf("Random item: %+v", items[len(items)/2])

	// Delete the user
	err = client.User().DeleteUser(user.Id)
	if err != nil {
		log.Fatal(err)
	}
}

func verifyEmby() {
	client := emby.NewEmbyApiClient("http://192.168.0.99:8096", nil)

	if err := client.System().Ping(); err != nil {
		log.Fatalf("failed to ping: %s", err)
	}

	adminKey, err := client.User().Authenticate(embyAdminUser, embyAdminPass)
	if err != nil {
		log.Fatalf("failed to authenticate")
	}

	client.SetApiKey(adminKey)

	resp, err := client.System().Info(true)
	if err != nil {
		log.Fatalf("failed to get system info: %s", err)
	}

	log.Printf("Emby info: %+v", resp)

	logsInfo, err := client.System().GetLogs()
	if err != nil {
		log.Fatalf("failed to get system logs: %s", err)
	}

	log.Printf("Emby logs: %+v", logsInfo)

	logName, logSize := logsInfo[0].Name, logsInfo[0].Size
	data, err := client.System().GetLogFile(logName)
	if err != nil {
		log.Fatalf("failed to get system log %s: %s", logName, err)
	}

	logData, _ := io.ReadAll(data)

	log.Printf("got log %s, size = %d, expected = %d", logName, len(logData), logSize)

	// Query public users
	_, err = client.User().GetUsers(true)
	if err != nil {
		log.Fatalf("failed to query users: %s", err)
	}

	// Query available users
	users, err := client.User().GetUsers(false)
	if err != nil {
		log.Fatalf("failed to query users: %s", err)
	}

	log.Printf("Users count: %d", len(users))

	// Create a new user
	user, err := client.User().CreateUser("test123")
	if err != nil {
		log.Fatalf("failed to create new user: %s", err)
	}

	log.Printf("User: %v", user)

	// Set user password
	err = client.User().UpdatePassword(user.Id, "", "abcd1234", false)
	log.Printf("%v", err)

	// Make the user an admin
	user.Policy.IsAdministrator = true
	err = client.User().UpdatePolicy(user.Id, &user.Policy)
	if err != nil {
		log.Fatal(err)
	}

	items, err := client.Library().GetItemsByUser(user.Id, map[string]string{
		"IncludeItemTypes": "Movie,Series",
	})
	if err != nil {
		log.Fatal(err)
	}

	log.Printf("Num items: %d", len(items))
	log.Printf("Random item: %+v", items[len(items)/2])

	// Delete the user
	err = client.User().DeleteUser(user.Id)
	if err != nil {
		log.Fatal(err)
	}
}

func verifyGelatinClient() {
	embyClient := emby.NewEmbyApiClient("http://192.168.0.99:8096", nil)
	embyKey, _ := embyClient.User().Authenticate(embyAdminUser, embyAdminPass)
	embyClient.SetApiKey(embyKey)

	jellyfinClient := jellyfin.NewJellyfinApiClient("http://192.168.0.99:8097", nil)
	jellyfinKey, _ := jellyfinClient.User().Authenticate(jellyfinAdminUser, jellyfinAdminPass)
	jellyfinClient.SetApiKey(jellyfinKey)

	opts := &gelatin.GelatinClientOpts{Interactive: true}
	client := gelatin.NewGelatinClient(embyClient, jellyfinClient, opts)

	userDiff, err := client.DiffUsers(false)
	if err != nil {
		log.Fatal(err)
	}

	log.Printf("User diff: %s", userDiff)

	if err := client.MigrateUsers(nil); err != nil {
		log.Fatal(err)
	}

	if err := client.MigrateUserWatchHistory("tksiksi"); err != nil {
		log.Print(err)
	}
}

func main() {
	flag.StringVar(&jellyfinAdminUser, "jellyfin-admin-user", "", "Jellyfin admin username")
	flag.StringVar(&jellyfinAdminPass, "jellyfin-admin-pass", "", "Jellyfin admin password")
	flag.StringVar(&embyAdminUser, "emby-admin-user", "", "Emby admin username")
	flag.StringVar(&embyAdminPass, "emby-admin-pass", "", "Emby admin password")
	flag.Parse()

	if jellyfinAdminUser == "" || jellyfinAdminPass == "" {
		log.Fatal("Jellyfin admin info must be specified")
	}

	if embyAdminUser == "" || embyAdminPass == "" {
		log.Fatal("Emby admin info must be specified")
	}

	verifyEmby()
	verifyJellyfin()
	verifyGelatinClient()
}