From 372554c86652921335518a2837182134d30f0295 Mon Sep 17 00:00:00 2001 From: Grant Linville Date: Mon, 12 Aug 2024 00:57:48 -0400 Subject: [PATCH] fix: support share credentials in context tools Signed-off-by: Grant Linville --- pkg/types/tool.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/pkg/types/tool.go b/pkg/types/tool.go index bb49e6f1..7e08f604 100644 --- a/pkg/types/tool.go +++ b/pkg/types/tool.go @@ -752,6 +752,16 @@ func (t Tool) GetCredentialTools(prg Program, agentGroup []ToolReference) ([]Too result.AddAll(referencedTool.GetToolRefsFromNames(referencedTool.ExportCredentials)) } + contextToolRefs, err := t.getDirectContextToolRefs(prg) + if err != nil { + return nil, err + } + + for _, contextToolRef := range contextToolRefs { + contextTool := prg.ToolSet[contextToolRef.ToolID] + result.AddAll(contextTool.GetToolRefsFromNames(contextTool.ExportCredentials)) + } + return result.List() }