137 lines
4.0 KiB
Go
137 lines
4.0 KiB
Go
package rest
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/cortezaproject/corteza-server/pkg/api"
|
|
"github.com/cortezaproject/corteza-server/pkg/filter"
|
|
"github.com/cortezaproject/corteza-server/pkg/payload"
|
|
"github.com/cortezaproject/corteza-server/system/rest/request"
|
|
"github.com/cortezaproject/corteza-server/system/service"
|
|
"github.com/cortezaproject/corteza-server/system/types"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
var _ = errors.Wrap
|
|
|
|
type (
|
|
SensitivityLevel struct {
|
|
svc sensitivityLevelService
|
|
}
|
|
|
|
sensitivityLevelSetPayload struct {
|
|
Filter types.DalSensitivityLevelFilter `json:"filter"`
|
|
Set []*sensitivityLevelPayload `json:"set"`
|
|
}
|
|
|
|
sensitivityLevelPayload struct {
|
|
*types.DalSensitivityLevel
|
|
}
|
|
|
|
sensitivityLevelService interface {
|
|
FindByID(ctx context.Context, ID uint64) (*types.DalSensitivityLevel, error)
|
|
Create(ctx context.Context, new *types.DalSensitivityLevel) (*types.DalSensitivityLevel, error)
|
|
Update(ctx context.Context, upd *types.DalSensitivityLevel) (*types.DalSensitivityLevel, error)
|
|
DeleteByID(ctx context.Context, ID uint64) error
|
|
UndeleteByID(ctx context.Context, ID uint64) error
|
|
Search(ctx context.Context, filter types.DalSensitivityLevelFilter) (types.DalSensitivityLevelSet, types.DalSensitivityLevelFilter, error)
|
|
}
|
|
)
|
|
|
|
func (SensitivityLevel) New() *SensitivityLevel {
|
|
return &SensitivityLevel{
|
|
svc: service.DefaultDalSensitivityLevel,
|
|
}
|
|
}
|
|
|
|
func (ctrl SensitivityLevel) List(ctx context.Context, r *request.DalSensitivityLevelList) (interface{}, error) {
|
|
var (
|
|
err error
|
|
set types.DalSensitivityLevelSet
|
|
|
|
f = types.DalSensitivityLevelFilter{
|
|
SensitivityLevelID: payload.ParseUint64s(r.SensitivityLevelID),
|
|
|
|
Deleted: filter.State(r.Deleted),
|
|
}
|
|
)
|
|
|
|
if f.Deleted == 0 {
|
|
f.Deleted = filter.StateExcluded
|
|
}
|
|
|
|
set, f, err = ctrl.svc.Search(ctx, f)
|
|
return ctrl.makeFilterPayload(ctx, set, f, err)
|
|
}
|
|
|
|
func (ctrl SensitivityLevel) Create(ctx context.Context, r *request.DalSensitivityLevelCreate) (interface{}, error) {
|
|
sensitivityLevel := &types.DalSensitivityLevel{
|
|
Handle: r.Handle,
|
|
Level: r.Level,
|
|
Meta: r.Meta,
|
|
}
|
|
|
|
return ctrl.svc.Create(ctx, sensitivityLevel)
|
|
}
|
|
|
|
func (ctrl SensitivityLevel) Update(ctx context.Context, r *request.DalSensitivityLevelUpdate) (interface{}, error) {
|
|
sensitivityLevel := &types.DalSensitivityLevel{
|
|
ID: r.SensitivityLevelID,
|
|
Handle: r.Handle,
|
|
Level: r.Level,
|
|
Meta: r.Meta,
|
|
}
|
|
|
|
return ctrl.svc.Update(ctx, sensitivityLevel)
|
|
}
|
|
|
|
func (ctrl SensitivityLevel) Read(ctx context.Context, r *request.DalSensitivityLevelRead) (interface{}, error) {
|
|
return ctrl.svc.FindByID(ctx, r.SensitivityLevelID)
|
|
}
|
|
|
|
func (ctrl SensitivityLevel) Delete(ctx context.Context, r *request.DalSensitivityLevelDelete) (interface{}, error) {
|
|
return api.OK(), ctrl.svc.DeleteByID(ctx, r.SensitivityLevelID)
|
|
}
|
|
|
|
func (ctrl SensitivityLevel) Undelete(ctx context.Context, r *request.DalSensitivityLevelUndelete) (interface{}, error) {
|
|
return api.OK(), ctrl.svc.UndeleteByID(ctx, r.SensitivityLevelID)
|
|
}
|
|
|
|
func (ctrl SensitivityLevel) makePayload(ctx context.Context, res *types.DalSensitivityLevel, err error) (*sensitivityLevelPayload, error) {
|
|
if err != nil || res == nil {
|
|
return nil, err
|
|
}
|
|
|
|
pl := &sensitivityLevelPayload{res}
|
|
|
|
return pl, nil
|
|
}
|
|
|
|
func (ctrl SensitivityLevel) makeFilterPayload(ctx context.Context, rr types.DalSensitivityLevelSet, f types.DalSensitivityLevelFilter, err error) (*sensitivityLevelSetPayload, error) {
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
out := &sensitivityLevelSetPayload{Filter: f, Set: make([]*sensitivityLevelPayload, len(rr))}
|
|
|
|
for i := range rr {
|
|
out.Set[i], _ = ctrl.makePayload(ctx, rr[i], nil)
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (ctrl SensitivityLevel) serve(ctx context.Context, fn string, archive io.ReadSeeker, err error) (interface{}, error) {
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return func(w http.ResponseWriter, req *http.Request) {
|
|
w.Header().Add("Content-Disposition", "attachment; filename="+fn)
|
|
|
|
http.ServeContent(w, req, fn, time.Now(), archive)
|
|
}, nil
|
|
}
|