Release catalyst

service/ticket_field.go

@@ -0,0 +1,74 @@
+package service
+
+import (
+	"context"
+
+	"github.com/SecurityBrewery/catalyst/generated/restapi/api"
+	"github.com/SecurityBrewery/catalyst/generated/restapi/operations/tickets"
+)
+
+func (s *Service) AddArtifact(ctx context.Context, params *tickets.AddArtifactParams) *api.Response {
+	return response(s.database.AddArtifact(ctx, params.ID, params.Artifact))
+}
+
+func (s *Service) RemoveArtifact(ctx context.Context, params *tickets.RemoveArtifactParams) *api.Response {
+	return response(s.database.RemoveArtifact(ctx, params.ID, params.Name))
+}
+
+func (s *Service) SetSchema(ctx context.Context, params *tickets.SetSchemaParams) *api.Response {
+	return response(s.database.SetTemplate(ctx, params.ID, params.Schema))
+}
+
+func (s *Service) AddComment(ctx context.Context, params *tickets.AddCommentParams) *api.Response {
+	return response(s.database.AddComment(ctx, params.ID, params.Comment))
+}
+
+func (s *Service) RemoveComment(ctx context.Context, params *tickets.RemoveCommentParams) *api.Response {
+	return response(s.database.RemoveComment(ctx, params.ID, params.CommentID))
+}
+
+func (s *Service) LinkTicket(ctx context.Context, params *tickets.LinkTicketParams) *api.Response {
+	err := s.database.RelatedCreate(ctx, params.ID, params.LinkedID)
+	if err != nil {
+		return response(nil, err)
+	}
+
+	return s.GetTicket(ctx, &tickets.GetTicketParams{ID: params.ID})
+}
+
+func (s *Service) UnlinkTicket(ctx context.Context, params *tickets.UnlinkTicketParams) *api.Response {
+	err := s.database.RelatedRemove(ctx, params.ID, params.LinkedID)
+	if err != nil {
+		return response(nil, err)
+	}
+
+	return s.GetTicket(ctx, &tickets.GetTicketParams{ID: params.ID})
+}
+
+func (s Service) SetReferences(ctx context.Context, params *tickets.SetReferencesParams) *api.Response {
+	return response(s.database.SetReferences(ctx, params.ID, params.References))
+}
+
+func (s Service) LinkFiles(ctx context.Context, params *tickets.LinkFilesParams) *api.Response {
+	return response(s.database.LinkFiles(ctx, params.ID, params.Files))
+}
+
+func (s Service) AddTicketPlaybook(ctx context.Context, params *tickets.AddTicketPlaybookParams) *api.Response {
+	return response(s.database.AddTicketPlaybook(ctx, params.ID, params.Playbook))
+}
+
+func (s Service) RemoveTicketPlaybook(ctx context.Context, params *tickets.RemoveTicketPlaybookParams) *api.Response {
+	return response(s.database.RemoveTicketPlaybook(ctx, params.ID, params.PlaybookID))
+}
+
+func (s Service) CompleteTask(ctx context.Context, params *tickets.CompleteTaskParams) *api.Response {
+	return response(s.database.TaskComplete(ctx, params.ID, params.PlaybookID, params.TaskID, params.Data))
+}
+
+func (s Service) SetTask(ctx context.Context, params *tickets.SetTaskParams) *api.Response {
+	return response(s.database.TaskUpdate(ctx, params.ID, params.PlaybookID, params.TaskID, params.Task))
+}
+
+func (s *Service) RunTask(ctx context.Context, params *tickets.RunTaskParams) *api.Response {
+	return response(nil, s.database.TaskRun(ctx, params.ID, params.PlaybookID, params.TaskID))
+}