package sqlite

import (
	"database/sql"

	"github.com/pkg/errors"

	_ "github.com/mutecomm/go-sqlcipher/v4" // We require go sqlcipher that overrides default implementation
	"github.com/status-im/migrate/v4/database/sqlcipher"
	bindata "github.com/status-im/migrate/v4/source/go_bindata"

	"github.com/status-im/status-go/internal/db/sqlite"
	"github.com/status-im/status-go/pkg/messaging"
)

var migrationsTable = "status_protocol_go_" + sqlcipher.DefaultMigrationsTable

func Migrate(database *sql.DB) error {
	lastProtocolMigrationVersion, _, err := sqlite.GetLastMigrationVersion(database, migrationsTable)
	if err != nil {
		return errors.Wrap(err, "failed to get current migration version")
	}

	err = messaging.SQLiteMigrate(database, lastProtocolMigrationVersion)
	if err != nil {
		return errors.Wrap(err, "failed to apply messaging migrations")
	}

	migrationNames, migrationGetter, err := prepareMigrations(defaultMigrations)
	if err != nil {
		return errors.Wrap(err, "failed to prepare status-go/protocol migrations")
	}

	// Ensure the protocol migration table version is synchronized with the latest migration after migrations are moved between modules.
	err = sqlite.UpdateMigrationTableVersion(database, migrationsTable, migrationNames, lastProtocolMigrationVersion)
	if err != nil {
		return errors.Wrap(err, "failed to update migration table version")
	}

	options := sqlite.MigrateOptions{
		MigrationTableName: migrationsTable,
	}
	err = sqlite.Migrate(database, bindata.Resource(migrationNames, bindata.AssetFunc(migrationGetter)), options)
	if err != nil {
		return errors.Wrap(err, "failed to apply status-go/protocol migrations")
	}

	return nil
}