glitternetwork--pinme/bin/utils/pinmeApi.ts

import chalk from 'chalk';
import { createCarApiClient, createPinmeApiClient } from './apiClient';
import { APP_CONFIG } from './config';
import { isDnsDomain } from './domainValidator';

// Token expired error codes
const TOKEN_EXPIRED_CODES = [
  401,
  403,
  10001,
  10002,
  20001,
  'TOKEN_EXPIRED',
  'AUTH_FAILED',
];
const TOKEN_EXPIRED_MESSAGES = [
  'token expired',
  'invalid token',
  'authentication failed',
  'auth failed',
  'unauthorized',
  '登录',
  '过期',
  'token',
  '鉴权',
];

function isTokenExpired(error: any): boolean {
  const status = error.response?.status;
  const message =
    error.response?.data?.msg ||
    error.response?.data?.message ||
    error.message ||
    '';
  const code = error.response?.data?.code;

  if (status && TOKEN_EXPIRED_CODES.includes(status)) {
    return true;
  }
  if (code && TOKEN_EXPIRED_CODES.includes(code)) {
    return true;
  }
  const lowerMessage = message.toLowerCase();
  return TOKEN_EXPIRED_MESSAGES.some((m) =>
    lowerMessage.includes(m.toLowerCase()),
  );
}

function showTokenExpiredHint(): void {
  console.log(chalk.red('\n⚠️  Token has expired or is invalid.'));
  console.log(chalk.yellow('Please re-run: pinme set-appkey <AppKey>\n'));
}

export async function bindAnonymousDevice(
  anonymousUid: string,
): Promise<boolean> {
  try {
    const client = createPinmeApiClient();
    const { data } = await client.post('/bind_anonymous', {
      anonymous_uid: anonymousUid,
    });
    return data?.code === 200;
  } catch (e: any) {
    if (isTokenExpired(e)) {
      showTokenExpiredHint();
      return false;
    }
    console.log(
      chalk.yellow(`Failed to trigger anonymous binding: ${e?.message || e}`),
    );
    return false;
  }
}

export interface CheckDomainResult {
  is_valid: boolean;
  error?: string;
}

export interface GetRootDomainResponse {
  code: number;
  msg: string;
  data: {
    domain: string;
  };
}

let rootDomainCache: string | null = null;

export async function getRootDomain(
  forceRefresh: boolean = false,
): Promise<string> {
  if (!forceRefresh && rootDomainCache) {
    return rootDomainCache;
  }

  const client = createPinmeApiClient();
  const { data } = await client.get<GetRootDomainResponse>('/root_domain');

  if (data?.code === 200 && data?.data?.domain) {
    rootDomainCache = data.data.domain;
    return rootDomainCache;
  }

  throw new Error(data?.msg || 'Failed to get root domain');
}

export async function checkDomainAvailable(
  domainName: string,
): Promise<CheckDomainResult> {
  if (isDnsDomain(domainName)) {
    return await {
      is_valid: true,
    };
  }
  const client = createPinmeApiClient();
  // Endpoint may not be fixed, prioritize environment variable, then try two common paths
  const configured = APP_CONFIG.pinmeCheckDomainPath;
  const fallbacks = [configured];
  let lastRecoverableError: any;

  for (const p of fallbacks) {
    try {
      const { data } = await client.post(p, { domain_name: domainName });
      if (typeof data?.is_valid === 'boolean') {
        return { is_valid: data.is_valid, error: data?.error };
      }
      if (data?.data && typeof data.data.is_valid === 'boolean') {
        return { is_valid: data.data.is_valid, error: data.data?.error };
      }
      // Unexpected structure, continue trying next path
    } catch (e: any) {
      if (isTokenExpired(e)) {
        showTokenExpiredHint();
        throw new Error('Token expired');
      }

      const status = e?.response?.status;
      if (status && ![404, 405].includes(status)) {
        throw e;
      }

      lastRecoverableError = e;
    }
  }

  if (lastRecoverableError) {
    throw lastRecoverableError;
  }

  // If all attempts fail silently, return unknown state and let the bind step decide.
  return { is_valid: true };
}

export async function bindPinmeDomain(
  domainName: string,
  hash: string,
  projectName?: string,
): Promise<boolean> {
  try {
    const client = createPinmeApiClient();
    const { data } = await client.post('/bind_pinme_domain', {
      domain_name: domainName,
      hash,
      ...(projectName ? { project_name: projectName } : {}),
    });
    return data?.code === 200;
  } catch (e: any) {
    if (isTokenExpired(e)) {
      showTokenExpiredHint();
      throw new Error('Token expired');
    }
    throw e;
  }
}

export interface MyDomainItem {
  domain_name: string;
  domain_type: number;
  bind_time: number;
  expire_time: number;
}

