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