|
|
@@ -24,6 +24,10 @@ import (
|
|
|
"strconv"
|
|
|
"strings"
|
|
|
"time"
|
|
|
+
|
|
|
+ "github.com/beego/beego/v2/server/web"
|
|
|
+ "github.com/beego/beego/v2/server/web/context"
|
|
|
+ "github.com/go-redis/redis/v8"
|
|
|
)
|
|
|
|
|
|
// AddReportAndChapter
|
|
|
@@ -1221,7 +1225,7 @@ func PublishReport(reportId int, reportUrl string, sysUser *system.Admin) (tips
|
|
|
|
|
|
// 生成报告pdf和长图
|
|
|
{
|
|
|
- reportPdfUrl := GetGeneralPdfUrl(reportInfo.ReportCode, reportInfo.ClassifyNameFirst, reportInfo.ReportLayout)
|
|
|
+ reportPdfUrl := GetGeneralPdfUrl(reportInfo.Id, reportInfo.ReportCode, reportInfo.ClassifyNameFirst, reportInfo.ReportLayout)
|
|
|
go Report2pdfAndJpeg(reportPdfUrl, reportId, 1)
|
|
|
}
|
|
|
|
|
|
@@ -1348,7 +1352,7 @@ func PublishChapterReport(reportInfo *models.Report, reportUrl string, sysUser *
|
|
|
|
|
|
// 生成报告pdf和长图
|
|
|
{
|
|
|
- reportPdfUrl := GetGeneralPdfUrl(reportInfo.ReportCode, reportInfo.ClassifyNameFirst, reportInfo.ReportLayout)
|
|
|
+ reportPdfUrl := GetGeneralPdfUrl(reportInfo.Id, reportInfo.ReportCode, reportInfo.ClassifyNameFirst, reportInfo.ReportLayout)
|
|
|
go Report2pdfAndJpeg(reportPdfUrl, reportId, 1)
|
|
|
}
|
|
|
|
|
|
@@ -1586,7 +1590,7 @@ func UpdateReportVideo(reportInfo *models.Report) {
|
|
|
// @param reportCode string
|
|
|
// @param reportLayout int8
|
|
|
// @return pdfUrl string
|
|
|
-func GetGeneralPdfUrl(reportCode, classifyFirstName string, reportLayout int8) (pdfUrl string) {
|
|
|
+func GetGeneralPdfUrl(reportId int, reportCode, classifyFirstName string, reportLayout int8) (pdfUrl string) {
|
|
|
// 如果是弘则，且是晨、周报，那么就不返回
|
|
|
if utils.InArrayByStr([]string{utils.BusinessCodeRelease, utils.BusinessCodeSandbox, utils.BusinessCodeDebug}, utils.BusinessCode) && utils.InArrayByStr([]string{"晨报", "周报"}, classifyFirstName) {
|
|
|
return
|
|
|
@@ -1615,6 +1619,14 @@ func GetGeneralPdfUrl(reportCode, classifyFirstName string, reportLayout int8) (
|
|
|
pdfUrl = fmt.Sprintf("%s/reportshare_smart_pdf?code=%s", reportUrl, reportCode)
|
|
|
}
|
|
|
|
|
|
+ if pdfUrl != "" {
|
|
|
+ token := utils.MD5(fmt.Sprint(pdfUrl, time.Now().UnixNano()/1e6))
|
|
|
+ e := generalReportAuthToken(token, ``, reportId)
|
|
|
+ if e == nil {
|
|
|
+ pdfUrl = fmt.Sprintf("%s&authToken=%s", pdfUrl, token)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
return
|
|
|
}
|
|
|
|
|
|
@@ -1909,3 +1921,407 @@ func processMapTable(data map[string]interface{}, reportId, fromScene int) error
|
|
|
}
|
|
|
return nil
|
|
|
}
|
|
|
+
|
|
|
+// HandleReportContent
|
|
|
+// @Description: 处理报告内容（动态图表/表格添加授权token）
|
|
|
+// @author: Roc
|
|
|
+// @datetime 2025-01-07 10:03:15
|
|
|
+// @param body string
|
|
|
+// @return newBody string
|
|
|
+func HandleReportContent(body string, opType string, tokenMap map[string]string) (newBody string) {
|
|
|
+ if body == `` {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ newBody = body
|
|
|
+
|
|
|
+ // 解析HTML
|
|
|
+ doc, err := html2.Parse(strings.NewReader(body))
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("Error parsing HTML:", err)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ replaceIframeSrc(doc, opType, tokenMap)
|
|
|
+
|
|
|
+ // 输出修改后的HTML
|
|
|
+ var modifiedHtml strings.Builder
|
|
|
+ err = html2.Render(&modifiedHtml, doc)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("Error rendering HTML:", err)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ newBody = modifiedHtml.String()
|
|
|
+ fmt.Println(newBody)
|
|
|
+
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+// replaceIframeSrc 遍历HTML节点，替换iframe的src属性
|
|
|
+func replaceIframeSrc(n *html2.Node, opType string, tokenMap map[string]string) {
|
|
|
+ if n.Type == html2.ElementNode && n.Data == "iframe" {
|
|
|
+ for i, attr := range n.Attr {
|
|
|
+ if attr.Key == "src" {
|
|
|
+ newLink := attr.Val
|
|
|
+ // 处理链接
|
|
|
+ switch opType {
|
|
|
+ case `add`:
|
|
|
+ newLink = linkAddToken(attr.Val, tokenMap)
|
|
|
+ case `del`:
|
|
|
+ newLink = linkDelToken(attr.Val)
|
|
|
+ }
|
|
|
+ // 替换原来的链接
|
|
|
+ n.Attr[i].Val = newLink
|
|
|
+ break
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ // 递归处理子节点
|
|
|
+ for c := n.FirstChild; c != nil; c = c.NextSibling {
|
|
|
+ replaceIframeSrc(c, opType, tokenMap)
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+// linkAddToken 链接添加token
|
|
|
+func linkAddToken(link string, tokenMap map[string]string) string {
|
|
|
+ var err error
|
|
|
+ defer func() {
|
|
|
+ if err != nil {
|
|
|
+ utils.FileLog.Info("处理链接失败，ERR：" + err.Error())
|
|
|
+ }
|
|
|
+ }()
|
|
|
+ if link == `` {
|
|
|
+ return link
|
|
|
+ }
|
|
|
+ parsedURL, err := url.Parse(link)
|
|
|
+ if err != nil {
|
|
|
+ return link
|
|
|
+ }
|
|
|
+
|
|
|
+ // 获取查询参数
|
|
|
+ queryParams := parsedURL.Query()
|
|
|
+
|
|
|
+ // 先移除authToken参数，避免莫名其妙的这个值入库了
|
|
|
+ queryParams.Del("authToken")
|
|
|
+
|
|
|
+ // 获取code参数
|
|
|
+ code := queryParams.Get("code")
|
|
|
+ if code == "" {
|
|
|
+ return link
|
|
|
+ }
|
|
|
+
|
|
|
+ showType := `chart`
|
|
|
+ if strings.Contains(parsedURL.Path, "sheetshow") {
|
|
|
+ showType = `excel`
|
|
|
+ }
|
|
|
+
|
|
|
+ // 避免报告里面一个图表/表格重复生成token
|
|
|
+ key := fmt.Sprint(showType, `:`, code)
|
|
|
+ if tokenMap != nil {
|
|
|
+ if token, ok := tokenMap[key]; ok {
|
|
|
+ // 在链接后面添加一个token值
|
|
|
+ return link + "&authToken=" + token
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ token, err := GeneralChartToken(showType, code)
|
|
|
+ if err != nil {
|
|
|
+ return link
|
|
|
+ }
|
|
|
+
|
|
|
+ if tokenMap != nil {
|
|
|
+ tokenMap[key] = token
|
|
|
+ }
|
|
|
+
|
|
|
+ // 在链接后面添加一个token值
|
|
|
+ return link + "&authToken=" + token
|
|
|
+}
|
|
|
+
|
|
|
+// linkDelToken 链接添加token
|
|
|
+func linkDelToken(link string) string {
|
|
|
+ var err error
|
|
|
+ defer func() {
|
|
|
+ if err != nil {
|
|
|
+ utils.FileLog.Info("处理链接失败，ERR：" + err.Error())
|
|
|
+ }
|
|
|
+ }()
|
|
|
+ parsedURL, err := url.Parse(link)
|
|
|
+ if err != nil {
|
|
|
+ return link
|
|
|
+ }
|
|
|
+
|
|
|
+ // 获取查询参数
|
|
|
+ queryParams := parsedURL.Query()
|
|
|
+
|
|
|
+ // 移除authToken参数
|
|
|
+ queryParams.Del("authToken")
|
|
|
+
|
|
|
+ // 更新URL的查询参数
|
|
|
+ parsedURL.RawQuery = queryParams.Encode()
|
|
|
+
|
|
|
+ return parsedURL.String()
|
|
|
+}
|
|
|
+
|
|
|
+// GeneralChartToken
|
|
|
+// @Description: 生成图表/表格授权token
|
|
|
+// @author: Roc
|
|
|
+// @datetime 2025-01-07 10:41:36
|
|
|
+// @param showType string
|
|
|
+// @param uniqueCode string
|
|
|
+// @return token string
|
|
|
+// @return err error
|
|
|
+func GeneralChartToken(showType, uniqueCode string) (token string, err error) {
|
|
|
+ // 缓存key
|
|
|
+ token = utils.MD5(fmt.Sprint(showType+`:`, uniqueCode, time.Now().UnixNano()/1e6))
|
|
|
+ key := fmt.Sprint(utils.CACHE_CHART_AUTH, token)
|
|
|
+ err = utils.Rc.Put(key, uniqueCode, utils.BusinessConfReportChartExpiredTime)
|
|
|
+
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+// GeneralReportToken
|
|
|
+// @Description: 生成报告授权token
|
|
|
+// @author: Roc
|
|
|
+// @datetime 2025-01-07 10:41:36
|
|
|
+// @param uniqueCode string
|
|
|
+// @return token string
|
|
|
+// @return err error
|
|
|
+func GeneralReportToken(linkToken string, reportId int) (token string, err error) {
|
|
|
+ // 图表授权token
|
|
|
+ token = utils.MD5(fmt.Sprint(linkToken, time.Now().UnixNano()/1e6))
|
|
|
+
|
|
|
+ // 缓存key
|
|
|
+ reportKey := getReportShareTokenKey(linkToken)
|
|
|
+ err = utils.Rc.Put(reportKey, token, utils.BusinessConfReportChartExpiredTime)
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ // 生成报告的图表授权token
|
|
|
+ err = generalReportAuthToken(token, ``, reportId)
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+// generalReportAuthToken
|
|
|
+// @Description: 生成报告的图表授权token
|
|
|
+// @author: Roc
|
|
|
+// @datetime 2025-03-17 17:47:07
|
|
|
+// @param token string
|
|
|
+// @param reportId int
|
|
|
+// @return err error
|
|
|
+func generalReportAuthToken(token, source string, reportId int) (err error) {
|
|
|
+ // 缓存key
|
|
|
+ reportTokenKey := getReportTokenKey(token, source)
|
|
|
+ err = utils.Rc.Put(reportTokenKey, reportId, utils.BusinessConfReportChartExpiredTime)
|
|
|
+
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+// getReportShareTokenKey
|
|
|
+// @Description:
|
|
|
+// @author: Roc
|
|
|
+// @datetime 2025-03-17 14:00:14
|
|
|
+// @param linkToken string
|
|
|
+// @return string
|
|
|
+func getReportShareTokenKey(linkToken string) string {
|
|
|
+ return fmt.Sprint(utils.CACHE_REPORT_SHARE_AUTH, utils.MD5(linkToken))
|
|
|
+}
|
|
|
+
|
|
|
+// GetReportAuthToken
|
|
|
+// @Description: 获取报告token
|
|
|
+// @author: Roc
|
|
|
+// @datetime 2025-03-17 16:48:38
|
|
|
+// @param linkToken string
|
|
|
+// @return string
|
|
|
+func GetReportAuthToken(linkToken string) string {
|
|
|
+ key := getReportShareTokenKey(linkToken)
|
|
|
+ return utils.Rc.GetStr(key)
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+// getReportTokenKey
|
|
|
+// @Description:
|
|
|
+// @author: Roc
|
|
|
+// @datetime 2025-03-17 14:00:14
|
|
|
+// @param linkToken string
|
|
|
+// @return string
|
|
|
+func getReportTokenKey(token, source string) string {
|
|
|
+ return fmt.Sprint(utils.CACHE_REPORT_AUTH, source, token)
|
|
|
+}
|
|
|
+
|
|
|
+// HandleReportContentStruct
|
|
|
+// @Description: 处理内容组件的链接
|
|
|
+// @author: Roc
|
|
|
+// @datetime 2025-01-07 13:38:39
|
|
|
+// @param body string
|
|
|
+// @param opType string
|
|
|
+// @return newBody string
|
|
|
+func HandleReportContentStruct(body string, opType string, tokenMap map[string]string) (newBody string) {
|
|
|
+ if body == `` {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ newBody = body
|
|
|
+
|
|
|
+ // 解析JSON数据到map[string]interface{}
|
|
|
+ var jsonData []map[string]interface{}
|
|
|
+ if err := json.Unmarshal([]byte(body), &jsonData); err != nil {
|
|
|
+ fmt.Println("Error parsing JSON:", err)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ // 处理每个组件
|
|
|
+ for i := range jsonData {
|
|
|
+ if err := processMap(jsonData[i], opType, tokenMap); err != nil {
|
|
|
+ fmt.Println("Error processing component:", err)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 将处理后的数据转换回JSON字符串
|
|
|
+ modifiedJSON, err := json.MarshalIndent(jsonData, "", " ")
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("Error marshaling JSON:", err)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ newBody = string(modifiedJSON)
|
|
|
+
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+// processMap 递归处理map中的content字段
|
|
|
+func processMap(data map[string]interface{}, opType string, tokenMap map[string]string) error {
|
|
|
+ for key, value := range data {
|
|
|
+ switch v := value.(type) {
|
|
|
+ case string:
|
|
|
+ if key == "content" {
|
|
|
+ contentSource, ok := data["compType"]
|
|
|
+ if !ok {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ contentSourceType, ok := contentSource.(string)
|
|
|
+ if !ok {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ if !utils.InArrayByStr([]string{`sheet`, `chart`}, contentSourceType) {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+
|
|
|
+ newContent := v
|
|
|
+ // 处理链接
|
|
|
+ switch opType {
|
|
|
+ case `add`:
|
|
|
+ newContent = linkAddToken(v, tokenMap)
|
|
|
+ case `del`:
|
|
|
+ newContent = linkDelToken(v)
|
|
|
+ }
|
|
|
+ data[key] = newContent
|
|
|
+ }
|
|
|
+ case map[string]interface{}:
|
|
|
+ if err := processMap(v, opType, tokenMap); err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ case []interface{}:
|
|
|
+ for i := range v {
|
|
|
+ if m, ok := v[i].(map[string]interface{}); ok {
|
|
|
+ if err := processMap(m, opType, tokenMap); err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+// GetReportShareUrlToken 获取报告分享链接token
|
|
|
+func GetReportShareUrlToken(req models.ReportShartUrlReq, adminId int) (linkToken string, err error) {
|
|
|
+ defer func() {
|
|
|
+ if err == nil && linkToken != `` {
|
|
|
+ GeneralReportToken(linkToken, req.ReportId)
|
|
|
+ }
|
|
|
+ }()
|
|
|
+ cacheLinkKey := utils.CACHE_REPORT_SHARE_SHORT_Url + strconv.Itoa(req.ReportId) + "userId:" + strconv.Itoa(adminId)
|
|
|
+ linkToken, _ = utils.Rc.RedisString(cacheLinkKey)
|
|
|
+ if linkToken != "" && utils.Rc.IsExist(fmt.Sprint(utils.CACHE_REPORT_SHARE_ORIGIN_Url, utils.MD5(linkToken))) {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ var tokenKey string
|
|
|
+
|
|
|
+ var ok bool
|
|
|
+ // 冲突检测
|
|
|
+ for i := 0; i < 3; i++ {
|
|
|
+ linkToken = req.Url
|
|
|
+ if i > 0 {
|
|
|
+ linkToken += "_" + utils.GetRandDigit(3)
|
|
|
+ }
|
|
|
+ hashUrl := utils.MurmurHash64([]byte(linkToken))
|
|
|
+ linkToken = utils.ConvertNumToBase62(hashUrl)
|
|
|
+ // 拼上报告标题
|
|
|
+ //linkToken = fmt.Sprintf("%s %s", linkToken, req.Title)
|
|
|
+
|
|
|
+ tokenKey = fmt.Sprint(utils.CACHE_REPORT_SHARE_ORIGIN_Url, utils.MD5(linkToken))
|
|
|
+ ok = utils.Rc.IsExist(tokenKey)
|
|
|
+ if !ok {
|
|
|
+ break
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if !ok {
|
|
|
+ after := time.Now().AddDate(0, 0, 7)
|
|
|
+ err = utils.Rc.Put(cacheLinkKey, linkToken, time.Until(after))
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ err = utils.Rc.Put(tokenKey, req.Url, time.Until(after))
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ linkToken = ""
|
|
|
+ err = errors.New("生成链接失败")
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func TransfromToOriginUrl(linkToken string) (originLink string, msg string, err error) {
|
|
|
+ cacheLinkKey := fmt.Sprint(utils.CACHE_REPORT_SHARE_ORIGIN_Url, utils.MD5(linkToken))
|
|
|
+ originLink, err = utils.Rc.RedisString(cacheLinkKey)
|
|
|
+ if err != nil {
|
|
|
+ if err == redis.Nil {
|
|
|
+ msg = "链接已失效, 请重新获取"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ msg = "获取链接失败"
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if originLink == "" {
|
|
|
+ msg = "链接已失效, 请重新获取"
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ reportToken := GetReportAuthToken(linkToken)
|
|
|
+ if reportToken != "" {
|
|
|
+ originLink += `&authToken=` + reportToken
|
|
|
+ }
|
|
|
+
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func FilterShareUrl() web.FilterFunc {
|
|
|
+ return func(c *context.Context) {
|
|
|
+ path := c.Input.Context.Request.URL.Path
|
|
|
+ tokenArr := strings.Split(path, "/")
|
|
|
+ token := tokenArr[len(tokenArr)-1]
|
|
|
+
|
|
|
+ newPath := "/adminapi/report/share/link"
|
|
|
+ q := c.Input.Context.Request.URL.Query()
|
|
|
+ q.Add("Token", token)
|
|
|
+ c.Input.Context.Request.URL.Path = newPath
|
|
|
+ c.Input.Context.Request.URL.RawQuery = q.Encode()
|
|
|
+
|
|
|
+ utils.ApiLog.Info(fmt.Sprintf("原始请求为：%s, 已修改请求路径为：%s?%s", path, newPath, q.Encode()))
|
|
|
+ }
|
|
|
+}
|
