streamplace / pkg / statedb / migrate.go

package statedb

import (
	"context"
	"os"
	"time"

	"github.com/lmittmann/tint"
	slogGorm "github.com/orandin/slog-gorm"
	"github.com/streamplace/oatproxy/pkg/oatproxy"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"stream.place/streamplace/pkg/config"
	"stream.place/streamplace/pkg/log"
)

func Migrate(cli *config.CLI) error {
	gormLogger := slogGorm.New(
		slogGorm.WithHandler(tint.NewHandler(os.Stderr, &tint.Options{
			TimeFormat: time.RFC3339,
		})),
		// slogGorm.WithTraceAll(),
	)

	newDB, err := MakeDB(cli, nil, nil)
	if err != nil {
		return err
	}

	oldDB, err := gorm.Open(sqlite.Open(cli.IndexDBPath), &gorm.Config{
		Logger: gormLogger,
	})
	if err != nil {
		return err
	}

	var sessions []oatproxy.OAuthSession
	if err := oldDB.Find(&sessions).Error; err != nil {
		return err
	}

	for _, session := range sessions {
		log.Log(context.Background(), "migrating session", "session", session.DownstreamDPoPJKT)
		err := newDB.DB.Save(&session).Error
		if err != nil {
			return err
		}
	}

	var notifications []Notification
	if err := oldDB.Find(&notifications).Error; err != nil {
		time.Sleep(1 * time.Second)
		return err
	}

	for _, notification := range notifications {
		log.Log(context.Background(), "migrating notification", "notification", notification)
		err := newDB.DB.Save(&notification).Error
		if err != nil {
			return err
		}
	}

	return nil
}