package tests

import (
	"database/sql"
	"log"
	"os"
	"strconv"
	"testing"

	_ "github.com/go-sql-driver/mysql"

	"github.com/localhots/bocadillo/binlog"
	"github.com/localhots/bocadillo/mysql/driver"
	"github.com/localhots/bocadillo/reader"
)

var suite *testSuite

func TestMain(m *testing.M) {
	dsn, conf := getConfig()

	conn, err := sql.Open("mysql", dsn)
	if err != nil {
		log.Fatal(err)
	}

	if conf.File == "" {
		pos := getLatestOffset(conn)
		log.Printf("File is not set, using latest from master: %s @ %d", pos.File, pos.Offset)
		conf.File = pos.File
		conf.Offset = uint32(pos.Offset)
	}

	rdr, err := reader.New(dsn, conf)
	if err != nil {
		log.Fatal(err)
	}

	suite = &testSuite{
		reader: rdr,
		conn:   conn,
	}

	exitCode := m.Run()

	os.Exit(exitCode)
}

func getConfig() (dsn string, conf driver.Config) {
	envOrDefault := func(name, def string) string {
		if val := os.Getenv(name); val != "" {
			return val
		}
		return def
	}
	makeUint32 := func(str string) uint32 {
		u64, err := strconv.ParseUint(str, 10, 32)
		if err != nil {
			log.Fatal(err)
		}
		return uint32(u64)
	}

	dsn = envOrDefault("TEST_DSN", "root@(127.0.0.1:3306)/loltest")
	conf.ServerID = makeUint32(envOrDefault("TEST_SERVER_ID", "1000"))
	conf.File = envOrDefault("TEST_FILE", "")
	conf.Offset = makeUint32(envOrDefault("TEST_OFFSET", "4"))
	hostname, _ := os.Hostname()
	conf.Hostname = envOrDefault("TEST_HOSTNAME", hostname)

	return
}

func getLatestOffset(conn *sql.DB) binlog.Position {
	var pos binlog.Position
	var _discard interface{}
	err := conn.QueryRow("SHOW MASTER STATUS").Scan(
		&pos.File, &pos.Offset, &_discard, &_discard, &_discard,
	)
	if err != nil {
		log.Fatal(err)
	}
	return pos
}