status-im/status-go

package sqlite

import (
    "strings"

    "github.com/pkg/errors"

    encryptmigrations "github.com/status-im/status-go/protocol/encryption/migrations"
    appmigrations "github.com/status-im/status-go/protocol/migrations"
    push_notification_client_migrations "github.com/status-im/status-go/protocol/pushnotificationclient/migrations"
    push_notification_server_migrations "github.com/status-im/status-go/protocol/pushnotificationserver/migrations"
    wakumigrations "github.com/status-im/status-go/protocol/transport/migrations"
)

type getter func(string) ([]byte, error)

type migrationsWithGetter struct {
    Names  []string
    Getter getter
}

var defaultMigrations = []migrationsWithGetter{
    {
        Names:  wakumigrations.AssetNames(),
        Getter: wakumigrations.Asset,
    },
    {
        Names:  encryptmigrations.AssetNames(),
        Getter: encryptmigrations.Asset,
    },
    {
        Names:  appmigrations.AssetNames(),
        Getter: appmigrations.Asset,
    },
    {
        Names:  push_notification_server_migrations.AssetNames(),
        Getter: push_notification_server_migrations.Asset,
    },
    {
        Names:  push_notification_client_migrations.AssetNames(),
        Getter: push_notification_client_migrations.Asset,
    },
}

func prepareMigrations(migrations []migrationsWithGetter) ([]string, getter, error) {
    var allNames []string
    nameToGetter := make(map[string]getter)

    for _, m := range migrations {
        for _, name := range m.Names {
            if !validateName(name) {
                continue
            }

            if _, ok := nameToGetter[name]; ok {
                return nil, nil, errors.Errorf("migration with name %s already exists", name)
            }
            allNames = append(allNames, name)
            nameToGetter[name] = m.Getter
        }
    }

    return allNames, func(name string) ([]byte, error) {
        getter, ok := nameToGetter[name]
        if !ok {
            return nil, errors.Errorf("no migration for name %s", name)
        }
        return getter(name)
    }, nil
}

// validateName verifies that only *.sql files are taken into consideration.
func validateName(name string) bool {
    return strings.HasSuffix(name, ".sql")
}