@softvisio/core/lib/net/proxy/upstream.js

Softisio core

import CacheLru from "#lib/cache/lru";
import Bucket from "./bucket.js";
import ProxyClient from "./client.js";

export default class extends ProxyClient {
    #buckets = new CacheLru( { "maxSize": 10_000 } );
    #upstreamUrl;

    async getRemoteAddress ( options ) {
        const proxy = await this.getProxy( options );

        if ( !proxy ) return;

        return proxy.getRemoteAddress();
    }

    async getPlaywrightProxy ( options ) {
        const proxy = await this.getProxy( options );

        if ( !proxy ) return;

        return proxy.getPlaywrightProxy();
    }

    // connect
    async connect ( url, options ) {
        if ( typeof url === "string" ) url = new URL( url );

        const proxy = await this.getProxy( options );

        if ( !proxy ) throw "Unable to get proxy";

        return proxy.connect( url, options );
    }

    // protected
    _updated () {
        super._updated();

        this.#buckets.clear();
    }

    _getBucket ( options = {} ) {
        const prefix = options.prefix ?? "";

        // merge options
        options = this._buildOptions( { ...this._options, ...options } );

        // build bucket key
        const id =
            prefix +
            ( options.session ||
                JSON.stringify( Object.keys( options )
                    .sort()
                    .reduce( ( result, key ) => {
                        result[ key ] = options[ key ];
                        return result;
                    }, {} ) ) );

        var bucket = this.#buckets.get( id );

        if ( bucket ) return bucket;

        bucket = new Bucket( id, options );

        this.#buckets.set( id, bucket );

        return bucket;
    }

    _buildProxy ( bucket ) {
        if ( !this.#upstreamUrl ) {
            const types = [];

            if ( this.isHttp ) types.push( "http" );
            if ( this.isSocks5 ) types.push( "socks5" );
            if ( this.isTls ) types.push( "tls" );

            const protocol = types.join( "+" );

            this.#upstreamUrl = new URL( protocol + "://local" );
        }

        const proxy = ProxyClient.new( this.#upstreamUrl, {
            "hostname": this.hostname,
            "port": this.port,
            "username": this.username,
            "password": this.password,
            "resolve": bucket.options.resolve,
        } );

        return proxy;
    }
}