openbilibili/app/admin/main/member/http/realname.go

package http

import (
	"bytes"
	"encoding/csv"
	"fmt"
	"io/ioutil"
	"net/http"
	"strconv"
	"time"

	"go-common/app/admin/main/member/model"
	"go-common/library/ecode"
	"go-common/library/log"
	bm "go-common/library/net/http/blademaster"

	"github.com/pkg/errors"
)

func realnameList(ctx *bm.Context) {
	var (
		arg   = &model.ArgRealnameList{}
		list  []*model.RespRealnameApply
		total int
		err   error
	)
	if err = ctx.Bind(arg); err != nil {
		return
	}
	if arg.PN <= 0 {
		arg.PN = 1
	}
	if arg.PS <= 0 || arg.PS > 100 {
		arg.PS = 10
	}
	// 没有指定实名认证申请状态State，默认查询所有
	if arg.State == "" {
		arg.State = model.RealnameApplyStateAll
	}
	// 如果没有使用 mid || card 则取默认7天前作为 TSFrom
	if arg.MID == 0 && arg.Card == "" {
		if arg.TSFrom <= 0 {
			arg.TSFrom = time.Now().Add(-time.Hour * 24 * 7).Unix()
		}
	}
	// 如果使用 mid 或 card 进行查询，则查询所有时间段对应的数据
	if arg.MID != 0 || arg.Card != "" {
		arg.TSFrom = 0
		arg.TSTo = 0
	}

	if list, total, err = svc.RealnameList(ctx, arg); err != nil {
		ctx.JSON(nil, err)
		return
	}
	res := map[string]interface{}{
		"list": list,
		"page": map[string]int{
			"num":   arg.PN,
			"size":  arg.PS,
			"total": total,
		},
	}
	ctx.JSON(res, nil)
}

func realnamePendingList(ctx *bm.Context) {
	var (
		arg   = &model.ArgRealnamePendingList{}
		list  []*model.RespRealnameApply
		total int
		err   error
	)
	if err = ctx.Bind(arg); err != nil {
		return
	}
	if arg.PN <= 0 {
		arg.PN = 1
	}
	if arg.PS <= 0 || arg.PS > 100 {
		arg.PS = 10
	}
	if list, total, err = svc.RealnamePendingList(ctx, arg); err != nil {
		ctx.JSON(nil, err)
		return
	}
	res := map[string]interface{}{
		"list": list,
		"page": map[string]int{
			"num":   arg.PN,
			"size":  arg.PS,
			"total": total,
		},
	}
	ctx.JSON(res, nil)
}

func realnameAuditApply(ctx *bm.Context) {
	var (
		arg       = &model.ArgRealnameAuditApply{}
		adminName string
		adminID   int64
		data      interface{}
		ok        bool
		err       error
	)
	if err = ctx.Bind(arg); err != nil {
		return
	}
	// 驳回必须有reason
	if arg.Action == model.RealnameActionReject && arg.Reason == "" {
		ctx.JSON(nil, ecode.RequestErr)
		return
	}
	if data, ok = ctx.Get("username"); !ok {
		ctx.JSON(nil, ecode.AccessDenied)
		return
	}
	if adminName, ok = data.(string); !ok {
		ctx.JSON(nil, ecode.ServerErr)
		return
	}
	if data, ok = ctx.Get("uid"); !ok {
		ctx.JSON(nil, ecode.AccessDenied)
		return
	}
	if adminID, ok = data.(int64); !ok {
		ctx.JSON(nil, ecode.ServerErr)
		return
	}
	ctx.JSON(nil, svc.RealnameAuditApply(ctx, arg, adminName, adminID))
}

func realnameReasonList(ctx *bm.Context) {
	var (
		arg   = &model.ArgRealnameReasonList{}
		list  []string
		total int
		err   error
	)
	if err = ctx.Bind(arg); err != nil {
		return
	}
	if arg.PN <= 0 {
		arg.PN = 1
	}
	if arg.PS <= 0 || arg.PS > 100 {
		arg.PS = 20
	}
	if list, total, err = svc.RealnameReasonList(ctx, arg); err != nil {
		ctx.JSON(nil, err)
		return
	}
	res := map[string]interface{}{
		"list": list,
		"page": map[string]int{
			"num":   arg.PN,
			"size":  arg.PS,
			"total": total,
		},
	}
	ctx.JSON(res, nil)
}

func realnameSetReason(ctx *bm.Context) {
	var (
		arg = &model.ArgRealnameSetReason{}
	)
	if err := ctx.Bind(arg); err != nil {
		return
	}
	ctx.JSON(nil, svc.RealnameSetReason(ctx, arg))
}

func realnameImage(ctx *bm.Context) {
	var (
		arg = &model.ArgRealnameImage{}
		img []byte
		err error
	)
	if err = ctx.Bind(arg); err != nil {
		return
	}
	if img, err = svc.FetchRealnameImage(ctx, arg.Token); err != nil {
		ctx.JSON(nil, err)
		return
	}
	ctx.Bytes(http.StatusOK, http.DetectContentType(img), img)
}

