Progress on database schema migration for account -> org refactor

pkg/services/sqlstore/migrator/migrations.go

@@ -0,0 +1,233 @@
+package migrator
+
+import (
+	"fmt"
+	"strings"
+)
+
+type MigrationBase struct {
+	id        string
+	Condition MigrationCondition
+}
+
+func (m *MigrationBase) Id() string {
+	return m.id
+}
+
+func (m *MigrationBase) SetId(id string) {
+	m.id = id
+}
+
+func (m *MigrationBase) GetCondition() MigrationCondition {
+	return m.Condition
+}
+
+type RawSqlMigration struct {
+	MigrationBase
+
+	sqlite string
+	mysql  string
+}
+
+func (m *RawSqlMigration) Sql(dialect Dialect) string {
+	switch dialect.DriverName() {
+	case MYSQL:
+		return m.mysql
+	case SQLITE:
+		return m.sqlite
+	}
+
+	panic("db type not supported")
+}
+
+func (m *RawSqlMigration) Sqlite(sql string) *RawSqlMigration {
+	m.sqlite = sql
+	return m
+}
+
+func (m *RawSqlMigration) Mysql(sql string) *RawSqlMigration {
+	m.mysql = sql
+	return m
+}
+
+type AddColumnMigration struct {
+	MigrationBase
+	tableName string
+	column    *Column
+}
+
+func (m *AddColumnMigration) Table(tableName string) *AddColumnMigration {
+	m.tableName = tableName
+	return m
+}
+
+func (m *AddColumnMigration) Column(col *Column) *AddColumnMigration {
+	m.column = col
+	return m
+}
+
+func (m *AddColumnMigration) Sql(dialect Dialect) string {
+	return dialect.AddColumnSql(m.tableName, m.column)
+}
+
+type AddIndexMigration struct {
+	MigrationBase
+	tableName string
+	index     Index
+}
+
+func (m *AddIndexMigration) Name(name string) *AddIndexMigration {
+	m.index.Name = name
+	return m
+}
+
+func (m *AddIndexMigration) Table(tableName string) *AddIndexMigration {
+	m.tableName = tableName
+	return m
+}
+
+func (m *AddIndexMigration) Unique() *AddIndexMigration {
+	m.index.Type = UniqueIndex
+	return m
+}
+
+func (m *AddIndexMigration) Columns(columns ...string) *AddIndexMigration {
+	m.index.Cols = columns
+	return m
+}
+
+func (m *AddIndexMigration) Sql(dialect Dialect) string {
+	if m.index.Name == "" {
+		m.index.Name = fmt.Sprintf("%s", strings.Join(m.index.Cols, "_"))
+	}
+	return dialect.CreateIndexSql(m.tableName, &m.index)
+}
+
+type DropIndexMigration struct {
+	MigrationBase
+	tableName string
+	index     Index
+}
+
+func (m *DropIndexMigration) Name(name string) *DropIndexMigration {
+	m.index.Name = name
+	return m
+}
+
+func (m *DropIndexMigration) Table(tableName string) *DropIndexMigration {
+	m.tableName = tableName
+	return m
+}
+
+func (m *DropIndexMigration) Unique() *DropIndexMigration {
+	m.index.Type = UniqueIndex
+	return m
+}
+
+func (m *DropIndexMigration) Columns(columns ...string) *DropIndexMigration {
+	m.index.Cols = columns
+	return m
+}
+
+func (m *DropIndexMigration) Sql(dialect Dialect) string {
+	if m.index.Name == "" {
+		m.index.Name = fmt.Sprintf("%s", strings.Join(m.index.Cols, "_"))
+	}
+	return dialect.DropIndexSql(m.tableName, &m.index)
+}
+
+type AddTableMigration struct {
+	MigrationBase
+	table Table
+}
+
+func (m *AddTableMigration) Sql(d Dialect) string {
+	return d.CreateTableSql(&m.table)
+}
+
+func (m *AddTableMigration) Name(name string) *AddTableMigration {
+	m.table.Name = name
+	return m
+}
+
+func (m *AddTableMigration) WithColumns(columns ...*Column) *AddTableMigration {
+	for _, col := range columns {
+		m.table.Columns = append(m.table.Columns, col)
+		if col.IsPrimaryKey {
+			m.table.PrimaryKeys = append(m.table.PrimaryKeys, col.Name)
+		}
+	}
+	return m
+}
+
+func (m *AddTableMigration) WithColumn(col *Column) *AddTableMigration {
+	m.table.Columns = append(m.table.Columns, col)
+	if col.IsPrimaryKey {
+		m.table.PrimaryKeys = append(m.table.PrimaryKeys, col.Name)
+	}
+	return m
+}
+
+type DropTableMigration struct {
+	MigrationBase
+	tableName string
+}
+
+func (m *DropTableMigration) Table(tableName string) *DropTableMigration {
+	m.tableName = tableName
+	return m
+}
+
+func (m *DropTableMigration) Sql(d Dialect) string {
+	return d.DropTable(m.tableName)
+}
+
+type RenameTableMigration struct {
+	MigrationBase
+	oldName string
+	newName string
+}
+
+func (m *RenameTableMigration) IfTableExists(tableName string) *RenameTableMigration {
+	m.Condition = &IfTableExistsCondition{TableName: tableName}
+	return m
+}
+
+func (m *RenameTableMigration) Rename(oldName string, newName string) *RenameTableMigration {
+	m.oldName = oldName
+	m.newName = newName
+	return m
+}
+
+func (m *RenameTableMigration) Sql(d Dialect) string {
+	return d.RenameTable(m.oldName, m.newName)
+}
+
+type CopyTableDataMigration struct {
+	MigrationBase
+	sourceTable string
+	targetTable string
+	sourceCols  string
+	targetCols  string
+}
+
+func (m *CopyTableDataMigration) Source(tableName string, cols string) *CopyTableDataMigration {
+	m.sourceTable = tableName
+	m.sourceCols = cols
+	return m
+}
+
+func (m *CopyTableDataMigration) Target(tableName string, cols string) *CopyTableDataMigration {
+	m.targetTable = tableName
+	m.targetCols = cols
+	return m
+}
+
+func (m *CopyTableDataMigration) IfTableExists(tableName string) *CopyTableDataMigration {
+	m.Condition = &IfTableExistsCondition{TableName: tableName}
+	return m
+}
+
+func (m *CopyTableDataMigration) Sql(d Dialect) string {
+	return d.CopyTableData(m.sourceTable, m.targetTable, m.sourceCols, m.targetCols)
+}