Knowledge_Base_Assistant/src/utils/request.ts

/* global
  IRequestSuite
*/
import type { AxiosInstance } from 'axios'
import axios from 'axios'
import Cookie from 'js-cookie'

import Router from '@/router'
import { currentHost } from '@/utils/location'


// redirect error
function errorRedirect(url: string) {
  Router.push(`/${ url }`)
}
// code Message
const codeMessage: {
  [key: number]: string
} = {
  200: '服务器成功返回请求的数据。',
  201: '新建或修改数据成功。',
  202: '一个请求已经进入后台排队（异步任务）。',
  204: '删除数据成功。',
  206: '进行范围请求成功。',
  400: '发出的请求有错误，服务器没有进行新建或修改数据的操作。',
  401: '用户没有权限（令牌、用户名、密码错误）。',
  403: '用户得到授权，但是访问是被禁止的。',
  404: '发出的请求针对的是不存在的记录，服务器没有进行操作。',
  405: '请求不允许。',
  406: '请求的格式不可得。',
  410: '请求的资源被永久删除，且不会再得到的。',
  422: '当创建一个对象时，发生一个验证错误。',
  500: '服务器发生错误，请检查服务器。',
  502: '网关错误。',
  503: '服务不可用，服务器暂时过载或维护。',
  504: '网关超时。'
}
// 创建axios实例
const service: AxiosInstance = axios.create({
  // api 的 base_url
  baseURL: currentHost.baseApi,
  // 请求超时时间
  timeout: 200000
})

// request拦截器
service.interceptors.request.use(
  request => {
    const token: string | undefined = Cookie.get('token')

    // Conversion of hump nomenclature

    /**
     * 让每个请求携带自定义 token
     * 请根据实际情况自行修改
     */
    if (request.url === '/login') {
      return request
    }
    request.headers['Content-Type'] = 'multipart/form-data'
    request.headers!.Authorization = token as string
    return request
  },
  error => {
    return Promise.reject(error)
  }
)

// respone拦截器
service.interceptors.response.use(
  response => {
    /**
     * response data
     *   {
     *     data: {},
     *     msg: "",
     *     error: 0      0 success | 1 error | 5000 failed | HTTP code
     *  }
     */

    const data: any = response.data
    const msg: string = data.msg || ''
    if (msg.indexOf('user not log in') !== -1 && data.error === -1) {
      // TODO 写死的  之后要根据语言跳转
      errorRedirect('login')
      return
    }
    if (response.config.autoDownLoadFile === undefined || response.config.autoDownLoadFile) {
      Promise.resolve().then(() => {
        useResHeadersAPI(response.headers, data)
      })
    }

    if (
      response.request.responseType === 'blob' &&
  /json$/gi.test(response.headers['content-type'])
    ) {
      return new Promise(resolve => {
        const reader = new FileReader()
        reader.readAsText(<Blob>response.data)

        reader.onload = () => {
          if (!reader.result || typeof reader.result !== 'string') return resolve(response.data)

          response.data = JSON.parse(reader.result)
          resolve(response.data)
        }

      })
    } else if (data instanceof Blob) {
      return {
        data,
        msg: '',
        error: 0
      }
    }

    if (data.code && data.data) {
      return {
        data: data.data,
        error: data.code === 200 ? 0 : -1,
        msg: 'ok'
      }
    }


    if (!data.data && !data.msg && !data.error) {
      return {
        data,
        error: 0,
        msg: 'ok'
      }
    }


    if (data.msg === null) {
      data.msg = 'Unknown error'
    }
    return data
  },
  error => {
    /**
     * 某些特定的接口 404 500 需要跳转
     * 在需要重定向的接口中传入 redirect字段  值为要跳转的路由
     *   redirect之后  调用接口的地方会继续执行
     *   因为此时 response error
     *   所以需要前端返回一个前端构造好的数据结构 避免前端业务部分逻辑出错
     * 不重定向的接口则不需要传
     */
    if (error.config.redirect) {
      errorRedirect(error.config.redirect)
    }
    if (error.response) {
      return {
        data: {},
        error: error.response.status,
        msg: codeMessage[error.response.status] || error.response.data.message
      }
    } else {
      // 某些特定的接口 failed 需要跳转
      console.log(error)
      return {
        data: {},
        error: 5000,
        aborted: error.config.signal?.aborted,
        msg: '服务请求不可用，请重试或检查您的网络。'
      }
    }
  }
)

export function sleep(time = 0) {
  return new Promise((resolve) => {
    setTimeout(() => {
      resolve({})
    }, time)
  })
}

function extractFileNameFromContentDispositionHeader(value: string) {
  const patterns = [
    /filename\*=[^']+'\w*'"([^"]+)";?/i,
    /filename\*=[^']+'\w*'([^;]+);?/i,
    /filename="([^;]*);?"/i,
    /filename=([^;]*);?/i
  ]

  let responseFilename: any = null
  patterns.some(regex => {
    responseFilename = regex.exec(value)
    return responseFilename !== null
  })

  if (responseFilename !== null && responseFilename.length > 1) {
    try {
      return decodeURIComponent(responseFilename[1])
    } catch (e) {
      console.error(e)
    }
  }

  return null
}

export function downloadFile(boldData: BlobPart, filename = '预设文件名称', type: any) {
  const blob = boldData instanceof Blob
    ? boldData
    : new Blob([boldData], {
      type
    })
  const url = window.URL.createObjectURL(blob)

  const link = document.createElement('a')
  link.style.display = 'none'
  link.href = url
  link.download = filename
  document.body.appendChild(link)

  link.click()

  document.body.removeChild(link)
}

export function useResHeadersAPI(headers: any, resData: any) {
  const disposition = headers['content-disposition']
  if (disposition) {
    let filename: string | null = ''

    filename = extractFileNameFromContentDispositionHeader(disposition)
    if (filename) {
      downloadFile(resData, filename, headers['content-type'])
    }
  }
}

const requestSuite: IRequestSuite = {
  get(uri, params, config) {
    return service.get(uri, {
      params,
      ...config
    })
  },
  post(uri, data, config) {
    return service.post(uri, data, config)
  },
  put(uri, data, config) {
    return service.put(uri, data, config)
  },
  patch(uri, data, config) {
    return service.patch(uri, data, config)
  },
  delete(uri, config) {
    return service.delete(uri, config)
  }
}

export default requestSuite