fix: add scansource to workspace scan command (#831)

Author: ShawkyZ

application/server/execute_command_test.go

@@ -122,6 +122,25 @@ func Test_executeWorkspaceScanCommand_shouldAskForTrust(t *testing.T) {
 	}, 2*time.Second, time.Millisecond)
 }
 
+func Test_executeWorkspaceScanCommand_shouldAcceptScanSourceParam(t *testing.T) {
+	c := testutil.UnitTest(t)
+	loc, jsonRPCRecorder := setupServerWithCustomDI(t, c, false)
+
+	s := &scanner.TestScanner{}
+	c.Workspace().AddFolder(workspace.NewFolder(c, "dummy", "dummy", s, di.HoverService(), di.ScanNotifier(), di.Notifier(), di.ScanPersister(), di.ScanStateAggregator()))
+	// explicitly enable folder trust which is disabled by default in tests
+	config.CurrentConfig().SetTrustedFolderFeatureEnabled(true)
+
+	params := lsp.ExecuteCommandParams{Command: types.WorkspaceScanCommand, Arguments: []any{"LLM"}}
+	_, err := loc.Client.Call(ctx, "workspace/executeCommand", params)
+	if err != nil {
+		t.Fatal(err)
+	}
+	assert.Eventually(t, func() bool {
+		return s.Calls() == 0 && checkTrustMessageRequest(jsonRPCRecorder, c)
+	}, 2*time.Second, time.Millisecond)
+}
+
 func Test_loginCommand_StartsAuthentication(t *testing.T) {
 	c := testutil.UnitTest(t)
 	loc, jsonRPCRecorder := setupServer(t, c)

domain/ide/command/workspace_scan.go

@@ -20,6 +20,7 @@ import (
 	"context"
 
 	"github.com/snyk/snyk-ls/application/config"
+	context2 "github.com/snyk/snyk-ls/internal/context"
 	"github.com/snyk/snyk-ls/internal/types"
 )
 
@@ -36,7 +37,27 @@ func (cmd *workspaceScanCommand) Command() types.CommandData {
 func (cmd *workspaceScanCommand) Execute(ctx context.Context) (any, error) {
 	w := cmd.c.Workspace()
 	w.Clear()
-	w.ScanWorkspace(ctx)
-	HandleUntrustedFolders(ctx, cmd.c, cmd.srv)
+	args := cmd.command.Arguments
+	enrichedCtx := cmd.enrichContextWithScanSource(ctx, args)
+	w.ScanWorkspace(enrichedCtx)
+	HandleUntrustedFolders(enrichedCtx, cmd.c, cmd.srv)
 	return nil, nil
 }
+
+func (cmd *workspaceScanCommand) enrichContextWithScanSource(ctx context.Context, args []any) context.Context {
+	if len(args) == 0 {
+		return ctx
+	}
+
+	sc, ok := args[0].(string)
+	if !ok {
+		return ctx
+	}
+
+	if sc != context2.IDE.String() && sc != context2.LLM.String() {
+		return ctx
+	}
+
+	scanSource := context2.ScanSource(sc)
+	return context2.NewContextWithScanSource(ctx, scanSource)
+}