mirror of
https://github.com/shanmiteko/LotteryAutoScript.git
synced 2026-06-04 21:01:17 +08:00
106 lines
3.3 KiB
JavaScript
106 lines
3.3 KiB
JavaScript
//@ts-check
|
|
/**
|
|
* @typedef {import("http").IncomingHttpHeaders} HttpHeaders 头部信息
|
|
*
|
|
* @typedef {object} Respond 返回的响应
|
|
* @property {HttpHeaders} headers 响应头
|
|
* @property {string} body 响应体
|
|
*
|
|
* @callback SuccessCb 成功回调
|
|
* @param {Respond} res
|
|
* @returns {void}
|
|
* @callback FailureCb 失败回调
|
|
* @param {string} err
|
|
* @returns {void}
|
|
*
|
|
* @typedef {object} RequestOptions Http请求选项
|
|
* @property {string} method 请求方法
|
|
* @property {string} url 完整链接
|
|
* @property {object} [query] 查询选项
|
|
* @property {object} [contents] 内容
|
|
* @property {HttpHeaders} [headers] 请求头
|
|
* @property {SuccessCb} [success] 成功回调
|
|
* @property {FailureCb} [failure] 失败回调
|
|
*/
|
|
|
|
const { request } = require('http');
|
|
const { stringify } = require('querystring');
|
|
/**
|
|
* 处理请求体
|
|
* 默认url编码字符串
|
|
* @private
|
|
* @param {HttpHeaders} headers 请求的内容格式
|
|
* @param {object} contents 请求体
|
|
* @returns {string} 格式化字符串
|
|
*/
|
|
function formatContents(headers, contents) {
|
|
const contentstype = headers['content-type'];
|
|
if (typeof contentstype === 'undefined') return '';
|
|
if (/application\/x-www-form-urlencoded/i.test(contentstype)) return stringify(contents);
|
|
if (/application\/json/i.test(contentstype)) return JSON.stringify(contents);
|
|
return '';
|
|
}
|
|
/**
|
|
* 测试url是否符合要求
|
|
* @private
|
|
* @param {string} url
|
|
* @returns {boolean}
|
|
*/
|
|
function isHttpR(url) {
|
|
const urlreg = /^([hH][tT]{2}[pP]:\/\/|[hH][tT]{2}[pP][sS]:\/\/)(([A-Za-z0-9-~]+).)+([A-Za-z0-9-~/])+$/;
|
|
return urlreg.test(url);
|
|
}
|
|
/**
|
|
* @private
|
|
* @param {import("http").IncomingMessage} res
|
|
* @param {SuccessCb} success
|
|
* @param {FailureCb} failure
|
|
* @returns {void}
|
|
*/
|
|
function resDataHandler(res, success, failure) {
|
|
let protodata = '';
|
|
const { statusCode, headers } = res;
|
|
if (statusCode < 400) {
|
|
res.setEncoding('utf8')
|
|
.on('data', chunk => { protodata += chunk })
|
|
.on('end', () => success({ headers: headers, body: protodata }))
|
|
} else {
|
|
failure(`网络错误: 错误码${statusCode}`);
|
|
}
|
|
}
|
|
/**
|
|
* @description 简化HTTP请求
|
|
* @param {RequestOptions} detail
|
|
* @returns {void}
|
|
*/
|
|
const HttpRequest = detail => {
|
|
const { method, url, query = {}, contents = {}, headers = {}, success, failure = e => console.warn(e) } = detail;
|
|
if (!isHttpR(url)) return failure('非http请求');
|
|
const
|
|
thisURL = new URL(url),
|
|
content = formatContents(headers, contents);
|
|
let options = {
|
|
method,
|
|
host: thisURL.host,
|
|
path: thisURL.pathname,
|
|
headers,
|
|
};
|
|
switch (method) {
|
|
case 'GET':
|
|
if (Object.keys(query).length !== 0) options.path += '?' + stringify(query);
|
|
break;
|
|
case 'POST':
|
|
options.headers['content-length'] = Buffer.byteLength(content, 'utf-8').toString();
|
|
break;
|
|
default:
|
|
return failure(`未实现的请求方法: ${method}`);
|
|
}
|
|
const req = request(options, res => resDataHandler(res, success, failure));
|
|
if (method === 'POST') req.write(content);
|
|
req.setTimeout(5000, req.abort)
|
|
.on('error', err => failure(`请求失败: ${err.message}`))
|
|
.end();
|
|
}
|
|
module.exports = {
|
|
HttpRequest
|
|
} |