Create & Init Project...

app/admin/main/vip/http/welfare.go

@@ -0,0 +1,147 @@
+package http
+
+import (
+	"io/ioutil"
+	"mime/multipart"
+	"net/http"
+
+	"go-common/app/admin/main/vip/model"
+	"go-common/library/ecode"
+	bm "go-common/library/net/http/blademaster"
+)
+
+const (
+	_maxFileSize = 1048576
+)
+
+func welfareTypeSave(c *bm.Context) {
+	arg := new(struct {
+		ID   int    `form:"id"`
+		Name string `form:"name" validate:"required"`
+	})
+	if err := c.Bind(arg); err != nil {
+		return
+	}
+	username, _ := c.Get("username")
+	c.JSON(nil, vipSvc.WelfareTypeSave(arg.ID, arg.Name, username.(string)))
+}
+
+func welfareTypeState(c *bm.Context) {
+	arg := new(struct {
+		ID int `form:"id" validate:"required"`
+	})
+	if err := c.Bind(arg); err != nil {
+		return
+	}
+	username, _ := c.Get("username")
+	c.JSON(nil, vipSvc.WelfareTypeState(c, arg.ID, username.(string)))
+}
+
+func welfareTypeList(c *bm.Context) {
+	c.JSON(vipSvc.WelfareTypeList())
+}
+
+func welfareSave(c *bm.Context) {
+	const _redirectThridPage = 3
+
+	wfReq := new(model.WelfareReq)
+	if err := c.Bind(wfReq); err != nil {
+		return
+	}
+	if wfReq.UsageForm == _redirectThridPage && wfReq.ReceiveUri == "" {
+		c.JSON(nil, ecode.VipWelfareRequestErr)
+		return
+	}
+	username, _ := c.Get("username")
+	c.JSON(nil, vipSvc.WelfareSave(username.(string), wfReq))
+}
+
+func welfareState(c *bm.Context) {
+	arg := new(struct {
+		ID int `form:"id" validate:"required"`
+	})
+	if err := c.Bind(arg); err != nil {
+		return
+	}
+	username, _ := c.Get("username")
+	c.JSON(nil, vipSvc.WelfareState(arg.ID, username.(string)))
+}
+
+func welfareList(c *bm.Context) {
+	arg := new(struct {
+		TID int `form:"tid"`
+	})
+	if err := c.Bind(arg); err != nil {
+		return
+	}
+	c.JSON(vipSvc.WelfareList(arg.TID))
+}
+
+func welfareBatchUpload(c *bm.Context) {
+	arg := new(struct {
+		WID      int    `form:"wid"`
+		Filename string `form:"filename"`
+		Vtime    int    `form:"vtime"`
+	})
+	if err := c.Bind(arg); err != nil {
+		return
+	}
+	fileBody, _, err := getFileBody(c, "file")
+	if err != nil {
+		c.JSON(nil, err)
+		return
+	}
+	username, _ := c.Get("username")
+	c.JSON(nil, vipSvc.WelfareBatchUpload(fileBody, arg.Filename, username.(string), arg.WID, arg.Vtime))
+}
+
+func welfareBatchList(c *bm.Context) {
+	arg := new(struct {
+		WID int `form:"wid" validate:"required"`
+	})
+	if err := c.Bind(arg); err != nil {
+		return
+	}
+	c.JSON(vipSvc.WelfareBatchList(arg.WID))
+}
+
+func welfareBatchState(c *bm.Context) {
+	arg := new(struct {
+		ID int `form:"id" validate:"required"`
+	})
+	if err := c.Bind(arg); err != nil {
+		return
+	}
+	username, _ := c.Get("username")
+	c.JSON(nil, vipSvc.WelfareBatchState(c, arg.ID, username.(string)))
+}
+
+func getFileBody(c *bm.Context, name string) (body []byte, filetype string, err error) {
+	var file multipart.File
+	if file, _, err = c.Request.FormFile(name); err != nil {
+		if err == http.ErrMissingFile {
+			err = nil
+			return
+		}
+		err = ecode.RequestErr
+		return
+	}
+	if file == nil {
+		return
+	}
+	defer file.Close()
+	if body, err = ioutil.ReadAll(file); err != nil {
+		err = ecode.RequestErr
+		return
+	}
+	filetype = http.DetectContentType(body)
+	if filetype != "text/plain; charset=utf-8" {
+		err = ecode.VipFileTypeErr
+		return
+	}
+	if len(body) > _maxFileSize {
+		err = ecode.FileTooLarge
+		return
+	}
+	return
+}