get_wx_list_logic.go 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247
  1. package Wx
  2. import (
  3. "context"
  4. "fmt"
  5. reqv3 "github.com/imroc/req/v3"
  6. "github.com/suyuan32/simple-admin-core/rpc/types/core"
  7. "strings"
  8. "wechat-api/ent"
  9. "wechat-api/ent/predicate"
  10. "wechat-api/ent/usagetotal"
  11. "wechat-api/ent/wx"
  12. "wechat-api/hook"
  13. "wechat-api/internal/svc"
  14. "wechat-api/internal/types"
  15. "wechat-api/internal/utils/dberrorhandler"
  16. "github.com/suyuan32/simple-admin-common/msg/errormsg"
  17. "github.com/suyuan32/simple-admin-common/utils/pointy"
  18. "github.com/zeromicro/go-zero/core/logx"
  19. )
  20. type GetWxListLogic struct {
  21. ctx context.Context
  22. svcCtx *svc.ServiceContext
  23. logx.Logger
  24. }
  25. func NewGetWxListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetWxListLogic {
  26. return &GetWxListLogic{
  27. ctx: ctx,
  28. svcCtx: svcCtx,
  29. Logger: logx.WithContext(ctx),
  30. }
  31. }
  32. func (l *GetWxListLogic) GetWxList(req *types.WxListReq) (*types.WxListResp, error) {
  33. organizationId := l.ctx.Value("organizationId").(uint64)
  34. isAdmin := l.ctx.Value("isAdmin").(bool)
  35. servers, err := l.svcCtx.DB.Server.Query().All(l.ctx)
  36. serverSet := make(map[uint64]*ent.Server, len(servers))
  37. serverSet[0] = &ent.Server{
  38. ID: 0,
  39. Name: "工作手机",
  40. Status: 1,
  41. }
  42. for _, s := range servers {
  43. serverSet[s.ID] = s
  44. }
  45. var predicates []predicate.Wx
  46. if req.Ctype != nil && *req.Ctype > 0 {
  47. predicates = append(predicates, wx.Ctype(*req.Ctype))
  48. }
  49. if !isAdmin {
  50. predicates = append(predicates, wx.OrganizationIDEQ(organizationId))
  51. } else {
  52. if req.OrganizationId != nil {
  53. predicates = append(predicates, wx.OrganizationIDEQ(*req.OrganizationId))
  54. }
  55. if req.OrganizationName != nil {
  56. departmentList, _ := l.svcCtx.CoreRpc.GetDepartmentList(l.ctx, &core.DepartmentListReq{Name: req.OrganizationName})
  57. organizationIds := make([]uint64, 0)
  58. for _, department := range departmentList.Data {
  59. organizationIds = append(organizationIds, *department.Id)
  60. }
  61. predicates = append(predicates, wx.OrganizationIDIn(organizationIds...))
  62. }
  63. }
  64. if req.ServerId != nil {
  65. predicates = append(predicates, wx.ServerIDEQ(*req.ServerId))
  66. }
  67. if req.Port != nil {
  68. predicates = append(predicates, wx.PortContains(*req.Port))
  69. }
  70. if req.ProcessId != nil {
  71. predicates = append(predicates, wx.ProcessIDContains(*req.ProcessId))
  72. }
  73. if req.Callback != nil {
  74. predicates = append(predicates, wx.CallbackContains(*req.Callback))
  75. }
  76. data, err := l.svcCtx.DB.Wx.Query().Where(predicates...).Order(ent.Desc(wx.FieldOrganizationID)).WithAgent().WithServer().Page(l.ctx, req.Page, req.PageSize)
  77. if err != nil {
  78. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  79. }
  80. resp := &types.WxListResp{}
  81. resp.Msg = errormsg.Success
  82. resp.Data.Total = data.PageDetails.Total
  83. var result types.WorkPhoneGetWeChatsResp
  84. res, err := reqv3.C().DevMode().R().SetSuccessResult(&result).Post("http://chat.gkscrm.com:13086/pc/GetWeChatsReq?id=13")
  85. if err != nil {
  86. return nil, err
  87. }
  88. if !res.IsSuccessState() {
  89. err = fmt.Errorf("GetWeChats failed with status code %d", res.StatusCode)
  90. return nil, err
  91. }
  92. workphoneWxList := make(map[string]types.WorkPhoneWeChat, len(result.Data))
  93. for _, v := range result.Data {
  94. workphoneWxList[v.Wechatid] = v
  95. }
  96. for _, v := range data.List {
  97. // 创建 hookClient 客户端
  98. serverInfo := serverSet[v.ServerID]
  99. serverName := serverInfo.Name
  100. if strings.HasPrefix(strings.ToUpper(v.Port), "PC-") {
  101. serverName = "本地版"
  102. }
  103. var loginStatus uint8 = 0
  104. hookClient := hook.NewHook(serverInfo.PrivateIP, serverInfo.AdminPort, v.Port)
  105. if v.ServerID > 0 {
  106. if serverInfo.Status == 1 {
  107. // 获取登录状态
  108. loginInfo, err := hookClient.IsLoginStatus()
  109. if err != nil {
  110. l.Error("退出登录失败", err)
  111. } else {
  112. if loginInfo.Onlinestatus == "3" {
  113. loginStatus = 1
  114. }
  115. }
  116. }
  117. } else {
  118. workphoneWx, exists := workphoneWxList[v.Wxid]
  119. if exists && workphoneWx.Isonline == 0 {
  120. loginStatus = 1
  121. } else {
  122. loginStatus = 0
  123. }
  124. }
  125. processID := v.ProcessID
  126. wxid := v.Wxid
  127. account := v.Account
  128. nickname := v.Nickname
  129. tel := v.Tel
  130. headBig := v.HeadBig
  131. departmentInfo, err := l.svcCtx.CoreRpc.GetDepartmentById(l.ctx, &core.IDReq{Id: v.OrganizationID})
  132. if err != nil {
  133. l.Error("获取部门信息失败", err)
  134. }
  135. if v.ServerID > 0 {
  136. if loginStatus == 1 {
  137. // 如果处于登录状态,获取登录信息
  138. wxInfo, _ := hookClient.GetSelfLoginInfo()
  139. if err != nil {
  140. l.Error("获取登录信息失败", err)
  141. } else {
  142. if wxid != wxInfo.Wxid {
  143. l.svcCtx.Rds.HDel(l.ctx, "wx_info", wxid)
  144. l.svcCtx.Rds.HDel(l.ctx, "wx_info", wxInfo.Wxid)
  145. l.svcCtx.Rds.HDel(l.ctx, "crontask_wx_server_info", wxid)
  146. l.svcCtx.Rds.HDel(l.ctx, "crontask_wx_server_info", wxInfo.Wxid)
  147. }
  148. processID = wxInfo.ProcessID
  149. wxid = wxInfo.Wxid
  150. account = wxInfo.Account
  151. nickname = wxInfo.Nickname
  152. tel = wxInfo.Tel
  153. headBig = wxInfo.HeadBig
  154. _ = l.svcCtx.DB.Wx.UpdateOneID(v.ID).
  155. SetNotNilStatus(&loginStatus).
  156. SetNotNilProcessID(&wxInfo.ProcessID).
  157. SetNotNilWxid(&wxInfo.Wxid).
  158. SetNotNilAccount(&wxInfo.Account).
  159. SetNotNilNickname(&wxInfo.Nickname).
  160. SetNotNilTel(&wxInfo.Tel).
  161. SetNotNilHeadBig(&wxInfo.HeadBig).
  162. Exec(l.ctx)
  163. }
  164. } else {
  165. if loginStatus != v.Status {
  166. _ = l.svcCtx.DB.Wx.UpdateOneID(v.ID).
  167. SetNotNilStatus(&loginStatus).
  168. Exec(l.ctx)
  169. }
  170. }
  171. }
  172. totalTokens := uint64(0)
  173. usageTotalInfo, _ := l.svcCtx.DB.UsageTotal.Query().
  174. Where(
  175. usagetotal.BotID(wxid),
  176. ).
  177. Only(l.ctx)
  178. if usageTotalInfo != nil {
  179. totalTokens = usageTotalInfo.TotalTokens
  180. }
  181. var agent types.AgentInfo
  182. if v.Edges.Agent != nil {
  183. agent = types.AgentInfo{
  184. BaseIDInfo: types.BaseIDInfo{
  185. Id: &v.AgentID,
  186. CreatedAt: pointy.GetPointer(v.Edges.Agent.CreatedAt.UnixMilli()),
  187. UpdatedAt: pointy.GetPointer(v.Edges.Agent.UpdatedAt.UnixMilli()),
  188. },
  189. Name: &v.Edges.Agent.Name,
  190. Role: &v.Edges.Agent.Role,
  191. Status: &v.Edges.Agent.Status,
  192. Background: &v.Edges.Agent.Background,
  193. Examples: &v.Edges.Agent.Examples,
  194. }
  195. }
  196. resp.Data.Data = append(resp.Data.Data,
  197. types.WxInfo{
  198. BaseIDInfo: types.BaseIDInfo{
  199. Id: &v.ID,
  200. CreatedAt: pointy.GetPointer(v.CreatedAt.UnixMilli()),
  201. UpdatedAt: pointy.GetPointer(v.UpdatedAt.UnixMilli()),
  202. },
  203. Status: &loginStatus,
  204. ServerId: &v.ServerID,
  205. ServerName: &serverName,
  206. Port: &v.Port,
  207. ProcessId: &processID,
  208. Callback: &v.Callback,
  209. Wxid: &wxid,
  210. Account: &account,
  211. Nickname: &nickname,
  212. Tel: &tel,
  213. HeadBig: &headBig,
  214. OrganizationId: &v.OrganizationID,
  215. OrganizationName: departmentInfo.Name,
  216. AgentId: &v.AgentID,
  217. AgentInfo: &agent,
  218. ApiBase: &v.APIBase,
  219. ApiKey: &v.APIKey,
  220. TotalTokens: &totalTokens,
  221. Ctype: &v.Ctype,
  222. })
  223. }
  224. return resp, nil
  225. }