package me.chanjar.weixin.cp.config.impl;

import com.tencent.wework.Finance;
import me.chanjar.weixin.common.bean.WxAccessToken;
import me.chanjar.weixin.common.util.http.apache.ApacheHttpClientBuilder;
import me.chanjar.weixin.cp.config.WxCpConfigStorage;
import me.chanjar.weixin.cp.constant.WxCpApiPathConsts;
import me.chanjar.weixin.cp.util.json.WxCpGsonBuilder;

import java.io.File;
import java.io.Serializable;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

/**
 * 基于内存的微信配置provider，在实际生产环境中应该将这些配置持久化.
 *
 * @author Daniel Qian
 */
public class WxCpDefaultConfigImpl implements WxCpConfigStorage, Serializable {
  private static final long serialVersionUID = 1154541446729462780L;
  /**
   * The Access token.
   */
  protected volatile String accessToken;
  /**
   * The Access token lock.
   */
  protected transient Lock accessTokenLock = new ReentrantLock();
  /**
   * The Agent id.
   */
  protected volatile Integer agentId;
  /**
   * The Jsapi ticket lock.
   */
  protected transient Lock jsapiTicketLock = new ReentrantLock();
  /**
   * The Agent jsapi ticket lock.
   */
  protected transient Lock agentJsapiTicketLock = new ReentrantLock();
  private volatile String corpId;
  private volatile String corpSecret;
  private volatile String token;
  private volatile String aesKey;
  private volatile long expiresTime;
  /**
   * 会话存档私钥以及sdk路径
   */
  private volatile String msgAuditSecret;
  private volatile String msgAuditPriKey;
  private volatile String msgAuditLibPath;
  /**
   * 会话存档access token及其过期时间
   */
  private volatile String msgAuditAccessToken;
  private volatile long msgAuditAccessTokenExpiresTime;
  /**
   * 会话存档access token锁
   */
  protected transient Lock msgAuditAccessTokenLock = new ReentrantLock();
  /**
   * 会话存档SDK及其过期时间
   */
  private volatile long msgAuditSdk;
  private volatile long msgAuditSdkExpiresTime;
  /**
   * 会话存档SDK引用计数，用于多线程安全的生命周期管理
   */
  private volatile int msgAuditSdkRefCount;
  private volatile String oauth2redirectUri;
  private volatile String httpProxyHost;
  private volatile int httpProxyPort;
  private volatile String httpProxyUsername;
  private volatile String httpProxyPassword;
  private volatile String jsapiTicket;
  private volatile long jsapiTicketExpiresTime;
  private volatile String agentJsapiTicket;
  private volatile long agentJsapiTicketExpiresTime;

  private volatile File tmpDirFile;

  private transient volatile ApacheHttpClientBuilder apacheHttpClientBuilder;

  private volatile String baseApiUrl;

  private volatile String webhookKey;

  @Override
  public void setBaseApiUrl(String baseUrl) {
    this.baseApiUrl = baseUrl;
  }

  @Override
  public String getApiUrl(String path) {
    if (baseApiUrl == null) {
      baseApiUrl = WxCpApiPathConsts.DEFAULT_CP_BASE_URL;
    }
    return baseApiUrl + path;
  }

  @Override
  public String getAccessToken() {
    return this.accessToken;
  }

  /**
   * Sets access token.
   *
   * @param accessToken the access token
   */
  public void setAccessToken(String accessToken) {
    this.accessToken = accessToken;
  }

  @Override
  public Lock getAccessTokenLock() {
    return this.accessTokenLock;
  }

  @Override
  public boolean isAccessTokenExpired() {
    return System.currentTimeMillis() > this.expiresTime;
  }

  @Override
  public void expireAccessToken() {
    this.expiresTime = 0;
  }

