Files
hoppscotch/packages/hoppscotch-app/helpers/strategies/AxiosStrategy.js

95 lines
2.3 KiB
JavaScript

import axios from "axios"
import { v4 } from "uuid"
import { decodeB64StringToArrayBuffer } from "../utils/b64"
import { settingsStore } from "~/newstore/settings"
import { JsonFormattedError } from "~/helpers/utils/JsonFormattedError"
let cancelSource = axios.CancelToken.source()
export const cancelRunningAxiosRequest = () => {
cancelSource.cancel()
// Create a new cancel token
cancelSource = axios.CancelToken.source()
}
const axiosWithProxy = async (req) => {
try {
let proxyReqPayload = {
...req,
wantsBinary: true,
}
const headers = {}
if (req.data instanceof FormData) {
const key = `proxyRequestData-${v4()}` // generate UniqueKey
headers["multipart-part-key"] = key
const formData = proxyReqPayload.data
proxyReqPayload.data = "" // discard formData
formData.append(key, JSON.stringify(proxyReqPayload)) // append axiosRequest to form
proxyReqPayload = formData
}
const { data } = await axios.post(
settingsStore.value.PROXY_URL || "https://proxy.hoppscotch.io",
proxyReqPayload,
{
headers,
cancelToken: cancelSource.token,
}
)
if (!data.success) {
throw new Error(data.data.message || "Proxy Error")
}
if (data.isBinary) {
data.data = decodeB64StringToArrayBuffer(data.data)
}
return data
} catch (e) {
// Check if the throw is due to a cancellation
if (axios.isCancel(e)) {
// eslint-disable-next-line no-throw-literal
throw "cancellation"
} else {
throw e
}
}
}
const axiosWithoutProxy = async (req, _store) => {
try {
const res = await axios({
...req,
cancelToken: (cancelSource && cancelSource.token) || "",
responseType: "arraybuffer",
})
return res
} catch (e) {
if (axios.isCancel(e)) {
// eslint-disable-next-line no-throw-literal
throw "cancellation"
} else if (e.response?.data) {
throw new JsonFormattedError(
JSON.parse(Buffer.from(e.response.data, "base64").toString("utf8"))
)
} else {
throw e
}
}
}
const axiosStrategy = (req) => {
if (settingsStore.value.PROXY_ENABLED) {
return axiosWithProxy(req)
}
return axiosWithoutProxy(req)
}
export const testables = {
cancelSource,
}
export default axiosStrategy