crawlab/frontend/crawlab-ui/src/utils/request.ts

export const getRequestBaseUrl = (): string => {
  return import.meta.env.VITE_APP_API_BASE_URL || 'http://localhost:8000';
};

export const getRequestBaseUrlWs = (): string => {
  if (import.meta.env.VITE_APP_API_BASE_URL.startsWith('http://')) {
    return (
      import.meta.env.VITE_APP_API_BASE_URL.replace(/https?/, 'ws') ||
      'ws://localhost:8000'
    );
  }
  return `ws://${window.location.hostname}${import.meta.env.VITE_APP_API_BASE_URL}`;
};

export const getEmptyResponseWithListData = <
  T = any,
>(): ResponseWithListData<T> => {
  return {
    total: 0,
    data: [] as T[],
  } as ResponseWithListData<T>;
};

export const downloadURI = (uri: string, name: string) => {
  const link = document.createElement('a');
  link.download = name;
  link.href = uri;
  link.click();
};

export const downloadData = (
  data: string | ArrayBuffer,
  name: string,
  type?: string
) => {
  let blobArr: any[] = [data];
  let options: BlobPropertyBag = {};
  if (type === 'json') {
    blobArr = [JSON.stringify(data)];
  } else if (type === 'csv') {
    const read = new Uint8Array([0xef, 0xbb, 0xbf]);
    blobArr = [read, data];
    options = { type: 'text/csv;charset=utf-8' };
  }
  const blob = new Blob(blobArr, options);
  const url = window.URL.createObjectURL(blob);
  downloadURI(url, name);
  window.URL.revokeObjectURL(url);
};