common.go 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
  1. package common
  2. import (
  3. "bytes"
  4. "context"
  5. "crypto/tls"
  6. "errors"
  7. "git.listensoft.net/tool/jspkit/common/lxhttp"
  8. "git.listensoft.net/tool/jspkit/common/models"
  9. "git.listensoft.net/tool/jspkit/common/variable"
  10. "git.listensoft.net/tool/jspkit/logger"
  11. "git.listensoft.net/tool/jspkit/taxerr"
  12. "github.com/go-kratos/kratos/v2/log"
  13. "github.com/go-rod/rod"
  14. "github.com/google/uuid"
  15. "go.uber.org/zap"
  16. "io"
  17. "mime/multipart"
  18. "net/http"
  19. "os"
  20. "strings"
  21. "time"
  22. )
  23. var ExcelServiceUrl = `http://47.105.103.181:4000/excel` // 新版解析excel服务
  24. var InfoCscsts bool = false
  25. const (
  26. ClickTimeOut = 5 * time.Second
  27. LoadTimeout = 20 * time.Second
  28. )
  29. // 手机号锁 time 锁定多长时间 传-1解锁
  30. func AddTelLockerX2(tel string, secend string) {
  31. c := lxhttp.NewHttpClient()
  32. bytes, _ := lxhttp.Get(c, variable.TaxTaskURL+"/api/v1/saveTelLock?tel="+tel+"&time="+secend)
  33. logger.Info("TelLock: " + string(bytes) + secend + "s")
  34. }
  35. func AddTelLockerX2Req(tel string, secend string, reqs string) {
  36. c := lxhttp.NewHttpClient()
  37. uri := variable.TaxTaskURL + "/api/v1/saveTelLock?tel=" + tel + "&time=" + secend
  38. if reqs != "" {
  39. uri += "&reqNos=" + reqs
  40. }
  41. bytes, _ := lxhttp.Get(c, uri)
  42. logger.Info(tel + " TelLock: " + string(bytes) + " " + secend + "s")
  43. }
  44. func HandleError(ctx context.Context, err error) error {
  45. if err == nil {
  46. return nil
  47. }
  48. uuid := ""
  49. _ = rod.Try(func() {
  50. if ctx != nil {
  51. uuid = ctx.Value(variable.UUID).(string)
  52. }
  53. })
  54. var usererr *taxerr.UserErr
  55. var taxerr2 *taxerr.SystemErr
  56. if errors.As(err, &taxerr2) {
  57. log.Info(uuid, " ", taxerr2.Error())
  58. if !strings.Contains(taxerr2.Error(), "[异常]:") {
  59. return errors.New("[异常]:" + taxerr2.Error())
  60. } else {
  61. return errors.New(taxerr2.Error())
  62. }
  63. } else if errors.As(err, &usererr) {
  64. log.Info(uuid, " ", usererr.Error())
  65. if !strings.Contains(usererr.Error(), "[错误]:") {
  66. return errors.New("[错误]:" + usererr.Error())
  67. } else {
  68. return errors.New(usererr.Error())
  69. }
  70. } else if err != nil {
  71. log.Error(uuid, " ", err.Error())
  72. return taxerr.NewWebStuckTitle(true)
  73. }
  74. return nil
  75. }
  76. // url 路径 filename 文件的上传参数
  77. func PostFile(url string, files map[string]string, ext map[string]string) ([]byte, error) {
  78. res := []byte{}
  79. //创建一个模拟的form中的一个选项,这个form项现在是空的
  80. bodyBuf := &bytes.Buffer{}
  81. bodyWriter := multipart.NewWriter(bodyBuf)
  82. for k, v := range files {
  83. //打开文件句柄操作
  84. file, err := os.Open(v)
  85. if err != nil {
  86. //zaplog.LoggerS.Info("error opening file")
  87. return res, err
  88. }
  89. defer file.Close()
  90. //相当于现在还没选择文件, form项里选择文件的选项
  91. fileWriter, err := bodyWriter.CreateFormFile(k, file.Name())
  92. if err != nil {
  93. //zaplog.LoggerS.Info("error writing to buffer")
  94. return res, err
  95. }
  96. //iocopy 这里相当于选择了文件,将文件放到form中
  97. _, err = io.Copy(fileWriter, file)
  98. if err != nil {
  99. return res, err
  100. }
  101. }
  102. //获取上传文件的类型,multipart/form-data; boundary=...
  103. contentType := bodyWriter.FormDataContentType()
  104. //上传的其他参数
  105. for key, val := range ext {
  106. _ = bodyWriter.WriteField(key, val)
  107. }
  108. //这个很关键,必须这样写关闭,不能使用defer关闭,不然会导致错误
  109. bodyWriter.Close()
  110. // 忽略证书
  111. c := http.Client{
  112. Transport: &http.Transport{
  113. TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
  114. },
  115. Timeout: time.Second * 60,
  116. }
  117. c.CloseIdleConnections()
  118. //发送post请求到服务端
  119. resp, err := c.Post(url, contentType, bodyBuf)
  120. if err != nil {
  121. return res, err
  122. }
  123. defer resp.Body.Close()
  124. return io.ReadAll(resp.Body)
  125. }
  126. func GetBaseUrl(p *rod.Page) string {
  127. URL := p.MustInfo().URL
  128. baseUrl := "https://" + URL[8:][:strings.Index(URL[8:], "/")]
  129. return baseUrl
  130. }
  131. func GetUUid() string {
  132. // 创建新的UUID
  133. newID := uuid.New().String()
  134. return strings.ReplaceAll(newID, "-", "")
  135. }
  136. // 拦截请求 for 代理
  137. func PageHijackReqForProxy(b *rod.Page, pattern string, ch chan []byte, c *http.Client) *rod.HijackRouter {
  138. router := b.HijackRequests()
  139. router.MustAdd(pattern, func(ctx *rod.Hijack) {
  140. rod.Try(func() {
  141. err := ctx.LoadResponse(c, true)
  142. if err != nil {
  143. return
  144. }
  145. ch <- []byte(ctx.Response.Body())
  146. time.Sleep(time.Second * 5)
  147. select {
  148. case <-ch:
  149. break
  150. }
  151. })
  152. })
  153. go router.Run()
  154. return router
  155. }
  156. // 拦截请求
  157. func BrowserHijackReq(b *rod.Browser, pattern string, ch chan []byte) *rod.HijackRouter {
  158. router := b.HijackRequests()
  159. router.MustAdd(pattern, func(ctx *rod.Hijack) {
  160. rod.Try(func() {
  161. ctx.MustLoadResponse()
  162. ch <- []byte(ctx.Response.Body())
  163. time.Sleep(time.Second * 5)
  164. select {
  165. case <-ch:
  166. break
  167. }
  168. })
  169. })
  170. go router.Run()
  171. return router
  172. }
  173. // DeleteTpassCookie 失效当前cookie
  174. func DeleteTpassCookie(info models.CompanyInfo) {
  175. //江西新版登录和旧版的tpass,分开来存储,防止新版登录的公司拿到旧版登录的tapss
  176. tsessionKey := info.Tel
  177. if !strings.Contains(tsessionKey, "_"+info.Area) {
  178. tsessionKey += "_" + info.Area
  179. }
  180. if strings.Contains(info.Dlfs, "代理") {
  181. tsessionKey += "Agent"
  182. }
  183. if (info.Area == "jiangxi" || info.Area == "heilongjiang" || info.Area == "guizhou" || info.Area == "chongqing") && info.Dlfs != "新版登录" {
  184. tsessionKey += "Other"
  185. }
  186. tsessionKey += "_tpass"
  187. client := lxhttp.NewHttpClient()
  188. body := map[string]string{
  189. "tsessionKey": tsessionKey,
  190. "pwd": info.Zzrmm,
  191. }
  192. logger.Info("req", zap.Any("body", body))
  193. bys, err := lxhttp.POSTJson(client, variable.SessionKeepURL+"/api/v1/session/TSession/delete", body, map[string]string{})
  194. if err != nil {
  195. logger.Error(err.Error())
  196. }
  197. logger.Info("删除key:" + tsessionKey + "结果: " + string(bys))
  198. }