UNPKG

windmill-utils-internal

Version:

Internal utility functions for Windmill

324 lines (323 loc) 11.7 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.request = exports.catchErrorCodes = exports.getResponseBody = exports.getResponseHeader = exports.sendRequest = exports.getRequestBody = exports.getHeaders = exports.resolve = exports.getFormData = exports.getQueryString = exports.base64 = exports.isFormData = exports.isBlob = exports.isStringWithValue = exports.isString = void 0; const ApiError_1 = require("./ApiError"); const CancelablePromise_1 = require("./CancelablePromise"); const isString = (value) => { return typeof value === 'string'; }; exports.isString = isString; const isStringWithValue = (value) => { return (0, exports.isString)(value) && value !== ''; }; exports.isStringWithValue = isStringWithValue; const isBlob = (value) => { return value instanceof Blob; }; exports.isBlob = isBlob; const isFormData = (value) => { return value instanceof FormData; }; exports.isFormData = isFormData; const base64 = (str) => { try { return btoa(str); } catch (err) { // @ts-ignore return Buffer.from(str).toString('base64'); } }; exports.base64 = base64; const getQueryString = (params) => { const qs = []; const append = (key, value) => { qs.push(`${encodeURIComponent(key)}=${encodeURIComponent(String(value))}`); }; const encodePair = (key, value) => { if (value === undefined || value === null) { return; } if (value instanceof Date) { append(key, value.toISOString()); } else if (Array.isArray(value)) { value.forEach(v => encodePair(key, v)); } else if (typeof value === 'object') { Object.entries(value).forEach(([k, v]) => encodePair(`${key}[${k}]`, v)); } else { append(key, value); } }; Object.entries(params).forEach(([key, value]) => encodePair(key, value)); return qs.length ? `?${qs.join('&')}` : ''; }; exports.getQueryString = getQueryString; const getUrl = (config, options) => { const encoder = config.ENCODE_PATH || encodeURI; const path = options.url .replace('{api-version}', config.VERSION) .replace(/{(.*?)}/g, (substring, group) => { if (options.path?.hasOwnProperty(group)) { return encoder(String(options.path[group])); } return substring; }); const url = config.BASE + path; return options.query ? url + (0, exports.getQueryString)(options.query) : url; }; const getFormData = (options) => { if (options.formData) { const formData = new FormData(); const process = (key, value) => { if ((0, exports.isString)(value) || (0, exports.isBlob)(value)) { formData.append(key, value); } else { formData.append(key, JSON.stringify(value)); } }; Object.entries(options.formData) .filter(([, value]) => value !== undefined && value !== null) .forEach(([key, value]) => { if (Array.isArray(value)) { value.forEach(v => process(key, v)); } else { process(key, value); } }); return formData; } return undefined; }; exports.getFormData = getFormData; const resolve = async (options, resolver) => { if (typeof resolver === 'function') { return resolver(options); } return resolver; }; exports.resolve = resolve; const getHeaders = async (config, options) => { const [token, username, password, additionalHeaders] = await Promise.all([ // @ts-ignore (0, exports.resolve)(options, config.TOKEN), // @ts-ignore (0, exports.resolve)(options, config.USERNAME), // @ts-ignore (0, exports.resolve)(options, config.PASSWORD), // @ts-ignore (0, exports.resolve)(options, config.HEADERS), ]); const headers = Object.entries({ Accept: 'application/json', ...additionalHeaders, ...options.headers, }) .filter(([, value]) => value !== undefined && value !== null) .reduce((headers, [key, value]) => ({ ...headers, [key]: String(value), }), {}); if ((0, exports.isStringWithValue)(token)) { headers['Authorization'] = `Bearer ${token}`; } if ((0, exports.isStringWithValue)(username) && (0, exports.isStringWithValue)(password)) { const credentials = (0, exports.base64)(`${username}:${password}`); headers['Authorization'] = `Basic ${credentials}`; } if (options.body !== undefined) { if (options.mediaType) { headers['Content-Type'] = options.mediaType; } else if ((0, exports.isBlob)(options.body)) { headers['Content-Type'] = options.body.type || 'application/octet-stream'; } else if ((0, exports.isString)(options.body)) { headers['Content-Type'] = 'text/plain'; } else if (!(0, exports.isFormData)(options.body)) { headers['Content-Type'] = 'application/json'; } } return new Headers(headers); }; exports.getHeaders = getHeaders; const getRequestBody = (options) => { if (options.body !== undefined) { if (options.mediaType?.includes('application/json') || options.mediaType?.includes('+json')) { return JSON.stringify(options.body); } else if ((0, exports.isString)(options.body) || (0, exports.isBlob)(options.body) || (0, exports.isFormData)(options.body)) { return options.body; } else { return JSON.stringify(options.body); } } return undefined; }; exports.getRequestBody = getRequestBody; const sendRequest = async (config, options, url, body, formData, headers, onCancel) => { const controller = new AbortController(); let request = { headers, body: body ?? formData, method: options.method, signal: controller.signal, }; if (config.WITH_CREDENTIALS) { request.credentials = config.CREDENTIALS; } for (const fn of config.interceptors.request._fns) { request = await fn(request); } onCancel(() => controller.abort()); return await fetch(url, request); }; exports.sendRequest = sendRequest; const getResponseHeader = (response, responseHeader) => { if (responseHeader) { const content = response.headers.get(responseHeader); if ((0, exports.isString)(content)) { return content; } } return undefined; }; exports.getResponseHeader = getResponseHeader; const getResponseBody = async (response) => { if (response.status !== 204) { try { const contentType = response.headers.get('Content-Type'); if (contentType) { const binaryTypes = ['application/octet-stream', 'application/pdf', 'application/zip', 'audio/', 'image/', 'video/']; if (contentType.includes('application/json') || contentType.includes('+json')) { return await response.json(); } else if (binaryTypes.some(type => contentType.includes(type))) { return await response.blob(); } else if (contentType.includes('multipart/form-data')) { return await response.formData(); } else if (contentType.includes('text/')) { return await response.text(); } } } catch (error) { console.error(error); } } return undefined; }; exports.getResponseBody = getResponseBody; const catchErrorCodes = (options, result) => { const errors = { 400: 'Bad Request', 401: 'Unauthorized', 402: 'Payment Required', 403: 'Forbidden', 404: 'Not Found', 405: 'Method Not Allowed', 406: 'Not Acceptable', 407: 'Proxy Authentication Required', 408: 'Request Timeout', 409: 'Conflict', 410: 'Gone', 411: 'Length Required', 412: 'Precondition Failed', 413: 'Payload Too Large', 414: 'URI Too Long', 415: 'Unsupported Media Type', 416: 'Range Not Satisfiable', 417: 'Expectation Failed', 418: 'Im a teapot', 421: 'Misdirected Request', 422: 'Unprocessable Content', 423: 'Locked', 424: 'Failed Dependency', 425: 'Too Early', 426: 'Upgrade Required', 428: 'Precondition Required', 429: 'Too Many Requests', 431: 'Request Header Fields Too Large', 451: 'Unavailable For Legal Reasons', 500: 'Internal Server Error', 501: 'Not Implemented', 502: 'Bad Gateway', 503: 'Service Unavailable', 504: 'Gateway Timeout', 505: 'HTTP Version Not Supported', 506: 'Variant Also Negotiates', 507: 'Insufficient Storage', 508: 'Loop Detected', 510: 'Not Extended', 511: 'Network Authentication Required', ...options.errors, }; const error = errors[result.status]; if (error) { throw new ApiError_1.ApiError(options, result, error); } if (!result.ok) { const errorStatus = result.status ?? 'unknown'; const errorStatusText = result.statusText ?? 'unknown'; const errorBody = (() => { try { return JSON.stringify(result.body, null, 2); } catch (e) { return undefined; } })(); throw new ApiError_1.ApiError(options, result, `Generic Error: status: ${errorStatus}; status text: ${errorStatusText}; body: ${errorBody}`); } }; exports.catchErrorCodes = catchErrorCodes; /** * Request method * @param config The OpenAPI configuration object * @param options The request options from the service * @returns CancelablePromise<T> * @throws ApiError */ const request = (config, options) => { return new CancelablePromise_1.CancelablePromise(async (resolve, reject, onCancel) => { try { const url = getUrl(config, options); const formData = (0, exports.getFormData)(options); const body = (0, exports.getRequestBody)(options); const headers = await (0, exports.getHeaders)(config, options); if (!onCancel.isCancelled) { let response = await (0, exports.sendRequest)(config, options, url, body, formData, headers, onCancel); for (const fn of config.interceptors.response._fns) { response = await fn(response); } const responseBody = await (0, exports.getResponseBody)(response); const responseHeader = (0, exports.getResponseHeader)(response, options.responseHeader); let transformedBody = responseBody; if (options.responseTransformer && response.ok) { transformedBody = await options.responseTransformer(responseBody); } const result = { url, ok: response.ok, status: response.status, statusText: response.statusText, body: responseHeader ?? transformedBody, }; (0, exports.catchErrorCodes)(options, result); resolve(result.body); } } catch (error) { reject(error); } }); }; exports.request = request;