package service

import (
	"time"

	"kra/internal/conf"
	"kra/internal/service/dto"

	"google.golang.org/protobuf/types/known/durationpb"
)

func applyAdminConfig(next *conf.AdminBackend, req *dto.SetSystemConfigRequest) {
	next.RouterPrefix = req.Config.Admin.RouterPrefix
	if next.System == nil {
		next.System = &conf.AdminBackend_System{}
	}
	next.System.UseRedis = req.Config.Admin.System.UseRedis
	next.System.UseMultipoint = req.Config.Admin.System.UseMultipoint
	next.System.UseStrictAuth = req.Config.Admin.System.UseStrictAuth
	next.System.DisableAutoMigrate = req.Config.Admin.System.DisableAutoMigrate
	next.System.UseMongo = req.Config.Admin.System.UseMongo
	if req.Config.Admin.Zap != nil {
		next.Zap = req.Config.Admin.Zap
	}
	if req.Config.Admin.Cors != nil {
		next.Cors = req.Config.Admin.Cors
	}
	if req.Config.Admin.App != nil {
		next.App = req.Config.Admin.App
	}
	applyJWTConfig(next, req)
	applyCaptchaConfig(next, req)
	applyMediaConfig(next, req)
	if req.Config.Admin.Storage != nil {
		preserveStorageSecrets(req.Config.Admin.Storage, next.Storage)
		next.Storage = req.Config.Admin.Storage
	}
	if next.Email != nil && req.Config.Email != nil {
		next.Email.To, next.Email.From, next.Email.Host = req.Config.Email.To, req.Config.Email.From, req.Config.Email.Host
		next.Email.Nickname, next.Email.Port = req.Config.Email.Nickname, req.Config.Email.Port
		next.Email.IsSsl, next.Email.IsLoginAuth = req.Config.Email.IsSSL, req.Config.Email.IsLoginAuth
		if req.Config.Email.Secret != "" && req.Config.Email.Secret != "******" {
			next.Email.Secret = req.Config.Email.Secret
		}
	}
}

func applyJWTConfig(next *conf.AdminBackend, req *dto.SetSystemConfigRequest) {
	if next.Jwt == nil {
		return
	}
	if req.Config.Admin.JWT.SigningKey != "" && req.Config.Admin.JWT.SigningKey != "******" {
		next.Jwt.SigningKey = req.Config.Admin.JWT.SigningKey
	}
	if req.Config.Admin.JWT.Issuer != "" {
		next.Jwt.Issuer = req.Config.Admin.JWT.Issuer
	}
	if value, err := time.ParseDuration(req.Config.Admin.JWT.ExpiresTime); err == nil && value > 0 {
		next.Jwt.ExpiresTime = durationpb.New(value)
	}
	if value, err := time.ParseDuration(req.Config.Admin.JWT.BufferTime); err == nil && value >= 0 {
		next.Jwt.BufferTime = durationpb.New(value)
	}
}

func applyCaptchaConfig(next *conf.AdminBackend, req *dto.SetSystemConfigRequest) {
	if next.Captcha != nil {
		if req.Config.Admin.Captcha.KeyLong > 0 {
			next.Captcha.KeyLong = req.Config.Admin.Captcha.KeyLong
		}
		if req.Config.Admin.Captcha.ImgWidth > 0 {
			next.Captcha.ImgWidth = req.Config.Admin.Captcha.ImgWidth
		}
		if req.Config.Admin.Captcha.ImgHeight > 0 {
			next.Captcha.ImgHeight = req.Config.Admin.Captcha.ImgHeight
		}
		if value, err := time.ParseDuration(req.Config.Admin.Captcha.StoreExpiration); err == nil && value > 0 {
			next.Captcha.StoreExpiration = durationpb.New(value)
		}
	}
	if next.Local != nil {
		if req.Config.Admin.Local.StorePath != "" {
			next.Local.StorePath = req.Config.Admin.Local.StorePath
		}
		if req.Config.Admin.Local.PathPrefix != "" {
			next.Local.PathPrefix = req.Config.Admin.Local.PathPrefix
		}
	}
}

func applyMediaConfig(next *conf.AdminBackend, req *dto.SetSystemConfigRequest) {
	if next.Media == nil {
		return
	}
	if req.Config.Admin.Media.SessionTTL > 0 {
		next.Media.SessionTtl = req.Config.Admin.Media.SessionTTL
	}
	next.Media.MaxFileSize = req.Config.Admin.Media.MaxFileSize
}

