rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix famework_id filer not working

Browse Source 
Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2026-04-09 12:35:07 +02:00
parent dee3f02c41
commit b005b92c77
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/server/api/mcp/v1/schema.resolvers.go
View File

@@ -1445,12 +1445,19 @@ func (r *Resolver) ListControlsTool(ctx context.Context, req *mcp.CallToolReques
controlFilter = coredata.NewControlFilter(input.Filter.Query) controlFilter = coredata.NewControlFilter(input.Filter.Query)
} }
page, err := prb.Controls.ListForOrganizationID(ctx, input.OrganizationID, cursor, controlFilter) var controlPage *page.Page[*coredata.Control, coredata.ControlOrderField]
var err error
if input.Filter != nil && input.Filter.FrameworkID != nil {
controlPage, err = prb.Controls.ListForFrameworkID(ctx, *input.Filter.FrameworkID, cursor, controlFilter)
} else {
controlPage, err = prb.Controls.ListForOrganizationID(ctx, input.OrganizationID, cursor, controlFilter)
}
if err != nil { if err != nil {
panic(fmt.Errorf("cannot list organization controls: %w", err)) panic(fmt.Errorf("cannot list organization controls: %w", err))
} }
return nil, types.NewListControlsOutput(page), nil return nil, types.NewListControlsOutput(controlPage), nil
} }
func (r *Resolver) GetControlTool(ctx context.Context, req *mcp.CallToolRequest, input *types.GetControlInput) (*mcp.CallToolResult, types.GetControlOutput, error) { func (r *Resolver) GetControlTool(ctx context.Context, req *mcp.CallToolRequest, input *types.GetControlInput) (*mcp.CallToolResult, types.GetControlOutput, error) {