package configgen

import (
	"errors"
	"fmt"
	"gitee.com/mosache/YFrame/tools/Y/vars"
	"os"
)

const (
	configTemp = `package config

import (
	"gitee.com/mosache/YFrame/core/stores/mysql"
	"gitee.com/mosache/YFrame/core/stores/redis"
	"github.com/zeromicro/go-zero/rest"
)

var (
	C Config
)

type Config struct {
	rest.RestConf

	Redis redis.Config
	Db    DBConnConfig
}

func (c Config) GetConfigs() []*mysql.Config {
	cs := []*mysql.Config{
		{
			Host:         c.Db.Main.Writer.Host,
			Port:         c.Db.Main.Writer.Port,
			UserName:     c.Db.Main.Writer.Username,
			PassWord:     c.Db.Main.Writer.Password,
			DataBaseName: c.Db.Main.Writer.DataBaseName,
		},
	}

	if c.Db.Main.Reader.Port != 0 {
		cs = append(cs, &mysql.Config{
			Host:         c.Db.Main.Reader.Host,
			Port:         c.Db.Main.Reader.Port,
			UserName:     c.Db.Main.Reader.Username,
			PassWord:     c.Db.Main.Reader.Password,
			DataBaseName: c.Db.Main.Reader.DataBaseName,
		})
	}
	return cs
}

type DBConnConfig struct {
	Main DBSourceConfig
}

type DBSourceConfig struct {
	Reader DBConfig
	Writer DBConfig
}

type DBSubSourceConfig struct {
	Name string
}

type DBConfig struct {
	Host         string
	Port         int
	Username     string
	Password     string
	DataBaseName string
}
`
)

func GenConfig() error {
	if _, err := os.Stat(vars.CodeConfigDir); err != nil {
		if os.IsNotExist(err) {
			/// create the code config dir
			err = os.MkdirAll(vars.CodeConfigDir, os.ModePerm)
			if err != nil {
				return err
			}

			/// create config.go
			if err = os.WriteFile(fmt.Sprintf("%s/%s", vars.CodeConfigDir, vars.CodeConfigFileName), []byte(configTemp), os.ModePerm); err != nil {
				return err
			}

			return nil

		} else {
			return err
		}
	}

	return errors.New("config dir is ready exist")
}