mirror of
https://github.com/zjk2017/ArcadiaScriptPublic.git
synced 2025-12-17 15:34:35 +08:00
90 lines
3.1 KiB
TypeScript
90 lines
3.1 KiB
TypeScript
/*
|
||
* @Author: renxia
|
||
* @Date: 2023-11-28 11:09:04
|
||
* @LastEditors: renxia
|
||
* @LastEditTime: 2024-03-05 09:42:24
|
||
* @Description:
|
||
*/
|
||
import { LiteStorage, Request, color, NLogger } from '@lzwme/fe-utils';
|
||
import { existsSync } from 'node:fs';
|
||
import { homedir } from 'node:os';
|
||
import { resolve, sep } from 'node:path';
|
||
|
||
export const logger = new NLogger('[ql-scripts]');
|
||
|
||
export function findFile(filename: string, dirs = [process.cwd(), __dirname]) {
|
||
const dirList = new Set([...dirs, process.cwd(), __dirname, homedir()]);
|
||
|
||
for (let dir of dirList) {
|
||
while (dir.length > 3 && dir.includes(sep)) {
|
||
const fullpath = resolve(dir, filename);
|
||
if (existsSync(fullpath)) return fullpath;
|
||
dir = dir.substring(0, dir.lastIndexOf(sep));
|
||
}
|
||
}
|
||
|
||
return '';
|
||
}
|
||
|
||
/** 获取本地持久缓存对象 */
|
||
export function getCacheStorage<T extends object = Record<string, any>>(uuid: string) {
|
||
return getConfigStorage<T>(uuid, resolve(process.cwd(), 'cache/lzwme_ql_cache.json'));
|
||
}
|
||
|
||
/** 获取配置持久化对象 */
|
||
export function getConfigStorage<T extends object = Record<string, any>>(uuid: string, filepath = process.env.LZWME_QL_CONFIG_FILE) {
|
||
if (!uuid) throw Error('请指定 uuid');
|
||
|
||
if (!filepath) {
|
||
filepath = findFile('lzwme_ql_config.json5') || findFile('lzwme_ql_config.json') || 'lzwme_ql_config.json';
|
||
}
|
||
|
||
// if (!existsSync(filepath) && !filepath.includes('cache.json')) console.warn(`未发现配置文件。请创建：${color.cyan(filepath)}`);
|
||
|
||
return new LiteStorage<T>({ filepath: resolve(process.cwd(), filepath), uuid });
|
||
}
|
||
|
||
interface SendNotifyParams extends Record<string, any> {
|
||
hasError?: boolean | number;
|
||
notifyType?: 0 | 1 | 2;
|
||
isPrint?: boolean;
|
||
exit?: boolean;
|
||
}
|
||
|
||
export async function sendNotify(title: string, body: string, params: SendNotifyParams = {}, author = '\n本通知 By：lzwme/ql-scripts') {
|
||
const notifyFilePath = findFile('sendNotify.js');
|
||
const { hasError, notifyType = Number(process.env.LZWME_QL_NOTIFY_TYPE) || 1, isPrint, exit = true } = params;
|
||
let needNotify = true;
|
||
|
||
if (notifyType === 0) needNotify = false; // 0 - 禁用通知
|
||
else if (notifyType === 1) { // 1 - 有异常才通知
|
||
needNotify = hasError === true;
|
||
} else if (notifyType === 2) { // 2 - 全通知
|
||
needNotify = true;
|
||
} else needNotify = process.env.LZWME_QL_NOTIFY !== 'false';
|
||
|
||
if (isPrint !== false && (isPrint || !notifyFilePath || !needNotify)) console.log(`[needNotify=false][${title}]\n${body}`);
|
||
|
||
if (notifyFilePath && needNotify) {
|
||
await require(notifyFilePath).sendNotify(title, body, params, author);
|
||
}
|
||
|
||
exit && process.exit(hasError ? 1 : 0);
|
||
}
|
||
|
||
/** 根据指定的位置返回附近位置及经纬度列表 */
|
||
export async function getGeoByGD(address: string, AMAP_KEY: string) {
|
||
const req = new Request();
|
||
const { data } = await req.get<{
|
||
status: string;
|
||
geocodes: {
|
||
province: string;
|
||
city: string;
|
||
formatted_address: string;
|
||
location: string;
|
||
}[];
|
||
}>(`https://restapi.amap.com/v3/geocode/geo?key=${AMAP_KEY}&output=json&address=${address.trim()}`);
|
||
|
||
return data.geocodes;
|
||
}
|