gosora/install/mssql.go

/*
*
* Gosora MSSQL Interface
* Copyright Azareal 2017 - 2018
*
 */
package install

import (
    "bytes"
    "database/sql"
    "fmt"
    "io/ioutil"
    "log"
    "net/url"
    "path/filepath"
    "strconv"
    "strings"

    "github.com/Azareal/Gosora/query_gen"
    _ "github.com/denisenkom/go-mssqldb"
)

func init() {
    adapters["mssql"] = &MssqlInstaller{dbHost: ""}
}

type MssqlInstaller struct {
    db         *sql.DB
    dbHost     string
    dbUsername string
    dbPassword string
    dbName     string
    dbInstance string
    dbPort     string
}

func (ins *MssqlInstaller) SetConfig(dbHost string, dbUsername string, dbPassword string, dbName string, dbPort string) {
    ins.dbHost = dbHost
    ins.dbUsername = dbUsername
    ins.dbPassword = dbPassword
    ins.dbName = dbName
    ins.dbInstance = "" // You can't set this from the installer right now, it allows you to connect to a named instance instead of a port
    ins.dbPort = dbPort
}

func (ins *MssqlInstaller) Name() string {
    return "mssql"
}

func (ins *MssqlInstaller) DefaultPort() string {
    return "1433"
}

func (ins *MssqlInstaller) InitDatabase() (err error) {
    query := url.Values{}
    query.Add("database", ins.dbName)
    u := &url.URL{
        Scheme:   "sqlserver",
        User:     url.UserPassword(ins.dbUsername, ins.dbPassword),
        Host:     ins.dbHost + ":" + ins.dbPort,
        Path:     ins.dbInstance,
        RawQuery: query.Encode(),
    }
    log.Print("u.String() ", u.String())

    db, err := sql.Open("mssql", u.String())
    if err != nil {
        return err
    }

    // Make sure that the connection is alive..
    err = db.Ping()
    if err != nil {
        return err
    }
    fmt.Println("Successfully connected to the database")

    // TODO: Create the database, if it doesn't exist

    // Ready the query builder
    ins.db = db
    qgen.Builder.SetConn(db)
    return qgen.Builder.SetAdapter("mssql")
}

func (ins *MssqlInstaller) TableDefs() (err error) {
    //fmt.Println("Creating the tables")
    files, _ := ioutil.ReadDir("./schema/mssql/")
    for _, f := range files {
        if !strings.HasPrefix(f.Name(), "query_") {
            continue
        }

        var table, ext string
        table = strings.TrimPrefix(f.Name(), "query_")
        ext = filepath.Ext(table)
        if ext != ".sql" {
            continue
        }
        table = strings.TrimSuffix(table, ext)

        // ? - This is mainly here for tests, although it might allow the installer to overwrite a production database, so we might want to proceed with caution
        _, err = ins.db.Exec("DROP TABLE IF EXISTS [" + table + "];")
        if err != nil {
            fmt.Println("Failed query:", "DROP TABLE IF EXISTS ["+table+"]")
            return err
        }

        fmt.Println("Creating table '" + table + "'")
        data, err := ioutil.ReadFile("./schema/mssql/" + f.Name())
        if err != nil {
            return err
        }
        data = bytes.TrimSpace(data)

        _, err = ins.db.Exec(string(data))
        if err != nil {
            fmt.Println("Failed query:", string(data))
            return err
        }
    }
    return nil
}

func (ins *MssqlInstaller) InitialData() (err error) {
    //fmt.Println("Seeding the tables")
    data, err := ioutil.ReadFile("./schema/mssql/inserts.sql")
    if err != nil {
        return err
    }
    data = bytes.TrimSpace(data)

    statements := bytes.Split(data, []byte(";"))
    for key, statement := range statements {
        if len(statement) == 0 {
            continue
        }

        fmt.Println("Executing query #" + strconv.Itoa(key) + " " + string(statement))
        _, err = ins.db.Exec(string(statement))
        if err != nil {
            return err
        }
    }
    return nil
}

func (ins *MssqlInstaller) CreateAdmin() error {
    return createAdmin()
}

func (ins *MssqlInstaller) DBHost() string {
    return ins.dbHost
}

func (ins *MssqlInstaller) DBUsername() string {
    return ins.dbUsername
}

func (ins *MssqlInstaller) DBPassword() string {
    return ins.dbPassword
}

func (ins *MssqlInstaller) DBName() string {
    return ins.dbName
}

func (ins *MssqlInstaller) DBPort() string {
    return ins.dbPort
}