package api

import (
	"bytes"
	"context"
	"fmt"
	"time"

	"github.com/sirupsen/logrus"
	"github.com/taosdata/taoskeeper/db"
	"github.com/taosdata/taoskeeper/infrastructure/config"
	"github.com/taosdata/taoskeeper/infrastructure/log"
	"github.com/taosdata/taoskeeper/util"
)

var commonLogger = log.GetLogger("CMN")

func CreateDatabase(username string, password string, host string, port int, usessl bool, dbname string, databaseOptions map[string]interface{}) {
	qid := util.GetQidOwn(config.Conf.InstanceID)
	commonLogger := commonLogger.WithFields(logrus.Fields{config.ReqIDKey: qid})

	ctx := context.Background()

	conn, err := db.NewConnector(username, password, host, port, usessl)
	if err != nil {
		commonLogger.Errorf("connect to adapter error, msg:%s", err)
		return
	}

	defer closeConn(conn)

	createDBSql := generateCreateDBSql(dbname, databaseOptions)
	commonLogger.Warningf("create database sql: %s", createDBSql)

	for i := 0; i < 3; i++ {
		if _, err := conn.Exec(ctx, createDBSql, util.GetQidOwn(config.Conf.InstanceID)); err != nil {
			commonLogger.Errorf("try %v times: create database %s error, msg:%v", i+1, dbname, err)
			time.Sleep(5 * time.Second)
			continue
		}
		return
	}
}

func generateCreateDBSql(dbname string, databaseOptions map[string]interface{}) string {
	var buf bytes.Buffer
	buf.WriteString("create database if not exists ")
	buf.WriteString(dbname)

	for k, v := range databaseOptions {
		buf.WriteString(" ")
		buf.WriteString(k)
		switch v := v.(type) {
		case string:
			buf.WriteString(fmt.Sprintf(" '%s'", v))
		default:
			buf.WriteString(fmt.Sprintf(" %v", v))
		}
		buf.WriteString(" ")
	}
	return buf.String()
}

func CreatTables(username string, password string, host string, port int, usessl bool, dbname string, createList []string) {
	ctx := context.Background()
	conn, err := db.NewConnectorWithDb(username, password, host, port, dbname, usessl)
	if err != nil {
		commonLogger.Errorf("connect to database error, msg:%s", err)
		return
	}
	defer closeConn(conn)

	for _, createSql := range createList {
		commonLogger.Infof("execute sql:%s", createSql)
		if _, err = conn.Exec(ctx, createSql, util.GetQidOwn(config.Conf.InstanceID)); err != nil {
			commonLogger.Errorf("execute sql: %s, error: %s", createSql, err)
		}
	}
}

func closeConn(conn *db.Connector) {
	if err := conn.Close(); err != nil {
		commonLogger.Errorf("close connection error, msg:%s", err)
	}
}