feat(rem): auto-resolve pipeline from agent-id in rem update interval

Allow `rem update interval --agent-id <id> <ms>` without requiring
--pipeline-id. The command queries PivotingContexts to find which
pipeline the agent belongs to. If the agent exists on multiple
pipelines, the user is prompted to disambiguate with --pipeline-id.

Author: h3zh1

client/command/pipeline/commands.go

@@ -223,13 +223,14 @@ rem delete rem_test
 		},
 		Example: `~~~
 rem update interval --session-id 08d6c05a 5000
+rem update interval --agent-id uDM0BgG6 5000
 rem update interval --pipeline-id rem_graph_api_03 --agent-id uDM0BgG6 5000
 ~~~`,
 	}
 	common.BindFlag(updateIntervalCmd, func(f *pflag.FlagSet) {
 		f.String("session-id", "", "Session ID to reconfigure (resolves pipeline and agent automatically)")
-		f.String("pipeline-id", "", "Pipeline name (used with --agent-id)")
-		f.String("agent-id", "", "REM agent ID (used with --pipeline-id)")
+		f.String("pipeline-id", "", "Pipeline name (required only when agent exists on multiple pipelines)")
+		f.String("agent-id", "", "REM agent ID (pipeline is auto-resolved if unique)")
 	})
 	common.BindFlagCompletions(updateIntervalCmd, func(comp carapace.ActionMap) {
 		comp["pipeline-id"] = common.RemPipelineCompleter(con)

client/command/pipeline/rem.go

@@ -10,6 +10,7 @@ import (
 	"github.com/chainreactors/malice-network/client/core"
 	"github.com/chainreactors/malice-network/helper/cryptography"
 	"github.com/chainreactors/malice-network/helper/third/rem"
+	"github.com/chainreactors/malice-network/helper/utils/output"
 	"github.com/chainreactors/tui"
 	"github.com/spf13/cobra"
 )
@@ -153,8 +154,40 @@ func RemUpdateIntervalCmd(cmd *cobra.Command, con *core.Console) error {
 		}
 	}
 
+	// Resolve pipeline from agent-id by querying PivotingContexts
+	if agentID != "" && pipelineID == "" {
+		ctxs, err := con.Rpc.GetContexts(con.Context(), &clientpb.Context{
+			Type: consts.ContextPivoting,
+		})
+		if err != nil {
+			return fmt.Errorf("failed to query pivot contexts: %w", err)
+		}
+		pivots, err := output.ToContexts[*output.PivotingContext](ctxs.Contexts)
+		if err != nil {
+			return fmt.Errorf("failed to parse pivot contexts: %w", err)
+		}
+		var matched []string
+		seen := make(map[string]struct{})
+		for _, p := range pivots {
+			if p.RemAgentID == agentID && p.Enable {
+				if _, dup := seen[p.Pipeline]; !dup {
+					seen[p.Pipeline] = struct{}{}
+					matched = append(matched, p.Pipeline)
+				}
+			}
+		}
+		switch len(matched) {
+		case 0:
+			return fmt.Errorf("agent %s not found in any active pipeline", agentID)
+		case 1:
+			pipelineID = matched[0]
+		default:
+			return fmt.Errorf("agent %s found in multiple pipelines %v, please specify --pipeline-id", agentID, matched)
+		}
+	}
+
 	if pipelineID == "" || agentID == "" {
-		return fmt.Errorf("either --session-id or both --pipeline-id and --agent-id are required")
+		return fmt.Errorf("either --session-id, --agent-id, or both --pipeline-id and --agent-id are required")
 	}
 
 	_, err = con.Rpc.RemAgentCtrl(con.Context(), &clientpb.REMAgent{