func realnameImagePreview(ctx *bm.Context) {
	var (
		arg = &model.ArgRealnameImagePreview{}
		img []byte
		err error
	)
	if err = ctx.Bind(arg); err != nil {
		return
	}
	if arg.BorderSize <= 0 {
		arg.BorderSize = 1000
	}
	if img, err = svc.RealnameImagePreview(ctx, arg.Token, arg.BorderSize); err != nil {
		ctx.JSON(nil, err)
		return
	}
	ctx.Bytes(http.StatusOK, http.DetectContentType(img), img)
}

func realnameSearchCard(ctx *bm.Context) {
	var (
		arg = &model.ArgRealnameSearchCard{}
		err error
	)
	if err = ctx.Bind(arg); err != nil {
		return
	}
	if len(arg.Cards) > 50 {
		err = ecode.RequestErr
		return
	}
	ctx.JSON(svc.RealnameSearchCard(ctx, arg.Cards, arg.CardType, arg.Country))
}

func realnameUnbind(ctx *bm.Context) {
	var (
		arg       = &model.ArgMid{}
		adminName string
		adminID   int64
		data      interface{}
		ok        bool
		err       error
	)
	if err = ctx.Bind(arg); err != nil {
		return
	}
	if data, ok = ctx.Get("username"); !ok {
		ctx.JSON(nil, ecode.AccessDenied)
		return
	}
	if adminName, ok = data.(string); !ok {
		ctx.JSON(nil, ecode.ServerErr)
		return
	}
	if data, ok = ctx.Get("uid"); !ok {
		ctx.JSON(nil, ecode.AccessDenied)
		return
	}
	if adminID, ok = data.(int64); !ok {
		ctx.JSON(nil, ecode.ServerErr)
		return
	}
	ctx.JSON(nil, svc.RealnameUnbind(ctx, arg.Mid, adminName, adminID))
}

func realnameExport(ctx *bm.Context) {
	arg := &model.ArgMids{}
	if err := ctx.Bind(arg); err != nil {
		return
	}
	if len(arg.Mid) > 1000 {
		ctx.JSON(nil, ecode.RequestErr)
		return
	}

	realnameExports, err := svc.RealnameExcel(ctx, arg.Mid)
	if err != nil {
		ctx.JSON(nil, err)
		return
	}
	data := make([][]string, 0, len(realnameExports))
	data = append(data, []string{"用户uid", "用户名", "昵称", "姓名", "手机号", "证件类型", "证件号"})
	for _, user := range realnameExports {
		fields := []string{
			strconv.FormatInt(user.Mid, 10),
			user.UserID,
			user.Uname,
			user.Realname,
			user.Tel,
			model.CardTypeString(user.CardType),
			avoidescape(user.CardNum),
		}
		data = append(data, fields)
	}
	buf := bytes.NewBuffer(nil)
	w := csv.NewWriter(buf)
	for _, record := range data {
		if err := w.Write(record); err != nil {
			ctx.JSON(nil, err)
			return
		}
	}
	w.Flush()
	res := buf.Bytes()
	ctx.Writer.Header().Set("Content-Type", "application/csv")
	ctx.Writer.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s.csv", "实名认证导出"))
	ctx.Writer.Write([]byte("\xEF\xBB\xBF")) // 写 UTF-8 的 BOM 头，别删，删了客服就找上门了
	ctx.Writer.Write(res)
}

func avoidescape(in string) string {
	return fmt.Sprintf("%s,", in)
}

func realnameSubmit(ctx *bm.Context) {
	arg := &model.ArgRealnameSubmit{}
	if err := ctx.Bind(arg); err != nil {
		return
	}
	ctx.JSON(nil, svc.RealnameSubmit(ctx, arg))
}

func realnameFileUpload(c *bm.Context) {
	arg := &model.ArgMid{}
	if err := c.Bind(arg); err != nil {
		return
	}
	mid := arg.Mid
	defer c.Request.Form.Del("img") // 防止日志不出现
	c.Request.ParseMultipartForm(32 << 20)
	imgBytes, err := func() ([]byte, error) {
		img := c.Request.FormValue("img")
		if img != "" {
			log.Info("Succeeded to parse img file from form value: mid: %d, length: %d", mid, len(img))
			return []byte(img), nil
		}
		log.Warn("Failed to parse img file from form value, fallback to form file: mid: %d", mid)
		f, _, err := c.Request.FormFile("img")
		if err != nil {
			return nil, errors.Wrapf(err, "parse img form file: mid: %d", mid)
		}
		defer f.Close()
		data, err := ioutil.ReadAll(f)
		if err != nil {
			return nil, errors.Wrapf(err, "read img form file: mid: %d", mid)
		}
		if len(data) <= 0 {
			return nil, errors.Wrapf(err, "form file data: mid: %d, length: %d", mid, len(data))
		}
		log.Info("Succeeded to parse file from form file: mid: %d, length: %d", mid, len(data))
		return data, nil
	}()
	if err != nil {
		log.Error("Failed to parse realname upload file: mid: %d: %+v", mid, err)
		c.JSON(nil, ecode.RequestErr)
		return
	}
	var resData struct {
		SRC string `json:"token"`
	}
	if resData.SRC, err = svc.RealnameFileUpload(c, arg.Mid, imgBytes); err != nil {
		log.Error("%+v", err)
		c.JSON(nil, err)
		return
	}
	c.JSON(resData, nil)
}