router.go

package api

import (
	"git.zgwit.com/zgwit/iot-admin/conf"
	"git.zgwit.com/zgwit/iot-admin/models"
	"github.com/gin-contrib/sessions"
	"github.com/gin-gonic/gin"
	"net/http"
	"reflect"
)

type paramFilter struct {
	Key    string        `form:"key"`
	Values []interface{} `form:"value"`
}

type paramSearch struct {
	Offset    int           `form:"offset"`
	Length    int           `form:"length"`
	SortKey   string        `form:"sortKey"`
	SortOrder string        `form:"sortOrder"`
	Filters   []paramFilter `form:"filters"`
	Keyword   string        `form:"keyword"`
}

type paramId struct {
	Id int64 `uri:"id"`
}

type paramId2 struct {
	Id  int64 `uri:"id"`
	Id2 int64 `uri:"id2"`
}

func mustLogin(c *gin.Context) {
	session := sessions.Default(c)
	if user := session.Get("user"); user != nil {
		c.Next()
	} else {
		//TODO 检查OAuth2返回的code，进一步获取用户信息，放置到session中

		c.JSON(http.StatusUnauthorized, gin.H{"ok": false, "error": "Unauthorized"})
		c.Abort()
	}
}

func RegisterRoutes(app *gin.RouterGroup) {

	if conf.Config.SysAdmin.Enable {
		//检查 session，必须登录
		app.Use(mustLogin)
	} else if conf.Config.BaseAuth.Enable {
		//检查HTTP认证
		app.Use(gin.BasicAuth(gin.Accounts(conf.Config.BaseAuth.Users)))
	} else {
		//支持匿名访问
	}

	//TODO 转移至子目录，并使用中间件，检查session及权限
	mod := reflect.TypeOf(models.Tunnel{})
	fields := []string{
		"name", "description", "type", "addr", "timeout",
		"register_enable", "register_regex", "register_min", "register_max",
		"heart_beat_enable", "heart_beat_interval", "heart_beat_content", "heart_beat_is_hex",
		"disabled"}
	app.POST("/project/:id/tunnels", curdApiListById(mod, "project_id"))
	app.POST("/tunnels", curdApiList(mod))
	app.POST("/tunnel", curdApiCreate(mod, nil))            //TODO 启动
	app.DELETE("/tunnel/:id", curdApiDelete(mod, nil))      //TODO 停止
	app.PUT("/tunnel/:id", curdApiModify(mod, fields, nil)) //TODO 重新启动
	app.GET("/tunnel/:id", curdApiGet(mod))

	app.GET("/tunnel/:id/start", tunnelStart)
	app.GET("/tunnel/:id/stop", tunnelStop)

	//app.POST("/channel/:id/links")

	//连接管理
	mod = reflect.TypeOf(models.Link{})
	fields = []string{"name"}
	app.POST("/tunnel/:id/links", curdApiListById(mod, "tunnel_id"))
	app.POST("/links", curdApiList(mod))
	app.DELETE("/link/:id", curdApiDelete(mod, nil)) //TODO 停止
	app.PUT("/link/:id", curdApiModify(mod, fields, nil))
	app.GET("/link/:id", curdApiGet(mod))

	mod = reflect.TypeOf(models.Device{})
	fields = []string{"name"}
	app.POST("/project/:id/devices", curdApiListById(mod, "project_id"))
	app.POST("/devices", curdApiList(mod))
	app.POST("/device", curdApiCreate(mod, nil))
	app.DELETE("/device/:id", curdApiDelete(mod, nil))
	app.PUT("/device/:id", curdApiModify(mod, fields, nil))
	app.GET("/device/:id", curdApiGet(mod))

	mod = reflect.TypeOf(models.Location{})
	fields = []string{"name"}
	app.POST("/device/:id/locations", curdApiListById(mod, "device_id"))
	//app.POST("/locations", curdApiList(mod))
	//app.POST("/location", curdApiCreate(mod, nil))
	app.DELETE("/location/:id", curdApiDelete(mod, nil))
	//app.PUT("/location/:id", curdApiModify(mod, fields, nil))
	app.GET("/location/:id", curdApiGet(mod))

	//插件管理
	mod = reflect.TypeOf(models.Plugin{})
	fields = []string{"name"}
	app.POST("/plugins", curdApiList(mod))
	app.POST("/plugin", curdApiCreate(mod, nil))
	app.DELETE("/plugin/:id", curdApiDelete(mod, nil))
	app.PUT("/plugin/:id", curdApiModify(mod, fields, nil))
	app.GET("/plugin/:id", curdApiGet(mod))

	//模型管理
	mod = reflect.TypeOf(models.Project{})
	fields = []string{"name"}
	app.POST("/projects", curdApiList(mod))
	app.POST("/project", curdApiCreate(mod, nil))
	app.DELETE("/project/:id", curdApiDelete(mod, nil))
	app.PUT("/project/:id", curdApiModify(mod, fields, nil))
	app.GET("/project/:id", curdApiGet(mod))

	//app.GET("/project/:id/tunnels", nop)
	//app.GET("/project/:id/variables", nop)
	//app.GET("/project/:id/batches", nop)
	//app.GET("/project/:id/jobs", nop)
	//app.GET("/project/:id/strategies", nop)

	app.POST("/project-import", projectImport)
	app.GET("/project/:id/export", projectExport)
	app.GET("/project/:id/deploy", projectDeploy)

	
	mod = reflect.TypeOf(models.ProjectElement{})
	fields = []string{"name"}
	app.POST("/project/:id/elements", curdApiListById(mod, "project_id"))
	//app.POST("/project-elements", curdApiList(mod))
	app.POST("/project-element", curdApiCreate(mod, nil))
	app.DELETE("/project-element/:id", curdApiDelete(mod, nil))
	app.PUT("/project-element/:id", curdApiModify(mod, fields, nil))
	app.GET("/project-element/:id", curdApiGet(mod))

	mod = reflect.TypeOf(models.ProjectJob{})
	fields = []string{"name"}
	app.POST("/project/:id/jobs", curdApiListById(mod, "project_id"))
	//app.POST("/project-jobs", curdApiList(mod))
	app.POST("/project-job", curdApiCreate(mod, nil))
	app.DELETE("/project-job/:id", curdApiDelete(mod, nil))
	app.PUT("/project-job/:id", curdApiModify(mod, fields, nil))
	app.GET("/project-job/:id", curdApiGet(mod))

	mod = reflect.TypeOf(models.ProjectStrategy{})
	fields = []string{"name"}
	app.POST("/project/:id/strategies", curdApiListById(mod, "project_id"))
	//app.POST("/project-strategies", curdApiList(mod))
	app.POST("/project-strategy", curdApiCreate(mod, nil))
	app.DELETE("/project-strategy/:id", curdApiDelete(mod, nil))
	app.PUT("/project-strategy/:id", curdApiModify(mod, fields, nil))
	app.GET("/project-strategy/:id", curdApiGet(mod))

	//元件管理
	mod = reflect.TypeOf(models.Element{})
	fields = []string{"name"}
	app.POST("/elements", curdApiList(mod))
	app.POST("/element", curdApiCreate(mod, nil))
	app.DELETE("/element/:id", curdApiDelete(mod, nil))
	app.PUT("/element/:id", curdApiModify(mod, fields, nil))
	app.GET("/element/:id", curdApiGet(mod))

	//元件变量
	mod = reflect.TypeOf(models.ElementVariable{})
	fields = []string{"name"}
	app.POST("/element/:id/variables", curdApiListById(mod, "element_id"))
	//app.POST("/element-variables", curdApiList(mod))
	app.POST("/element-variable", curdApiCreate(mod, nil))
	app.DELETE("/element-variable/:id", curdApiDelete(mod, nil))
	app.PUT("/element-variable/:id", curdApiModify(mod, fields, nil))
	app.GET("/element-variable/:id", curdApiGet(mod))

	//元件批量操作
	mod = reflect.TypeOf(models.ElementBatch{})
	fields = []string{"name"}
	app.POST("/element/:id/batches", curdApiListById(mod, "element_id"))
	//app.POST("/element-batches", curdApiList(mod))
	app.POST("/element-batch", curdApiCreate(mod, nil))
	app.DELETE("/element-batch/:id", curdApiDelete(mod, nil))
	app.PUT("/element-batch/:id", curdApiModify(mod, fields, nil))
	app.GET("/element-batch/:id", curdApiGet(mod))

}

func replyOk(ctx *gin.Context, data interface{}) {
	ctx.JSON(http.StatusOK, gin.H{
		"ok":   true,
		"data": data,
	})
}

func replyFail(ctx *gin.Context, err string) {
	ctx.JSON(http.StatusOK, gin.H{
		"ok":    false,
		"error": err,
	})
}

func replyError(ctx *gin.Context, err error) {
	ctx.JSON(http.StatusOK, gin.H{
		"ok":    false,
		"error": err.Error(),
	})
}

func nop(c *gin.Context) {
	c.String(http.StatusForbidden, "Unsupported")
}