Files
catalyst/service/playbook.go
Jonas Plum 2bad1f5f28 Migrate to Go 1.18 (#45)
* Migrate to Go 1.18 and add linters
2022-03-20 03:17:18 +01:00

50 lines
1.5 KiB
Go

package service
import (
"context"
"fmt"
"github.com/arangodb/go-driver"
"github.com/SecurityBrewery/catalyst/database"
"github.com/SecurityBrewery/catalyst/generated/model"
)
func playbookResponseID(playbook *model.PlaybookTemplateResponse) []driver.DocumentID {
if playbook == nil {
return nil
}
return playbookID(playbook.ID)
}
func playbookID(id string) []driver.DocumentID {
return []driver.DocumentID{driver.DocumentID(fmt.Sprintf("%s/%s", database.PlaybookCollectionName, id))}
}
func (s *Service) ListPlaybooks(ctx context.Context) ([]*model.PlaybookTemplateResponse, error) {
return s.database.PlaybookList(ctx)
}
func (s *Service) CreatePlaybook(ctx context.Context, form *model.PlaybookTemplateForm) (doc *model.PlaybookTemplateResponse, err error) {
defer s.publishRequest(ctx, err, "CreatePlaybook", playbookResponseID(doc))
return s.database.PlaybookCreate(ctx, form)
}
func (s *Service) GetPlaybook(ctx context.Context, id string) (*model.PlaybookTemplateResponse, error) {
return s.database.PlaybookGet(ctx, id)
}
func (s *Service) UpdatePlaybook(ctx context.Context, id string, form *model.PlaybookTemplateForm) (doc *model.PlaybookTemplateResponse, err error) {
defer s.publishRequest(ctx, err, "UpdatePlaybook", playbookResponseID(doc))
return s.database.PlaybookUpdate(ctx, id, form)
}
func (s *Service) DeletePlaybook(ctx context.Context, id string) (err error) {
defer s.publishRequest(ctx, err, "DeletePlaybook", playbookID(id))
return s.database.PlaybookDelete(ctx, id)
}