118 lines
3.6 KiB
Go
118 lines
3.6 KiB
Go
package rest
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/crusttech/crust/internal/payload"
|
|
"github.com/crusttech/crust/internal/payload/outgoing"
|
|
"github.com/crusttech/crust/sam/rest/request"
|
|
"github.com/crusttech/crust/sam/service"
|
|
"github.com/crusttech/crust/sam/types"
|
|
)
|
|
|
|
var _ = errors.Wrap
|
|
|
|
type (
|
|
Message struct {
|
|
svc struct {
|
|
msg service.MessageService
|
|
}
|
|
}
|
|
)
|
|
|
|
func (Message) New() *Message {
|
|
ctrl := &Message{}
|
|
ctrl.svc.msg = service.DefaultMessage
|
|
return ctrl
|
|
}
|
|
|
|
func (ctrl *Message) Create(ctx context.Context, r *request.MessageCreate) (interface{}, error) {
|
|
return ctrl.wrap(ctx)(ctrl.svc.msg.With(ctx).Create(&types.Message{
|
|
ChannelID: r.ChannelID,
|
|
Message: r.Message,
|
|
}))
|
|
}
|
|
|
|
func (ctrl *Message) ReplyCreate(ctx context.Context, r *request.MessageReplyCreate) (interface{}, error) {
|
|
return ctrl.wrap(ctx)(ctrl.svc.msg.With(ctx).Create(&types.Message{
|
|
ChannelID: r.ChannelID,
|
|
ReplyTo: r.MessageID,
|
|
Message: r.Message,
|
|
}))
|
|
}
|
|
|
|
func (ctrl *Message) ReplyGet(ctx context.Context, r *request.MessageReplyGet) (interface{}, error) {
|
|
return ctrl.wrapSet(ctx)(ctrl.svc.msg.With(ctx).Find(&types.MessageFilter{
|
|
ChannelID: r.ChannelID,
|
|
RepliesTo: r.MessageID,
|
|
}))
|
|
}
|
|
|
|
func (ctrl *Message) History(ctx context.Context, r *request.MessageHistory) (interface{}, error) {
|
|
return ctrl.wrapSet(ctx)(ctrl.svc.msg.With(ctx).Find(&types.MessageFilter{
|
|
ChannelID: r.ChannelID,
|
|
FirstID: r.LastMessageID,
|
|
}))
|
|
}
|
|
|
|
func (ctrl *Message) Edit(ctx context.Context, r *request.MessageEdit) (interface{}, error) {
|
|
return ctrl.wrap(ctx)(ctrl.svc.msg.With(ctx).Update(&types.Message{
|
|
ID: r.MessageID,
|
|
ChannelID: r.ChannelID,
|
|
Message: r.Message,
|
|
}))
|
|
}
|
|
|
|
func (ctrl *Message) Delete(ctx context.Context, r *request.MessageDelete) (interface{}, error) {
|
|
return nil, ctrl.svc.msg.With(ctx).Delete(r.MessageID)
|
|
}
|
|
|
|
func (ctrl *Message) MarkAsUnread(ctx context.Context, r *request.MessageMarkAsUnread) (interface{}, error) {
|
|
return true, ctrl.svc.msg.With(ctx).MarkAsUnread(r.MessageID)
|
|
}
|
|
|
|
func (ctrl *Message) PinCreate(ctx context.Context, r *request.MessagePinCreate) (interface{}, error) {
|
|
return nil, ctrl.svc.msg.With(ctx).Pin(r.MessageID)
|
|
}
|
|
|
|
func (ctrl *Message) PinRemove(ctx context.Context, r *request.MessagePinRemove) (interface{}, error) {
|
|
return nil, ctrl.svc.msg.With(ctx).RemovePin(r.MessageID)
|
|
}
|
|
|
|
func (ctrl *Message) BookmarkCreate(ctx context.Context, r *request.MessageBookmarkCreate) (interface{}, error) {
|
|
return nil, ctrl.svc.msg.With(ctx).Bookmark(r.MessageID)
|
|
}
|
|
|
|
func (ctrl *Message) BookmarkRemove(ctx context.Context, r *request.MessageBookmarkRemove) (interface{}, error) {
|
|
return nil, ctrl.svc.msg.With(ctx).RemoveBookmark(r.MessageID)
|
|
}
|
|
|
|
func (ctrl *Message) ReactionCreate(ctx context.Context, r *request.MessageReactionCreate) (interface{}, error) {
|
|
return nil, ctrl.svc.msg.With(ctx).React(r.MessageID, r.Reaction)
|
|
}
|
|
|
|
func (ctrl *Message) ReactionRemove(ctx context.Context, r *request.MessageReactionRemove) (interface{}, error) {
|
|
return nil, ctrl.svc.msg.With(ctx).RemoveReaction(r.MessageID, r.Reaction)
|
|
}
|
|
func (ctrl *Message) wrap(ctx context.Context) func(m *types.Message, err error) (*outgoing.Message, error) {
|
|
return func(m *types.Message, err error) (*outgoing.Message, error) {
|
|
if err != nil || m == nil {
|
|
return nil, err
|
|
} else {
|
|
return payload.Message(ctx, m), nil
|
|
}
|
|
}
|
|
}
|
|
|
|
func (ctrl *Message) wrapSet(ctx context.Context) func(mm types.MessageSet, err error) (*outgoing.MessageSet, error) {
|
|
return func(mm types.MessageSet, err error) (*outgoing.MessageSet, error) {
|
|
if err != nil {
|
|
return nil, err
|
|
} else {
|
|
return payload.Messages(ctx, mm), nil
|
|
}
|
|
}
|
|
}
|