package config

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

	"github.com/gofiber/fiber/v2"
	"github.com/ilyakaznacheev/cleanenv"
	"github.com/joho/godotenv"
	"github.com/oarkflow/log"
	"github.com/sujit-baniya/flash"
	"github.com/sujit-baniya/ip"
)

// Config is a application configuration structure
type AppConfig struct {
	Auth       AuthConfig `yaml:"auth"`
	Mail       Mail       `yaml:"mail"`
	Hash       Hash
	View       ViewConfig     `yaml:"view"`
	Cache      CacheConfig    `yaml:"cache"`
	Database   DatabaseConfig `yaml:"database"`
	Session    SessionConfig  `yaml:"session"`
	Queue      QueueConfig    `yaml:"queue"`
	JwtSecrets JwtSecrets     `yaml:"jwt"`
	Storage    StorageConfig  `yaml:"storage"`
	Server     ServerConfig   `yaml:"server"`
	Log        LogConfig      `yaml:"log"`
	Token      Token          `yaml:"token"`
	Profiler   ProfilerConfig `yaml:"profiler"`
	Flash      *flash.Flash
	GeoIP      *ip.GeoIpDB
	ConfigFile string
}

func (cfg *AppConfig) Setup() {
	err := godotenv.Load()
	// read configuration from the file and environment variables
	if err = cleanenv.ReadConfig(cfg.ConfigFile, cfg); err != nil {
		fmt.Println(err)
		os.Exit(2)
	}
	cfg.Server.LoadPath()
	cfg.View.Load(cfg.Server.Path)
	cfg.Mail.View = &cfg.View
	cfg.Server.TemplateEngine = cfg.View.Template.TemplateEngine
	cfg.Server.Setup()
	cfg.LoadComponents()
	if cfg.Auth.Type == "casbin" {
		modelFile := filepath.Join(cfg.Server.AssetPath, "rbac_model.conf")
		cfg.Auth.Setup(cfg.Database.DB, modelFile)
	}
	path := MakeDir(filepath.Join(cfg.Server.AssetPath, "GeoLite2-City.mmdb"))
	cfg.GeoIP = ip.NewGeoIpDB(path)
}

func (cfg *AppConfig) PrepareLog() {
	writer := &log.MultiWriter{}
	path := MakeDir(filepath.Join(cfg.Server.Path, cfg.Log.InfoLevel.Path))
	writer.InfoWriter = &log.FileWriter{Filename: filepath.Join(path, "INFO.log"), EnsureFolder: true, TimeFormat: cfg.Log.InfoLevel.TimeFormat}

	path = MakeDir(filepath.Join(cfg.Server.Path, cfg.Log.WarnLevel.Path))
	writer.WarnWriter = &log.FileWriter{Filename: filepath.Join(cfg.Server.Path, cfg.Log.WarnLevel.Path, "WARN.log"), EnsureFolder: true, TimeFormat: cfg.Log.WarnLevel.TimeFormat}

	path = MakeDir(filepath.Join(cfg.Server.Path, cfg.Log.ErrorLevel.Path))
	writer.ErrorWriter = &log.FileWriter{Filename: filepath.Join(cfg.Server.Path, cfg.Log.ErrorLevel.Path, "ERROR.log"), EnsureFolder: true, TimeFormat: cfg.Log.ErrorLevel.TimeFormat}
	if cfg.Log.ConsoleLog.Show {
		writer.ConsoleWriter = &log.IOWriter{Writer: os.Stderr}
		writer.ConsoleLevel = log.InfoLevel
	}
	log.DefaultLogger = log.Logger{
		TimeField:  cfg.Log.TimeField,
		TimeFormat: cfg.Log.TimeFormat,
		Writer:     writer,
	}
}

func (cfg *AppConfig) Route404() {
	cfg.Server.Use(func(c *fiber.Ctx) error {
		return c.Status(fiber.StatusNotFound).SendString("Page not found")
	})
}

func (cfg *AppConfig) LoadComponents() {
	cfg.Flash = flash.New(flash.Config{
		Name:     "fiber",
		HTTPOnly: true,
	})
	cfg.LoadStatic()
	cfg.PrepareLog()
	_ = cfg.Database.Setup()
	_ = cfg.Session.Setup()
	cfg.Cache.Setup()
	cfg.Storage.Setup()
}

func (cfg *AppConfig) LoadStatic() {
	cfg.Server.Static("/websocket", "./resources/views/websocket.html")
	cfg.Server.Static("/", filepath.Join(cfg.Server.Path, cfg.Server.PublicPath), fiber.Static{
		Compress:      true,
		ByteRange:     true,
		CacheDuration: 24 * time.Hour,
	})
}

func (cfg *AppConfig) LoadSpamDetectionEngine() {

}