CSEC_PUBLIC/catalyst
1
0
Fork 0
mirror of https://github.com/SecurityBrewery/catalyst.git synced 2025-12-07 07:42:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

Change code generator (#4)

Browse Source 
* Change code generator
* Remove gin
This commit is contained in:
Jonas Plum
2022-01-08 00:48:44 +01:00
committed by GitHub
parent b5dd0cfacd
commit 8333ea88a8
148 changed files with 3077 additions and 23976 deletions
Download Patch File Download Diff File Expand all files Collapse all files

210
service/ticket.go
View File

@@ -5,18 +5,25 @@ import (
"fmt"
"github.com/arangodb/go-driver"
"github.com/google/uuid"
"github.com/SecurityBrewery/catalyst/database"
"github.com/SecurityBrewery/catalyst/generated/models"
"github.com/SecurityBrewery/catalyst/generated/restapi/api"
"github.com/SecurityBrewery/catalyst/generated/restapi/operations/tickets"
"github.com/SecurityBrewery/catalyst/generated/model"
)
func ticketID(id int64) []driver.DocumentID {
return []driver.DocumentID{driver.DocumentID(fmt.Sprintf("%s/%d", database.TicketCollectionName, id))}
func ticketWithTicketsID(ticketResponse *model.TicketWithTickets) []driver.DocumentID {
if ticketResponse == nil {
return nil
}
return ticketID(ticketResponse.ID)
}
func ticketIDs(ticketResponses []*models.TicketResponse) []driver.DocumentID {
func ticketID(ticketID int64) []driver.DocumentID {
id := fmt.Sprintf("%s/%d", database.TicketCollectionName, ticketID)
return []driver.DocumentID{driver.DocumentID(id)}
}
func ticketIDs(ticketResponses []*model.TicketResponse) []driver.DocumentID {
var ids []driver.DocumentID
for _, ticketResponse := range ticketResponses {
ids = append(ids, ticketID(ticketResponse.ID)...)
@@ -24,48 +31,161 @@ func ticketIDs(ticketResponses []*models.TicketResponse) []driver.DocumentID {
return ids
}
func (s *Service) CreateTicket(ctx context.Context, params *tickets.CreateTicketParams) *api.Response {
createdTickets, err := s.database.TicketBatchCreate(ctx, []*models.TicketForm{params.Ticket})
if len(createdTickets) > 0 {
return s.response(ctx, "CreateTicket", ticketIDs(createdTickets), createdTickets[0], err)
}
return s.response(ctx, "CreateTicket", ticketIDs(createdTickets), nil, err)
}
func (s *Service) CreateTicketBatch(ctx context.Context, params *tickets.CreateTicketBatchParams) *api.Response {
ticketBatch, err := s.database.TicketBatchCreate(ctx, params.Ticket)
return s.response(ctx, "CreateTicketBatch", ticketIDs(ticketBatch), nil, err)
}
func (s *Service) GetTicket(ctx context.Context, params *tickets.GetTicketParams) *api.Response {
ticket, err := s.database.TicketGet(ctx, params.ID)
return s.response(ctx, "GetTicket", nil, ticket, err)
}
func (s *Service) UpdateTicket(ctx context.Context, params *tickets.UpdateTicketParams) *api.Response {
ticket, err := s.database.TicketUpdate(ctx, params.ID, params.Ticket)
return s.response(ctx, "UpdateTicket", ticketID(ticket.ID), ticket, err)
}
func (s *Service) DeleteTicket(ctx context.Context, params *tickets.DeleteTicketParams) *api.Response {
if err := s.database.TicketDelete(ctx, params.ID); err != nil {
return s.response(ctx, "DeleteTicket", ticketID(params.ID), nil, err)
}
_ = s.storage.DeleteBucket(fmt.Sprint(params.ID))
return s.response(ctx, "DeleteTicket", ticketID(params.ID), nil, nil)
}
func (s *Service) ListTickets(ctx context.Context, params *tickets.ListTicketsParams) *api.Response {
func (s *Service) ListTickets(ctx context.Context, s3 *string, i *int, i2 *int, strings []string, bools []bool, s2 *string) (*model.TicketList, error) {
q := ""
if params.Query != nil && *params.Query != "" {
q = *params.Query
if s2 != nil && *s2 != "" {
q = *s2
}
t := ""
if params.Type != nil && *params.Type != "" {
t = *params.Type
if s3 != nil && *s3 != "" {
t = *s3
}
ticketList, err := s.database.TicketList(ctx, t, q, params.Sort, params.Desc, *params.Offset, *params.Count)
return s.response(ctx, "ListTickets", nil, ticketList, err)
offset := int64(0)
if i != nil {
offset = int64(*i)
}
count := int64(25)
if i2 != nil {
count = int64(*i2)
}
return s.database.TicketList(ctx, t, q, strings, bools, offset, count)
}
func (s *Service) CreateTicket(ctx context.Context, form *model.TicketForm) (doc *model.TicketResponse, err error) {
createdTickets, err := s.database.TicketBatchCreate(ctx, []*model.TicketForm{form})
defer s.publishRequest(ctx, err, "CreateTicket", ticketIDs(createdTickets))
if len(createdTickets) > 0 {
return createdTickets[0], err
}
return nil, err
}
func (s *Service) CreateTicketBatch(ctx context.Context, forms []*model.TicketForm) error {
createdTickets, err := s.database.TicketBatchCreate(ctx, forms)
defer s.publishRequest(ctx, err, "CreateTicket", ticketIDs(createdTickets))
return err
}
func (s *Service) GetTicket(ctx context.Context, i int64) (*model.TicketWithTickets, error) {
return s.database.TicketGet(ctx, i)
}
func (s *Service) UpdateTicket(ctx context.Context, i int64, ticket *model.Ticket) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "UpdateTicket", ticketWithTicketsID(doc))
return s.database.TicketUpdate(ctx, i, ticket)
}
func (s *Service) DeleteTicket(ctx context.Context, i int64) (err error) {
defer s.publishRequest(ctx, err, "DeleteTicket", ticketID(i))
return s.database.TicketDelete(ctx, i)
}
func (s *Service) AddArtifact(ctx context.Context, i int64, artifact *model.Artifact) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "AddArtifact", ticketWithTicketsID(doc))
return s.database.AddArtifact(ctx, i, artifact)
}
func (s *Service) GetArtifact(ctx context.Context, i int64, s2 string) (*model.Artifact, error) {
return s.database.ArtifactGet(ctx, i, s2)
}
func (s *Service) SetArtifact(ctx context.Context, i int64, s2 string, artifact *model.Artifact) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "SetArtifact", ticketWithTicketsID(doc))
return s.database.ArtifactUpdate(ctx, i, s2, artifact)
}
func (s *Service) RemoveArtifact(ctx context.Context, i int64, s2 string) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "RemoveArtifact", ticketWithTicketsID(doc))
return s.database.RemoveArtifact(ctx, i, s2)
}
func (s *Service) EnrichArtifact(ctx context.Context, i int64, s2 string, form *model.EnrichmentForm) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "EnrichArtifact", ticketWithTicketsID(doc))
return s.database.EnrichArtifact(ctx, i, s2, form)
}
func (s *Service) RunArtifact(ctx context.Context, id int64, name string, automation string) error {
artifact, err := s.database.ArtifactGet(ctx, id, name)
if err != nil {
return err
}
defer s.publishRequest(ctx, err, "RunArtifact", ticketID(id))
jobID := uuid.NewString()
origin := &model.Origin{ArtifactOrigin: &model.ArtifactOrigin{TicketId: id, Artifact: name}}
return s.bus.PublishJob(jobID, automation, name, &model.Context{Artifact: artifact}, origin)
}
func (s *Service) AddComment(ctx context.Context, i int64, form *model.CommentForm) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "AddComment", ticketWithTicketsID(doc))
return s.database.AddComment(ctx, i, form)
}
func (s *Service) RemoveComment(ctx context.Context, i int64, i2 int) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "RemoveComment", ticketWithTicketsID(doc))
return s.database.RemoveComment(ctx, i, int64(i2))
}
func (s *Service) LinkFiles(ctx context.Context, i int64, files []*model.File) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "LinkFiles", ticketWithTicketsID(doc))
return s.database.LinkFiles(ctx, i, files)
}
func (s *Service) AddTicketPlaybook(ctx context.Context, i int64, form *model.PlaybookTemplateForm) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "AddTicketPlaybook", ticketWithTicketsID(doc))
return s.database.AddTicketPlaybook(ctx, i, form)
}
func (s *Service) RemoveTicketPlaybook(ctx context.Context, i int64, s2 string) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "RemoveTicketPlaybook", ticketWithTicketsID(doc))
return s.database.RemoveTicketPlaybook(ctx, i, s2)
}
func (s *Service) SetTask(ctx context.Context, i int64, s3 string, s2 string, task *model.Task) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "SetTask", ticketWithTicketsID(doc))
return s.database.TaskUpdate(ctx, i, s3, s2, task)
}
func (s *Service) CompleteTask(ctx context.Context, i int64, s3 string, s2 string, m map[string]interface{}) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "CompleteTask", ticketWithTicketsID(doc))
return s.database.TaskComplete(ctx, i, s3, s2, m)
}
func (s *Service) RunTask(ctx context.Context, i int64, s3 string, s2 string) (err error) {
defer s.publishRequest(ctx, err, "RunTask", ticketID(i))
return s.database.TaskRun(ctx, i, s3, s2)
}
func (s *Service) SetReferences(ctx context.Context, i int64, references []*model.Reference) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "SetReferences", ticketID(i))
return s.database.SetReferences(ctx, i, references)
}
func (s *Service) SetSchema(ctx context.Context, i int64, s2 string) (doc *model.TicketWithTickets, err error) {
defer s.publishRequest(ctx, err, "SetSchema", ticketID(i))
return s.database.SetTemplate(ctx, i, s2)
}
func (s *Service) LinkTicket(ctx context.Context, i int64, i2 int64) (*model.TicketWithTickets, error) {
err := s.database.RelatedCreate(ctx, i, i2)
if err != nil {
return nil, err
}
defer s.publishRequest(ctx, err, "LinkTicket", ticketID(i))
return s.GetTicket(ctx, i)
}
func (s *Service) UnlinkTicket(ctx context.Context, i int64, i2 int64) (*model.TicketWithTickets, error) {
err := s.database.RelatedRemove(ctx, i, i2)
if err != nil {
return nil, err
}
defer s.publishRequest(ctx, err, "UnlinkTicket", ticketID(i))
return s.GetTicket(ctx, i)
}