  @Override
  public synchronized void updateAccessToken(WxAccessToken accessToken) {
    updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn());
  }

  @Override
  public synchronized void updateAccessToken(String accessToken, int expiresInSeconds) {
    this.accessToken = accessToken;
    this.expiresTime = System.currentTimeMillis() + (expiresInSeconds - 200) * 1000L;
  }

  @Override
  public String getJsapiTicket() {
    return this.jsapiTicket;
  }

  /**
   * Sets jsapi ticket.
   *
   * @param jsapiTicket the jsapi ticket
   */
  public void setJsapiTicket(String jsapiTicket) {
    this.jsapiTicket = jsapiTicket;
  }

  @Override
  public Lock getJsapiTicketLock() {
    return this.jsapiTicketLock;
  }

  /**
   * Gets jsapi ticket expires time.
   *
   * @return the jsapi ticket expires time
   */
  public long getJsapiTicketExpiresTime() {
    return this.jsapiTicketExpiresTime;
  }

  /**
   * Sets jsapi ticket expires time.
   *
   * @param jsapiTicketExpiresTime the jsapi ticket expires time
   */
  public void setJsapiTicketExpiresTime(long jsapiTicketExpiresTime) {
    this.jsapiTicketExpiresTime = jsapiTicketExpiresTime;
  }

  @Override
  public boolean isJsapiTicketExpired() {
    return System.currentTimeMillis() > this.jsapiTicketExpiresTime;
  }

  @Override
  public synchronized void updateJsapiTicket(String jsapiTicket, int expiresInSeconds) {
    this.jsapiTicket = jsapiTicket;
    // 预留200秒的时间
    this.jsapiTicketExpiresTime = System.currentTimeMillis() + (expiresInSeconds - 200) * 1000L;
  }

  @Override
  public String getAgentJsapiTicket() {
    return this.agentJsapiTicket;
  }

  @Override
  public Lock getAgentJsapiTicketLock() {
    return this.agentJsapiTicketLock;
  }

  @Override
  public boolean isAgentJsapiTicketExpired() {
    return System.currentTimeMillis() > this.agentJsapiTicketExpiresTime;
  }

  @Override
  public void expireAgentJsapiTicket() {
    this.agentJsapiTicketExpiresTime = 0;
  }

  @Override
  public void updateAgentJsapiTicket(String jsapiTicket, int expiresInSeconds) {
    this.agentJsapiTicket = jsapiTicket;
    // 预留200秒的时间
    this.agentJsapiTicketExpiresTime = System.currentTimeMillis() + (expiresInSeconds - 200) * 1000L;
  }

  @Override
  public void expireJsapiTicket() {
    this.jsapiTicketExpiresTime = 0;
  }

  @Override
  public String getCorpId() {
    return this.corpId;
  }

  /**
   * Sets corp id.
   *
   * @param corpId the corp id
   */
  public void setCorpId(String corpId) {
    this.corpId = corpId;
  }

  @Override
  public String getCorpSecret() {
    return this.corpSecret;
  }

  /**
   * Sets corp secret.
   *
   * @param corpSecret the corp secret
   */
  public void setCorpSecret(String corpSecret) {
    this.corpSecret = corpSecret;
  }

  @Override
  public String getToken() {
    return this.token;
  }

  /**
   * Sets token.
   *
   * @param token the token
   */
  public void setToken(String token) {
    this.token = token;
  }

  @Override
  public long getExpiresTime() {
    return this.expiresTime;
  }

  /**
   * Sets expires time.
   *
   * @param expiresTime the expires time
   */
  public void setExpiresTime(long expiresTime) {
    this.expiresTime = expiresTime;
  }

  @Override
  public String getAesKey() {
    return this.aesKey;
  }

  @Override
  public String getMsgAuditPriKey() {
    return this.msgAuditPriKey;
  }

  @Override
  public String getMsgAuditLibPath() {
    return this.msgAuditLibPath;
  }

  /**
   * Sets aes key.
   *
   * @param aesKey the aes key
   */
  public void setAesKey(String aesKey) {
    this.aesKey = aesKey;
  }

  @Override
  public Integer getAgentId() {
    return this.agentId;
  }

  /**
   * Sets agent id.
   *
   * @param agentId the agent id
   */
  public void setAgentId(Integer agentId) {
    this.agentId = agentId;
  }

  /**
   * 设置企微会话存档路径.
   *
   * @param msgAuditLibPath 会话存档具体路径
   */
  public void setMsgAuditLibPath(String msgAuditLibPath) {
    this.msgAuditLibPath = msgAuditLibPath;
  }

  /**
   * 设置会话存档私钥
   *
   * @param msgAuditPriKey 会话存档私钥
   */
  public void setMsgAuditPriKey(String msgAuditPriKey) {
    this.msgAuditPriKey = msgAuditPriKey;
  }

  @Override
  public String getOauth2redirectUri() {
    return this.oauth2redirectUri;
  }

  /**
   * Sets oauth 2 redirect uri.
   *
   * @param oauth2redirectUri the oauth 2 redirect uri
   */
  public void setOauth2redirectUri(String oauth2redirectUri) {
    this.oauth2redirectUri = oauth2redirectUri;
  }

  @Override
  public String getHttpProxyHost() {
    return this.httpProxyHost;
  }

  /**
   * Sets http proxy host.
   *
   * @param httpProxyHost the http proxy host
   */
  public void setHttpProxyHost(String httpProxyHost) {
    this.httpProxyHost = httpProxyHost;
  }

  @Override
  public int getHttpProxyPort() {
    return this.httpProxyPort;
  }

  /**
   * Sets http proxy port.
   *
   * @param httpProxyPort the http proxy port
   */
  public void setHttpProxyPort(int httpProxyPort) {
    this.httpProxyPort = httpProxyPort;
  }

  @Override
  public String getHttpProxyUsername() {
    return this.httpProxyUsername;
  }

  /**
   * Sets http proxy username.
   *
   * @param httpProxyUsername the http proxy username
   */
  public void setHttpProxyUsername(String httpProxyUsername) {
    this.httpProxyUsername = httpProxyUsername;
  }

  @Override
  public String getHttpProxyPassword() {
    return this.httpProxyPassword;
  }

  /**
   * Sets http proxy password.
   *
   * @param httpProxyPassword the http proxy password
   */
  public void setHttpProxyPassword(String httpProxyPassword) {
    this.httpProxyPassword = httpProxyPassword;
  }

  @Override
  public String toString() {
    return WxCpGsonBuilder.create().toJson(this);
  }

  @Override
  public File getTmpDirFile() {
    return this.tmpDirFile;
  }

  /**
   * Sets tmp dir file.
   *
   * @param tmpDirFile the tmp dir file
   */
  public void setTmpDirFile(File tmpDirFile) {
    this.tmpDirFile = tmpDirFile;
  }

  @Override
  public ApacheHttpClientBuilder getApacheHttpClientBuilder() {
    return this.apacheHttpClientBuilder;
  }

  /**
   * Sets apache http client builder.
   *
   * @param apacheHttpClientBuilder the apache http client builder
   */
  public void setApacheHttpClientBuilder(ApacheHttpClientBuilder apacheHttpClientBuilder) {
    this.apacheHttpClientBuilder = apacheHttpClientBuilder;
  }

  @Override
  public boolean autoRefreshToken() {
    return true;
  }

  @Override
  public String getWebhookKey() {
    return this.webhookKey;
  }

  /**
   * Sets webhook key.
   *
   * @param webhookKey the webhook key
   * @return the webhook key
   */
  public WxCpDefaultConfigImpl setWebhookKey(String webhookKey) {
    this.webhookKey = webhookKey;
    return this;
  }

  @Override
  public String getMsgAuditSecret() {
    return this.msgAuditSecret;
  }

  /**
   * 设置会话存档secret
   *
   * @param msgAuditSecret the msg audit secret
   * @return this
   */
  public WxCpDefaultConfigImpl setMsgAuditSecret(String msgAuditSecret) {
    this.msgAuditSecret = msgAuditSecret;
    return this;
  }

  @Override
  public String getMsgAuditAccessToken() {
    return this.msgAuditAccessToken;
  }

  @Override
  public Lock getMsgAuditAccessTokenLock() {
    return this.msgAuditAccessTokenLock;
  }

  @Override
  public boolean isMsgAuditAccessTokenExpired() {
    return System.currentTimeMillis() > this.msgAuditAccessTokenExpiresTime;
  }

  @Override
  public void expireMsgAuditAccessToken() {
    this.msgAuditAccessTokenExpiresTime = 0;
  }

  @Override
  public synchronized void updateMsgAuditAccessToken(String accessToken, int expiresInSeconds) {
    this.msgAuditAccessToken = accessToken;
    // 预留200秒的时间
    this.msgAuditAccessTokenExpiresTime = System.currentTimeMillis() + (expiresInSeconds - 200) * 1000L;
  }

  @Override
  public long getMsgAuditSdk() {
    return this.msgAuditSdk;
  }

  @Override
  public boolean isMsgAuditSdkExpired() {
    return System.currentTimeMillis() > this.msgAuditSdkExpiresTime;
  }

  @Override
  public synchronized void updateMsgAuditSdk(long sdk, int expiresInSeconds) {
    // 如果有旧的SDK且不同于新的SDK，需要销毁旧的SDK
    if (this.msgAuditSdk > 0 && this.msgAuditSdk != sdk) {
      // 无论旧SDK是否仍有引用，都需要销毁它以避免资源泄漏
      // 如果有飞行中的请求使用旧SDK，这些请求可能会失败，但这比资源泄漏更安全
      Finance.DestroySdk(this.msgAuditSdk);
    }
    this.msgAuditSdk = sdk;
    // 预留200秒的时间
    this.msgAuditSdkExpiresTime = System.currentTimeMillis() + (expiresInSeconds - 200) * 1000L;
    // 重置引用计数，因为这是一个全新的SDK
    this.msgAuditSdkRefCount = 0;
  }

  @Override
  public void expireMsgAuditSdk() {
    this.msgAuditSdkExpiresTime = 0;
  }

  @Override
  public synchronized int incrementMsgAuditSdkRefCount(long sdk) {
    if (this.msgAuditSdk == sdk && sdk > 0) {
      return ++this.msgAuditSdkRefCount;
    }
    return -1; // SDK不匹配，返回-1表示错误
  }

  @Override
  public synchronized int decrementMsgAuditSdkRefCount(long sdk) {
    if (this.msgAuditSdk == sdk && this.msgAuditSdkRefCount > 0) {
      int newCount = --this.msgAuditSdkRefCount;
      // 当引用计数降为0时，自动销毁SDK以释放资源
      // 再次检查SDK是否仍然是当前缓存的SDK（防止并发重新初始化）
      if (newCount == 0 && this.msgAuditSdk == sdk) {
        Finance.DestroySdk(sdk);
        this.msgAuditSdk = 0;
        this.msgAuditSdkExpiresTime = 0;
      }
      return newCount;
    }
    return -1; // SDK不匹配或引用计数已为0，返回-1表示错误
  }

  @Override
  public synchronized int getMsgAuditSdkRefCount(long sdk) {
    if (this.msgAuditSdk == sdk && sdk > 0) {
      return this.msgAuditSdkRefCount;
    }
    return -1; // SDK不匹配，返回-1表示错误
  }

  @Override
  public synchronized long acquireMsgAuditSdk() {
    // 检查SDK是否有效（已初始化且未过期）
    if (this.msgAuditSdk > 0 && !isMsgAuditSdkExpired()) {
      this.msgAuditSdkRefCount++;
      return this.msgAuditSdk;
    }
    return 0; // SDK未初始化或已过期
  }

  @Override
  public synchronized void releaseMsgAuditSdk(long sdk) {
    if (this.msgAuditSdk == sdk && this.msgAuditSdkRefCount > 0) {
      int newCount = --this.msgAuditSdkRefCount;
      // 当引用计数降为0时，自动销毁SDK以释放资源
      // 再次检查SDK是否仍然是当前缓存的SDK（防止并发重新初始化）
      if (newCount == 0 && this.msgAuditSdk == sdk) {
        Finance.DestroySdk(sdk);
        this.msgAuditSdk = 0;
        this.msgAuditSdkExpiresTime = 0;
      }
    }
  }
}
