controller

controllers/menu/menu.go

@@ -17,6 +17,7 @@ type MenuController struct {
 // @Description 初始化 Controller, 系统自动调用
 func (c *MenuController) Constructor() {
 	c.dao = menu.NewMenuServ(c.Context)
+
 }
 
 func (c *MenuController) GetMenuList() {

controllers/wechatimg/wechatimt.go

@@ -0,0 +1,121 @@
+package wechatimg
+
+import (
+	"mime/multipart"
+	"strconv"
+	"time"
+	"unsafe"
+	"wechat-conf/controllers"
+	"wechat-conf/models/model"
+	"wechat-conf/service/sysuser"
+	"wechat-conf/service/wechatimg"
+	"wechat-conf/utils"
+)
+
+// WechatImgController 信息
+type WechatImgController struct {
+	dao  *wechatimg.WechatImgServ
+	sdao *sysuser.SysuserServ
+	controllers.BaseController
+}
+
+// Constructor 初始化 Controller
+// @Title Constructor
+// @Description 初始化 Controller, 系统自动调用
+func (c *WechatImgController) Constructor() {
+	c.dao = wechatimg.NewWechatImgServ(c.Context)
+	c.sdao = sysuser.NewSysuserServ(c.Context)
+}
+
+func (c *WechatImgController) GetWechatImgList() {
+	user := c.Context.Get("user").(*model.SysUser)
+	appId, err := c.sdao.GetAppIdByOrgId(user.OrgId)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	client, err := utils.Component.GetWxClient(appId)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	page, _ := c.GetInt("page")
+	pageSize, _ := c.GetInt("pagesize")
+	count, _ := c.GetInt("count")
+	offset := page * (pageSize - 1)
+	var data map[string]string
+	data["offset"] = strconv.Itoa(offset)
+	data["type"] = "image"
+	data["count"] = strconv.Itoa(count)
+	wechatImg, err := client.GetMaterialList(data)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	c.ResponseJSON(wechatImg)
+}
+func (c *WechatImgController) SyncWechatImg() {
+	user := c.Context.Get("user").(*model.SysUser)
+	appId, err := c.sdao.GetAppIdByOrgId(user.OrgId)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	client, err := utils.Component.GetWxClient(appId)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	for i := 0; i < 200; i++ {
+		var data map[string]string
+		offset := i * 20
+		data["offset"] = strconv.Itoa(offset)
+		data["type"] = "image"
+		data["count"] = strconv.Itoa(20)
+		wechatImg, err := client.GetMaterialList(data)
+		if err != nil {
+			c.ResponseError(err)
+		}
+		if wechatImg == nil {
+			break
+		}
+		err = c.dao.SyncWechatImg(wechatImg, user.OrgId)
+
+		if err != nil {
+			c.ResponseError(err)
+		}
+	}
+	c.ResponseJSON("同步成功")
+}
+func (c *WechatImgController) AddNewImg() {
+	user := c.Context.Get("user").(*model.SysUser)
+	fromFront := c.GetString("fromFront")
+	req := c.Ctx.Request
+	file, fh, err := req.FormFile(fromFront)
+	appId, err := c.sdao.GetAppIdByOrgId(user.OrgId)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	client, err := utils.Component.GetWxClient(appId)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	token := utils.Component.GetToken()
+	type wechatMatrial struct {
+		access_token string
+		media        multipart.File
+	}
+	var media = wechatMatrial{
+		access_token: token,
+		media:        file,
+	}
+	byteDate := *(*[]byte)(unsafe.Pointer(&media))
+	data, err := client.AddMaterial(byteDate, "image")
+	var img = model.TaWechatImg{
+		MediaId:    data["media_id"].(string),
+		Url:        data["url"].(string),
+		Name:       fh.Filename,
+		UpdateTime: time.Now(),
+	}
+	err = c.dao.AddWechatImg(img)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	c.ResponseJSON("上传成功")
+
+}

controllers/wechatmenu/wechatmenu.go

@@ -0,0 +1,51 @@
+package wechatmenu
+
+import (
+	"wechat-conf/controllers"
+	"wechat-conf/models/model"
+	"wechat-conf/service/sysuser"
+	"wechat-conf/service/wechatmenu"
+	"wechat-conf/utils"
+)
+
+// MenuController 信息
+type MenuController struct {
+	dao  *wechatmenu.WechatmenuServ
+	sdao *sysuser.SysuserServ
+	controllers.BaseController
+}
+
+// Constructor 初始化 Controller
+// @Title Constructor
+// @Description 初始化 Controller, 系统自动调用
+func (c *MenuController) Constructor() {
+	c.dao = wechatmenu.NewWechatmenuServ(c.Context)
+	c.sdao = sysuser.NewSysuserServ(c.Context)
+}
+
+func (c *MenuController) GetMenuList() {
+	user := c.Context.Get("user").(*model.SysUser)
+	appId, err := c.sdao.GetAppIdByOrgId(user.OrgId)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	client, err := utils.Component.GetWxClient(appId)
+	if err != nil {
+		c.ResponseError(err)
+	}
+	menu, err := client.GetMenu()
+	c.ResponseJSON(menu)
+}
+
+// func (c *MenuController) RefreshMenu() {
+// 	user := c.Context.Get("user").(*model.SysUser)
+// 	appId, err := c.sdao.GetAppIdByOrgId(user.OrgId)
+// 	if err != nil {
+// 		c.ResponseError(err)
+// 	}
+// 	client, err := utils.Component.GetWxClient(appId)
+// 	if err != nil {
+// 		c.ResponseError(err)
+// 	}
+// 	err = client.RefreshMenu()
+// }

models/sysuser/sysuser.go

@@ -89,3 +89,19 @@ func (m *SysuserDAO) BindUserToOrg(userId, orgId string) error {
 	_, err := m.db.Cols(cols...).Where("user_id=?", user.UserId).Update(user)
 	return err
 }
+
+func (m *SysuserDAO) GetAppIdByOrgId(orgId string) (string, error) {
+	var config []model.SysWechatConf
+	sql := `SELECT
+	* 
+FROM
+	sys_wechat_conf a
+	INNER JOIN sys_org b ON b.wechat_id = a.conf_id 
+WHERE
+	org_id = '` + orgId + `'`
+	err := m.db.Sql(sql).Find(&config)
+	if len(config) > 0 {
+		return config[0].Appid, err
+	}
+	return "", err
+}

models/wechatimg/wechatimg.go

@@ -65,3 +65,7 @@ func (m *WechatImgDAO) DeleteWechatImg(mediaId string) error {
 	_, err := m.db.Cols(cols...).Where("media_id = ?", img.MediaId).Update(img)
 	return err
 }
+func (m *WechatImgDAO) DeleteBatchByOrgId(orgId string) error {
+	_, err := m.db.Delete(&model.TaWechatImg{OrgId: orgId})
+	return err
+}

service/sysuser/sysuser.go

@@ -74,3 +74,11 @@ func (s *SysuserServ) DeleteUser(userId string) error {
 	}
 	return nil
 }
+func (s *SysuserServ) GetAppIdByOrgId(orgId string) (string, error) {
+	appId, err := s.dao.GetAppIdByOrgId(orgId)
+	if err != nil {
+		utils.LogError("获取APPID失败: " + err.Error())
+		return "", errors.New("获取APPID失败")
+	}
+	return appId, nil
+}

service/wechatimg/wechatimg.go

@@ -0,0 +1,64 @@
+package wechatimg
+
+import (
+	"errors"
+	"time"
+	"wechat-conf/models/model"
+	"wechat-conf/models/wechatimg"
+	"wechat-conf/utils"
+
+	"github.com/astaxie/beego"
+)
+
+// WechatImgServ 用户
+type WechatImgServ struct {
+	ctx *utils.Context
+	dao *wechatimg.WechatImgDAO
+}
+
+// NewWechatImgServ 初始化
+func NewWechatImgServ(ctx *utils.Context) *WechatImgServ {
+	return &WechatImgServ{
+		ctx: ctx,
+		dao: wechatimg.NewWechatImgDAO(ctx),
+	}
+}
+
+func (s *WechatImgServ) AddWechatImg(img model.TaWechatImg) error {
+	err := s.dao.AddWechatImg(img)
+	if err != nil {
+		utils.LogError("新增图片失败: " + err.Error())
+		return errors.New("新增图片失败")
+	}
+	return nil
+}
+
+func (s *WechatImgServ) SyncWechatImg(data interface{}, orgId string) error {
+	err := s.dao.DeleteBatchByOrgId(orgId)
+	if err != nil {
+		utils.LogError("同步图片失败: " + err.Error())
+		return errors.New("同步图片失败")
+	}
+	type wechatimg struct {
+		media_id    string
+		name        string
+		update_time time.Time
+		url         string
+	}
+	img := data.([]wechatimg)
+	for _, val := range img {
+		beego.Error(val)
+		var wechat model.TaWechatImg
+		wechat.MediaId = val.media_id
+		wechat.Name = val.name
+		wechat.OrgId = orgId
+		wechat.Url = val.url
+		wechat.UpdateTime = val.update_time
+		err := s.dao.AddWechatImg(wechat)
+		if err != nil {
+			utils.LogError("同步图片失败: " + err.Error())
+			return errors.New("同步图片失败")
+		}
+	}
+	return nil
+}

service/wechatmenu/wechatmenu.go

@@ -0,0 +1,20 @@
+package wechatmenu
+
+import (
+	"wechat-conf/models/wechatmenu"
+	"wechat-conf/utils"
+)
+
+// WechatmenuServ 用户
+type WechatmenuServ struct {
+	ctx *utils.Context
+	dao *wechatmenu.WechatMenuDAO
+}
+
+// NewWechatmenuServ 初始化
+func NewWechatmenuServ(ctx *utils.Context) *WechatmenuServ {
+	return &WechatmenuServ{
+		ctx: ctx,
+		dao: wechatmenu.NewWechatMenuDAO(ctx),
+	}
+}