nginx-ui/server/api/config.go

package api

import (
    "github.com/0xJacky/Nginx-UI/tool"
    "github.com/gin-gonic/gin"
    "io/ioutil"
    "log"
    "net/http"
    "os"
    "path/filepath"
)

func GetConfigs(c *gin.Context) {
    orderBy := c.Query("order_by")
    sort := c.DefaultQuery("sort", "desc")

    mySort := map[string]string{
        "name": "string",
        "modify": "time",
    }

    configFiles, err := ioutil.ReadDir(tool.GetNginxConfPath("/"))

	if err != nil {
		ErrorHandler(c, err)
		return
	}

	var configs []gin.H

	for i := range configFiles {
		file := configFiles[i]

		if !file.IsDir() && "." != file.Name()[0:1] {
			configs = append(configs, gin.H{
				"name":   file.Name(),
				"size":   file.Size(),
				"modify": file.ModTime(),
			})
		}
	}

    configs = tool.Sort(orderBy, sort, mySort[orderBy], configs)

	c.JSON(http.StatusOK, gin.H{
		"configs": configs,
	})
}

func GetConfig(c *gin.Context) {
	name := c.Param("name")
	path := filepath.Join(tool.GetNginxConfPath("/"), name)

	content, err := ioutil.ReadFile(path)

	if err != nil {
		ErrorHandler(c, err)
		return
	}

	c.JSON(http.StatusOK, gin.H{
		"config": string(content),
	})

}

type AddConfigJson struct {
	Name    string `json:"name" binding:"required"`
	Content string `json:"content" binding:"required"`
}

func AddConfig(c *gin.Context) {
	var request AddConfigJson
	err := c.BindJSON(&request)
	if err != nil {
		ErrorHandler(c, err)
		return
	}

	name := request.Name
	content := request.Content

	path := filepath.Join(tool.GetNginxConfPath("/"), name)

	log.Println(path)
	if _, err = os.Stat(path); err == nil {
		c.JSON(http.StatusNotAcceptable, gin.H{
			"message": "config exist",
		})
		return
	}

	if content != "" {
		err := ioutil.WriteFile(path, []byte(content), 0644)
		if err != nil {
			ErrorHandler(c, err)
			return
		}
	}

    output := tool.ReloadNginx()

    if output != "" {
        c.JSON(http.StatusInternalServerError, gin.H{
            "message": output,
        })
        return
    }

	c.JSON(http.StatusOK, gin.H{
		"name":    name,
		"content": content,
	})

}

type EditConfigJson struct {
	Content string `json:"content" binding:"required"`
}

func EditConfig(c *gin.Context) {
	name := c.Param("name")
	var request EditConfigJson
	err := c.BindJSON(&request)
	if err != nil {
		ErrorHandler(c, err)
		return
	}
	path := filepath.Join(tool.GetNginxConfPath("/"), name)
	content := request.Content

	origContent, err := ioutil.ReadFile(path)
	if err != nil {
		ErrorHandler(c, err)
		return
	}

	if content != "" && content != string(origContent) {
		// model.CreateBackup(path)
		err := ioutil.WriteFile(path, []byte(content), 0644)
		if err != nil {
			ErrorHandler(c, err)
			return
		}
	}

    output := tool.ReloadNginx()

    if output != "" {
        c.JSON(http.StatusInternalServerError, gin.H{
            "message": output,
        })
        return
    }

	GetConfig(c)
}