func applyDataConfig(data *conf.Data, req *dto.SetSystemConfigRequest) *conf.Data {
	if req.Config.Data == nil {
		return data
	}
	if data == nil {
		data = &conf.Data{}
	}
	applyDatabaseConfig(data, req)
	applyRedisConfig(data, req)
	if req.Config.Data.DatabaseList != nil {
		for i, item := range req.Config.Data.DatabaseList {
			if item != nil && (item.Password == "" || item.Password == "******") && i < len(data.DatabaseList) && data.DatabaseList[i] != nil {
				item.Password = data.DatabaseList[i].Password
			}
		}
		data.DatabaseList = req.Config.Data.DatabaseList
	}
	if req.Config.Data.RedisList != nil {
		for i, item := range req.Config.Data.RedisList {
			if item != nil && (item.Password == "" || item.Password == "******") && i < len(data.RedisList) && data.RedisList[i] != nil {
				item.Password = data.RedisList[i].Password
			}
		}
		data.RedisList = req.Config.Data.RedisList
	}
	if req.Config.Data.Mongo != nil {
		if data.Mongo != nil && (req.Config.Data.Mongo.Password == "" || req.Config.Data.Mongo.Password == "******") {
			req.Config.Data.Mongo.Password = data.Mongo.Password
		}
		data.Mongo = req.Config.Data.Mongo
	}
	return data
}

func applyDatabaseConfig(data *conf.Data, req *dto.SetSystemConfigRequest) {
	value := req.Config.Data.Database
	if value == nil {
		return
	}
	password := value.Password
	if data.Database != nil && (password == "" || password == "******") {
		password = data.Database.Password
	}
	data.Database = &conf.Data_Database{Driver: value.Driver, Source: value.Source, Host: value.Host, Port: value.Port, User: value.User, Password: password, Name: value.Name, Config: value.Config, Path: value.Path, Prefix: value.Prefix, Engine: value.Engine, LogMode: value.LogMode, MaxIdleConns: value.MaxIdleConns, MaxOpenConns: value.MaxOpenConns, ConnMaxLifetime: value.ConnMaxLifetime, Singular: value.Singular}
}

func applyRedisConfig(data *conf.Data, req *dto.SetSystemConfigRequest) {
	value := req.Config.Data.Redis
	if value == nil {
		return
	}
	password := value.Password
	if data.Redis != nil && (password == "" || password == "******") {
		password = data.Redis.Password
	}
	redis := &conf.Data_Redis{Network: value.Network, Addr: value.Addr, Name: value.Name, Password: password, Db: value.DB, UseCluster: value.UseCluster, ClusterAddrs: value.ClusterAddrs}
	if duration, err := time.ParseDuration(value.ReadTimeout); err == nil && duration >= 0 {
		redis.ReadTimeout = durationpb.New(duration)
	}
	if duration, err := time.ParseDuration(value.WriteTimeout); err == nil && duration >= 0 {
		redis.WriteTimeout = durationpb.New(duration)
	}
	data.Redis = redis
}

func objectStores(storage *conf.AdminBackend_Storage) []*conf.AdminBackend_ObjectStore {
	if storage == nil {
		return nil
	}
	return []*conf.AdminBackend_ObjectStore{storage.AliyunOss, storage.HuaweiObs, storage.TencentCos, storage.AwsS3, storage.CloudflareR2, storage.Minio}
}

func maskStorageSecrets(storage *conf.AdminBackend_Storage) {
	if storage.Qiniu != nil {
		storage.Qiniu.SecretKey = "******"
	}
	for _, item := range objectStores(storage) {
		if item != nil {
			item.SecretKey = "******"
		}
	}
}

func preserveStorageSecrets(next, current *conf.AdminBackend_Storage) {
	if next == nil || current == nil {
		return
	}
	if next.Qiniu != nil && current.Qiniu != nil && (next.Qiniu.SecretKey == "" || next.Qiniu.SecretKey == "******") {
		next.Qiniu.SecretKey = current.Qiniu.SecretKey
	}
	nextItems, currentItems := objectStores(next), objectStores(current)
	for i := range nextItems {
		if nextItems[i] != nil && currentItems[i] != nil && (nextItems[i].SecretKey == "" || nextItems[i].SecretKey == "******") {
			nextItems[i].SecretKey = currentItems[i].SecretKey
		}
	}
}