Files
catalyst/service/playbook.go
Jonas Plum 8333ea88a8 Change code generator (#4)
* Change code generator
* Remove gin
2022-01-08 00:48:44 +01:00

46 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)
}