export async function getMyDomains(): Promise<MyDomainItem[]> {
  try {
    const client = createPinmeApiClient();
    const { data } = await client.get('/my_domains');
    if (data?.code === 200) {
      if (Array.isArray(data?.data)) {
        // v4: data is array
        return data.data as MyDomainItem[];
      }
      if (data?.data?.list && Array.isArray(data.data.list)) {
        // fallback: sometimes wrapped in { list: [] }
        return data.data.list as MyDomainItem[];
      }
    }
    if (data?.code === 401 || data?.code === 403) {
      showTokenExpiredHint();
      throw new Error('Token expired');
    }
    return [];
  } catch (e: any) {
    if (isTokenExpired(e)) {
      showTokenExpiredHint();
      throw new Error('Token expired');
    }
    throw e;
  }
}

export interface BindDnsDomainV4Response {
  code: number;
  msg: string;
  data?: {
    domain_name: string;
    hash: string;
    bind_time?: number;
  };
}

export async function bindDnsDomainV4(
  domainName: string,
  hash: string,
  tokenAddress: string,
  authToken: string,
  projectName?: string,
): Promise<BindDnsDomainV4Response> {
  try {
    const client = createPinmeApiClient();
    const { data } = await client.post<BindDnsDomainV4Response>(
      '/bind_dns',
      {
        domain_name: domainName,
        hash: hash,
        ...(projectName ? { project_name: projectName } : {}),
      },
      {
        headers: {
          'x-auth-token': authToken,
          'x-token-address': tokenAddress,
        },
      },
    );
    return data as BindDnsDomainV4Response;
  } catch (e: any) {
    if (isTokenExpired(e)) {
      showTokenExpiredHint();
      throw new Error('Token expired');
    }
    throw e;
  }
}

export interface WalletBalanceResponse {
  code: number;
  msg: string;
  data?: {
    wallet_balance_usd?: number;
  };
}

export async function getWalletBalance(
  tokenAddress: string,
  authToken: string,
): Promise<WalletBalanceResponse> {
  try {
    const client = createPinmeApiClient();
    const { data } = await client.get<WalletBalanceResponse>(
      '/pay/wallet/balance',
      {
        headers: {
          'authentication-tokens': authToken,
          'token-address': tokenAddress,
        },
      },
    );
    return data as WalletBalanceResponse;
  } catch (e: any) {
    if (isTokenExpired(e)) {
      showTokenExpiredHint();
      throw new Error('Token expired');
    }
    throw e;
  }
}

export interface IsVipResponse {
  code: number;
  msg: string;
  data?: {
    is_vip: boolean;
    vip_expire_time?: number;
  };
}

export async function isVip(
  tokenAddress: string,
  authToken: string,
): Promise<IsVipResponse> {
  try {
    const client = createPinmeApiClient();
    const { data } = await client.get<IsVipResponse>('/is_vip', {
      headers: {
        'x-auth-token': authToken,
        'x-token-address': tokenAddress,
      },
    });
    return data as IsVipResponse;
  } catch (e: any) {
    if (isTokenExpired(e)) {
      showTokenExpiredHint();
      throw new Error('Token expired');
    }
    throw e;
  }
}

export interface CarExportResponse {
  code: number;
  msg: string;
  data: {
    cid: string;
    status: string;
    task_id: string;
  };
}

export interface CarExportStatusResponse {
  code: number;
  msg: string;
  data: {
    task_id: string;
    cid: string;
    status: 'processing' | 'completed' | 'failed';
    download_url?: string;
  };
}

export async function requestCarExport(
  cid: string,
  uid: string,
): Promise<CarExportResponse['data']> {
  try {
    const client = createCarApiClient();
    // Use POST method as shown in the example
    const { data } = await client.post<CarExportResponse>('/car/export', null, {
      params: {
        cid,
        uid,
      },
    });
    if (data?.code === 200 && data?.data) {
      return data.data;
    }
    throw new Error(data?.msg || 'Failed to request CAR export');
  } catch (e: any) {
    if (isTokenExpired(e)) {
      showTokenExpiredHint();
      throw new Error('Token expired');
    }
    if (e.response?.data?.msg) {
      throw new Error(e.response.data.msg);
    }
    throw new Error(`Failed to request CAR export: ${e?.message || e}`);
  }
}

export interface BindDnsDomainV4Response {
  code: number;
  msg: string;
  data?: {
    domain_name: string;
    hash: string;
    bind_time?: number;
  };
}
export async function checkCarExportStatus(
  taskId: string,
): Promise<CarExportStatusResponse['data']> {
  try {
    const client = createCarApiClient();
    const { data } = await client.get<CarExportStatusResponse>(
      '/car/export/status',
      {
        params: {
          task_id: taskId,
        },
      },
    );
    if (data?.code === 200 && data?.data) {
      return data.data;
    }
    throw new Error(data?.msg || 'Failed to check export status');
  } catch (e: any) {
    if (isTokenExpired(e)) {
      showTokenExpiredHint();
      throw new Error('Token expired');
    }
    if (e.response?.data?.msg) {
      throw new Error(e.response.data.msg);
    }
    throw new Error(`Failed to check export status: ${e?.message || e}`);
  }
}