zpan/internal/app/api/system.go

package api

import (
	"fmt"

	"github.com/gin-gonic/gin"
	"github.com/saltbo/gopkg/ginutil"
	"github.com/saltbo/gopkg/strutil"
	"github.com/spf13/viper"

	"github.com/saltbo/zpan/internal/app/dao"
	"github.com/saltbo/zpan/internal/app/model"
	"github.com/saltbo/zpan/internal/app/service"
	"github.com/saltbo/zpan/internal/pkg/bind"
	"github.com/saltbo/zpan/internal/pkg/middleware"
	"github.com/saltbo/zpan/internal/pkg/provider"
)

type Option struct {
	sOption *service.Option
}

func NewOptionResource() *Option {
	return &Option{
		sOption: service.NewOption(),
	}
}

func (rs *Option) Register(router *gin.RouterGroup) {
	router.PUT("/system/database", rs.setupDatabase)
	router.PUT("/system/account", rs.createAdministrator)

	router.Use(middleware.Installer)
	router.Use(middleware.LoginAuth())
	router.GET("/system/providers", rs.providers)
	router.GET("/system/matter-path-envs", rs.matterPathEnvs)
	router.GET("/system/options/:name", rs.find)
	router.PUT("/system/options/:name", rs.update)
}

func (rs *Option) setupDatabase(c *gin.Context) {
	if viper.IsSet("installed") {
		ginutil.JSONBadRequest(c, fmt.Errorf("datebase config already installed"))
		return
	}

	p := make(map[string]string)
	if err := c.ShouldBind(&p); err != nil {
		ginutil.JSONBadRequest(c, err)
		return
	}

	if err := dao.Init(p["driver"], p["dsn"]); err != nil {
		ginutil.JSONServerError(c, err)
		return
	}

	if err := dao.NewOption().Init(); err != nil {
		ginutil.JSONServerError(c, err)
		return
	}

	if err := service.EnsureJWTSecret(); err != nil {
		ginutil.JSONServerError(c, err)
		return
	}

	viper.Set("database.driver", p["driver"])
	viper.Set("database.dsn", p["dsn"])
	cfgFile := viper.ConfigFileUsed()
	if cfgFile == "" {
		cfgFile = "/etc/zpan/config.yml"
	}
	if err := viper.WriteConfigAs(cfgFile); err != nil {
		ginutil.JSONServerError(c, err)
		return
	}

	ginutil.JSON(c)
}

func (rs *Option) createAdministrator(c *gin.Context) {
	if viper.IsSet("installed") {
		ginutil.JSONBadRequest(c, fmt.Errorf("datebase config already installed"))
		return
	}

	p := new(bind.BodyUserCreation)
	if err := c.ShouldBind(&p); err != nil {
		ginutil.JSONBadRequest(c, err)
		return
	}
	// 创建基本信息
	user := &model.User{
		Email:    p.Email,
		Username: "admin",
		Password: strutil.Md5Hex(p.Password),
		Roles:    "admin",
		Ticket:   strutil.RandomText(6),
		Status:   model.StatusActivated,
	}
	if _, err := dao.NewUser().Create(user, 0); err != nil {
		ginutil.JSONServerError(c, err)
		return
	}

	viper.Set("installed", true)
	ginutil.JSON(c)

}

func (rs *Option) providers(c *gin.Context) {
	ginutil.JSONData(c, provider.GetProviders())
}

func (rs *Option) find(c *gin.Context) {
	ret, err := dao.NewOption().Get(c.Param("name"))
	if err != nil {
		ginutil.JSONBadRequest(c, err)
		return
	}

	ginutil.JSONData(c, ret)
}

func (rs *Option) update(c *gin.Context) {
	p := make(map[string]interface{})
	if err := c.ShouldBind(&p); err != nil {
		ginutil.JSONBadRequest(c, err)
		return
	}

	if err := rs.sOption.Update(c.Param("name"), p); err != nil {
		ginutil.JSONServerError(c, err)
		return
	}

	ginutil.JSON(c)
}

func (rs *Option) matterPathEnvs(c *gin.Context) {
	ginutil.JSONData(c, model.SupportEnvs)
}