nginx-ui/server/modules/auth_token/controller.go

package auth_token

import (
	"github.com/astaxie/beego/logs"
	"nginx-ui/server/base"
	"nginx-ui/server/models"
	"nginx-ui/server/vo"
)

type Controller struct {
	base.Controller
	service *Service
}

func NewController() *Controller {
	return &Controller{
		service: NewService(),
	}
}

var AuthController = NewController()

// List 获取全部用户信息
func (c *Controller) List() {
	user := c.RequiredUser()
	if user == nil {
		return
	}
	req := vo.PageReq{}
	if !c.ReadBody(&req) {
		return
	}
	query := &ListQuery{Uid: user.Id}
	resp, err := c.service.List(query, &req)
	if err != nil {
		logs.Warn("Users get fail: %v", err)
		c.ErrorJson(err)
		return
	}
	c.SetData(resp).Json()
}

// Create 用户注册
func (c *Controller) Create() {
	user := c.RequiredUser()
	if user == nil {
		return
	}
	req := models.AuthToken{}
	if !c.ReadBody(&req) {
		return
	}
	req.Uid = user.Id
	err := c.service.Creat(&req)
	if err != nil {
		logs.Warn("Users get fail: %v", err)
		c.ErrorJson(err)
		return
	}
	c.SetData(req).Json()
}

// Update 更新Token
func (c *Controller) Update() {
	user := c.RequiredUser()
	if user == nil {
		return
	}
	req := models.AuthToken{}
	if !c.ReadBody(&req) {
		return
	}
	err := c.service.Update(&req)
	if err != nil {
		logs.Warn("Users get fail: %v", err)
		c.ErrorJson(err)
		return
	}
	c.SetData(req).Json()
}

func (c *Controller) GetDetail() {
	// 获取全部用户信息
	user := c.RequiredUser()
	if user == nil {
		return
	}
	id, err := c.GetIntQuery("id")
	if err != nil {
		logs.Warn("AuthToken get fail: %v", err)
		c.ErrorJson(err)
		return
	}
	query, err := c.service.GetById(id)
	if err != nil {
		logs.Warn("AuthToken get fail: %v", err)
		c.ErrorJson(err)
		return
	}
	c.SetData(query).Json()
}

func (c *Controller) DeleteById() {
	req := models.AuthToken{}
	if !c.ReadBody(&req) {
		return
	}
	err := c.service.DeleteById(req.Id)
	if err != nil {
		logs.Warn("AuthToken delete fail: %v", err)
		c.ErrorJson(err)
		return
	}
	c.Json()
}