jellyfin-user-clone/main.go

package main

import (
	"database/sql"
	"fmt"
	"io"
	"io/ioutil"
	"log"
	"os"
	"path"
	"slices"
	"strings"

	_ "github.com/glebarez/sqlite"
	"github.com/google/uuid"
)

const (
	jellyDB   = "jellyfin.db"
	libraryDB = "library.db"
)

func main() {
	data := os.Args[1]
	fromU := os.Args[2]
	toU := os.Args[3]
	_ = toU

	workd, err := ioutil.TempDir(os.TempDir(), "jellyfin-user-clone.*")
	if err != nil {
		log.Fatalf("%v", err)
	}

	{
		if err := cp(path.Join(data, jellyDB), path.Join(workd, jellyDB)); err != nil {
			log.Fatalf("%v", err)
		} else if err := cp(path.Join(data, libraryDB), path.Join(workd, libraryDB)); err != nil {
			log.Fatalf("%v", err)
		}
	}

	jellyfinDB, err := sql.Open("sqlite", path.Join(workd, jellyDB))
	if err != nil {
		log.Fatalf("%v", err)
	}
	defer jellyfinDB.Close()

	fromUUID, err := SelectOne[string](jellyfinDB, `SELECT Id FROM Users WHERE Username = $1`, fromU)
	if err != nil {
		log.Fatalf("%v", err)
	}
	fromID, err := SelectOne[int](jellyfinDB, `SELECT InternalId FROM Users WHERE Id = $1`, fromUUID)
	if err != nil {
		log.Fatalf("%v", err)
	}
	log.Println(fromU, fromUUID, fromID)

	if n, err := SelectOne[int](jellyfinDB, `SELECT COUNT(*) FROM Users WHERE Username = $1`, toU); err != nil {
		log.Fatalf("%v", err)
	} else if n == 1 {
	} else if err := CloneForColumn(jellyfinDB, `Users`, `Username`, fromU, toU, `Id`, `InternalId`); err != nil {
		log.Fatalf("%v", err)
	} else if n, err := SelectOne[int](jellyfinDB, `SELECT COUNT(*) FROM Users WHERE Username = $1`, toU); err != nil {
		log.Fatalf("%v", err)
	} else if n != 1 {
		log.Fatalf("still no username=%q after insert", toU)
	}

	toUUID, err := SelectOne[string](jellyfinDB, `SELECT Id FROM Users WHERE Username = $1`, toU)
	if err != nil {
		log.Fatalf("%v", err)
	}
	toID, err := SelectOne[int](jellyfinDB, `SELECT InternalId FROM Users WHERE Id = $1`, toUUID)
	if err != nil {
		log.Fatalf("%v", err)
	}
	log.Println(toU, toUUID, toID)

	log.Fatalf("not impl get toU, toUUID")

	tables, err := Tables(jellyfinDB)
	if err != nil {
		log.Fatalf("%v", err)
	}

	for _, table := range tables {
		columns, err := Columns(jellyfinDB, table)
		if err != nil {
			log.Fatalf("%v", err)
		}
		log.Println(table, columns)

		userColumns := slices.DeleteFunc(slices.Clone(columns), func(s string) bool {
			return !slices.Contains([]string{
				"user",
				"userid",
			}, strings.ToLower(s))
		})
		if len(userColumns) == 0 {
			continue
		}

		for _, column := range userColumns {
			if n, err := SelectOne[int](jellyfinDB, fmt.Sprintf(`SELECT COUNT(*) FROM %q WHERE %q = $1`, table, column), fromID); err != nil {
				log.Fatalf("%v", err)
			} else if n > 0 {
				if err := CloneForColumn(jellyfinDB, table, column, fromID, toID); err != nil {
					log.Fatalf("%v", err)
				}
			} else if n, err := SelectOne[int](jellyfinDB, fmt.Sprintf(`SELECT COUNT(*) FROM %q WHERE %q = $1`, table, column), fromUUID); err != nil {
				log.Fatalf("%v", err)
			} else if n > 0 {
				if err := CloneForColumn(jellyfinDB, table, column, fromUUID, toUUID); err != nil {
					log.Fatalf("%v", err)
				}
			}
		}

		log.Println(table, userColumns)
		log.Fatalf("not impl")
	}
}

