# coding: utf-8
require "awesome_print"
require 'huobi_client/request'
require 'active_support/all'

module HuobiClient
  class Client
    attr_reader :access, :secret, :account_id

    include HuobiClient::Request

    def initialize(access, secret, account_id=nil)
      @access = access
      @secret = secret
      @account_id = account_id
    end

    # 行情API
    def kline(symbol:, period:, size: 150) # 获取K线数据
      get '/market/history/kline', fun_params(__method__, binding)
    end

    def ticker(symbol:)
      get '/market/detail/merged', fun_params(__method__, binding)
    end

    def depth(symbol:, type:)
      get '/market/depth', fun_params(__method__, binding)
    end

    def trade_detail(symbol:)
      get '/market/trade', fun_params(__method__, binding)
    end

    def trade_list(symbol:, size: 1)
      get '/market/history/trade', fun_params(__method__, binding)
    end

    def market_detail(symbol:)
      get '/market/detail', fun_params(__method__, binding)
    end

    # 公共API
    def symbols
      get '/v1/common/symbols'
    end

    def hadax_symbols
      get '/v1/hadax/common/symbols'
    end

    def currencys
      get '/v1/common/currencys'
    end

    def hadax_currencys
      get '/v1/hadax/common/currencys'
    end

    def timestamp
      get '/v1/common/timestamp'
    end

    # 用户资产API
    def accounts
      get '/v1/account/accounts'
    end

    def balance(account_id: @account_id)
      get "/v1/account/accounts/#{account_id}/balance"
    end

    def hadax_balance(account_id: @account_id)
      get "/v1/hadax/account/accounts/#{account_id}/balance"
    end

    # 交易API
    def place(account_id: @account_id, symbol:, type:, amount:, price: nil, source: 'api') # Pro站下单
      # types
      # buy-market：市价买,
      # sell-market：市价卖,
      # buy-limit：限价买,
      # sell-limit：限价卖,
      # buy-ioc：IOC买单,
      # sell-ioc：IOC卖单
      post '/v1/order/orders/place', fun_params(__method__, binding).transform_keys {|key| key.to_s.sub('_', '-')}
    end

    def hadax_place(account_id: @account_id, symbol:, type:, amount:, price: nil, source: 'api') # HADAX站下单
      post '/v1/hadax/order/orders/place', fun_params(__method__, binding).transform_keys {|key| key.to_s.sub('_', '-')}
    end

    def submit_cancel(order_id:) # 申请撤销一个订单请求
      post "/v1/order/orders/#{order_id}/submitcancel"
    end

    def batch_cancel(order_ids:) # 批量撤销订单
      post '/v1/order/orders/batchcancel', fun_params(__method__, binding).transform_keys {|key| key.to_s.sub('_', '-')}
    end

    def order_detail(order_id:) # 查询某个订单详情
      get "/v1/order/orders/#{order_id}"
    end

    def match_results(order_id:) # 查询某个订单的成交明细
      get "/v1/order/orders/#{order_id}/matchresults"
    end

    def orders(symbol: nil, states:, types: nil, start_date: nil, end_date: nil, from: nil, direct: nil, size: nil) # 查询当前委托、历史委托
      # direct	false	string	查询方向		prev 向前，next 向后
      # states	true	string	查询的订单状态组合，使用','分割		pre-submitted 准备提交, submitted 已提交, partial-filled 部分成交, partial-canceled 部分成交撤销, filled 完全成交, canceled 已撤销
      get '/v1/order/orders', fun_params(__method__, binding).transform_keys {|key| key.to_s.sub('_', '-')}
    end

    def all_match_results(symbol: nil, states:, types: nil, start_date: nil, end_date: nil, from: nil, direct: nil, size: nil) # 查询当前成交、历史成交
      get '/v1/order/matchresults', fun_params(__method__, binding).transform_keys {|key| key.to_s.sub('_', '-')}
    end

    # 借贷交易API （重要：如果使用借贷资产交易，请在下单接口/v1/order/orders/place请求参数source中填写‘margin-api’）

    def dw_transfer_in(symbol:, currency:, amount:) # 现货账户划入至借贷账户
      post '/v1/dw/transfer-in/margin', fun_params(__method__, binding)
    end

    def dw_transfer_out(symbol:, currency:, amount:) # 借贷账户划出至现货账户
      post '/v1/dw/transfer-out/margin', fun_params(__method__, binding)
    end

    def margin_apply(symbol:, currency:, amount:) # 申请借贷
      post '/v1/margin/orders', fun_params(__method__, binding)
    end

    def margin_repay(amount:, order_id:) # 归还借贷
      post "/v1/margin/orders/#{order_id}/repay", fun_params(__method__, binding)
    end

    def loan_orders(symbol:, states: nil, start_date: nil, end_date: nil, from: nil, direct: nil, size: nil) # 借贷订单
      get '/v1/margin/loan-orders', fun_params(__method__, binding).transform_keys {|key| key.to_s.sub('_', '-')}
    end

    def margin_balance(symbol:) # 借贷账户详情
      get '/v1/margin/accounts/balance', fun_params(__method__, binding)
    end

    # 虚拟币提现API
    def withdraw(address:, amount:, currency:, fee: nil, addr_tag: nil) # 申请提现虚拟币
      post '/v1/dw/withdraw/api/create', fun_params(__method__, binding).transform_keys {|key| key.to_s.sub('_', '-')}
    end

    def withdraw_cancel # 申请取消提现虚拟币
      post '/v1/dw/withdraw-virtual/{withdraw-id}/cancel'
    end

    def withdraw_query(currency:, type:, from: nil, size: nil) # 查询虚拟币充提记录
      # type	true	string	'deposit' or 'withdraw'
      get '/v1/query/deposit-withdraw', fun_params(__method__, binding)
    end

    private
    def fun_params(_method, _binding)
      Hash[method(_method).parameters.map.collect { |_, name| [name, _binding.local_variable_get(name)] }]
    end

  end
end