package sdk

import "context"

const (
	OpBuy           = 23
	OpSell          = 24
	OrderTypeVolume = 1101
	PrTypeLatest    = 5
	QuickTradeNow   = 2
)

type PassorderRequest struct {
	OpType       int     `json:"opType"`
	OrderType    int     `json:"orderType,omitempty"`
	Stock        string  `json:"stock"`
	PrType       int     `json:"prType,omitempty"`
	Price        float64 `json:"price"`
	Volume       int     `json:"volume"`
	QuickTrade   int     `json:"quickTrade,omitempty"`
	StrategyName string  `json:"strategyName,omitempty"`
}

func (c *Client) Passorder(ctx context.Context, req PassorderRequest) (*OrderRefResult, error) {
	var out OrderRefResult
	if err := c.post(ctx, "/api/trade/passorder", req, &out); err != nil {
		return nil, err
	}
	return &out, nil
}

// PassorderLatest 按最新价下单，不附加策略订单号。
func (c *Client) PassorderLatest(ctx context.Context, side int, stock string, volume int) (*OrderRefResult, error) {
	return c.PassorderLatestTagged(ctx, side, stock, volume, "")
}

// PassorderLatestTagged 使用 strategyName 将本地唯一订单号传给 QMT。
func (c *Client) PassorderLatestTagged(ctx context.Context, side int, stock string, volume int, orderID string) (*OrderRefResult, error) {
	return c.Passorder(ctx, PassorderRequest{
		OpType:       side,
		OrderType:    OrderTypeVolume,
		Stock:        stock,
		PrType:       PrTypeLatest,
		Price:        -1,
		Volume:       volume,
		QuickTrade:   QuickTradeNow,
		StrategyName: orderID,
	})
}

type AlgoPassorderRequest struct {
	OpType         int            `json:"opType"`
	OrderType      int            `json:"orderType,omitempty"`
	Stock          string         `json:"stock"`
	PrType         int            `json:"prType"`
	Price          float64        `json:"price"`
	Volume         int            `json:"volume"`
	StrategyName   string         `json:"strategyName,omitempty"`
	QuickTrade     int            `json:"quickTrade,omitempty"`
	UserOrderID    string         `json:"userOrderId,omitempty"`
	UserOrderParam map[string]any `json:"userOrderParam,omitempty"`
}

func (c *Client) AlgoPassorder(ctx context.Context, req AlgoPassorderRequest) (*OrderRefResult, error) {
	var out OrderRefResult
	if err := c.post(ctx, "/api/trade/algo_passorder", req, &out); err != nil {
		return nil, err
	}
	return &out, nil
}

type SmartAlgoPassorderRequest struct {
	OpType            int     `json:"opType"`
	OrderType         int     `json:"orderType,omitempty"`
	Stock             string  `json:"stock"`
	PrType            int     `json:"prType"`
	Price             float64 `json:"price"`
	Volume            int     `json:"volume"`
	SmartAlgoType     string  `json:"smartAlgoType"`
	LimitOverRate     int     `json:"limitOverRate"`
	MinAmountPerOrder int     `json:"minAmountPerOrder"`
	StartTime         string  `json:"startTime,omitempty"`
	EndTime           string  `json:"endTime,omitempty"`
}

func (c *Client) SmartAlgoPassorder(ctx context.Context, req SmartAlgoPassorderRequest) (*OrderRefResult, error) {
	var out OrderRefResult
	if err := c.post(ctx, "/api/trade/smart_algo_passorder", req, &out); err != nil {
		return nil, err
	}
	return &out, nil
}

type StyleOrderResult struct {
	Status string `json:"status"`
	Action string `json:"action"`
	Stock  string `json:"stock"`
}

func (c *Client) styleOrder(ctx context.Context, path string, body map[string]any) (*StyleOrderResult, error) {
	var out StyleOrderResult
	if err := c.post(ctx, path, body, &out); err != nil {
		return nil, err
	}
	return &out, nil
}

func (c *Client) OrderLots(ctx context.Context, stock string, lots int, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/order_lots", map[string]any{"stock": stock, "lots": lots, "style": style, "price": price})
}

func (c *Client) OrderValue(ctx context.Context, stock string, value float64, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/order_value", map[string]any{"stock": stock, "value": value, "style": style, "price": price})
}

func (c *Client) OrderPercent(ctx context.Context, stock string, percent float64, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/order_percent", map[string]any{"stock": stock, "percent": percent, "style": style, "price": price})
}

func (c *Client) OrderTargetValue(ctx context.Context, stock string, tarValue float64, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/order_target_value", map[string]any{"stock": stock, "tar_value": tarValue, "style": style, "price": price})
}

func (c *Client) OrderTargetPercent(ctx context.Context, stock string, tarPercent float64, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/order_target_percent", map[string]any{"stock": stock, "tar_percent": tarPercent, "style": style, "price": price})
}

func (c *Client) OrderShares(ctx context.Context, stock string, shares int, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/order_shares", map[string]any{"stock": stock, "shares": shares, "style": style, "price": price})
}

