Add log export for audit logs and SCIM events

Route audit-log and SCIM-event exports through export_jobs with typed
arguments, an iam BuildAndUploadExport/SendExportEmail implementation,
and a concurrent export-job worker with stale recovery. Stream JSONL via
page.WalkAll into S3, and expose the request flow on console, connect,
MCP, and CLI.

Co-authored-by: Bryan Frimin <bryan@getprobo.com>
Signed-off-by: Sacha Al Himdani <sacha@probo.com>
This commit is contained in:
Sacha Al Himdani
2026-07-29 16:12:23 +02:00
parent b2e2d15582
commit cd6c46212a
45 changed files with 2340 additions and 153 deletions

View File

@@ -7351,3 +7351,69 @@ func (r *Resolver) DeleteCommitmentTool(ctx context.Context, req *mcp.CallToolRe
return nil, types.DeleteCommitmentOutput{DeletedCommitmentID: input.ID}, nil
}
func (r *Resolver) RequestAuditLogExportTool(ctx context.Context, req *mcp.CallToolRequest, input *types.RequestAuditLogExportInput) (*mcp.CallToolResult, types.RequestAuditLogExportOutput, error) {
scope, err := r.Authorize(ctx, input.OrganizationID, iam.ActionAuditLogExport)
if err != nil {
return nil, types.RequestAuditLogExportOutput{}, err
}
identity := authn.IdentityFromContext(ctx)
logExport, err := r.iamSvc.OrganizationService.RequestLogExport(
ctx,
scope,
iam.RequestLogExportRequest{
OrganizationID: input.OrganizationID,
Type: coredata.ExportJobTypeAuditLog,
FromTime: input.FromTime,
ToTime: input.ToTime,
RecipientEmail: identity.EmailAddress,
RecipientName: identity.FullName,
},
)
if err != nil {
if _, ok := errors.AsType[*iam.ErrInvalidLogExportTimeRange](err); ok {
return nil, types.RequestAuditLogExportOutput{}, err
}
return nil, types.RequestAuditLogExportOutput{}, fmt.Errorf("cannot request audit log export: %w", err)
}
return nil, types.RequestAuditLogExportOutput{
ExportJobID: logExport.ID,
}, nil
}
func (r *Resolver) RequestSCIMEventExportTool(ctx context.Context, req *mcp.CallToolRequest, input *types.RequestSCIMEventExportInput) (*mcp.CallToolResult, types.RequestSCIMEventExportOutput, error) {
scope, err := r.Authorize(ctx, input.OrganizationID, iam.ActionSCIMEventExport)
if err != nil {
return nil, types.RequestSCIMEventExportOutput{}, err
}
identity := authn.IdentityFromContext(ctx)
logExport, err := r.iamSvc.OrganizationService.RequestLogExport(
ctx,
scope,
iam.RequestLogExportRequest{
OrganizationID: input.OrganizationID,
Type: coredata.ExportJobTypeSCIMEvent,
FromTime: input.FromTime,
ToTime: input.ToTime,
RecipientEmail: identity.EmailAddress,
RecipientName: identity.FullName,
},
)
if err != nil {
if _, ok := errors.AsType[*iam.ErrInvalidLogExportTimeRange](err); ok {
return nil, types.RequestSCIMEventExportOutput{}, err
}
return nil, types.RequestSCIMEventExportOutput{}, fmt.Errorf("cannot request SCIM event export: %w", err)
}
return nil, types.RequestSCIMEventExportOutput{
ExportJobID: logExport.ID,
}, nil
}

View File

@@ -8487,6 +8487,66 @@ components:
items:
$ref: "#/components/schemas/AuditLogEntry"
RequestAuditLogExportInput:
type: object
required:
- organization_id
- from_time
- to_time
properties:
organization_id:
$ref: "#/components/schemas/GID"
description: Organization ID
from_time:
type: string
format: date-time
go.probo.inc/mcpgen/type: time.Time
description: Start of the time range (inclusive). The range must not exceed 1 year.
to_time:
type: string
format: date-time
go.probo.inc/mcpgen/type: time.Time
description: End of the time range (exclusive). The range must not exceed 1 year.
RequestAuditLogExportOutput:
type: object
required:
- export_job_id
properties:
export_job_id:
$ref: "#/components/schemas/GID"
description: ID of the created log export
RequestSCIMEventExportInput:
type: object
required:
- organization_id
- from_time
- to_time
properties:
organization_id:
$ref: "#/components/schemas/GID"
description: Organization ID
from_time:
type: string
format: date-time
go.probo.inc/mcpgen/type: time.Time
description: Start of the time range (inclusive). The range must not exceed 1 year.
to_time:
type: string
format: date-time
go.probo.inc/mcpgen/type: time.Time
description: End of the time range (exclusive). The range must not exceed 1 year.
RequestSCIMEventExportOutput:
type: object
required:
- export_job_id
properties:
export_job_id:
$ref: "#/components/schemas/GID"
description: ID of the created log export
AuditLogEntry:
type: object
required:
@@ -13955,6 +14015,24 @@ tools:
$ref: "#/components/schemas/ListAuditLogEntriesInput"
outputSchema:
$ref: "#/components/schemas/ListAuditLogEntriesOutput"
- name: requestAuditLogExport
description: Request an export of audit log entries for the organization within a time range. The export will be emailed as a JSONL download link.
hints:
readonly: false
idempotent: false
inputSchema:
$ref: "#/components/schemas/RequestAuditLogExportInput"
outputSchema:
$ref: "#/components/schemas/RequestAuditLogExportOutput"
- name: requestSCIMEventExport
description: Request an export of SCIM events for the organization within a time range. The export will be emailed as a JSONL download link.
hints:
readonly: false
idempotent: false
inputSchema:
$ref: "#/components/schemas/RequestSCIMEventExportInput"
outputSchema:
$ref: "#/components/schemas/RequestSCIMEventExportOutput"
- name: listWebhookSubscriptions
description: List all webhook subscriptions for the organization
hints: