mirror of
https://github.com/sub-store-org/Sub-Store.git
synced 2025-08-10 00:52:40 +00:00
419 lines
14 KiB
JavaScript
419 lines
14 KiB
JavaScript
import { ProxyUtils } from '@/core/proxy-utils';
|
||
import { RuleUtils } from '@/core/rule-utils';
|
||
import download from '@/utils/download';
|
||
import Gist from '@/utils/gist';
|
||
import $ from '@/core/app';
|
||
|
||
import {
|
||
SUBS_KEY,
|
||
ARTIFACTS_KEY,
|
||
ARTIFACT_REPOSITORY_KEY,
|
||
COLLECTIONS_KEY,
|
||
RULES_KEY,
|
||
SETTINGS_KEY,
|
||
} from './constants';
|
||
|
||
export default function register($app) {
|
||
// Initialization
|
||
if (!$.read(ARTIFACTS_KEY)) $.write({}, ARTIFACTS_KEY);
|
||
|
||
// RESTful APIs
|
||
$app.route('/api/artifacts').get(getAllArtifacts).post(createArtifact);
|
||
|
||
$app.route('/api/artifact/:name')
|
||
.get(getArtifact)
|
||
.patch(updateArtifact)
|
||
.delete(deleteArtifact);
|
||
|
||
// sync all artifacts
|
||
$app.get('/api/cron/sync-artifacts', cronSyncArtifacts);
|
||
}
|
||
|
||
function getAllArtifacts(req, res) {
|
||
const allArtifacts = $.read(ARTIFACTS_KEY);
|
||
res.json({
|
||
status: 'success',
|
||
data: allArtifacts,
|
||
});
|
||
}
|
||
|
||
async function getArtifact(req, res) {
|
||
let { name } = req.params;
|
||
name = decodeURIComponent(name);
|
||
const action = req.query.action;
|
||
const allArtifacts = $.read(ARTIFACTS_KEY);
|
||
const artifact = allArtifacts[name];
|
||
|
||
if (artifact) {
|
||
if (action) {
|
||
let item;
|
||
switch (artifact.type) {
|
||
case 'subscription':
|
||
item = $.read(SUBS_KEY)[artifact.source];
|
||
break;
|
||
case 'collection':
|
||
item = $.read(COLLECTIONS_KEY)[artifact.source];
|
||
break;
|
||
case 'rule':
|
||
item = $.read(RULES_KEY)[artifact.source];
|
||
break;
|
||
}
|
||
const output = await produceArtifact({
|
||
type: artifact.type,
|
||
item,
|
||
platform: artifact.platform,
|
||
});
|
||
if (action === 'preview') {
|
||
res.send(output);
|
||
} else if (action === 'sync') {
|
||
$.info(`正在上传配置：${artifact.name}\n>>>`);
|
||
console.log(JSON.stringify(artifact, null, 2));
|
||
try {
|
||
const resp = await syncArtifact({
|
||
[encodeURIComponent(artifact.name)]: {
|
||
content: output,
|
||
},
|
||
});
|
||
artifact.updated = new Date().getTime();
|
||
const body = JSON.parse(resp.body);
|
||
artifact.url = body.files[
|
||
encodeURIComponent(artifact.name)
|
||
].raw_url.replace(/\/raw\/[^/]*\/(.*)/, '/raw/$1');
|
||
$.write(allArtifacts, ARTIFACTS_KEY);
|
||
res.json({
|
||
status: 'success',
|
||
});
|
||
} catch (err) {
|
||
res.status(500).json({
|
||
status: 'failed',
|
||
message: err,
|
||
});
|
||
}
|
||
}
|
||
} else {
|
||
res.json({
|
||
status: 'success',
|
||
data: artifact,
|
||
});
|
||
}
|
||
} else {
|
||
res.status(404).json({
|
||
status: 'failed',
|
||
message: '未找到对应的配置！',
|
||
});
|
||
}
|
||
}
|
||
|
||
function createArtifact(req, res) {
|
||
const artifact = req.body;
|
||
$.info(`正在创建远程配置：${artifact.name}`);
|
||
const allArtifacts = $.read(ARTIFACTS_KEY);
|
||
if (allArtifacts[artifact.name]) {
|
||
res.status(500).json({
|
||
status: 'failed',
|
||
message: `远程配置${artifact.name}已存在！`,
|
||
});
|
||
} else {
|
||
allArtifacts[artifact.name] = artifact;
|
||
$.write(allArtifacts, ARTIFACTS_KEY);
|
||
res.status(201).json({
|
||
status: 'success',
|
||
data: artifact,
|
||
});
|
||
}
|
||
}
|
||
|
||
function updateArtifact(req, res) {
|
||
const allArtifacts = $.read(ARTIFACTS_KEY);
|
||
let oldName = req.params.name;
|
||
oldName = decodeURIComponent(oldName);
|
||
const artifact = allArtifacts[oldName];
|
||
if (artifact) {
|
||
$.info(`正在更新远程配置：${artifact.name}`);
|
||
const newArtifact = req.body;
|
||
if (
|
||
typeof newArtifact.name !== 'undefined' &&
|
||
!/^[\w-_.]*$/.test(newArtifact.name)
|
||
) {
|
||
res.status(500).json({
|
||
status: 'failed',
|
||
message: `远程配置名称 ${newArtifact.name} 中含有非法字符！名称中只能包含英文字母、数字、下划线、横杠。`,
|
||
});
|
||
} else {
|
||
const merged = {
|
||
...artifact,
|
||
...newArtifact,
|
||
};
|
||
allArtifacts[merged.name] = merged;
|
||
if (merged.name !== oldName) delete allArtifacts[oldName];
|
||
$.write(allArtifacts, ARTIFACTS_KEY);
|
||
res.json({
|
||
status: 'success',
|
||
data: merged,
|
||
});
|
||
}
|
||
} else {
|
||
res.status(404).json({
|
||
status: 'failed',
|
||
message: '未找到对应的远程配置！',
|
||
});
|
||
}
|
||
}
|
||
|
||
async function deleteArtifact(req, res) {
|
||
let { name } = req.params;
|
||
name = decodeURIComponent(name);
|
||
$.info(`正在删除远程配置：${name}`);
|
||
const allArtifacts = $.read(ARTIFACTS_KEY);
|
||
try {
|
||
const artifact = allArtifacts[name];
|
||
if (!artifact) throw new Error(`远程配置：${name}不存在！`);
|
||
if (artifact.updated) {
|
||
// delete gist
|
||
const files = {};
|
||
files[encodeURIComponent(artifact.name)] = {
|
||
content: '',
|
||
};
|
||
await syncArtifact(files);
|
||
}
|
||
// delete local cache
|
||
delete allArtifacts[name];
|
||
$.write(allArtifacts, ARTIFACTS_KEY);
|
||
res.json({
|
||
status: 'success',
|
||
});
|
||
} catch (err) {
|
||
$.error(`无法删除远程配置：${name}，原因：${err}`);
|
||
res.status(500).json({
|
||
status: 'failed',
|
||
message: `无法删除远程配置：${name}, 原因：${err}`,
|
||
});
|
||
}
|
||
}
|
||
|
||
async function cronSyncArtifacts(_, res) {
|
||
$.info('开始同步所有远程配置...');
|
||
const allArtifacts = $.read(ARTIFACTS_KEY);
|
||
const files = {};
|
||
|
||
try {
|
||
await Promise.all(
|
||
Object.values(allArtifacts).map(async (artifact) => {
|
||
if (artifact.sync) {
|
||
$.info(`正在同步云配置：${artifact.name}...`);
|
||
let item;
|
||
switch (artifact.type) {
|
||
case 'subscription':
|
||
item = $.read(SUBS_KEY)[artifact.source];
|
||
break;
|
||
case 'collection':
|
||
item = $.read(COLLECTIONS_KEY)[artifact.source];
|
||
break;
|
||
case 'rule':
|
||
item = $.read(RULES_KEY)[artifact.source];
|
||
break;
|
||
}
|
||
const output = await produceArtifact({
|
||
type: artifact.type,
|
||
item,
|
||
platform: artifact.platform,
|
||
});
|
||
|
||
files[artifact.name] = {
|
||
content: output,
|
||
};
|
||
}
|
||
}),
|
||
);
|
||
|
||
const resp = await syncArtifact(files);
|
||
const body = JSON.parse(resp.body);
|
||
|
||
for (const artifact of Object.values(allArtifacts)) {
|
||
artifact.updated = new Date().getTime();
|
||
// extract real url from gist
|
||
artifact.url = body.files[artifact.name].raw_url.replace(
|
||
/\/raw\/[^/]*\/(.*)/,
|
||
'/raw/$1',
|
||
);
|
||
}
|
||
|
||
$.write(allArtifacts, ARTIFACTS_KEY);
|
||
$.info('全部订阅同步成功！');
|
||
res.status(200).end();
|
||
} catch (err) {
|
||
res.status(500).json({
|
||
error: err,
|
||
});
|
||
$.info(`同步订阅失败，原因：${err}`);
|
||
}
|
||
}
|
||
|
||
async function syncArtifact(files) {
|
||
const { gistToken } = $.read(SETTINGS_KEY);
|
||
if (!gistToken) {
|
||
return Promise.reject('未设置Gist Token！');
|
||
}
|
||
const manager = new Gist({
|
||
token: gistToken,
|
||
key: ARTIFACT_REPOSITORY_KEY,
|
||
});
|
||
return manager.upload(files);
|
||
}
|
||
|
||
async function produceArtifact({ type, item, platform, noProcessor }) {
|
||
platform = platform || 'JSON';
|
||
noProcessor = noProcessor || false;
|
||
|
||
if (type === 'subscription') {
|
||
const sub = item;
|
||
let raw;
|
||
if (sub.source === 'local') {
|
||
raw = sub.content;
|
||
} else {
|
||
raw = await download(sub.url, sub.ua);
|
||
}
|
||
// parse proxies
|
||
let proxies = ProxyUtils.parse(raw);
|
||
if (!noProcessor) {
|
||
console.log('Processing proxy...');
|
||
// apply processors
|
||
proxies = await ProxyUtils.process(
|
||
proxies,
|
||
sub.process || [],
|
||
platform,
|
||
);
|
||
}
|
||
// check duplicate
|
||
const exist = {};
|
||
for (const proxy of proxies) {
|
||
if (exist[proxy.name]) {
|
||
$.notify(
|
||
'🌍 『 𝑺𝒖𝒃-𝑺𝒕𝒐𝒓𝒆 』',
|
||
'⚠️ 订阅包含重复节点！',
|
||
'请仔细检测配置！',
|
||
{
|
||
'media-url':
|
||
'https://cdn3.iconfinder.com/data/icons/seo-outline-1/512/25_code_program_programming_develop_bug_search_developer-512.png',
|
||
},
|
||
);
|
||
break;
|
||
}
|
||
exist[proxy.name] = true;
|
||
}
|
||
// produce
|
||
return ProxyUtils.produce(proxies, platform);
|
||
} else if (type === 'collection') {
|
||
const allSubs = $.read(SUBS_KEY);
|
||
const collection = item;
|
||
const subnames = collection['subscriptions'];
|
||
const results = {};
|
||
let processed = 0;
|
||
|
||
await Promise.all(
|
||
subnames.map(async (name) => {
|
||
const sub = allSubs[name];
|
||
try {
|
||
$.info(`正在处理子订阅：${sub.name}...`);
|
||
let raw;
|
||
if (sub.source === 'local') {
|
||
raw = sub.content;
|
||
} else {
|
||
raw = await download(sub.url, sub.ua);
|
||
}
|
||
// parse proxies
|
||
let currentProxies = ProxyUtils.parse(raw);
|
||
if (!noProcessor) {
|
||
// apply processors
|
||
currentProxies = await ProxyUtils.process(
|
||
currentProxies,
|
||
sub.process || [],
|
||
platform,
|
||
);
|
||
}
|
||
results[name] = currentProxies;
|
||
processed++;
|
||
$.info(
|
||
`✅ 子订阅：${sub.name}加载成功，进度--${
|
||
100 * (processed / subnames.length).toFixed(1)
|
||
}% `,
|
||
);
|
||
} catch (err) {
|
||
processed++;
|
||
$.error(
|
||
`❌ 处理组合订阅中的子订阅: ${
|
||
sub.name
|
||
}时出现错误：${err}，该订阅已被跳过！进度--${
|
||
100 * (processed / subnames.length).toFixed(1)
|
||
}%`,
|
||
);
|
||
}
|
||
}),
|
||
);
|
||
|
||
// merge proxies with the original order
|
||
let proxies = Array.prototype.concat.apply(
|
||
[],
|
||
subnames.map((name) => results[name]),
|
||
);
|
||
|
||
if (!noProcessor) {
|
||
// apply own processors
|
||
proxies = await ProxyUtils.process(
|
||
proxies,
|
||
collection.process || [],
|
||
platform,
|
||
);
|
||
}
|
||
if (proxies.length === 0) {
|
||
throw new Error(`组合订阅中不含有效节点！`);
|
||
}
|
||
// check duplicate
|
||
const exist = {};
|
||
for (const proxy of proxies) {
|
||
if (exist[proxy.name]) {
|
||
$.notify(
|
||
'🌍 『 𝑺𝒖𝒃-𝑺𝒕𝒐𝒓𝒆 』',
|
||
'⚠️ 订阅包含重复节点！',
|
||
'请仔细检测配置！',
|
||
{
|
||
'media-url':
|
||
'https://cdn3.iconfinder.com/data/icons/seo-outline-1/512/25_code_program_programming_develop_bug_search_developer-512.png',
|
||
},
|
||
);
|
||
break;
|
||
}
|
||
exist[proxy.name] = true;
|
||
}
|
||
return ProxyUtils.produce(proxies, platform);
|
||
} else if (type === 'rule') {
|
||
const rule = item;
|
||
let rules = [];
|
||
for (let i = 0; i < rule.urls.length; i++) {
|
||
const url = rule.urls[i];
|
||
$.info(
|
||
`正在处理URL：${url}，进度--${
|
||
100 * ((i + 1) / rule.urls.length).toFixed(1)
|
||
}% `,
|
||
);
|
||
try {
|
||
const { body } = await download(url);
|
||
const currentRules = RuleUtils.parse(body);
|
||
rules = rules.concat(currentRules);
|
||
} catch (err) {
|
||
$.error(
|
||
`处理分流订阅中的URL: ${url}时出现错误：${err}! 该订阅已被跳过。`,
|
||
);
|
||
}
|
||
}
|
||
// remove duplicates
|
||
rules = await RuleUtils.process(rules, [
|
||
{ type: 'Remove Duplicate Filter' },
|
||
]);
|
||
// produce output
|
||
return RuleUtils.produce(rules, platform);
|
||
}
|
||
}
|
||
|
||
export { syncArtifact, produceArtifact };
|