caddydash/api/config.go

package api

import (
	"caddydash/config"
	"caddydash/db"
	"caddydash/gen"
	"fmt"
	"os"

	"github.com/infinite-iroha/touka"
)

func GetConfig(cdb *db.ConfigDB) touka.HandlerFunc {
	return func(c *touka.Context) {
		filename := c.Param("filename")
		params, err := cdb.GetParams(filename)
		if err != nil {
			c.JSON(500, touka.H{"error": err.Error()})
			return
		}
		// 解码[]byte的gob数据
		var config gen.CaddyUniConfig
		err = gen.DecodeGobConfig(params.ParamsOrigin, &config)
		if err != nil {
			c.JSON(500, touka.H{"error": err.Error()})
			return
		}
		c.JSON(200, config)
	}
}

func PutConfig(cdb *db.ConfigDB, cfg *config.Config) touka.HandlerFunc {
	return func(c *touka.Context) {
		filename := c.Param("filename")
		var config gen.CaddyUniConfig
		err := c.ShouldBindJSON(&config)
		if err != nil {
			c.JSON(500, touka.H{"error": err.Error()})
			return
		}

		var paramsGOB []byte
		var paramsOrigin []byte

		// Mode标识符固定为uni, 模板已被统合为只有uni
		paramsGOB, err = gen.EncodeGobConfig(config)
		if err != nil {
			c.Warnf("encode gob config error: %v", err)
			c.JSON(500, touka.H{"error": err.Error()})
			return
		}
		// 把json变为gob []byte
		paramsOrigin, err = gen.EncodeGobConfig(config)
		if err != nil {
			c.Warnf("encode origin config error: %v", err)
			c.JSON(500, touka.H{"error": err.Error()})
			return
		}
		paramsEntry := db.ParamsEntry{
			Filename:     filename,
			TemplateType: config.Mode,
			ParamsGOB:    paramsGOB,
			ParamsOrigin: paramsOrigin,
		}

		err = WriteConfig(cdb, paramsEntry, cfg, filename)
		if err != nil {
			c.Warnf("write config error: %v", err)
			c.JSON(500, touka.H{"error": err.Error()})
			return
		}

		c.JSON(200, touka.H{"message": "config saved and rendered"})
		return
	}
}

// 渲染并写入配置
func WriteConfig(cdb *db.ConfigDB, paramsEntry db.ParamsEntry, cfg *config.Config, filename string) error {
	var err error
	err = cdb.SaveParams(paramsEntry)
	if err != nil {
		err = fmt.Errorf("save params error: %w", err)
		return err
	}
	err = gen.RenderConfig(filename, cdb)
	if err != nil {
		err = fmt.Errorf("render config error: %w", err)
		return err
	}
	// 写入文件
	renderedEntry, err := cdb.GetRenderedConfig(filename)
	if err != nil {
		err = fmt.Errorf("get rendered config error: %w", err)
		return err
	}
	err = os.WriteFile(cfg.Server.CaddyDir+"config.d/"+filename, renderedEntry.RenderedContent, 0644)
	if err != nil {
		err = fmt.Errorf("write rendered config file error: %w", err)
		return err
	}
	return nil
}

func DeleteConfig(cdb *db.ConfigDB, cfg *config.Config) touka.HandlerFunc {
	return func(c *touka.Context) {
		filename := c.Param("filename")
		err := cdb.DeleteParams(filename)
		if err != nil {
			c.JSON(500, touka.H{"error": err.Error()})
			return
		}
		// 删除文件
		err = os.Remove(cfg.Server.CaddyDir + "config.d/" + filename)
		if err != nil {
			c.Warnf("delete rendered config file error: %v", err)
		}
		c.JSON(200, touka.H{"message": "config deleted"})
	}
}

/*
func PutConfig(cdb *db.ConfigDB, cfg *config.Config) touka.HandlerFunc {
	return func(c *touka.Context) {
		filename := c.Param("filename")
		var config gen.CaddyUniConfig
		err := c.ShouldBindJSON(&config)
		if err != nil {
			c.JSON(500, touka.H{"error": err.Error()})
			return
		}
		var (
			paramsGOB    []byte
			paramsOrigin []byte
		)
		switch config.TmplType {
		case "file_server":
			caddyfscfg := gen.CaddyFileServerConfig{
				Domain:          config.Domain,
				FileDirPath:     config.FileServer.FileDirPath,
				EnableBrowser:   config.FileServer.EnableBrowser,
				Headers:         gen.HeadersMapToHeadersUp(config.Headers),
				EnableLog:       config.Log.EnableLog,
				LogDomain:       config.Log.LogDomain,
				EnableErrorPage: config.ErrorPage.EnableErrorPage,
				EnableEncode:    config.Encode.EnableEncode,
			}
			paramsGOB, err = gen.EncodeGobConfig(caddyfscfg)
			if err != nil {
				c.Warnf("encode gob config error: %v", err)
				c.JSON(500, touka.H{"error": err.Error()})
				return
			}
			// 把json变为gob []byte
			paramsOrigin, err = gen.EncodeGobConfig(config)
			if err != nil {
				c.Warnf("encode origin config error: %v", err)
				c.JSON(500, touka.H{"error": err.Error()})
				return
			}
			paramsEntry := db.ParamsEntry{
				Filename:     filename,
				TemplateType: config.TmplType,
				ParamsGOB:    paramsGOB,
				ParamsOrigin: paramsOrigin,
			}

			err = WriteConfig(cdb, paramsEntry, cfg, filename)
			if err != nil {
				c.Warnf("write config error: %v", err)
				c.JSON(500, touka.H{"error": err.Error()})
				return
			}

			c.JSON(200, touka.H{"message": "config saved and rendered"})
			return

		case "reverse_proxy":
			caddyrpCfg := gen.CaddyReverseProxyConfig{
				Domain:          config.Domain,
				ReverseProxy:    config.UpStream.UpStream,
				Headers:         gen.HeadersMapToHeadersUp(config.Headers),
				HeadersUp:       gen.HeadersMapToHeadersUp(config.UpStream.UpStreamHeaders),
				EnableLog:       config.Log.EnableLog,
				LogDomain:       config.Log.LogDomain,
				EnableErrorPage: config.ErrorPage.EnableErrorPage,
				EnableEncode:    config.Encode.EnableEncode,
			}
			paramsGOB, err = gen.EncodeGobConfig(caddyrpCfg)
			if err != nil {
				c.Warnf("encode gob config error: %v", err)
				c.JSON(500, touka.H{"error": err.Error()})
				return
			}
			// 把json变为gob []byte
			paramsOrigin, err = gen.EncodeGobConfig(config)
			if err != nil {
				c.Warnf("encode origin config error: %v", err)
				c.JSON(500, touka.H{"error": err.Error()})
				return
			}
			paramsEntry := db.ParamsEntry{
				Filename:     filename,
				TemplateType: config.TmplType,
				ParamsGOB:    paramsGOB,
				ParamsOrigin: paramsOrigin,
			}

			err = WriteConfig(cdb, paramsEntry, cfg, filename)
			if err != nil {
				c.Warnf("write config error: %v", err)
				c.JSON(500, touka.H{"error": err.Error()})
				return
			}

			c.JSON(200, touka.H{"message": "config saved and rendered"})
			return

		default:
			c.JSON(500, touka.H{"error": "unknown template type"})
			return
		}
	}
}
*/