package service

import (
	"context"
	"matchmaking-system/internal/pkg/flags"
	"matchmaking-system/internal/service/order"
	"matchmaking-system/internal/service/share"

	v1 "matchmaking-system/api/matchmaking/v1/share"
)

// GetBotStockTrade
//
//	@Description: 美股订单列表
//	@receiver s
//	@param ctx
//	@param req
//	@return *v1.GetUsBotStockTradeReply
//	@return error
func (s *ConduitService) GetBotStockTrade(ctx context.Context, req *v1.GetUsBotStockTradeRequest) (*v1.GetUsBotStockTradeReply, error) {
	if !share.VerificationBotStockTrade(req) {
		return share.BotStockTradeReply(false, flags.ErrIsParameter.Error(), req, nil, 0), nil
	}

	stockTradeList, totalCount, err := s.us.BotStockTradeList(ctx, req.PageSize, req.PageCount, req.Status)
	if err != nil {
		return share.BotStockTradeReply(false, err.Error(), req, nil, 0), nil
	}

	return share.BotStockTradeReply(true, flags.SetNull, req, stockTradeList, totalCount), nil
}

// SharePlaceOrder
//
//	@Description: 美股下单
//	@receiver s
//	@param ctx
//	@param req
//	@return *v1.UsOrderReply
//	@return error
func (s *ConduitService) SharePlaceOrder(ctx context.Context, req *v1.UsOrderRequest) (*v1.UsOrderReply, error) {
	if !share.VerificationShareOrderQuery(req) {
		return share.ShareUsPlaceOrderReply(false, flags.ErrIsParameter.Error(), flags.SetNull), nil
	}

	orderId, err := s.us.SharePlaceOrder(ctx, share.ShareUsOrderQuery(req))
	if err != nil {
		return share.ShareUsPlaceOrderReply(false, order.CheckShareError(err), flags.SetNull), nil
	}

	return share.ShareUsPlaceOrderReply(true, flags.SetNull, orderId), nil
}

// ShareUpdateOrder
//
//	@Description: 美股设置止盈止损
//	@receiver s
//	@param ctx
//	@param req
//	@return *v1.UsOrderReply
//	@return error
func (s *ConduitService) ShareUpdateOrder(ctx context.Context, req *v1.UpdateUsOrderRequest) (*v1.UsOrderReply, error) {
	if !share.VerificationStopOrderQuery(req) {
		return share.ShareUpdateOrderReply(false, flags.ErrIsParameter.Error(), req), nil
	}

	_, err := s.us.ShareUpdateOrder(ctx, share.StopOrderQuery(req))
	if err != nil {
		return share.ShareUpdateOrderReply(false, order.CheckShareError(err), req), nil
	}

	return share.ShareUpdateOrderReply(true, flags.SetNull, req), nil
}

// SharePosition
//
//	@Description: 美股平仓
//	@receiver s
//	@param ctx
//	@param req
//	@return *v1.UsOrderReply
//	@return error
func (s *ConduitService) SharePosition(ctx context.Context, req *v1.CancelUsOrderRequest) (*v1.UsOrderReply, error) {
	if !share.VerificationSharePosition(req) {
		return share.SharePositionOrderReply(false, flags.ErrIsParameter.Error(), req), nil
	}

	_, err := s.us.SharePosition(ctx, req.OrderId)
	if err != nil {
		return share.SharePositionOrderReply(false, order.CheckShareError(err), req), nil
	}

	return share.SharePositionOrderReply(true, flags.SetNull, req), nil
}

// ShareAllPosition
//
//	@Description: 美股一键平仓
//	@receiver s
//	@param ctx
//	@param req
//	@return *v1.AllUsOrderReply
//	@return error
func (s *ConduitService) ShareAllPosition(ctx context.Context, req *v1.AllUsOrderRequest) (*v1.AllUsOrderReply, error) {
	if err := s.us.ShareAllPosition(ctx); err != nil {
		return share.ShareAllPositionRequest(false, order.CheckShareError(err)), nil
	}

	return share.ShareAllPositionRequest(true, flags.SetNull), nil
}

// ShareCancel
//
//	@Description: 美股撤单
//	@receiver s
//	@param ctx
//	@param req
//	@return *v1.UsOrderReply
//	@return error
func (s *ConduitService) ShareCancel(ctx context.Context, req *v1.CancelUsOrderRequest) (*v1.UsOrderReply, error) {
	if !share.VerificationShareCancel(req) {
		return share.ShareCancelReply(false, flags.ErrIsParameter.Error(), req), nil
	}

	_, err := s.us.ShareCancel(ctx, req.OrderId)
	if err != nil {
		return share.ShareCancelReply(false, order.CheckShareError(err), req), nil
	}

	return share.ShareCancelReply(true, flags.SetNull, req), nil
}