proxy-cmd/bin/proxy-url.js

Toggle proxy environments (http_proxy and https_proxy) in command line.

#!/usr/bin/env node
"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.proxyUrl = void 0;
const child_process_1 = require("child_process");
const path_1 = require("path");
const fs_extra_1 = require("fs-extra");
const os_1 = __importDefault(require("os"));
const chalk_1 = __importDefault(require("chalk"));
/**
 * Set or show proxy url
 */
const proxyUrl = (url = '') => {
    const proxyCmd = (0, path_1.join)(os_1.default.homedir(), 'proxy-cmd');
    const pxUrlFile = (0, path_1.join)(proxyCmd, '.proxy-url');
    (0, fs_extra_1.ensureFileSync)(pxUrlFile);
    // If no url is provided, read from command line arguments
    if (!url) {
        let [x, cmd, u, u2] = process.argv;
        x = (0, path_1.basename)(x);
        if (x === 'proxy-cmd' || x === 'proxy')
            u2 = u;
        url = u2;
    }
    // Set proxy url
    if (url && url.match(/^https?:\/\/[\d.:]+$/gm)) {
        (0, fs_extra_1.writeFileSync)(pxUrlFile, url);
        // Windows
        if (os_1.default.platform() === 'win32') {
            // Set PROXY_URL
            try {
                (0, child_process_1.execSync)(`setx PROXY_URL "${url}" /M`, { stdio: 'inherit' });
            }
            catch { }
        }
        // MacOS or Linux
        else {
            const wtAliasInRcFile = (rcFile) => {
                (0, fs_extra_1.ensureFileSync)(rcFile);
                let rcTpl = (0, fs_extra_1.readFileSync)(rcFile, 'utf-8');
                // Set PROXY_URL
                if (rcTpl.match(/^(export\s*)?PROXY_URL\s*=.*$/igm)) {
                    rcTpl = rcTpl.replace(/^(export\s*)?PROXY_URL\s*=.*$/igm, `export PROXY_URL='${url}'`);
                }
                else {
                    if (rcTpl && !rcTpl.startsWith('\n'))
                        rcTpl = `${rcTpl}\n`;
                    rcTpl = `${rcTpl}export PROXY_URL='${url}'`;
                }
                if (rcTpl.match(/^\s*#?\s*export http_proxy/igm)) {
                    rcTpl = rcTpl.replace(/^\s*(#?\s*)export http_proxy\s*=\s*.*/img, `$1export http_proxy="${url}"`);
                }
                if (rcTpl.match(/^\s*#?\s*export https_proxy/igm)) {
                    rcTpl = rcTpl.replace(/^\s*(#?\s*)export https_proxy\s*=\s*.*/img, `$1export https_proxy="${url}"`);
                }
                (0, fs_extra_1.writeFileSync)(rcFile, rcTpl);
                try {
                    (0, child_process_1.execSync)(`type srouce > /dev/null 2>&1 && source ${rcFile}`, { stdio: 'inherit' });
                }
                catch { }
            };
            // MacOS
            if (os_1.default.platform() === 'darwin') {
                const zshrc = (0, path_1.join)(os_1.default.homedir(), '.zshrc');
                const bashrc = (0, path_1.join)(os_1.default.homedir(), '.bashrc');
                wtAliasInRcFile(zshrc);
                wtAliasInRcFile(bashrc);
            }
            // Linux
            else {
                const bashrc = (0, path_1.join)(os_1.default.homedir(), '.bashrc');
                wtAliasInRcFile(bashrc);
            }
        }
        console.log(`Proxy url set to: ${chalk_1.default.cyan(url)}`);
    }
    // Show current proxy url
    else {
        const purl = (0, fs_extra_1.readFileSync)(pxUrlFile, 'utf-8');
        console.log(`Current proxy url is: ${chalk_1.default.cyan(purl)}`);
    }
};
exports.proxyUrl = proxyUrl;