Goit

package goit

import (
	"database/sql"
	"fmt"
	"log"

	"github.com/Jamozed/Goit/src/util"
)

/*
Current database schema:
users:
	id INTEGER PRIMARY KEY AUTOINCREMENT
	name TEXT UNIQUE NOT NULL
	name_full TEXT NOT NULL
	pass BLOB NOT NULL
	pass_algo TEXT NOT NULL
	salt BLOB NOT NULL
	is_admin BOOLEAN NOT NULL

repos:
	id INTEGER PRIMARY KEY AUTOINCREMENT
	owner_id INTEGER NOT NULL
	name TEXT UNIQUE NOT NULL
	name_lower TEXT UNIQUE NOT NULL
	description TEXT NOT NULL
	default_branch TEXT NOT NULL DEFAULT 'master'
	upstream TEXT NOT NULL
	visibility INTEGER NOT NULL DEFAULT 0
	is_mirror BOOLEAN NOT NULL
*/

func updateDatabase(db *sql.DB) error {
	const LATEST_VERSION = 3

	tx, err := db.Begin()
	if err != nil {
		return err
	}
	defer tx.Rollback()

	var version int
	if err := tx.QueryRow("PRAGMA user_version").Scan(&version); err != nil {
		return err
	}

	if version > LATEST_VERSION {
		return fmt.Errorf("database version is newer than supported (%d > %d)", version, LATEST_VERSION)
	}

	logMigration := true

	if version <= 0 {
		/* Database is empty or new, initialise from scratch */
		log.Println("Initialising database at version", LATEST_VERSION)
		logMigration = false

		if _, err := tx.Exec(
			`CREATE TABLE IF NOT EXISTS users (
				id INTEGER PRIMARY KEY AUTOINCREMENT,
				name TEXT UNIQUE NOT NULL,
				name_full TEXT NOT NULL,
				pass BLOB NOT NULL,
				pass_algo TEXT NOT NULL,
				salt BLOB NOT NULL,
				is_admin BOOLEAN NOT NULL
			)`,
		); err != nil {
			return err
		}

		if _, err := tx.Exec(
			`CREATE TABLE IF NOT EXISTS repos (
				id INTEGER PRIMARY KEY AUTOINCREMENT,
				owner_id INTEGER NOT NULL,
				name TEXT UNIQUE NOT NULL,
				name_lower TEXT UNIQUE NOT NULL,
				description TEXT NOT NULL,
				upstream TEXT NOT NULL,
				is_private BOOLEAN NOT NULL,
				is_mirror BOOLEAN NOT NULL
			)`,
		); err != nil {
			return err
		}

		version = 1
	}
	if version <= 1 {
		if logMigration {
			log.Println("Migrating database from version 1 to 2")
		}

		if _, err := tx.Exec("ALTER TABLE repos ADD COLUMN default_branch TEXT NOT NULL DEFAULT 'master'"); err != nil {
			return err
		}

		version = 2
	}
	if version <= 2 {
		if logMigration {
			log.Println("Migrating database from version 2 to 3")
		}

		if _, err := tx.Exec("ALTER TABLE repos ADD COLUMN visibility INTEGER NOT NULL DEFAULT 0"); err != nil {
			return err
		}

		/* Set values for each repo according to is_private */
		var visibilities = map[int64]Visibility{}

		if rows, err := tx.Query("SELECT id, is_private FROM repos"); err != nil {
			return err
		} else {
			for rows.Next() {
				var id int64
				var isPrivate bool

				if err := rows.Scan(&id, &isPrivate); err != nil {
					return err
				}

				visibilities[id] = util.If(isPrivate, Private, Public)
			}

			rows.Close()
		}

		for id, visibility := range visibilities {
			if _, err := tx.Exec("UPDATE repos SET visibility = ? WHERE id = ?", visibility, id); err != nil {
				return err
			}
		}

		/* Remove is_private column */
		if _, err := tx.Exec("ALTER TABLE repos DROP COLUMN is_private"); err != nil {
			return err
		}

		version = 3
	}

	if _, err := tx.Exec(fmt.Sprint("PRAGMA user_version = ", version)); err != nil {
		return err
	}

	if err := tx.Commit(); err != nil {
		return err
	}

	return nil
}