From 7fa0cff1f758d6d25a8c1430f43dc800dd538ed6 Mon Sep 17 00:00:00 2001
From: taolx0 <taolx25@foxmail.com>
Date: Thu, 24 Oct 2024 18:03:13 +0800
Subject: [PATCH 1/2] chore:add app extract link tips

---
 sqle/api/controller/v1/instance_audit_plan.go | 40 ++++++++++++-------
 sqle/server/auditplan/meta.go                 |  5 ++-
 2 files changed, 28 insertions(+), 17 deletions(-)

diff --git a/sqle/api/controller/v1/instance_audit_plan.go b/sqle/api/controller/v1/instance_audit_plan.go
index 72ab7d371..556e388d3 100644
--- a/sqle/api/controller/v1/instance_audit_plan.go
+++ b/sqle/api/controller/v1/instance_audit_plan.go
@@ -834,22 +834,32 @@ func GetAuditPlanExecCmd(projectName string, iap *model.InstanceAuditPlan, ap *m
 		return ""
 	}
 
-	scannerd, err := scannerCmd.GetScannerdCmd(ap.Type)
-	if err != nil {
-		logger.Infof("get scannerd %s failed %s", ap.Type, err)
-		return ""
-	}
-	cmd, err := scannerd.GenCommand("./scannerd", map[string]string{
-		scannerCmd.FlagHost:        ip,
-		scannerCmd.FlagPort:        port,
-		scannerCmd.FlagProject:     projectName,
-		scannerCmd.FlagToken:       iap.Token,
-		scannerCmd.FlagAuditPlanID: fmt.Sprint(ap.ID),
-	})
-	if err != nil {
-		logger.Infof("generate scannerd %s command failed %s", ap.Type, err)
-		return ""
+	var cmd string
+	switch ap.Type {
+	case auditplan.TypeAllAppExtract:
+		cmd = fmt.Sprintf(`SQLE_PROJECT_NAME=%s \
+PROJECT_APP_NAME=<Please provide the business parameters here> \
+java -<Please provide the java agent name parameters here> \
+-jar <Please provide the java app name parameters here>`, projectName)
+	default:
+		scannerd, err := scannerCmd.GetScannerdCmd(ap.Type)
+		if err != nil {
+			logger.Infof("get scannerd %s failed %s", ap.Type, err)
+			return ""
+		}
+		cmd, err = scannerd.GenCommand("./scannerd", map[string]string{
+			scannerCmd.FlagHost:        ip,
+			scannerCmd.FlagPort:        port,
+			scannerCmd.FlagProject:     projectName,
+			scannerCmd.FlagToken:       iap.Token,
+			scannerCmd.FlagAuditPlanID: fmt.Sprint(ap.ID),
+		})
+		if err != nil {
+			logger.Infof("generate scannerd %s command failed %s", ap.Type, err)
+			return ""
+		}
 	}
+
 	return cmd
 }
 
diff --git a/sqle/server/auditplan/meta.go b/sqle/server/auditplan/meta.go
index 2a823b1fa..f03c88c8c 100644
--- a/sqle/server/auditplan/meta.go
+++ b/sqle/server/auditplan/meta.go
@@ -181,8 +181,9 @@ func GetMeta(typ string) (Meta, error) {
 }
 
 var supportedCmdTypeList = map[string]struct{}{
-	TypeMySQLSlowLog: {},
-	TypeTiDBAuditLog: {},
+	TypeMySQLSlowLog:  {},
+	TypeTiDBAuditLog:  {},
+	TypeAllAppExtract: {},
 }
 
 func GetSupportedScannerAuditPlanType() map[string]struct{} {

From 40a9f704680526dfff2016810db1f9267f02f0fd Mon Sep 17 00:00:00 2001
From: taolx0 <taolx25@foxmail.com>
Date: Fri, 25 Oct 2024 15:14:39 +0800
Subject: [PATCH 2/2] chore:add token response

---
 sqle/api/controller/v1/instance_audit_plan.go | 6 ++++--
 sqle/model/instance_audit_plan_list.go        | 1 +
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/sqle/api/controller/v1/instance_audit_plan.go b/sqle/api/controller/v1/instance_audit_plan.go
index 556e388d3..d539cb862 100644
--- a/sqle/api/controller/v1/instance_audit_plan.go
+++ b/sqle/api/controller/v1/instance_audit_plan.go
@@ -418,6 +418,7 @@ type AuditPlanTypeResBase struct {
 	AuditPlanId       uint   `json:"audit_plan_id"`
 	AuditPlanType     string `json:"type"`
 	AuditPlanTypeDesc string `json:"desc"`
+	Token             string `json:"token"`
 }
 
 type GetInstanceAuditPlansReqV1 struct {
@@ -519,7 +520,7 @@ func GetInstanceAuditPlans(c echo.Context) error {
 		typeBases := make([]AuditPlanTypeResBase, 0, len(auditPlanIds))
 		for _, auditPlanId := range auditPlanIds {
 			if auditPlanId != "" {
-				typeBase, err := ConvertAuditPlanTypeToResByID(c.Request().Context(), auditPlanId)
+				typeBase, err := ConvertAuditPlanTypeToResByID(c.Request().Context(), auditPlanId, v.Token)
 				if err != nil {
 					return controller.JSONBaseErrorReq(c, err)
 				}
@@ -547,7 +548,7 @@ func GetInstanceAuditPlans(c echo.Context) error {
 	})
 }
 
-func ConvertAuditPlanTypeToResByID(ctx context.Context, id string) (AuditPlanTypeResBase, error) {
+func ConvertAuditPlanTypeToResByID(ctx context.Context, id string, token string) (AuditPlanTypeResBase, error) {
 	auditPlanID, err := strconv.Atoi(id)
 	if err != nil {
 		return AuditPlanTypeResBase{}, err
@@ -566,6 +567,7 @@ func ConvertAuditPlanTypeToResByID(ctx context.Context, id string) (AuditPlanTyp
 				AuditPlanType:     auditPlan.Type,
 				AuditPlanTypeDesc: locale.Bundle.LocalizeMsgByCtx(ctx, meta.Desc),
 				AuditPlanId:       auditPlan.ID,
+				Token:             token,
 			}, nil
 		}
 	}
diff --git a/sqle/model/instance_audit_plan_list.go b/sqle/model/instance_audit_plan_list.go
index 69917150d..f9e7f90f3 100644
--- a/sqle/model/instance_audit_plan_list.go
+++ b/sqle/model/instance_audit_plan_list.go
@@ -28,6 +28,7 @@ SELECT
     instance_audit_plans.active_status,
     instance_audit_plans.create_user_id,
     instance_audit_plans.created_at,
+	instance_audit_plans.token,
     audit_plans.audit_plan_ids