From 37548b4e6b1570e31b462f23b8a9afa7eb653dbe Mon Sep 17 00:00:00 2001
From: Ethan <fsk741@gmail.com>
Date: Wed, 16 Nov 2022 17:45:26 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E9=98=BF=E9=87=8C?=
 =?UTF-8?q?=E4=BA=91=20trace=20url=20(#241)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: fusikai <fusikai@douyu.tv>
---
 api/apiv1/proxyintegrat/group.go              |  4 +-
 api/apiv1/proxyintegrat/trace_url.go          | 22 +++++++
 app/adminengine/router_admin.go               |  1 +
 .../components/ModalCreate/index.tsx          |  1 +
 assets/src/pages/proxy/index.tsx              |  2 +-
 assets/src/pages/proxy/proxyView/index.tsx    | 62 +++++++++++++------
 .../src/pages/proxy/services/proxyintegrat.ts |  5 ++
 config/resource.yaml                          |  3 +
 .../aliyunlog/complete_log_search_url.go      | 58 ++++++++++++++++-
 .../pkg/service/loggerplatform/default.go     |  6 +-
 pkg/model/view/loggerplatform.go              | 10 +--
 11 files changed, 143 insertions(+), 31 deletions(-)
 create mode 100644 api/apiv1/proxyintegrat/trace_url.go

diff --git a/api/apiv1/proxyintegrat/group.go b/api/apiv1/proxyintegrat/group.go
index 26ded5351..d84adaa16 100644
--- a/api/apiv1/proxyintegrat/group.go
+++ b/api/apiv1/proxyintegrat/group.go
@@ -2,9 +2,10 @@ package proxyintegrat
 
 import "github.com/labstack/echo/v4"
 
-//Group api group
+// Group api group
 func Group(g *echo.Group) {
 	g.GET("/uilist", UIList)
+	g.GET("/traceURL", TraceURL)
 	g.GET("/proxyMenuList", ProxyMenuList)
 	g.POST("/proxyMenuCreateOrUpdate", ProxyMenuCreateOrUpdate)
 	g.POST("/proxyMenuDelete", ProxyMenuDelete)
@@ -12,4 +13,5 @@ func Group(g *echo.Group) {
 	g.GET("/proxyManageList", ProxyManageList)
 	g.POST("/proxyManageCreateOrUpdate", ProxyManageCreateOrUpdate)
 	g.POST("/proxyManageDelete", ProxyManageDelete)
+
 }
diff --git a/api/apiv1/proxyintegrat/trace_url.go b/api/apiv1/proxyintegrat/trace_url.go
new file mode 100644
index 000000000..1b76d50e4
--- /dev/null
+++ b/api/apiv1/proxyintegrat/trace_url.go
@@ -0,0 +1,22 @@
+package proxyintegrat
+
+import (
+	"github.com/douyu/juno/internal/pkg/packages/contrib/output"
+	"github.com/douyu/juno/internal/pkg/service/aliyunlog"
+	"github.com/douyu/juno/internal/pkg/service/user"
+	"github.com/labstack/echo/v4"
+)
+
+// TraceUrl 阿里云trace地址获取
+func TraceURL(c echo.Context) (err error) {
+	u := user.GetUser(c)
+	toURL := c.QueryParam("toURL")
+	data, err := aliyunlog.Instance.CompleteTraceSearchUrl(c.Request().Context(), &aliyunlog.CompleteTraceSearchUrlRequest{
+		Region: "华北2（北京）",
+		ToURL:  toURL,
+	}, u)
+	if err != nil {
+		return output.JSON(c, output.MsgErr, err.Error())
+	}
+	return output.JSON(c, output.MsgOk, "", map[string]interface{}{"url": data})
+}
diff --git a/app/adminengine/router_admin.go b/app/adminengine/router_admin.go
index 929a6ae5d..74ea83ef9 100644
--- a/app/adminengine/router_admin.go
+++ b/app/adminengine/router_admin.go
@@ -413,4 +413,5 @@ func apiAdmin(server *xecho.Server) {
 	{
 		loggerGroup.GET("/logstore", core.Handle(loggerplatform.LogStore))
 	}
+
 }
diff --git a/assets/src/pages/analysis/proxy/proxymenu/components/ModalCreate/index.tsx b/assets/src/pages/analysis/proxy/proxymenu/components/ModalCreate/index.tsx
index 992823455..8ed631229 100644
--- a/assets/src/pages/analysis/proxy/proxymenu/components/ModalCreate/index.tsx
+++ b/assets/src/pages/analysis/proxy/proxymenu/components/ModalCreate/index.tsx
@@ -20,6 +20,7 @@ export interface ModalCreateResourceInterface {
 export const PanelTypes = [
   { title: 'Grafana', val: 'grafana' },
   { title: 'Pyroscope', val: 'pyroscope' },
+  { title: 'AliTrace', val: 'alitrace' },
 ];
 
 function ModalCreateResource(props: ModalCreateResourceInterface) {
diff --git a/assets/src/pages/proxy/index.tsx b/assets/src/pages/proxy/index.tsx
index 18e66ecc2..51874ec67 100644
--- a/assets/src/pages/proxy/index.tsx
+++ b/assets/src/pages/proxy/index.tsx
@@ -24,7 +24,7 @@ const renderPanel = (config) => {
     case 'welcome':
       return <WelcomeView {...config.config} />;
     default:
-      return <ProxyView {...config.config} />;
+      return <ProxyView {...config.config} panelType={config.panelType} />;
   }
 };
 
diff --git a/assets/src/pages/proxy/proxyView/index.tsx b/assets/src/pages/proxy/proxyView/index.tsx
index 757424ebb..431684e11 100644
--- a/assets/src/pages/proxy/proxyView/index.tsx
+++ b/assets/src/pages/proxy/proxyView/index.tsx
@@ -1,25 +1,51 @@
-import React from 'react';
+import React, { useState, useEffect } from 'react';
+import { traceURL } from "../services/proxyintegrat"
 
-export default class Proxy extends React.Component<any, any> {
-  render() {
-    const { proxyURL } = this.props;
-    return (
-      <div
+export default function (props) {
+  let [dynamicUrl, updateDynamicUrl] = useState('');
+  const { proxyURL, panelType } = props;
+  //支持动态嵌入页面
+  if (panelType == "alitrace") {
+    useEffect(() => {
+      traceURL({ toURL: proxyURL }).then((res) => {
+        if (res.code == 0) {
+          updateDynamicUrl(res.data && res.data.url);
+        }
+      })
+    }, [])
+    return dynamicUrl !== '' ? <div
+      style={{
+        width: '100%',
+        display: 'flex',
+        flex: 1,
+      }}
+    >
+      <iframe
+        src={dynamicUrl}
         style={{
+          border: 'none',
           width: '100%',
-          display: 'flex',
           flex: 1,
         }}
-      >
-        <iframe
-          src={proxyURL}
-          style={{
-            border: 'none',
-            width: '100%',
-            flex: 1,
-          }}
-        />
-      </div>
-    );
+      />
+    </div > : <div>请设置代理页面</div>
   }
+  return (
+    <div
+      style={{
+        width: '100%',
+        display: 'flex',
+        flex: 1,
+      }}
+    >
+      <iframe
+        src={proxyURL}
+        style={{
+          border: 'none',
+          width: '100%',
+          flex: 1,
+        }}
+      />
+    </div>
+  );
 }
diff --git a/assets/src/pages/proxy/services/proxyintegrat.ts b/assets/src/pages/proxy/services/proxyintegrat.ts
index f034ce517..2fecc4fb5 100644
--- a/assets/src/pages/proxy/services/proxyintegrat.ts
+++ b/assets/src/pages/proxy/services/proxyintegrat.ts
@@ -1,5 +1,10 @@
 import request from '@/utils/request';
+import { stringify } from 'qs';
 
 export async function uilist() {
   return request('/api/admin/proxyintegrat/uilist');
 }
+
+export async function traceURL(params) {
+  return request(`/api/admin/proxyintegrat/traceURL?${stringify(params)}`);
+}
diff --git a/config/resource.yaml b/config/resource.yaml
index 3d92c59ac..c68b0a0d3 100644
--- a/config/resource.yaml
+++ b/config/resource.yaml
@@ -135,6 +135,9 @@ permission:
           - path: /api/admin/proxyintegrat/uilist
             method: GET
             name: 功能菜单
+          - path: /api/admin/proxyintegrat/traceURL
+            method: GET
+            name: trace授权地址 
       - name: 依赖拓扑
         path: /analysis/topology
         api:
diff --git a/internal/pkg/service/aliyunlog/complete_log_search_url.go b/internal/pkg/service/aliyunlog/complete_log_search_url.go
index 4b706587a..26d8cd907 100644
--- a/internal/pkg/service/aliyunlog/complete_log_search_url.go
+++ b/internal/pkg/service/aliyunlog/complete_log_search_url.go
@@ -9,7 +9,9 @@ import (
 	"io/ioutil"
 	"net/http"
 	"net/url"
+	"strconv"
 
+	"github.com/aliyun/alibaba-cloud-sdk-go/sdk/requests"
 	"github.com/aliyun/alibaba-cloud-sdk-go/services/sts"
 	"github.com/douyu/juno/pkg/model/db"
 	"github.com/douyu/jupiter/pkg/conf"
@@ -23,13 +25,18 @@ type CompleteLogSearchUrlRequest struct {
 	StartTime string `query:"startTime"`
 	EndTime   string `query:"endTime"`
 }
+
+type CompleteTraceSearchUrlRequest struct {
+	Region string `query:"region"`
+	ToURL  string `query:"toURL"`
+}
 type TokenBody struct {
 	SigninToken string `json:"SigninToken"`
 }
 
 //参考:https://help.aliyun.com/document_detail/74971.html
 
-//CompleteLogSearchUrl 获取
+// CompleteLogSearchUrl 获取
 func (s *Service) CompleteLogSearchUrl(ctx context.Context, req *CompleteLogSearchUrlRequest, u *db.User) (signInUrl string, err error) {
 	client, ok := s.StsClients[req.Region]
 	if !ok {
@@ -43,6 +50,7 @@ func (s *Service) CompleteLogSearchUrl(ctx context.Context, req *CompleteLogSear
 	//设置参数。关于参数含义和设置方法，请参见《API参考》。
 	request.RoleArn = conf.GetString("aliyun.roleArn")
 	request.RoleSessionName = u.Username
+	request.DurationSeconds = requests.Integer(strconv.Itoa(3600 * 12))
 
 	//发起请求，并得到响应。
 	response, err := client.AssumeRole(request)
@@ -79,10 +87,54 @@ func (s *Service) CompleteLogSearchUrl(ctx context.Context, req *CompleteLogSear
 		query.Add("startTime", req.StartTime)
 		query.Add("endTime", req.EndTime)
 	}
-	// query.Add("readOnly", "true")
-	// query.Add("hiddenBack", "true")
+	query.Add("readOnly", "true")
+	query.Add("hiddenBack", "true")
 	toURL = toURL + "?" + query.Encode()
 	signInUrl = fmt.Sprintf("https://signin.aliyun.com/federation?Action=Login&LoginUrl=%s&Destination=%s&SigninToken=%s", conf.GetString("server.http.rootUrl"), url.QueryEscape(toURL), tokenItem.SigninToken)
 	return
 
 }
+
+//参考:https://help.aliyun.com/document_detail/123491.html
+
+// CompleteLogSearchUrl 获取
+func (s *Service) CompleteTraceSearchUrl(ctx context.Context, req *CompleteTraceSearchUrlRequest, u *db.User) (signInUrl string, err error) {
+	client, ok := s.StsClients[req.Region]
+	if !ok {
+		err = errors.New("no client")
+		return
+	}
+	//构建请求对象。
+	request := sts.CreateAssumeRoleRequest()
+	request.Scheme = "https"
+
+	//设置参数。关于参数含义和设置方法，请参见《API参考》。
+	request.RoleArn = conf.GetString("aliyun.roleArn")
+	request.RoleSessionName = u.Username
+	request.DurationSeconds = requests.Integer(strconv.Itoa(3600 * 12))
+
+	//发起请求，并得到响应。
+	response, err := client.AssumeRole(request)
+	if err != nil {
+		fmt.Print(err.Error())
+	}
+	tokenURl := fmt.Sprintf("https://signin.aliyun.com/federation?Action=GetSigninToken&AccessKeyId=%s&AccessKeySecret=%s&SecurityToken=%s&TicketType=mini",
+		url.QueryEscape(response.Credentials.AccessKeyId),
+		url.QueryEscape(response.Credentials.AccessKeySecret),
+		url.QueryEscape(response.Credentials.SecurityToken))
+	dataRes, err := http.Get(tokenURl)
+	if err != nil {
+		return
+	}
+	dataBytes, err := ioutil.ReadAll(dataRes.Body)
+	if err != nil {
+		return
+	}
+	tokenItem := new(TokenBody)
+	err = json.Unmarshal(dataBytes, tokenItem)
+	if err != nil {
+		return
+	}
+	signInUrl = fmt.Sprintf("https://signin.aliyun.com/federation?Action=Login&LoginUrl=%s&Destination=%s&SigninToken=%s", conf.GetString("server.http.rootUrl"), url.QueryEscape(req.ToURL), tokenItem.SigninToken)
+	return
+}
diff --git a/internal/pkg/service/loggerplatform/default.go b/internal/pkg/service/loggerplatform/default.go
index 4662e8119..c30841474 100644
--- a/internal/pkg/service/loggerplatform/default.go
+++ b/internal/pkg/service/loggerplatform/default.go
@@ -35,7 +35,7 @@ const (
 	LogTypBiz     = "biz"
 )
 
-//newAppLogD ..
+// newAppLogD ..
 func newAppLogD() {
 	var ald appLogDefault
 	ald.dashboardUrl = cfg.Cfg.AppLog.Default.DashboardUrl
@@ -60,7 +60,7 @@ func newAppLogD() {
 	return
 }
 
-//LogStore get log store
+// LogStore get log store
 func (a *appLogDefault) LogStore(env, query, typ, appName, aid string, u *db.User) (string, error) {
 	project, logStore := a.GetLogStoreName(env, typ, appName)
 	query = a.genQuery(env, typ, aid, query, appName)
@@ -81,7 +81,7 @@ func (a *appLogDefault) genQuery(env, typ, aid, query, appName string) string {
 	return query
 }
 
-//getLogStoreName ...
+// getLogStoreName ...
 func (a *appLogDefault) GetLogStoreName(env, typ, appName string) (string, string) {
 	var project AppLogDefaultEnvData
 	//业务日志进行拦截
diff --git a/pkg/model/view/loggerplatform.go b/pkg/model/view/loggerplatform.go
index f9f411f5f..b3d9215fe 100644
--- a/pkg/model/view/loggerplatform.go
+++ b/pkg/model/view/loggerplatform.go
@@ -9,10 +9,10 @@ type (
 	}
 	//ReqAliyunLogDefault env, query, typ, appNam
 	ReqAliyunLogDefault struct {
-		Env     string `json:"env" valid:"required"`
-		Query   string `json:"query" valid:"required"`
-		Typ     string `json:"typ" valid:"required"`
-		AppName string `json:"app_name" query:"app_name" valid:"required"`
-		Aid     string `json:"aid" query:"aid" valid:"required"`
+		Env     string `query:"env" json:"env" valid:"required"`
+		Query   string `query:"query" json:"query" valid:"required"`
+		Typ     string `query:"typ" json:"typ" valid:"required"`
+		AppName string `query:"app_name"  json:"app_name" valid:"required"`
+		Aid     string `query:"aid" json:"aid" valid:"required"`
 	}
 )