wechat_platform.go 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663
  1. package rag
  2. import (
  3. "encoding/json"
  4. "eta/eta_api/controllers"
  5. "eta/eta_api/models"
  6. "eta/eta_api/models/rag"
  7. "eta/eta_api/models/rag/request"
  8. "eta/eta_api/models/rag/response"
  9. "eta/eta_api/models/system"
  10. "eta/eta_api/services/llm"
  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 like ?`, 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 llm.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 like ?`, 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. obj := new(rag.WechatPlatformUserMapping)
  191. list, err := obj.GetListByCondition(condition, pars, startSize, pageSize)
  192. if err != nil {
  193. br.Msg = "获取失败"
  194. br.ErrMsg = "获取失败,Err:" + err.Error()
  195. return
  196. }
  197. br.Ret = 200
  198. br.Success = true
  199. br.Msg = "获取成功"
  200. br.Data = list
  201. }
  202. // PublicList
  203. // @Title 公共列表
  204. // @Description 公共列表
  205. // @Param PageSize query int true "每页数据条数"
  206. // @Param CurrentIndex query int true "当前页页码,从1开始"
  207. // @Param KeyWord query string true "搜索关键词"
  208. // @Success 200 {object} models.WechatPlatformListResp
  209. // @router /wechat_platform/list/public [get]
  210. func (c *WechatPlatformController) PublicList() {
  211. br := new(models.BaseResponse).Init()
  212. defer func() {
  213. c.Data["json"] = br
  214. c.ServeJSON()
  215. }()
  216. sysUser := c.SysUser
  217. if sysUser == nil {
  218. br.Msg = "请登录"
  219. br.ErrMsg = "请登录,SysUser Is Empty"
  220. return
  221. }
  222. pageSize, _ := c.GetInt("PageSize")
  223. currentIndex, _ := c.GetInt("CurrentIndex")
  224. keyWord := c.GetString("KeyWord")
  225. var startSize int
  226. if pageSize <= 0 {
  227. pageSize = utils.PageSize20
  228. }
  229. if currentIndex <= 0 {
  230. currentIndex = 1
  231. }
  232. startSize = utils.StartIndex(currentIndex, pageSize)
  233. var condition string
  234. var pars []interface{}
  235. if keyWord != "" {
  236. condition += fmt.Sprintf(` AND %s = ?`, rag.WechatPlatformColumns.Nickname)
  237. pars = append(pars, `%`+keyWord+`%`)
  238. }
  239. obj := new(rag.WechatPlatformUserMapping)
  240. list, err := obj.GetListByCondition(condition, pars, startSize, 100000)
  241. if err != nil {
  242. br.Msg = "获取失败"
  243. br.ErrMsg = "获取失败,Err:" + err.Error()
  244. return
  245. }
  246. resp := make([]response.WechatPlatformPublicListResp, 0)
  247. if list != nil && len(list) > 0 {
  248. userIdList := make([]int, 0)
  249. uerIdMap := make(map[int]bool)
  250. userFollowIndexMap := make(map[int]int)
  251. for _, v := range list {
  252. if _, ok := uerIdMap[v.FollowUserId]; !ok {
  253. userIdList = append(userIdList, v.FollowUserId)
  254. uerIdMap[v.FollowUserId] = true
  255. }
  256. index, ok := userFollowIndexMap[v.FollowUserId]
  257. if !ok {
  258. userFollowIndexMap[v.FollowUserId] = len(resp)
  259. resp = append(resp, response.WechatPlatformPublicListResp{
  260. UserId: v.FollowUserId,
  261. List: []*rag.UserFollowWechatPlatform{v},
  262. })
  263. } else {
  264. resp[index].List = append(resp[index].List, v)
  265. }
  266. }
  267. userList, err := system.GetAdminListByIdList(userIdList)
  268. if err != nil {
  269. br.Msg = "获取失败"
  270. br.ErrMsg = "获取失败,Err:" + err.Error()
  271. return
  272. }
  273. userNameMap := make(map[int]*system.Admin)
  274. for _, v := range userList {
  275. userNameMap[v.AdminId] = v
  276. }
  277. for k, v := range resp {
  278. userInfo, ok := userNameMap[v.UserId]
  279. if !ok {
  280. continue
  281. }
  282. resp[k].Name = userInfo.RealName + `关注`
  283. }
  284. }
  285. br.Ret = 200
  286. br.Success = true
  287. br.Msg = "获取成功"
  288. br.Data = resp
  289. }
  290. // PublicList
  291. // @Title 公共列表
  292. // @Description 公共列表
  293. // @Param PageSize query int true "每页数据条数"
  294. // @Param CurrentIndex query int true "当前页页码,从1开始"
  295. // @Param KeyWord query string true "搜索关键词"
  296. // @Success 200 {object} models.WechatPlatformListResp
  297. // @router /wechat_platform/op [post]
  298. func (c *WechatPlatformController) Op() {
  299. br := new(models.BaseResponse).Init()
  300. defer func() {
  301. c.Data["json"] = br
  302. c.ServeJSON()
  303. }()
  304. sysUser := c.SysUser
  305. if sysUser == nil {
  306. br.Msg = "请登录"
  307. br.ErrMsg = "请登录,SysUser Is Empty"
  308. return
  309. }
  310. var req request.OpWechatPlatformReq
  311. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  312. if err != nil {
  313. br.Msg = "参数解析异常!"
  314. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  315. return
  316. }
  317. if req.WechatPlatformId <= 0 {
  318. br.Msg = "参数错误"
  319. return
  320. }
  321. if req.Status < 0 || req.Status > 1 {
  322. br.Msg = "参数错误"
  323. return
  324. }
  325. obj := rag.WechatPlatform{}
  326. wechatPlatform, err := obj.GetByID(req.WechatPlatformId)
  327. if err != nil {
  328. br.Msg = "修改失败"
  329. br.ErrMsg = "修改失败,Err:" + err.Error()
  330. if utils.IsErrNoRow(err) {
  331. br.Msg = "公众号不存在"
  332. br.IsSendEmail = false
  333. }
  334. return
  335. }
  336. wechatPlatform.Enabled = req.Status
  337. wechatPlatform.ModifyTime = time.Now()
  338. err = wechatPlatform.Update([]string{"enabled", `modify_time`})
  339. if err != nil {
  340. br.Msg = "修改失败"
  341. br.ErrMsg = "修改失败,Err:" + err.Error()
  342. return
  343. }
  344. br.Ret = 200
  345. br.Success = true
  346. br.Msg = "修改成功"
  347. }
  348. // Refresh
  349. // @Title 公共列表
  350. // @Description 公共列表
  351. // @Success 200 {object} models.WechatPlatformListResp
  352. // @router /wechat_platform/refresh [post]
  353. func (c *WechatPlatformController) Refresh() {
  354. br := new(models.BaseResponse).Init()
  355. defer func() {
  356. c.Data["json"] = br
  357. c.ServeJSON()
  358. }()
  359. sysUser := c.SysUser
  360. if sysUser == nil {
  361. br.Msg = "请登录"
  362. br.ErrMsg = "请登录,SysUser Is Empty"
  363. return
  364. }
  365. var req request.RefreshWechatPlatformReq
  366. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  367. if err != nil {
  368. br.Msg = "参数解析异常!"
  369. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  370. return
  371. }
  372. if req.WechatPlatformId <= 0 {
  373. br.Msg = "参数错误"
  374. return
  375. }
  376. obj := rag.WechatPlatform{}
  377. wechatPlatform, err := obj.GetByID(req.WechatPlatformId)
  378. if err != nil {
  379. br.Msg = "修改失败"
  380. br.ErrMsg = "修改失败,Err:" + err.Error()
  381. if utils.IsErrNoRow(err) {
  382. br.Msg = "公众号不存在"
  383. br.IsSendEmail = false
  384. }
  385. return
  386. }
  387. if wechatPlatform.FakeId != `` {
  388. br.Msg = "公众号已添加成功"
  389. br.ErrMsg = "公众号已添加成功"
  390. br.IsSendEmail = false
  391. return
  392. }
  393. go llm.AddWechatArticleOpToCache(wechatPlatform.WechatPlatformId, `add`)
  394. br.Ret = 200
  395. br.Success = true
  396. br.Msg = "刷新成功"
  397. }
  398. // ArticleList
  399. // @Title 我关注的接口
  400. // @Description 我关注的接口
  401. // @Param PageSize query int true "每页数据条数"
  402. // @Param CurrentIndex query int true "当前页页码,从1开始"
  403. // @Param WechatPlatformId query int true "微信公众号id"
  404. // @Param KeyWord query string true "搜索关键词"
  405. // @Success 200 {object} []*rag.WechatPlatform
  406. // @router /wechat_platform/article/list [get]
  407. func (c *WechatPlatformController) ArticleList() {
  408. br := new(models.BaseResponse).Init()
  409. defer func() {
  410. c.Data["json"] = br
  411. c.ServeJSON()
  412. }()
  413. sysUser := c.SysUser
  414. if sysUser == nil {
  415. br.Msg = "请登录"
  416. br.ErrMsg = "请登录,SysUser Is Empty"
  417. return
  418. }
  419. pageSize, _ := c.GetInt("PageSize")
  420. currentIndex, _ := c.GetInt("CurrentIndex")
  421. wechatPlatformId, _ := c.GetInt("WechatPlatformId")
  422. keyWord := c.GetString("KeyWord")
  423. var startSize int
  424. if pageSize <= 0 {
  425. pageSize = utils.PageSize20
  426. }
  427. if currentIndex <= 0 {
  428. currentIndex = 1
  429. }
  430. startSize = utils.StartIndex(currentIndex, pageSize)
  431. var condition string
  432. var pars []interface{}
  433. if keyWord != "" {
  434. condition += fmt.Sprintf(` AND (b.%s like ? or a.%s like ? ) `, rag.WechatPlatformColumns.Nickname, rag.WechatArticleColumns.Title)
  435. pars = append(pars, `%`+keyWord+`%`, `%`+keyWord+`%`)
  436. }
  437. if wechatPlatformId > 0 {
  438. condition += fmt.Sprintf(` AND a.%s = ?`, rag.WechatArticleColumns.WechatPlatformID)
  439. pars = append(pars, wechatPlatformId)
  440. }
  441. condition += fmt.Sprintf(` AND b.%s = ? `, rag.WechatPlatformColumns.Enabled)
  442. pars = append(pars, 1)
  443. obj := new(rag.WechatArticle)
  444. total, list, err := obj.GetPageListByPlatformCondition(condition, pars, startSize, pageSize)
  445. if err != nil {
  446. br.Msg = "获取失败"
  447. br.ErrMsg = "获取失败,Err:" + err.Error()
  448. return
  449. }
  450. viewList := make([]rag.WechatArticleView, 0)
  451. if list != nil && len(list) > 0 {
  452. viewList = obj.ArticleAndPlatformListToViewList(list)
  453. }
  454. page := paging.GetPaging(currentIndex, pageSize, total)
  455. resp := response.WechatArticleListListResp{
  456. List: viewList,
  457. Paging: page,
  458. }
  459. br.Ret = 200
  460. br.Success = true
  461. br.Msg = "获取成功"
  462. br.Data = resp
  463. }
  464. // ArticleList
  465. // @Title 我关注的接口
  466. // @Description 我关注的接口
  467. // @Param WechatArticleId query int true "文章id"
  468. // @Success 200 {object} []*rag.WechatArticle
  469. // @router /wechat_platform/article/detail [get]
  470. func (c *WechatPlatformController) ArticleDetail() {
  471. br := new(models.BaseResponse).Init()
  472. defer func() {
  473. c.Data["json"] = br
  474. c.ServeJSON()
  475. }()
  476. sysUser := c.SysUser
  477. if sysUser == nil {
  478. br.Msg = "请登录"
  479. br.ErrMsg = "请登录,SysUser Is Empty"
  480. return
  481. }
  482. wechatArticleId, _ := c.GetInt("WechatArticleId")
  483. if wechatArticleId <= 0 {
  484. br.Msg = "请选择文章"
  485. br.IsSendEmail = false
  486. return
  487. }
  488. obj := new(rag.WechatArticle)
  489. item, err := obj.GetById(wechatArticleId)
  490. if err != nil {
  491. br.Msg = "获取失败"
  492. br.ErrMsg = "获取失败,Err:" + err.Error()
  493. return
  494. }
  495. if item.IsDeleted == 1 {
  496. br.Msg = "文章已删除"
  497. br.IsSendEmail = false
  498. return
  499. }
  500. resp := item.ToView()
  501. resp.Content = html.UnescapeString(item.Content)
  502. // 获取摘要信息
  503. {
  504. abstractObj := rag.WechatArticleAbstract{}
  505. abstractItem, err := abstractObj.GetByWechatArticleId(wechatArticleId)
  506. if err != nil && !utils.IsErrNoRow(err) {
  507. br.Msg = "获取失败"
  508. br.ErrMsg = "获取失败,Err:" + err.Error()
  509. return
  510. }
  511. resp.Abstract = abstractItem.Content
  512. }
  513. br.Ret = 200
  514. br.Success = true
  515. br.Msg = "获取成功"
  516. br.Data = resp
  517. }
  518. // ArticleList
  519. // @Title 我关注的接口
  520. // @Description 我关注的接口
  521. // @Param WechatArticleId query int true "文章id"
  522. // @Success 200 {object} []*rag.WechatPlatform
  523. // @router /wechat_platform/article/del [get]
  524. func (c *WechatPlatformController) ArticleDel() {
  525. br := new(models.BaseResponse).Init()
  526. defer func() {
  527. c.Data["json"] = br
  528. c.ServeJSON()
  529. }()
  530. sysUser := c.SysUser
  531. if sysUser == nil {
  532. br.Msg = "请登录"
  533. br.ErrMsg = "请登录,SysUser Is Empty"
  534. return
  535. }
  536. wechatArticleId, _ := c.GetInt("WechatArticleId")
  537. if wechatArticleId <= 0 {
  538. br.Msg = "请选择文章"
  539. br.IsSendEmail = false
  540. return
  541. }
  542. obj := new(rag.WechatArticle)
  543. item, err := obj.GetById(wechatArticleId)
  544. if err != nil {
  545. br.Msg = "获取失败"
  546. br.ErrMsg = "获取失败,Err:" + err.Error()
  547. return
  548. }
  549. if item.IsDeleted == 1 {
  550. br.Msg = "文章已删除"
  551. br.IsSendEmail = false
  552. return
  553. }
  554. item.IsDeleted = 1
  555. err = item.Update([]string{"is_deleted"})
  556. if err != nil {
  557. br.Msg = "删除失败"
  558. br.ErrMsg = "删除失败,Err:" + err.Error()
  559. return
  560. }
  561. br.Ret = 200
  562. br.Success = true
  563. br.Msg = "删除成功"
  564. }
  565. //func init() {
  566. // //obj := rag.WechatPlatform{}
  567. // //item, _ := obj.GetByID(2)
  568. // //fmt.Println(llm.BeachAddWechatPlatform(item))
  569. //
  570. // obj := rag.WechatArticle{}
  571. // //item, _ := obj.GetById(30)
  572. // list, _ := obj.GetListByCondition(``, ` `, []interface{}{}, 0, 1000)
  573. // //llm.ArticleToTmpFile(item.TextContent)
  574. // for _, item := range list {
  575. // //llm.ArticleToKnowledge(item)
  576. // llm.GenerateArticleAbstract(item)
  577. // }
  578. //}
  579. //func init() {
  580. // //obj := rag.WechatPlatform{}
  581. // //item, _ := obj.GetByID(2)
  582. // //fmt.Println(llm.BeachAddWechatPlatform(item))
  583. //
  584. // obj := rag.WechatArticle{}
  585. // list, _ := obj.GetListByCondition(`wechat_article_id,cover_url`, ` `, []interface{}{}, 0, 1000)
  586. // //obj := rag.WechatPlatform{}
  587. // //list, _ := obj.GetListByCondition(` AND wechat_platform_id !=1 `, []interface{}{}, 0, 100)
  588. // //llm.ArticleToTmpFile(item.TextContent)
  589. // for _, item := range list {
  590. // //llm.ArticleToKnowledge(item)
  591. // services.ReplaceWechatArticlePic(item)
  592. // }
  593. //}