tmrgo/routers/api/user.go

package api

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"

	"../../pkg/app"
	"../../pkg/e"
	"../../pkg/util"
	"../../routers/restful"
	"../../service/user_service"
	"github.com/Anderson-Lu/gofasion/gofasion"
	"github.com/astaxie/beego/logs"
	"github.com/astaxie/beego/validation"
	"github.com/gin-gonic/gin"
)

type auth struct {
	Id         int    `json:"id"`
	Username   string `json:"username"`
	Password   string `json:"password"`
	CaptchaKey string `json:"CaptchaKey"`
	Role       int    `json:"role_id"`
	Imei       string `json:"imei"`
	PastureId  string `json:"pastureid"`
	TypeIn     int    `json:"typein"`
}

// @Summary   获取登录token 信息
// @Tags auth
// @Accept json
// @Produce  json
// @Param username formData string true  "admin"
// @Param password formData string true  "123456"
// @Success 200 {string} json "{ "code": 200e, "data": { "token": "xxx" }, "msg": "ok" }"
// @Failure 400 {string} json "{"code":400,  "data":null,"msg":"请求参数错误"}"
// @Failure 404 {string} json "{ "code": 404, "data":null,"msg":"请求参数错误"}"
// @Router /auth  [POST]
func Auth(c *gin.Context) {

	appG := app.Gin{C: c}
	var reqInfo auth
	err := c.BindJSON(&reqInfo)
	//c.Request.ParseForm()
	//reqInfo.Username = c.PostForm("username")
	//reqInfo.Password = c.PostForm("pwd")

	valid := validation.Validation{}
	valid.MaxSize(reqInfo.Username, 100, "username").Message("最长为100字符")
	valid.MaxSize(reqInfo.Password, 100, "password").Message("最长为100字符")

	if valid.HasErrors() {
		app.MarkErrors(valid.Errors)
		appG.Response(http.StatusInternalServerError, e.ERROR_ADD_FAIL, valid.Errors)
		return
	}

	authService := user_service.User{Username: reqInfo.Username, Password: reqInfo.Password}
	isExist, err := authService.Check()

	if err != nil {
		appG.Response(http.StatusInternalServerError, e.ERROR_AUTH_CHECK_TOKEN_FAIL, err)
		return
	}

	if !isExist {
		//appG.Response(http.StatusUnauthorized, e.ERROR_AUTH, nil)
		appG.Response(http.StatusOK, e.ERROR_AUTH, "用户名或密码错误")
		return
	}

	token, err := util.GenerateToken(reqInfo.Username, reqInfo.Password)
	if err != nil {
		appG.Response(http.StatusInternalServerError, e.ERROR_AUTH_TOKEN, err)
		return
	}

	appG.Response(http.StatusOK, e.SUCCESS, map[string]string{
		"token": token,
	})
}

// @Summary   获取登录token 信息
// @Tags auth
// @Accept json
// @Produce  json
// @Param username formData string true  "admin"
// @Param password formData string true  "123456"
// @Success 200 {string} json "{ "code": 200e, "data": { "token": "xxx" }, "msg": "ok" }"
// @Failure 400 {string} json "{"code":400,  "data":null,"msg":"请求参数错误"}"
// @Failure 404 {string} json "{ "code": 404, "data":null,"msg":"请求参数错误"}"
// @Router /auth  [POST]
func AuthLogin(c *gin.Context) {

	appG := app.Gin{C: c}
	var reqInfo auth
	err := c.BindJSON(&reqInfo)
	fmt.Println("++++++++++++++++", reqInfo)

	valid := validation.Validation{}
	valid.MaxSize(reqInfo.Username, 100, "username").Message("最长为100字符")

	if valid.HasErrors() {
		app.MarkErrors(valid.Errors)
		appG.Response(http.StatusInternalServerError, e.ERROR_ADD_FAIL, valid.Errors)
		return
	}
	err = restful.CheckUserFace(reqInfo.Username, reqInfo.Imei, reqInfo.TypeIn)
	if err != nil {
		appG.Response(http.StatusInternalServerError, e.ERROR_AUTH_CHECK_TOKEN_FAIL, err.Error())
		return
	}
	token, err := util.GenerateToken(reqInfo.Username, reqInfo.Password)
	if err != nil {
		appG.Response(http.StatusInternalServerError, e.ERROR_AUTH_TOKEN, err.Error())
		return
	}

	appG.Response(http.StatusOK, e.SUCCESS, map[string]string{
		"token": token,
	})
}

// @Summary   获取单个用户信息
// @Tags  users
// @Accept json
// @Produce  json
// @Param  id  body  int true "id"
// @Success 200 {string} json "{ "code": 200, "data": {}, "msg": "ok" }"
// @Router /authdata/userinfo  [GET]
func UserLogout(c *gin.Context) {
	appG := app.Gin{C: c}
	data := make(map[string]interface{})
	data["name"] = ""
	data["avatar"] = ""
	data["introduction"] = ""
	appG.Response(http.StatusOK, e.SUCCESS, data)
}

// @Summary   获取单个用户信息
// @Tags  users
// @Accept json
// @Produce  json
// @Param  id  path  int true "iddd"
// @Success 200 {string} json "{ "code": 200, "data": {}, "msg": "ok" }"
// @Router /authdata/userinfo  [GET]
func GetUserinfo(c *gin.Context) {
	appG := app.Gin{C: c}
	data := restful.GetUserInfo(c.MustGet("jwt_username").(string))
	appG.Response(http.StatusOK, e.SUCCESS, data)
}

func GetOpenID(c *gin.Context) {
	code := c.Param("code")
	appG := app.Gin{C: c}
	openid, err := util.SendWxAuthAPI(code)
	if err != nil {
		appG.Response(http.StatusOK, e.SUCCESS, map[string]string{
			"err": "openid 获取失败 :" + err.Error(),
		})
		return
	}
	appG.Response(http.StatusOK, e.SUCCESS, map[string]string{
		"openid": openid,
	})
}

func UserWXOpenIDBinding(c *gin.Context) {
	appG := app.Gin{C: c}
	dataByte, _ := ioutil.ReadAll(c.Request.Body)
	fsion := gofasion.NewFasion(string(dataByte))
	openid := fsion.Get("openid").ValueStr()
	// pastureid := fsion.Get("pastureid").ValueStr()
	userinfo := fsion.Get("userinfo").ValueStr()
	tx := restful.Engine.NewSession()
	defer tx.Close()

	userlist := strings.Split(userinfo, ".")
	if len(userlist) < 3 {
		logs.Error("UserWXOpenIDBinding-error1:", nil)
		appG.Response(http.StatusInternalServerError, e.ERROR, false)
		return
	}

	_, err := tx.SQL(` replace into user_wx(userid,pastureid,openid,name) VALUES (?,(select  pastureid  from user where username = ?),?,?)`, userlist[1], userlist[1], openid, userlist[2]).Execute()
	if err != nil {
		logs.Error("UserWXOpenIDBinding-error2:", err)
		appG.Response(http.StatusInternalServerError, e.ERROR, false)
		return
	}

	appG.Response(http.StatusOK, e.SUCCESS, true)
}