import { adapter } from './adapter.ts';
import { QueryWs, QueryWsOpts } from './ws.ts';
import { Query, ClientQuery } from './query.ts';
import { BaseQuery, QueryOptions, wrapperError } from './query.ts';

export { QueryOpts, QueryWs, ClientQuery, Query, QueryWsOpts, adapter, BaseQuery, wrapperError };
export { QueryOptions }
export type { DataOpts, Result, Data } from './query.ts';

type QueryOpts = {
  url?: string;
  adapter?: typeof adapter;
  headers?: Record<string, string>;
  timeout?: number;
  isClient?: boolean;
};

/**
 * 前端调用后端QueryRouter, 封装 beforeRequest 和 wss
 */
export class QueryClient extends Query {
  tokenName: string;
  storage: Storage;
  token: string;
  constructor(opts?: QueryOptions & { tokenName?: string; storage?: Storage; io?: boolean }) {
    super(opts);
    this.tokenName = opts?.tokenName || 'token';
    this.storage = opts?.storage || globalThis.localStorage;
    this.beforeRequest = async (opts) => {
      const token = this.token || this.getToken();
      if (token) {
        opts.headers = {
          ...opts.headers,
          Authorization: `Bearer ${token}`,
        };
      }
      return opts;
    };
    if (opts?.io) {
      this.createWs();
    }
  }
  createWs(opts?: QueryWsOpts) {
    this.qws = new QueryWs({ url: this.url, ...opts });
  }
  getToken() {
    return this.storage.getItem(this.tokenName);
  }
  saveToken(token: string) {
    this.storage.setItem(this.tokenName, token);
  }
  removeToken() {
    this.storage.removeItem(this.tokenName);
  }
}
// 移除默认生成的实例
// export const client = new QueryClient();