surgio/build/provider/TrojanProvider.js

Generating rules for Surge, Clash, Quantumult like a PRO

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.getTrojanSubscription = void 0;
const assert_1 = __importDefault(require("assert"));
const zod_1 = require("zod");
const utils_1 = require("../utils");
const relayable_url_1 = __importDefault(require("../utils/relayable-url"));
const trojan_1 = require("../utils/trojan");
const Provider_1 = __importDefault(require("./Provider"));
class TrojanProvider extends Provider_1.default {
    #originalUrl;
    udpRelay;
    tls13;
    constructor(name, config) {
        super(name, config);
        const schema = zod_1.z.object({
            url: zod_1.z.string().url(),
            udpRelay: zod_1.z.boolean().optional(),
            tls13: zod_1.z.boolean().optional(),
        });
        const result = schema.safeParse(config);
        // istanbul ignore next
        if (!result.success) {
            throw new utils_1.SurgioError('TrojanProvider 配置校验失败', {
                cause: result.error,
                providerName: name,
            });
        }
        this.#originalUrl = result.data.url;
        this.udpRelay = result.data.udpRelay;
        this.tls13 = result.data.tls13;
        this.supportGetSubscriptionUserInfo = true;
    }
    // istanbul ignore next
    get url() {
        return (0, relayable_url_1.default)(this.#originalUrl, this.config.relayUrl);
    }
    getSubscriptionUserInfo = async (params = {}) => {
        const requestUserAgent = this.determineRequestUserAgent(params.requestUserAgent);
        const { subscriptionUserinfo } = await (0, exports.getTrojanSubscription)({
            url: this.url,
            udpRelay: this.udpRelay,
            tls13: this.tls13,
            requestUserAgent: requestUserAgent || this.config.requestUserAgent,
        });
        if (subscriptionUserinfo) {
            return subscriptionUserinfo;
        }
        return undefined;
    };
    getNodeList = async (params = {}) => {
        const requestUserAgent = this.determineRequestUserAgent(params.requestUserAgent);
        const { nodeList } = await (0, exports.getTrojanSubscription)({
            url: this.url,
            udpRelay: this.udpRelay,
            tls13: this.tls13,
            requestUserAgent: requestUserAgent,
        });
        if (this.config.hooks?.afterNodeListResponse) {
            const newList = await this.config.hooks.afterNodeListResponse(nodeList, params);
            if (newList) {
                return newList;
            }
        }
        return nodeList;
    };
}
exports.default = TrojanProvider;
/**
 * @see https://github.com/trojan-gfw/trojan-url/blob/master/trojan-url.py
 */
const getTrojanSubscription = async ({ url, udpRelay, tls13, requestUserAgent, }) => {
    (0, assert_1.default)(url, '未指定订阅地址 url');
    const response = await Provider_1.default.requestCacheableResource(url, {
        requestUserAgent: requestUserAgent || 'shadowrocket',
    });
    const config = (0, utils_1.fromBase64)(response.body);
    const nodeList = config
        .split('\n')
        .filter((item) => !!item && item.startsWith('trojan://'))
        .map((item) => {
        const nodeConfig = (0, trojan_1.parseTrojanUri)(item);
        return {
            ...nodeConfig,
            udpRelay,
            tls13,
        };
    });
    return {
        nodeList,
        subscriptionUserinfo: response.subscriptionUserinfo,
    };
};
exports.getTrojanSubscription = getTrojanSubscription;
//# sourceMappingURL=TrojanProvider.js.map