func CloneForColumn[T any](db *sql.DB, table, column string, from, to T, omit ...string) error {
	columns, err := Columns(db, table)
	if err != nil {
		return err
	}

	// TODO SELECT idx_name FROM PRAGMA_INDEX_LIST('Users') WHERE "unique" to get unique constraints
	// TODO SELECT col_name FROM PRAGMA_INDEX_INFO(idx_name);
	// TODO how is .Users.InternalId maintained? by chance?

	notTheseColumns := slices.DeleteFunc(slices.Clone(columns), func(s string) bool { return s == column || slices.Contains(omit, s) })
	for i := range notTheseColumns {
		notTheseColumns[i] = fmt.Sprintf("%q", notTheseColumns[i])
	}

	notNullTextColumns, err := NotNullTextColumns(db, table)
	if err != nil {
		return err
	}
	uuidGenColumns := slices.DeleteFunc(notNullTextColumns, func(s string) bool { return !slices.Contains(omit, s) })
	for i := range uuidGenColumns {
		uuidGenColumns[i] = fmt.Sprintf("%q", uuidGenColumns[i])
	}

	notNullIntColumns, err := NotNullIntColumns(db, table)
	if err != nil {
		return err
	}
	incrGenColumns := slices.DeleteFunc(notNullIntColumns, func(s string) bool { return !slices.Contains(omit, s) })
	for i := range incrGenColumns {
		incrGenColumns[i] = fmt.Sprintf("%q", incrGenColumns[i])
	}

	q := fmt.Sprintf(
		`INSERT INTO %q (%q, %s) SELECT $1, %s %s %s FROM %q WHERE %q = $2`,
		table, column, strings.Join(append(incrGenColumns, append(uuidGenColumns, notTheseColumns...)...), ", "),
		func() string {
			incrs := make([]string, len(incrGenColumns))
			for i := range incrs {
				incrs[i] = fmt.Sprintf(`%s+100`, incrGenColumns[i])
			}
			s := strings.Join(incrs, ", ")
			if len(incrs) > 0 {
				s += ", "
			}
			return s
		}(),
		func() string {
			uuids := make([]string, len(uuidGenColumns))
			for i := range uuids {
				uuids[i] = fmt.Sprintf("'%s'", strings.ToUpper(uuid.New().String()))
			}
			s := strings.Join(uuids, ", ")
			if len(uuids) > 0 {
				s += ", "
			}
			return s
		}(), strings.Join(notTheseColumns, ", "), table, column,
	)
	log.Printf("EXEC | %s (%v, %v)", q, to, from)
	return Insert(db, q, to, from)
}

func cp(from, to string) error {
	fromF, err := os.Open(from)
	if err != nil {
		return err
	}
	defer fromF.Close()

	toF, err := os.Create(to)
	if err != nil {
		return err
	}
	defer toF.Close()

	_, err = io.Copy(toF, fromF)
	return err
}

func Tables(db *sql.DB) ([]string, error) {
	return Select[string](db, `SELECT name FROM sqlite_schema WHERE type = 'table' AND name NOT LIKE 'sqlite_%'`)
}

func Columns(db *sql.DB, table string) ([]string, error) {
	return Select[string](db, `SELECT name FROM PRAGMA_TABLE_INFO($1)`, table)
}

func NotNullTextColumns(db *sql.DB, table string) ([]string, error) {
	return Select[string](db, `SELECT name FROM PRAGMA_TABLE_INFO($1) WHERE "notnull" = 1 AND "type" = 'TEXT'`, table)
}

func NotNullIntColumns(db *sql.DB, table string) ([]string, error) {
	return Select[string](db, `SELECT name FROM PRAGMA_TABLE_INFO($1) WHERE "notnull" = 1 AND "type" = 'INTEGER'`, table)
}

func Insert(db *sql.DB, q string, args ...any) error {
	_, err := db.Exec(q, args...)
	return err
}

func Exec(db *sql.DB, q string, args ...any) error {
	_, err := db.Exec(q, args...)
	return err
}

func SelectOne[T any](db *sql.DB, q string, args ...any) (T, error) {
	var some T

	results, err := Select[T](db, q, args...)
	if err != nil {
		return some, err
	}

	if len(results) != 1 {
		return some, fmt.Errorf("expected 1 result but got %d (%+v)", len(results), results)
	}

	return results[0], nil
}

func Select[T any](db *sql.DB, q string, args ...any) ([]T, error) {
	rows, err := db.Query(q, args...)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	results := []T{}
	for rows.Next() {
		var some T
		if err := rows.Scan(&some); err != nil {
			return nil, err
		}
		results = append(results, some)
	}

	return results, rows.Err()
}