func (c *Client) FuturesBuyOpen(ctx context.Context, stock string, amount int, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/futures/buy_open", map[string]any{"stock": stock, "amount": amount, "style": style, "price": price})
}
func (c *Client) FuturesBuyCloseTdayFirst(ctx context.Context, stock string, amount int, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/futures/buy_close_tdayfirst", map[string]any{"stock": stock, "amount": amount, "style": style, "price": price})
}
func (c *Client) FuturesBuyCloseYdayFirst(ctx context.Context, stock string, amount int, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/futures/buy_close_ydayfirst", map[string]any{"stock": stock, "amount": amount, "style": style, "price": price})
}
func (c *Client) FuturesSellOpen(ctx context.Context, stock string, amount int, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/futures/sell_open", map[string]any{"stock": stock, "amount": amount, "style": style, "price": price})
}
func (c *Client) FuturesSellCloseTdayFirst(ctx context.Context, stock string, amount int, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/futures/sell_close_tdayfirst", map[string]any{"stock": stock, "amount": amount, "style": style, "price": price})
}
func (c *Client) FuturesSellCloseYdayFirst(ctx context.Context, stock string, amount int, style string, price float64) (*StyleOrderResult, error) {
	return c.styleOrder(ctx, "/api/trade/futures/sell_close_ydayfirst", map[string]any{"stock": stock, "amount": amount, "style": style, "price": price})
}

type TaskResult struct {
	Status string `json:"status"`
	TaskID any    `json:"taskId"`
}

func (c *Client) CancelTask(ctx context.Context, taskID string) (*TaskResult, error) {
	return c.task(ctx, "/api/trade/cancel_task", taskID)
}
func (c *Client) PauseTask(ctx context.Context, taskID string) (*TaskResult, error) {
	return c.task(ctx, "/api/trade/pause_task", taskID)
}
func (c *Client) ResumeTask(ctx context.Context, taskID string) (*TaskResult, error) {
	return c.task(ctx, "/api/trade/resume_task", taskID)
}

func (c *Client) task(ctx context.Context, path, taskID string) (*TaskResult, error) {
	var out TaskResult
	if err := c.post(ctx, path, map[string]any{"taskId": taskID, "accountType": c.accountType}, &out); err != nil {
		return nil, err
	}
	return &out, nil
}

func (c *Client) DoOrder(ctx context.Context) (map[string]any, error) {
	var out map[string]any
	if err := c.post(ctx, "/api/trade/do_order", nil, &out); err != nil {
		return nil, err
	}
	return out, nil
}

func (c *Client) TradeDetailData(ctx context.Context, datatype string) ([]map[string]string, error) {
	var out struct {
		Data []map[string]string `json:"data"`
	}
	if err := c.post(ctx, "/api/trade/trade_detail_data", map[string]any{"account": c.accountType, "datatype": datatype}, &out); err != nil {
		return nil, err
	}
	if out.Data == nil {
		return []map[string]string{}, nil
	}
	return out.Data, nil
}

func (c *Client) ValueByOrderID(ctx context.Context, orderID, datatype string) (map[string]string, error) {
	var out struct {
		Data map[string]string `json:"data"`
	}
	body := map[string]any{"orderId": orderID, "accountType": c.accountType, "datatype": datatype}
	if err := c.post(ctx, "/api/trade/value_by_order_id", body, &out); err != nil {
		return nil, err
	}
	return out.Data, nil
}

func (c *Client) LastOrderID(ctx context.Context, datatype string) (any, error) {
	var out struct {
		LastOrderID any `json:"last_order_id"`
	}
	if err := c.post(ctx, "/api/trade/last_order_id", map[string]any{"account": c.accountType, "datatype": datatype}, &out); err != nil {
		return nil, err
	}
	return out.LastOrderID, nil
}

func (c *Client) CanCancelOrder(ctx context.Context, orderID string) (any, error) {
	var out struct {
		CanCancel any `json:"can_cancel"`
	}
	if err := c.post(ctx, "/api/trade/can_cancel_order", map[string]any{"orderId": orderID, "accountType": c.accountType}, &out); err != nil {
		return nil, err
	}
	return out.CanCancel, nil
}

func (c *Client) contractList(ctx context.Context, path string) ([]map[string]string, error) {
	var out struct {
		Data []map[string]string `json:"data"`
	}
	if err := c.post(ctx, path, nil, &out); err != nil {
		return nil, err
	}
	return out.Data, nil
}

func (c *Client) DebtContract(ctx context.Context) ([]map[string]string, error) {
	return c.contractList(ctx, "/api/trade/debt_contract")
}
func (c *Client) AssureContract(ctx context.Context) ([]map[string]string, error) {
	return c.contractList(ctx, "/api/trade/assure_contract")
}
func (c *Client) EnableShortContract(ctx context.Context) ([]map[string]string, error) {
	return c.contractList(ctx, "/api/trade/enable_short_contract")
}

func (c *Client) IPOData(ctx context.Context, typ string) (any, error) {
	return c.postField(ctx, "/api/trade/ipo_data", map[string]any{"type": typ}, "data")
}

func (c *Client) NewPurchaseLimit(ctx context.Context) (any, error) {
	return c.postField(ctx, "/api/trade/new_purchase_limit", nil, "data")
}