package com.dsh.other.util.wx;

import com.google.gson.annotations.SerializedName;
import okhttp3.*;

import java.io.IOException;
import java.io.UncheckedIOException;
import java.security.PrivateKey;
import java.security.PublicKey;
import java.util.List;

/**
 * App下单
 */
public class PartnerAppPrepay {
  private static String HOST = "https://api.mch.weixin.qq.com";
  private static String METHOD = "POST";
  private static String PATH = "/v3/pay/partner/transactions/app";

  public PartnerAPIv3AppPrepayResponse run(PartnerAPIv3CommonPrepayRequest request) {
    String uri = PATH;
    String reqBody = WXPayUtility.toJson(request);

    Request.Builder reqBuilder = new Request.Builder().url(HOST + uri);
    reqBuilder.addHeader("Accept", "application/json");
    reqBuilder.addHeader("Wechatpay-Serial", wechatPayPublicKeyId);
    reqBuilder.addHeader("Authorization", WXPayUtility.buildAuthorization(mchid, certificateSerialNo, privateKey, METHOD, uri, reqBody));
    reqBuilder.addHeader("Content-Type", "application/json");
    RequestBody requestBody = RequestBody.create(MediaType.parse("application/json; charset=utf-8"), reqBody);
    reqBuilder.method(METHOD, requestBody);
    Request httpRequest = reqBuilder.build();

    // 发送HTTP请求
    OkHttpClient client = new OkHttpClient.Builder().build();
    try (Response httpResponse = client.newCall(httpRequest).execute()) {
      String respBody = WXPayUtility.extractBody(httpResponse);
      if (httpResponse.code() >= 200 && httpResponse.code() < 300) {
        // 2XX 成功，验证应答签名
        WXPayUtility.validateResponse(this.wechatPayPublicKeyId, this.wechatPayPublicKey,
                                      httpResponse.headers(), respBody);
        // 从HTTP应答报文构建返回数据
        return WXPayUtility.fromJson(respBody, PartnerAPIv3AppPrepayResponse.class);
      } else {
        throw new WXPayUtility.ApiException(httpResponse.code(), respBody, httpResponse.headers());
      }
    } catch (IOException e) {
      throw new UncheckedIOException("Sending request to " + uri + " failed.", e);
    }
  }

  private final String mchid;
  private final String certificateSerialNo;
  private final PrivateKey privateKey;
  private final String wechatPayPublicKeyId;
  private final PublicKey wechatPayPublicKey;

  public PartnerAppPrepay(String mchid, String certificateSerialNo, String privateKeyFilePath, String wechatPayPublicKeyId, String wechatPayPublicKeyFilePath) {
    this.mchid = mchid;
    this.certificateSerialNo = certificateSerialNo;
    this.privateKey = WXPayUtility.loadPrivateKeyFromPath(privateKeyFilePath);
    this.wechatPayPublicKeyId = wechatPayPublicKeyId;
    this.wechatPayPublicKey = WXPayUtility.loadPublicKeyFromPath(wechatPayPublicKeyFilePath);
  }

  public static class PartnerSettleInfo {
    @SerializedName("profit_sharing")
    public Boolean profitSharing;
  }

  public static class CommonAmountInfo {
    @SerializedName("total")
    public Long total;

    @SerializedName("currency")
    public String currency;
  }

  public static class CommonSceneInfo {
    @SerializedName("payer_client_ip")
    public String payerClientIp;

    @SerializedName("device_id")
    public String deviceId;

    @SerializedName("store_info")
    public StoreInfo storeInfo;
  }

  public static class GoodsDetail {
    @SerializedName("merchant_goods_id")
    public String merchantGoodsId;

    @SerializedName("wechatpay_goods_id")
    public String wechatpayGoodsId;

    @SerializedName("goods_name")
    public String goodsName;

    @SerializedName("quantity")
    public Long quantity;

    @SerializedName("unit_price")
    public Long unitPrice;
  }

  public static class PartnerAPIv3AppPrepayResponse {
    @SerializedName("prepay_id")
    public String prepayId;
  }

  public static class PartnerAPIv3CommonPrepayRequest {
    @SerializedName("sp_appid")
    public String spAppid;

    @SerializedName("sp_mchid")
    public String spMchid;

    @SerializedName("sub_appid")
    public String subAppid;

    @SerializedName("sub_mchid")
    public String subMchid;

    @SerializedName("description")
    public String description;

    @SerializedName("out_trade_no")
    public String outTradeNo;

    @SerializedName("time_expire")
    public String timeExpire;

    @SerializedName("attach")
    public String attach;

    @SerializedName("notify_url")
    public String notifyUrl;

    @SerializedName("goods_tag")
    public String goodsTag;

    @SerializedName("settle_info")
    public PartnerSettleInfo settleInfo;

    @SerializedName("support_fapiao")
    public Boolean supportFapiao;

    @SerializedName("amount")
    public CommonAmountInfo amount;

    @SerializedName("detail")
    public CouponInfo detail;

    @SerializedName("scene_info")
    public CommonSceneInfo sceneInfo;
  }

  public static class CouponInfo {
    @SerializedName("cost_price")
    public Long costPrice;

    @SerializedName("invoice_id")
    public String invoiceId;

    @SerializedName("goods_detail")
    public List<GoodsDetail> goodsDetail;
  }

  public static class StoreInfo {
    @SerializedName("id")
    public String id;

    @SerializedName("name")
    public String name;

    @SerializedName("area_code")
    public String areaCode;

    @SerializedName("address")
    public String address;
  }
}