package controllers import ( "fmt" "io" "log" "net/http" "os" "path" "time" "git.rosy.net.cn/jx-callback/business/jxstore/event" "git.rosy.net.cn/jx-callback/business/jxutils/jxcontext" "git.rosy.net.cn/jx-callback/globals" "git.rosy.net.cn/jx-callback/business/jxutils" "git.rosy.net.cn/jx-callback/business/model/dao" "git.rosy.net.cn/jx-callback/business/model" "github.com/gorilla/websocket" "git.rosy.net.cn/baseapi/utils" "github.com/astaxie/beego" ) // 操作事件明细相关API type EventController struct { beego.Controller } const ( audioPath = "/jxdata/cthrgw/dist/audio/" ) // 配置升级程序(升级为websocket) var upgrader = websocket.Upgrader{} // @Title 测试websocket // @Description 测试websocket // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /TestWebsocket [get] func (c *EventController) TestWebsocket() { // 解决跨域问题(微信小程序) upgrader.CheckOrigin = func(r *http.Request) bool { return true } //升级将HTTP服务器连接升级到WebSocket协议。 //responseHeader包含在对客户端升级的响应中 //请求。使用responseHeader指定Cookie(设置Cookie)和 //应用程序协商的子目录(Sec WebSocket协议)。 //如果升级失败,则升级将向客户端答复一个HTTP错误 ws, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil) if err != nil { log.Fatal(err) } var ( userID = c.GetString("userID") ) if userID == "" { return } if globals.IsProductEnv() { _, _, err = jxcontext.New(nil, c.GetString("token"), c.Ctx.ResponseWriter, c.Ctx.Request) if err != nil { msg := &CallResult{ Code: model.ErrCodeGeneralFailed, Desc: err.Error(), } ws.WriteJSON(&msg) } globals.SugarLogger.Debugf("TestWebsocket connection...") } c.EnableRender = false //Beego不启用渲染 go event.ImMessage(userID, ws) } // @Title 查询聊天记录 // @Description 查询聊天记录 // @Param token header string true "认证token" // @Param groupID query int true "组ID" // @Param fromTime query string false "开始时间" // @Param toTime query string false "结束时间" // @Param offset query int false "门店列表起始序号(以0开始,缺省为0)" // @Param pageSize query int false "门店列表页大小(缺省为50,-1表示全部)" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /GetImMessageRecord [get] func (c *EventController) GetImMessageRecord() { c.callGetImMessageRecord(func(params *tEventGetImMessageRecordParams) (retVal interface{}, errCode string, err error) { var db = dao.GetDB() retVal, err = dao.GetImMessageRecord(db, params.GroupID, "", params.Ctx.GetUserID(), 0, -1, utils.Str2Time(params.FromTime), utils.Str2Time(params.ToTime), params.Offset, params.PageSize) //清除此用户组所有的未读标记 if messageGroupReads, err := dao.GetMessageGroupRead(db, params.Ctx.GetUserID(), params.GroupID); err == nil { for _, v := range messageGroupReads { v.UnReadCount = 0 dao.UpdateEntity(db, v, "UnReadCount") } } return retVal, "", err }) } // @Title 用户未读消息设置 // @Description 用户未读消息设置,用户在退出登录,ws关闭,以及关闭小程序或app时调用 // @Param token header string true "认证token" // @Param payload query string true "messageGroupRead 格式[{"groupID":,"unReadCount":}]" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /UpdateUserMessageGroupRead [post] func (c *EventController) UpdateUserMessageGroupRead() { c.callUpdateUserMessageGroupRead(func(params *tEventUpdateUserMessageGroupReadParams) (retVal interface{}, errCode string, err error) { var reads []*model.MessageGroupRead if err = jxutils.Strings2Objs(params.Payload, &reads); err == nil { err = event.UpdateUserMessageGroupRead(params.Ctx, reads) } return retVal, "", err }) } // @Title 发送聊天消息(限定系统消息) // @Description 发送聊天消息(限定系统消息) // @Param token header string true "认证token" // @Param payload formData string true "immessageRecord 类型" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /SendSysMessage [post] func (c *EventController) SendSysMessage() { c.callSendSysMessage(func(params *tEventSendSysMessageParams) (retVal interface{}, errCode string, err error) { var imMessageRecord *model.ImMessageRecord if err = jxutils.Strings2Objs(params.Payload, &imMessageRecord); err == nil { err = event.SendSysMessage(params.Ctx, imMessageRecord) } return retVal, "", err }) } // @Title 创建聊天组 // @Description 创建聊天组 // @Param token header string true "认证token" // @Param userID formData string true "创建者id" // @Param userID2 formData string false "被拉的id 如果userID2为空就默认为是创建的群聊" // @Param name formData string false "如果是群聊,则要传入群名" // @Param dividePercentage formData int false "如果是群聊,则要传入分成比例" // @Param quitPrice formData int false "如果是群聊,则要传入退团金额" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /CreateMessageGroup [post] func (c *EventController) CreateMessageGroup() { c.callCreateMessageGroup(func(params *tEventCreateMessageGroupParams) (retVal interface{}, errCode string, err error) { retVal, err = event.CreateMessageGroup(params.Ctx, params.UserID, params.UserID2, params.Name, params.DividePercentage, params.QuitPrice) return retVal, "", err }) } // @Title 查询某个用户所有聊天组 // @Description 查询某个用户所有聊天组 // @Param token header string true "认证token" // @Param userID query string true "userid" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /GetMessageGroupByUser [get] func (c *EventController) GetMessageGroupByUser() { c.callGetMessageGroupByUser(func(params *tEventGetMessageGroupByUserParams) (retVal interface{}, errCode string, err error) { retVal, err = event.GetMessageGroupByUser(params.Ctx, params.UserID) return retVal, "", err }) } // @Title 查询聊天组 // @Description 查询聊天组 // @Param token header string true "认证token" // @Param groupID query int false "groupID" // @Param isMember query bool true "是否查询组员" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /GetMessageGroups [get] func (c *EventController) GetMessageGroups() { c.callGetMessageGroups(func(params *tEventGetMessageGroupsParams) (retVal interface{}, errCode string, err error) { retVal, err = dao.GetMessageGroups(dao.GetDB(), "", params.GroupID, model.GroupTypeMulit, params.IsMember, "") return retVal, "", err }) } // @Title 加入用户组 // @Description 加入用户组 // @Param token header string true "认证token" // @Param groupID formData int true "组号" // @Param userID formData string true "被邀请人ID" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /AddMessageGroup [post] func (c *EventController) AddMessageGroup() { c.callAddMessageGroup(func(params *tEventAddMessageGroupParams) (retVal interface{}, errCode string, err error) { err = event.AddMessageGroup(params.Ctx, params.GroupID, params.UserID) return retVal, "", err }) } // @Title 修改群组信息 // @Description 修改群组信息 // @Param token header string true "认证token" // @Param groupID formData int true "组号" // @Param payload formData string true "群组 payload" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /UpdateMessageGroup [put] func (c *EventController) UpdateMessageGroup() { c.callUpdateMessageGroup(func(params *tEventUpdateMessageGroupParams) (retVal interface{}, errCode string, err error) { payload := make(map[string]interface{}) if err = utils.UnmarshalUseNumber([]byte(params.Payload), &payload); err == nil { retVal, err = event.UpdateMessageGroup(params.Ctx, params.GroupID, payload) } return retVal, "", err }) } // @Title 退出用户组(踢人) // @Description 退出用户组(踢人) // @Param token header string true "认证token" // @Param groupID formData int true "组号" // @Param userID formData string true "userID" // @Param flag formData bool false "是否是解散群" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /DeleteMessageGroup [post] func (c *EventController) DeleteMessageGroup() { c.callDeleteMessageGroup(func(params *tEventDeleteMessageGroupParams) (retVal interface{}, errCode string, err error) { errCode, err = event.DeleteMessageGroup(params.Ctx, params.GroupID, params.UserID, params.Flag) return retVal, errCode, err }) } // @Title 转让群主 // @Description 转让群主 // @Param token header string true "认证token" // @Param groupID formData int true "组号" // @Param userID formData string true "userID" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /TransferMessageGroupMaster [post] func (c *EventController) TransferMessageGroupMaster() { c.callTransferMessageGroupMaster(func(params *tEventTransferMessageGroupMasterParams) (retVal interface{}, errCode string, err error) { err = event.TransferMessageGroupMaster(params.Ctx, params.GroupID, params.UserID) return retVal, "", err }) } // @Title 上传图片 // @Description 上传图片 // @Param token header string true "认证token" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /UploadImg [post] func (c *EventController) UploadImg() { c.callUploadImg(func(params *tEventUploadImgParams) (retVal interface{}, errCode string, err error) { file, head, err := c.GetFile("rsmImg") defer file.Close() if head.Size > 1024*1024*5 { err = fmt.Errorf("图片太大,请重新选择!") } if path.Ext(head.Filename) != ".jpg" && path.Ext(head.Filename) != ".png" { err = fmt.Errorf("不支持的图片格式,请重新选择!") } if err != nil { return retVal, "", err } fileName := utils.GetUUID() + "_" + time.Now().Format("20060102") + path.Ext(head.Filename) c.SaveToFile("rsmImg", "/jxdata/cthrgw/dist/img/"+fileName) return "https://www.jxcs.net/img/" + fileName, "", err }) } // @Title 上传音频 // @Description 上传音频 // @Param token header string true "认证token" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /UploadAudio [post] func (c *EventController) UploadAudio() { c.callUploadAudio(func(params *tEventUploadAudioParams) (retVal interface{}, errCode string, err error) { var ( timeStr = time.Now().Format("20060102") filePath = audioPath + timeStr ) files := c.Ctx.Request.MultipartForm.File["rsmAudio"] head := files[0] file, err := head.Open() defer file.Close() if path.Ext(head.Filename) != ".mp3" { err = fmt.Errorf("不支持的音频格式,请重新选择!") } if err != nil { return retVal, "", err } fileName := utils.GetUUID() + "_" + timeStr + path.Ext(head.Filename) if _, err = os.Stat(filePath); os.IsNotExist(err) { os.MkdirAll(filePath, os.ModePerm) os.Chmod(filePath, os.ModePerm) } f, err := os.OpenFile(filePath+"/"+fileName, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666) if err != nil { return retVal, "", err } defer f.Close() io.Copy(f, file) return &model.ImMessageRecord{ Content: "https://www.jxcs.net/audio/" + timeStr + "/" + fileName, }, "", err }) } // @Title 得到用户统计数据 // @Description 得到用户统计数据 // @Param token header string true "认证token" // @Success 200 {object} controllers.CallResult // @Failure 200 {object} controllers.CallResult // @router /GetUserStatistics [get] func (c *EventController) GetUserStatistics() { c.callGetUserStatistics(func(params *tEventGetUserStatisticsParams) (retVal interface{}, errCode string, err error) { retVal, err = event.GetUserStatistics(params.Ctx) return retVal, "", err }) }