129 lines
3.2 KiB
Go
129 lines
3.2 KiB
Go
package rest
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/titpetric/factory/resputil"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/crusttech/crust/compose/internal/service"
|
|
"github.com/crusttech/crust/compose/rest/request"
|
|
"github.com/crusttech/crust/compose/types"
|
|
)
|
|
|
|
var _ = errors.Wrap
|
|
|
|
type (
|
|
recordPayload struct {
|
|
*types.Record
|
|
|
|
CanUpdateRecord bool `json:"canUpdateRecord"`
|
|
CanDeleteRecord bool `json:"canDeleteRecord"`
|
|
}
|
|
|
|
recordSetPayload struct {
|
|
Filter types.RecordFilter `json:"filter"`
|
|
Set []*recordPayload `json:"set"`
|
|
}
|
|
|
|
Record struct {
|
|
record service.RecordService
|
|
attachment service.AttachmentService
|
|
permissions service.PermissionsService
|
|
}
|
|
)
|
|
|
|
func (Record) New() *Record {
|
|
return &Record{
|
|
record: service.DefaultRecord,
|
|
attachment: service.DefaultAttachment,
|
|
permissions: service.DefaultPermissions,
|
|
}
|
|
}
|
|
|
|
func (ctrl *Record) Report(ctx context.Context, r *request.RecordReport) (interface{}, error) {
|
|
return ctrl.record.With(ctx).Report(r.NamespaceID, r.ModuleID, r.Metrics, r.Dimensions, r.Filter)
|
|
}
|
|
|
|
func (ctrl *Record) List(ctx context.Context, r *request.RecordList) (interface{}, error) {
|
|
set, filter, err := ctrl.record.With(ctx).Find(types.RecordFilter{
|
|
NamespaceID: r.NamespaceID,
|
|
ModuleID: r.ModuleID,
|
|
Filter: r.Filter,
|
|
Sort: r.Sort,
|
|
PerPage: r.PerPage,
|
|
Page: r.Page,
|
|
})
|
|
return ctrl.makeFilterPayload(ctx, set, filter, err)
|
|
}
|
|
|
|
func (ctrl *Record) Read(ctx context.Context, r *request.RecordRead) (interface{}, error) {
|
|
return ctrl.record.With(ctx).FindByID(r.NamespaceID, r.RecordID)
|
|
}
|
|
|
|
func (ctrl *Record) Create(ctx context.Context, r *request.RecordCreate) (interface{}, error) {
|
|
return ctrl.record.With(ctx).Create(&types.Record{ModuleID: r.ModuleID, Values: r.Values})
|
|
}
|
|
|
|
func (ctrl *Record) Update(ctx context.Context, r *request.RecordUpdate) (interface{}, error) {
|
|
return ctrl.record.With(ctx).Update(&types.Record{
|
|
ID: r.RecordID,
|
|
ModuleID: r.ModuleID,
|
|
Values: r.Values})
|
|
}
|
|
|
|
func (ctrl *Record) Delete(ctx context.Context, r *request.RecordDelete) (interface{}, error) {
|
|
return resputil.OK(), ctrl.record.With(ctx).DeleteByID(r.NamespaceID, r.RecordID)
|
|
}
|
|
|
|
func (ctrl *Record) Upload(ctx context.Context, r *request.RecordUpload) (interface{}, error) {
|
|
file, err := r.Upload.Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer file.Close()
|
|
|
|
a, err := ctrl.attachment.With(ctx).CreateRecordAttachment(
|
|
r.NamespaceID,
|
|
r.Upload.Filename,
|
|
r.Upload.Size,
|
|
file,
|
|
r.ModuleID,
|
|
r.RecordID,
|
|
r.FieldName,
|
|
)
|
|
|
|
return makeAttachmentPayload(ctx, a, err)
|
|
}
|
|
|
|
func (ctrl Record) makePayload(ctx context.Context, c *types.Record, err error) (*recordPayload, error) {
|
|
if err != nil || c == nil {
|
|
return nil, err
|
|
}
|
|
|
|
perm := ctrl.permissions.With(ctx)
|
|
|
|
return &recordPayload{
|
|
Record: c,
|
|
|
|
CanUpdateRecord: perm.CanUpdateRecord(c),
|
|
CanDeleteRecord: perm.CanDeleteRecord(c),
|
|
}, nil
|
|
}
|
|
|
|
func (ctrl Record) makeFilterPayload(ctx context.Context, nn types.RecordSet, f types.RecordFilter, err error) (*recordSetPayload, error) {
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
modp := &recordSetPayload{Filter: f, Set: make([]*recordPayload, len(nn))}
|
|
|
|
for i := range nn {
|
|
modp.Set[i], _ = ctrl.makePayload(ctx, nn[i], nil)
|
|
}
|
|
|
|
return modp, nil
|
|
}
|