#!/usr/bin/env node
import yargs from 'yargs';
import { hideBin } from 'yargs/helpers';
import { pipenet } from './pipenet.js';
import { createServer } from './server/index.js';
async function runClient(opts) {
    let headers;
    if (opts.headers) {
        try {
            headers = JSON.parse(opts.headers);
        }
        catch (err) {
            console.error('Invalid headers JSON:', err);
            process.exit(1);
        }
    }
    const tunnel = await pipenet({
        allowInvalidCert: opts['allow-invalid-cert'],
        headers,
        host: opts.host,
        localCa: opts['local-ca'],
        localCert: opts['local-cert'],
        localHost: opts['local-host'],
        localHttps: opts['local-https'],
        localKey: opts['local-key'],
        port: opts.port,
        subdomain: opts.subdomain,
    });
    console.log('your url is: %s', tunnel.url);
    tunnel.on('error', (err) => {
        console.error('tunnel error:', err.message);
    });
    tunnel.on('close', () => {
        console.log('tunnel closed');
        process.exit(0);
    });
    if (opts['print-requests']) {
        tunnel.on('request', (info) => {
            console.log('%s %s', info.method, info.path);
        });
    }
    process.on('SIGINT', () => {
        tunnel.close();
    });
}
async function runServer(opts) {
    const server = createServer({
        domains: opts.domain,
        landing: opts.landing,
        maxTcpSockets: opts['max-sockets'],
        secure: opts.secure,
        tunnelPort: opts['tunnel-port'],
    });
    // Start the tunnel server if configured
    if (server.tunnelServer && opts['tunnel-port']) {
        await server.tunnelServer.listen(opts['tunnel-port'], opts.address);
        console.log('tunnel server listening on port %d', opts['tunnel-port']);
    }
    const listenCallback = () => {
        console.log('pipenet server listening on port %d', opts.port);
        if (opts.address) {
            console.log('bound to address: %s', opts.address);
        }
        if (opts.domain && opts.domain.length > 0) {
            console.log('tunnel domain(s): %s', opts.domain.join(', '));
        }
    };
    if (opts.address) {
        server.listen(opts.port, opts.address, listenCallback);
    }
    else {
        server.listen(opts.port, listenCallback);
    }
    process.on('SIGINT', () => {
        console.log('shutting down server...');
        if (server.tunnelServer) {
            server.tunnelServer.close();
        }
        server.close(() => {
            process.exit(0);
        });
    });
}
yargs(hideBin(process.argv))
    .usage('Usage: pipenet <command> [options]')
    .env(true)
    .demandCommand(1, 'You must specify a command: client or server')
    .command('client', 'Start a tunnel client', (yargs) => {
    return yargs
        .option('port', {
        alias: 'p',
        demandOption: true,
        describe: 'Internal HTTP server port',
        type: 'number',
    })
        .option('host', {
        alias: 'h',
        default: 'https://pipenet.dev',
        describe: 'Upstream server providing forwarding',
        type: 'string',
    })
        .option('subdomain', {
        alias: 's',
        describe: 'Request this subdomain',
        type: 'string',
    })
        .option('local-host', {
        alias: 'l',
        default: 'localhost',
        describe: 'Tunnel traffic to this host instead of localhost',
        type: 'string',
    })
        .option('local-https', {
        describe: 'Tunnel traffic to a local HTTPS server',
        type: 'boolean',
    })
        .option('local-cert', {
        describe: 'Path to certificate PEM file for local HTTPS server',
        type: 'string',
    })
        .option('local-key', {
        describe: 'Path to certificate key file for local HTTPS server',
        type: 'string',
    })
        .option('local-ca', {
        describe: 'Path to certificate authority file for self-signed certificates',
        type: 'string',
    })
        .option('allow-invalid-cert', {
        describe: 'Disable certificate checks for your local HTTPS server',
        type: 'boolean',
    })
        .option('print-requests', {
        describe: 'Print basic request info',
        type: 'boolean',
    })
        .option('headers', {
        describe: 'Custom headers to send with tunnel connection (JSON format)',
        type: 'string',
    });
}, (argv) => {
    runClient(argv).catch((err) => {
        console.error(err);
        process.exit(1);
    });
})
    .command('server', 'Start a tunnel server', (yargs) => {
    return yargs
        .option('port', {
        alias: 'p',
        default: 3000,
        describe: 'Port for the server to listen on',
        type: 'number',
    })
        .option('address', {
        alias: 'a',
        describe: 'Address to bind the server to (e.g., 0.0.0.0)',
        type: 'string',
    })
        .option('domain', {
        alias: 'd',
        array: true,
        describe: 'Custom domain(s) for the tunnel server (can be specified multiple times)',
        type: 'string',
    })
        .option('secure', {
        default: false,
        describe: 'Require HTTPS connections',
        type: 'boolean',
    })
        .option('landing', {
        describe: 'URL to redirect root requests to',
        type: 'string',
    })
        .option('max-sockets', {
        default: 10,
        describe: 'Maximum number of TCP sockets per client',
        type: 'number',
    })
        .option('tunnel-port', {
        alias: 't',
        describe: 'Port for tunnel connections (enables single-port mode for cloud deployments)',
        type: 'number',
    });
}, (argv) => {
    runServer(argv);
})
    .help('help')
    .version()
    .parse();
//# sourceMappingURL=cli.js.map