package database

import (
	"fmt"
	"log"
	"math/rand"
	"os"
	"time"

	_ "github.com/go-sql-driver/mysql"
	"xorm.io/xorm"
	xlog "xorm.io/xorm/log"
	"xorm.io/xorm/names"
)

type DB struct {
	User   string
	Pass   string
	Schema string
}

type LogFiles struct {
	DebugFile *os.File
	ErrorFile *os.File
	TgLogFile *os.File
	DBLogFile *os.File
}

func OpenLogs() (files LogFiles) {
	return LogFiles{
		DebugFile: CreateLog("messages"),
		ErrorFile: CreateLog("error"),
		TgLogFile: CreateLog("tg"),
		DBLogFile: CreateLog("sql"),
	}
}

func (files *LogFiles) CloseAll() {
	files.DebugFile.Close()
	files.ErrorFile.Close()
	files.TgLogFile.Close()
	files.DBLogFile.Close()
}

func Connect(db DB, logger *os.File) (*xorm.Engine, error) {
	engine, err := xorm.NewEngine("mysql", fmt.Sprintf("%s:%s@tcp(localhost:3306)/%s?charset=utf8", db.User, db.Pass, db.Schema))
	if err != nil {
		return nil, err
	}
	sqlLogger := xlog.NewSimpleLogger(logger)
	engine.SetLogger(sqlLogger)
	engine.ShowSQL(true)
	engine.SetMapper(names.SameMapper{})

	err = engine.Sync(
		&User{},
		&TgUser{},
		&Group{},
		&Lesson{},
		&Teacher{},
		&ShedulesInUser{},
		&File{},
		&TempMsg{},
	)
	if err != nil {
		return nil, err
	}
	return engine, nil
}

func GenerateID(engine *xorm.Engine) (int64, error) {
	id := rand.Int63n(899999999) + 100000000

	exists, err := engine.ID(id).Exist(&User{})
	if err != nil {
		return 0, err
	}

	if exists {
		return GenerateID(engine)
	} else {
		return id, nil
	}
}

// TODO: изобрести раздорбление логов по дате
func CreateLog(name string) *os.File {
	if _, err := os.Stat("logs"); os.IsNotExist(err) {
		err = os.Mkdir("logs", os.ModePerm)
		if err != nil {
			log.Fatal("failed create log folder")
		}
	}
	fileName := "./logs/" + name + ".log"
	if _, err := os.Stat(fileName); !os.IsNotExist(err) {
		new := fmt.Sprintf("./logs/%s.before.%s.log", name, time.Now().Format("06-02-01-15-04-05"))
		err := os.Rename(fileName, new)
		if err != nil {
			log.Fatal(err)
		}
	}
	logFile, err := os.Create(fileName)
	if err != nil {
		log.Fatalf("failed open %s.log file: %s", name, err)
	}
	//defer logFile.Close()
	return logFile
}