package sql

import (
	"fmt"
	"io"
	"log"
	"os"
	"path"
	"path/filepath"
	"time"

	"git.apinb.com/bsm-sdk/engine/utils"
	"git.apinb.com/bsm-sdk/engine/vars"
	"gorm.io/gorm/logger"
)

var (
	F *os.File
)

func setLogger(serviceName string, stdout bool) logger.Interface {

	// create log dir
	absPath := path.Join(utils.GetCurrentPath(), vars.SqlLogDir, serviceName)
	if !utils.PathExists(absPath) {
		os.MkdirAll(absPath, os.ModePerm)
	}

	// open log file
	fullPath := getLogFileFullPath(absPath, getLogFileName(serviceName))
	F = openLogFile(fullPath)

	// write log
	var multiOutput io.Writer
	if stdout {
		multiOutput = io.MultiWriter(os.Stdout, F)
	} else {
		multiOutput = io.MultiWriter(F)
	}

	return logger.New(log.New(multiOutput, "\r\n", log.LstdFlags), logger.Config{
		SlowThreshold:             200 * time.Millisecond,
		IgnoreRecordNotFoundError: false,
		Colorful:                  false,
	})
}

func getLogFileName(serviceName string) string {
	return fmt.Sprintf("sql_%s.%s", time.Now().Format(vars.YYYYMMDD), vars.SqlLogFileExt)
}

func getLogFileFullPath(dir, fn string) string {
	return filepath.Join(dir, fn)
}

func openLogFile(filePath string) *os.File {
	handle, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
	if err != nil {
		log.Printf("Fail to OpenFile :%v", err)
	}

	return handle
}