wechat_platform.go 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652
  1. package rag
  2. import (
  3. "encoding/json"
  4. "eta/eta_api/cache"
  5. "eta/eta_api/controllers"
  6. "eta/eta_api/models"
  7. "eta/eta_api/models/rag"
  8. "eta/eta_api/models/rag/request"
  9. "eta/eta_api/models/rag/response"
  10. "eta/eta_api/models/system"
  11. "eta/eta_api/utils"
  12. "fmt"
  13. "github.com/rdlucklib/rdluck_tools/paging"
  14. "html"
  15. "strings"
  16. "time"
  17. )
  18. // WechatPlatformController
  19. // @Description: 微信公众号管理
  20. type WechatPlatformController struct {
  21. controllers.BaseAuthController
  22. }
  23. // TagList
  24. // @Title 获取ppt列表
  25. // @Description 获取ppt列表接口
  26. // @Param PageSize query int true "每页数据条数"
  27. // @Param CurrentIndex query int true "当前页页码,从1开始"
  28. // @Param KeyWord query string true "搜索关键词"
  29. // @Success 200 {object} models.TagListResp
  30. // @router /tag/list [get]
  31. func (c *WechatPlatformController) TagList() {
  32. br := new(models.BaseResponse).Init()
  33. defer func() {
  34. c.Data["json"] = br
  35. c.ServeJSON()
  36. }()
  37. sysUser := c.SysUser
  38. if sysUser == nil {
  39. br.Msg = "请登录"
  40. br.ErrMsg = "请登录,SysUser Is Empty"
  41. return
  42. }
  43. pageSize, _ := c.GetInt("PageSize")
  44. currentIndex, _ := c.GetInt("CurrentIndex")
  45. keyWord := c.GetString("KeyWord")
  46. var startSize int
  47. if pageSize <= 0 {
  48. pageSize = utils.PageSize20
  49. }
  50. if currentIndex <= 0 {
  51. currentIndex = 1
  52. }
  53. startSize = utils.StartIndex(currentIndex, pageSize)
  54. var condition string
  55. var pars []interface{}
  56. if keyWord != "" {
  57. condition = fmt.Sprintf(` AND %s = ?`, rag.WechatPlatformColumns.Nickname)
  58. pars = append(pars, `%`+keyWord+`%`)
  59. }
  60. obj := new(rag.Tag)
  61. total, list, err := obj.GetPageListByCondition(condition, pars, startSize, pageSize)
  62. if err != nil {
  63. br.Msg = "获取失败"
  64. br.ErrMsg = "获取失败,Err:" + err.Error()
  65. return
  66. }
  67. page := paging.GetPaging(currentIndex, pageSize, total)
  68. resp := new(response.TagListResp)
  69. resp.Paging = page
  70. resp.List = list
  71. br.Ret = 200
  72. br.Success = true
  73. br.Msg = "获取成功"
  74. br.Data = resp
  75. }
  76. // Add
  77. // @Title 新增公众号
  78. // @Description 新增公众号
  79. // @Param request body request.AddWechatPlatformReq true "type json string"
  80. // @Success 200 Ret=200 新增成功
  81. // @router /wechat_platform/add [post]
  82. func (c *WechatPlatformController) Add() {
  83. br := new(models.BaseResponse).Init()
  84. defer func() {
  85. c.Data["json"] = br
  86. c.ServeJSON()
  87. }()
  88. var req request.AddWechatPlatformReq
  89. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  90. if err != nil {
  91. br.Msg = "参数解析异常!"
  92. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  93. return
  94. }
  95. req.Name = strings.TrimSpace(req.Name)
  96. if req.Name == "" {
  97. br.Msg = "请输入公众号名称"
  98. br.IsSendEmail = false
  99. return
  100. }
  101. req.Link = strings.TrimSpace(req.Link)
  102. if req.Link == "" {
  103. br.Msg = "请输入文章链接"
  104. br.IsSendEmail = false
  105. return
  106. }
  107. var condition string
  108. var pars []interface{}
  109. condition = fmt.Sprintf(` AND %s = ?`, rag.WechatPlatformColumns.Nickname)
  110. pars = append(pars, req.Name)
  111. obj := new(rag.WechatPlatform)
  112. item, err := obj.GetByCondition(condition, pars)
  113. if err != nil && !utils.IsErrNoRow(err) {
  114. br.Msg = "公众号信息获取失败"
  115. br.ErrMsg = "公众号信息获取失败,Err:" + err.Error()
  116. return
  117. }
  118. if item.WechatPlatformId > 0 {
  119. br.Msg = "公众号名称重复"
  120. br.IsSendEmail = false
  121. return
  122. }
  123. item = &rag.WechatPlatform{
  124. WechatPlatformId: 0,
  125. FakeId: "",
  126. Nickname: req.Name,
  127. Alias: "",
  128. RoundHeadImg: "",
  129. ServiceType: 0,
  130. Signature: "",
  131. Verified: 0,
  132. ArticleLink: req.Link,
  133. Enabled: 0,
  134. SysUserId: c.SysUser.AdminId,
  135. ModifyTime: time.Now(),
  136. CreateTime: time.Now(),
  137. }
  138. err = item.Add(req.TagIdList)
  139. if err != nil {
  140. br.Msg = "添加失败"
  141. br.ErrMsg = "添加失败,Err:" + err.Error()
  142. return
  143. }
  144. // 异步新增公众号
  145. //go llm.AddWechatPlatform(item)
  146. go cache.AddWechatArticleOpToCache(item.WechatPlatformId, `add`)
  147. br.Ret = 200
  148. br.Success = true
  149. br.Msg = `添加成功`
  150. }
  151. // FollowList
  152. // @Title 我关注的接口
  153. // @Description 我关注的接口
  154. // @Param PageSize query int true "每页数据条数"
  155. // @Param CurrentIndex query int true "当前页页码,从1开始"
  156. // @Param KeyWord query string true "搜索关键词"
  157. // @Success 200 {object} []*rag.WechatPlatform
  158. // @router /wechat_platform/list/follow [get]
  159. func (c *WechatPlatformController) FollowList() {
  160. br := new(models.BaseResponse).Init()
  161. defer func() {
  162. c.Data["json"] = br
  163. c.ServeJSON()
  164. }()
  165. sysUser := c.SysUser
  166. if sysUser == nil {
  167. br.Msg = "请登录"
  168. br.ErrMsg = "请登录,SysUser Is Empty"
  169. return
  170. }
  171. pageSize, _ := c.GetInt("PageSize")
  172. currentIndex, _ := c.GetInt("CurrentIndex")
  173. keyWord := c.GetString("KeyWord")
  174. var startSize int
  175. if pageSize <= 0 {
  176. pageSize = utils.PageSize20
  177. }
  178. if currentIndex <= 0 {
  179. currentIndex = 1
  180. }
  181. startSize = utils.StartIndex(currentIndex, pageSize)
  182. var condition string
  183. var pars []interface{}
  184. if keyWord != "" {
  185. condition = fmt.Sprintf(` AND %s = ?`, rag.WechatPlatformColumns.Nickname)
  186. pars = append(pars, `%`+keyWord+`%`)
  187. }
  188. condition = fmt.Sprintf(` AND b.%s = ?`, rag.WechatPlatformUserMappingColumns.SysUserID)
  189. pars = append(pars, c.SysUser.AdminId)
  190. if keyWord != "" {
  191. condition = fmt.Sprintf(` AND %s = ?`, rag.WechatPlatformColumns.Nickname)
  192. pars = append(pars, `%`+keyWord+`%`)
  193. }
  194. obj := new(rag.WechatPlatformUserMapping)
  195. list, err := obj.GetListByCondition(condition, pars, startSize, pageSize)
  196. if err != nil {
  197. br.Msg = "获取失败"
  198. br.ErrMsg = "获取失败,Err:" + err.Error()
  199. return
  200. }
  201. br.Ret = 200
  202. br.Success = true
  203. br.Msg = "获取成功"
  204. br.Data = list
  205. }
  206. // PublicList
  207. // @Title 公共列表
  208. // @Description 公共列表
  209. // @Param PageSize query int true "每页数据条数"
  210. // @Param CurrentIndex query int true "当前页页码,从1开始"
  211. // @Param KeyWord query string true "搜索关键词"
  212. // @Success 200 {object} models.WechatPlatformListResp
  213. // @router /wechat_platform/list/public [get]
  214. func (c *WechatPlatformController) PublicList() {
  215. br := new(models.BaseResponse).Init()
  216. defer func() {
  217. c.Data["json"] = br
  218. c.ServeJSON()
  219. }()
  220. sysUser := c.SysUser
  221. if sysUser == nil {
  222. br.Msg = "请登录"
  223. br.ErrMsg = "请登录,SysUser Is Empty"
  224. return
  225. }
  226. pageSize, _ := c.GetInt("PageSize")
  227. currentIndex, _ := c.GetInt("CurrentIndex")
  228. keyWord := c.GetString("KeyWord")
  229. var startSize int
  230. if pageSize <= 0 {
  231. pageSize = utils.PageSize20
  232. }
  233. if currentIndex <= 0 {
  234. currentIndex = 1
  235. }
  236. startSize = utils.StartIndex(currentIndex, pageSize)
  237. var condition string
  238. var pars []interface{}
  239. if keyWord != "" {
  240. condition = fmt.Sprintf(` AND %s = ?`, rag.WechatPlatformColumns.Nickname)
  241. pars = append(pars, `%`+keyWord+`%`)
  242. }
  243. obj := new(rag.WechatPlatformUserMapping)
  244. list, err := obj.GetListByCondition(condition, pars, startSize, 100000)
  245. if err != nil {
  246. br.Msg = "获取失败"
  247. br.ErrMsg = "获取失败,Err:" + err.Error()
  248. return
  249. }
  250. resp := make([]response.WechatPlatformPublicListResp, 0)
  251. if list != nil && len(list) > 0 {
  252. userIdList := make([]int, 0)
  253. uerIdMap := make(map[int]bool)
  254. userFollowIndexMap := make(map[int]int)
  255. for _, v := range list {
  256. if _, ok := uerIdMap[v.FollowUserId]; !ok {
  257. userIdList = append(userIdList, v.FollowUserId)
  258. uerIdMap[v.FollowUserId] = true
  259. }
  260. index, ok := userFollowIndexMap[v.FollowUserId]
  261. if !ok {
  262. userFollowIndexMap[v.FollowUserId] = len(resp)
  263. resp = append(resp, response.WechatPlatformPublicListResp{
  264. UserId: v.FollowUserId,
  265. List: []*rag.UserFollowWechatPlatform{v},
  266. })
  267. } else {
  268. resp[index].List = append(resp[index].List, v)
  269. }
  270. }
  271. userList, err := system.GetAdminListByIdList(userIdList)
  272. if err != nil {
  273. br.Msg = "获取失败"
  274. br.ErrMsg = "获取失败,Err:" + err.Error()
  275. return
  276. }
  277. userNameMap := make(map[int]*system.Admin)
  278. for _, v := range userList {
  279. userNameMap[v.AdminId] = v
  280. }
  281. for k, v := range resp {
  282. userInfo, ok := userNameMap[v.UserId]
  283. if !ok {
  284. continue
  285. }
  286. resp[k].Name = userInfo.RealName + `关注`
  287. }
  288. }
  289. br.Ret = 200
  290. br.Success = true
  291. br.Msg = "获取成功"
  292. br.Data = resp
  293. }
  294. // PublicList
  295. // @Title 公共列表
  296. // @Description 公共列表
  297. // @Param PageSize query int true "每页数据条数"
  298. // @Param CurrentIndex query int true "当前页页码,从1开始"
  299. // @Param KeyWord query string true "搜索关键词"
  300. // @Success 200 {object} models.WechatPlatformListResp
  301. // @router /wechat_platform/op [post]
  302. func (c *WechatPlatformController) Op() {
  303. br := new(models.BaseResponse).Init()
  304. defer func() {
  305. c.Data["json"] = br
  306. c.ServeJSON()
  307. }()
  308. sysUser := c.SysUser
  309. if sysUser == nil {
  310. br.Msg = "请登录"
  311. br.ErrMsg = "请登录,SysUser Is Empty"
  312. return
  313. }
  314. var req request.OpWechatPlatformReq
  315. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  316. if err != nil {
  317. br.Msg = "参数解析异常!"
  318. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  319. return
  320. }
  321. if req.WechatPlatformId <= 0 {
  322. br.Msg = "参数错误"
  323. return
  324. }
  325. if req.Status < 0 || req.Status > 1 {
  326. br.Msg = "参数错误"
  327. return
  328. }
  329. obj := rag.WechatPlatform{}
  330. wechatPlatform, err := obj.GetByID(req.WechatPlatformId)
  331. if err != nil {
  332. br.Msg = "修改失败"
  333. br.ErrMsg = "修改失败,Err:" + err.Error()
  334. if utils.IsErrNoRow(err) {
  335. br.Msg = "公众号不存在"
  336. br.IsSendEmail = false
  337. }
  338. return
  339. }
  340. wechatPlatform.Enabled = req.Status
  341. wechatPlatform.ModifyTime = time.Now()
  342. err = wechatPlatform.Update([]string{"enabled", `modify_time`})
  343. if err != nil {
  344. br.Msg = "修改失败"
  345. br.ErrMsg = "修改失败,Err:" + err.Error()
  346. return
  347. }
  348. br.Ret = 200
  349. br.Success = true
  350. br.Msg = "修改成功"
  351. }
  352. // Refresh
  353. // @Title 公共列表
  354. // @Description 公共列表
  355. // @Success 200 {object} models.WechatPlatformListResp
  356. // @router /wechat_platform/refresh [post]
  357. func (c *WechatPlatformController) Refresh() {
  358. br := new(models.BaseResponse).Init()
  359. defer func() {
  360. c.Data["json"] = br
  361. c.ServeJSON()
  362. }()
  363. sysUser := c.SysUser
  364. if sysUser == nil {
  365. br.Msg = "请登录"
  366. br.ErrMsg = "请登录,SysUser Is Empty"
  367. return
  368. }
  369. var req request.RefreshWechatPlatformReq
  370. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  371. if err != nil {
  372. br.Msg = "参数解析异常!"
  373. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  374. return
  375. }
  376. if req.WechatPlatformId <= 0 {
  377. br.Msg = "参数错误"
  378. return
  379. }
  380. obj := rag.WechatPlatform{}
  381. wechatPlatform, err := obj.GetByID(req.WechatPlatformId)
  382. if err != nil {
  383. br.Msg = "修改失败"
  384. br.ErrMsg = "修改失败,Err:" + err.Error()
  385. if utils.IsErrNoRow(err) {
  386. br.Msg = "公众号不存在"
  387. br.IsSendEmail = false
  388. }
  389. return
  390. }
  391. if wechatPlatform.FakeId != `` {
  392. br.Msg = "公众号已添加成功"
  393. br.ErrMsg = "公众号已添加成功"
  394. br.IsSendEmail = false
  395. return
  396. }
  397. go cache.AddWechatArticleOpToCache(wechatPlatform.WechatPlatformId, `add`)
  398. br.Ret = 200
  399. br.Success = true
  400. br.Msg = "刷新成功"
  401. }
  402. // ArticleList
  403. // @Title 我关注的接口
  404. // @Description 我关注的接口
  405. // @Param PageSize query int true "每页数据条数"
  406. // @Param CurrentIndex query int true "当前页页码,从1开始"
  407. // @Param WechatPlatformId query int true "微信公众号id"
  408. // @Param KeyWord query string true "搜索关键词"
  409. // @Success 200 {object} []*rag.WechatPlatform
  410. // @router /wechat_platform/article/list [get]
  411. func (c *WechatPlatformController) ArticleList() {
  412. br := new(models.BaseResponse).Init()
  413. defer func() {
  414. c.Data["json"] = br
  415. c.ServeJSON()
  416. }()
  417. sysUser := c.SysUser
  418. if sysUser == nil {
  419. br.Msg = "请登录"
  420. br.ErrMsg = "请登录,SysUser Is Empty"
  421. return
  422. }
  423. pageSize, _ := c.GetInt("PageSize")
  424. currentIndex, _ := c.GetInt("CurrentIndex")
  425. wechatPlatformId, _ := c.GetInt("WechatPlatformId")
  426. keyWord := c.GetString("KeyWord")
  427. var startSize int
  428. if pageSize <= 0 {
  429. pageSize = utils.PageSize20
  430. }
  431. if currentIndex <= 0 {
  432. currentIndex = 1
  433. }
  434. startSize = utils.StartIndex(currentIndex, pageSize)
  435. var condition string
  436. var pars []interface{}
  437. if keyWord != "" {
  438. condition = fmt.Sprintf(` AND a.%s = ?`, rag.WechatPlatformColumns.Nickname)
  439. pars = append(pars, `%`+keyWord+`%`)
  440. }
  441. if wechatPlatformId > 0 {
  442. condition = fmt.Sprintf(` AND a.%s = ?`, rag.WechatArticleColumns.WechatPlatformID)
  443. pars = append(pars, wechatPlatformId)
  444. }
  445. condition = fmt.Sprintf(` AND b.%s = ?`, rag.WechatPlatformColumns.Enabled)
  446. pars = append(pars, 1)
  447. obj := new(rag.WechatArticle)
  448. total, list, err := obj.GetPageListByPlatformCondition(condition, pars, startSize, pageSize)
  449. if err != nil {
  450. br.Msg = "获取失败"
  451. br.ErrMsg = "获取失败,Err:" + err.Error()
  452. return
  453. }
  454. viewList := make([]rag.WechatArticleView, 0)
  455. if list != nil && len(list) > 0 {
  456. viewList = obj.ArticleAndPlatformListToViewList(list)
  457. }
  458. page := paging.GetPaging(currentIndex, pageSize, total)
  459. resp := response.WechatArticleListListResp{
  460. List: viewList,
  461. Paging: page,
  462. }
  463. br.Ret = 200
  464. br.Success = true
  465. br.Msg = "获取成功"
  466. br.Data = resp
  467. }
  468. // ArticleList
  469. // @Title 我关注的接口
  470. // @Description 我关注的接口
  471. // @Param WechatArticleId query int true "文章id"
  472. // @Success 200 {object} []*rag.WechatArticle
  473. // @router /wechat_platform/article/detail [get]
  474. func (c *WechatPlatformController) ArticleDetail() {
  475. br := new(models.BaseResponse).Init()
  476. defer func() {
  477. c.Data["json"] = br
  478. c.ServeJSON()
  479. }()
  480. sysUser := c.SysUser
  481. if sysUser == nil {
  482. br.Msg = "请登录"
  483. br.ErrMsg = "请登录,SysUser Is Empty"
  484. return
  485. }
  486. wechatArticleId, _ := c.GetInt("WechatArticleId")
  487. if wechatArticleId <= 0 {
  488. br.Msg = "请选择文章"
  489. br.IsSendEmail = false
  490. return
  491. }
  492. obj := new(rag.WechatArticle)
  493. item, err := obj.GetById(wechatArticleId)
  494. if err != nil {
  495. br.Msg = "获取失败"
  496. br.ErrMsg = "获取失败,Err:" + err.Error()
  497. return
  498. }
  499. if item.IsDeleted == 1 {
  500. br.Msg = "文章已删除"
  501. br.IsSendEmail = false
  502. return
  503. }
  504. resp := item.ToView()
  505. resp.Content = html.UnescapeString(item.Content)
  506. // 获取摘要信息
  507. {
  508. abstractObj := rag.WechatArticleAbstract{}
  509. abstractItem, err := abstractObj.GetByWechatArticleId(wechatArticleId)
  510. if err != nil && !utils.IsErrNoRow(err) {
  511. br.Msg = "获取失败"
  512. br.ErrMsg = "获取失败,Err:" + err.Error()
  513. return
  514. }
  515. resp.Abstract = abstractItem.Content
  516. }
  517. br.Ret = 200
  518. br.Success = true
  519. br.Msg = "获取成功"
  520. br.Data = resp
  521. }
  522. // ArticleList
  523. // @Title 我关注的接口
  524. // @Description 我关注的接口
  525. // @Param WechatArticleId query int true "文章id"
  526. // @Success 200 {object} []*rag.WechatPlatform
  527. // @router /wechat_platform/article/del [get]
  528. func (c *WechatPlatformController) ArticleDel() {
  529. br := new(models.BaseResponse).Init()
  530. defer func() {
  531. c.Data["json"] = br
  532. c.ServeJSON()
  533. }()
  534. sysUser := c.SysUser
  535. if sysUser == nil {
  536. br.Msg = "请登录"
  537. br.ErrMsg = "请登录,SysUser Is Empty"
  538. return
  539. }
  540. wechatArticleId, _ := c.GetInt("WechatArticleId")
  541. if wechatArticleId <= 0 {
  542. br.Msg = "请选择文章"
  543. br.IsSendEmail = false
  544. return
  545. }
  546. obj := new(rag.WechatArticle)
  547. item, err := obj.GetById(wechatArticleId)
  548. if err != nil {
  549. br.Msg = "获取失败"
  550. br.ErrMsg = "获取失败,Err:" + err.Error()
  551. return
  552. }
  553. if item.IsDeleted == 1 {
  554. br.Msg = "文章已删除"
  555. br.IsSendEmail = false
  556. return
  557. }
  558. item.IsDeleted = 1
  559. err = item.Update([]string{"is_deleted"})
  560. if err != nil {
  561. br.Msg = "删除失败"
  562. br.ErrMsg = "删除失败,Err:" + err.Error()
  563. return
  564. }
  565. br.Ret = 200
  566. br.Success = true
  567. br.Msg = "删除成功"
  568. }
  569. //func init() {
  570. // //obj := rag.WechatPlatform{}
  571. // //item, _ := obj.GetByID(2)
  572. // //fmt.Println(llm.BeachAddWechatPlatform(item))
  573. //
  574. // obj := rag.WechatArticle{}
  575. // //item, _ := obj.GetById(30)
  576. // list, _ := obj.GetListByCondition(``, ` `, []interface{}{}, 0, 1)
  577. // //llm.ArticleToTmpFile(item.TextContent)
  578. // for _, item := range list {
  579. // //llm.ArticleToKnowledge(item)
  580. // llm.GenerateArticleAbstract(item)
  581. // }
  582. //}