From 6d92974e0c1c976a023a4b2e240c2e1e990c8212 Mon Sep 17 00:00:00 2001 From: Darren Shepherd Date: Thu, 5 Sep 2024 20:42:18 -0700 Subject: [PATCH] bug: "share tools:" should support all tool types, not just tool --- pkg/types/tool.go | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/pkg/types/tool.go b/pkg/types/tool.go index d9d59837..f10788b4 100644 --- a/pkg/types/tool.go +++ b/pkg/types/tool.go @@ -602,13 +602,27 @@ func (t Tool) GetToolsByType(prg *Program, toolType ToolType) ([]ToolReference, case ToolTypeInput: exportRefs = tool.ExportInputFilters case ToolTypeTool: - exportRefs = tool.Export case ToolTypeCredential: exportRefs = tool.ExportCredentials default: return nil, fmt.Errorf("unknown tool type %v", toolType) } toolSet.AddAll(tool.GetToolRefsFromNames(exportRefs)) + + toolRefs, err := tool.GetToolRefsFromNames(tool.Export) + if err != nil { + return nil, err + } + + for _, toolRef := range toolRefs { + tool, ok := prg.ToolSet[toolRef.ToolID] + if !ok { + continue + } + if slices.Contains(toolsListFilterType, tool.Type) { + toolSet.Add(toolRef) + } + } } return toolSet.List()