mirror of
https://github.com/SecurityBrewery/catalyst.git
synced 2025-12-06 15:22:47 +01:00
39 lines
1.5 KiB
Go
39 lines
1.5 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/SecurityBrewery/catalyst/generated/models"
|
|
"github.com/SecurityBrewery/catalyst/generated/restapi/api"
|
|
"github.com/SecurityBrewery/catalyst/generated/restapi/operations/tickets"
|
|
)
|
|
|
|
func (s *Service) RunArtifact(ctx context.Context, params *tickets.RunArtifactParams) (r *api.Response) {
|
|
artifact, err := s.database.ArtifactGet(ctx, params.ID, params.Name)
|
|
if err != nil {
|
|
return s.response(ctx, "RunArtifact", ticketID(params.ID), nil, err)
|
|
}
|
|
|
|
jobID := uuid.NewString()
|
|
origin := &models.Origin{ArtifactOrigin: &models.ArtifactOrigin{TicketId: params.ID, Artifact: params.Name}}
|
|
err = s.bus.PublishJob(jobID, params.Automation, params.Name, &models.Context{Artifact: artifact}, origin)
|
|
return s.response(ctx, "RunArtifact", ticketID(params.ID), nil, err)
|
|
}
|
|
|
|
func (s *Service) EnrichArtifact(ctx context.Context, params *tickets.EnrichArtifactParams) *api.Response {
|
|
i, err := s.database.EnrichArtifact(ctx, params.ID, params.Name, params.Data)
|
|
return s.response(ctx, "EnrichArtifact", ticketID(params.ID), i, err)
|
|
}
|
|
|
|
func (s *Service) SetArtifact(ctx context.Context, params *tickets.SetArtifactParams) *api.Response {
|
|
i, err := s.database.ArtifactUpdate(ctx, params.ID, params.Name, params.Artifact)
|
|
return s.response(ctx, "SetArtifact", ticketID(params.ID), i, err)
|
|
}
|
|
|
|
func (s *Service) GetArtifact(ctx context.Context, params *tickets.GetArtifactParams) *api.Response {
|
|
i, err := s.database.ArtifactGet(ctx, params.ID, params.Name)
|
|
return s.response(ctx, "GetArtifact", nil, i, err)
|
|
}
|