@push.rocks/smartproxy 16.0.2 → 16.0.3

package/ts/proxies/smart-proxy/smart-proxy.ts

@@ -6,7 +6,7 @@ import { SecurityManager } from './security-manager.js';
 import { TlsManager } from './tls-manager.js';
 import { NetworkProxyBridge } from './network-proxy-bridge.js';
 import { TimeoutManager } from './timeout-manager.js';
-// import { PortRangeManager } from './port-range-manager.js';
+import { PortManager } from './port-manager.js';
 import { RouteManager } from './route-manager.js';
 import { RouteConnectionHandler } from './route-connection-handler.js';
 
@@ -39,7 +39,8 @@ import type { IRouteConfig } from './models/route-types.js';
  * - Advanced options (timeout, headers, etc.)
  */
 export class SmartProxy extends plugins.EventEmitter {
-  private netServers: plugins.net.Server[] = [];
+  // Port manager handles dynamic listener management
+  private portManager: PortManager;
   private connectionLogger: NodeJS.Timeout | null = null;
   private isShuttingDown: boolean = false;
   
@@ -49,8 +50,7 @@ export class SmartProxy extends plugins.EventEmitter {
   private tlsManager: TlsManager;
   private networkProxyBridge: NetworkProxyBridge;
   private timeoutManager: TimeoutManager;
-  // private portRangeManager: PortRangeManager;
-  private routeManager: RouteManager;
+  public routeManager: RouteManager; // Made public for route management
   private routeConnectionHandler: RouteConnectionHandler;
   
   // Port80Handler for ACME certificate management
@@ -151,8 +151,6 @@ export class SmartProxy extends plugins.EventEmitter {
     // Create the route manager
     this.routeManager = new RouteManager(this.settings);
 
-    // Create port range manager
-    // this.portRangeManager = new PortRangeManager(this.settings);
     
     // Create other required components
     this.tlsManager = new TlsManager(this.settings);
@@ -168,6 +166,9 @@ export class SmartProxy extends plugins.EventEmitter {
       this.timeoutManager,
       this.routeManager
     );
+
+    // Initialize port manager
+    this.portManager = new PortManager(this.settings, this.routeConnectionHandler);
   }
   
   /**
@@ -271,33 +272,8 @@ export class SmartProxy extends plugins.EventEmitter {
     // Get listening ports from RouteManager
     const listeningPorts = this.routeManager.getListeningPorts();
 
-    // Create servers for each port
-    for (const port of listeningPorts) {
-      const server = plugins.net.createServer((socket) => {
-        // Check if shutting down
-        if (this.isShuttingDown) {
-          socket.end();
-          socket.destroy();
-          return;
-        }
-        
-        // Delegate to route connection handler
-        this.routeConnectionHandler.handleConnection(socket);
-      }).on('error', (err: Error) => {
-        console.log(`Server Error on port ${port}: ${err.message}`);
-      });
-      
-      server.listen(port, () => {
-        const isNetworkProxyPort = this.settings.useNetworkProxy?.includes(port);
-        console.log(
-          `SmartProxy -> OK: Now listening on port ${port}${
-            isNetworkProxyPort ? ' (NetworkProxy forwarding enabled)' : ''
-          }`
-        );
-      });
-      
-      this.netServers.push(server);
-    }
+    // Start port listeners using the PortManager
+    await this.portManager.addPorts(listeningPorts);
 
     // Set up periodic connection logging and inactivity checks
     this.connectionLogger = setInterval(() => {
@@ -383,6 +359,7 @@ export class SmartProxy extends plugins.EventEmitter {
   public async stop() {
     console.log('SmartProxy shutting down...');
     this.isShuttingDown = true;
+    this.portManager.setShuttingDown(true);
     
     // Stop CertProvisioner if active
     if (this.certProvisioner) {
@@ -401,31 +378,14 @@ export class SmartProxy extends plugins.EventEmitter {
       }
     }
 
-    // Stop accepting new connections
-    const closeServerPromises: Promise<void>[] = this.netServers.map(
-      (server) =>
-        new Promise<void>((resolve) => {
-          if (!server.listening) {
-            resolve();
-            return;
-          }
-          server.close((err) => {
-            if (err) {
-              console.log(`Error closing server: ${err.message}`);
-            }
-            resolve();
-          });
-        })
-    );
-
     // Stop the connection logger
     if (this.connectionLogger) {
       clearInterval(this.connectionLogger);
       this.connectionLogger = null;
     }
 
-    // Wait for servers to close
-    await Promise.all(closeServerPromises);
+    // Stop all port listeners
+    await this.portManager.closeAll();
     console.log('All servers closed. Cleaning up active connections...');
 
     // Clean up all active connections
@@ -434,8 +394,6 @@ export class SmartProxy extends plugins.EventEmitter {
     // Stop NetworkProxy
     await this.networkProxyBridge.stop();
 
-    // Clear all servers
-    this.netServers = [];
 
     console.log('SmartProxy shutdown complete.');
   }
@@ -479,6 +437,12 @@ export class SmartProxy extends plugins.EventEmitter {
     // Update routes in RouteManager
     this.routeManager.updateRoutes(newRoutes);
 
+    // Get the new set of required ports
+    const requiredPorts = this.routeManager.getListeningPorts();
+
+    // Update port listeners to match the new configuration
+    await this.portManager.updatePorts(requiredPorts);
+
     // If NetworkProxy is initialized, resync the configurations
     if (this.networkProxyBridge.getNetworkProxy()) {
       await this.networkProxyBridge.syncRoutesToNetworkProxy(newRoutes);
@@ -609,6 +573,41 @@ export class SmartProxy extends plugins.EventEmitter {
     return true;
   }
   
+  /**
+   * Add a new listening port without changing the route configuration
+   *
+   * This allows you to add a port listener without updating routes.
+   * Useful for preparing to listen on a port before adding routes for it.
+   *
+   * @param port The port to start listening on
+   * @returns Promise that resolves when the port is listening
+   */
+  public async addListeningPort(port: number): Promise<void> {
+    return this.portManager.addPort(port);
+  }
+
+  /**
+   * Stop listening on a specific port without changing the route configuration
+   *
+   * This allows you to stop a port listener without updating routes.
+   * Useful for temporary maintenance or port changes.
+   *
+   * @param port The port to stop listening on
+   * @returns Promise that resolves when the port is closed
+   */
+  public async removeListeningPort(port: number): Promise<void> {
+    return this.portManager.removePort(port);
+  }
+
+  /**
+   * Get a list of all ports currently being listened on
+   *
+   * @returns Array of port numbers
+   */
+  public getListeningPorts(): number[] {
+    return this.portManager.getListeningPorts();
+  }
+
   /**
    * Get statistics about current connections
    */
@@ -638,7 +637,9 @@ export class SmartProxy extends plugins.EventEmitter {
       terminationStats,
       acmeEnabled: !!this.port80Handler,
       port80HandlerPort: this.port80Handler ? this.settings.acme?.port : null,
-      routes: this.routeManager.getListeningPorts().length
+      routes: this.routeManager.getListeningPorts().length,
+      listeningPorts: this.portManager.getListeningPorts(),
+      activePorts: this.portManager.getListeningPorts().length
     };
   }
   

package/ts/proxies/smart-proxy/utils/route-helpers.ts

@@ -14,9 +14,11 @@
  * - Static file server routes (createStaticFileRoute)
  * - API routes (createApiRoute)
  * - WebSocket routes (createWebSocketRoute)
+ * - Port mapping routes (createPortMappingRoute, createOffsetPortMappingRoute)
+ * - Dynamic routing (createDynamicRoute, createSmartLoadBalancer)
  */
 
-import type { IRouteConfig, IRouteMatch, IRouteAction, IRouteTarget, TPortRange } from '../models/route-types.js';
+import type { IRouteConfig, IRouteMatch, IRouteAction, IRouteTarget, TPortRange, IRouteContext } from '../models/route-types.js';
 
 /**
  * Create an HTTP-only route configuration
@@ -452,4 +454,168 @@ export function createWebSocketRoute(
     priority: options.priority || 100, // Higher priority for WebSocket routes
     ...options
   };
+}
+
+/**
+ * Create a helper function that applies a port offset
+ * @param offset The offset to apply to the matched port
+ * @returns A function that adds the offset to the matched port
+ */
+export function createPortOffset(offset: number): (context: IRouteContext) => number {
+  return (context: IRouteContext) => context.port + offset;
+}
+
+/**
+ * Create a port mapping route with context-based port function
+ * @param options Port mapping route options
+ * @returns Route configuration object
+ */
+export function createPortMappingRoute(options: {
+  sourcePortRange: TPortRange;
+  targetHost: string | string[] | ((context: IRouteContext) => string | string[]);
+  portMapper: (context: IRouteContext) => number;
+  name?: string;
+  domains?: string | string[];
+  priority?: number;
+  [key: string]: any;
+}): IRouteConfig {
+  // Create route match
+  const match: IRouteMatch = {
+    ports: options.sourcePortRange,
+    domains: options.domains
+  };
+
+  // Create route action
+  const action: IRouteAction = {
+    type: 'forward',
+    target: {
+      host: options.targetHost,
+      port: options.portMapper
+    }
+  };
+
+  // Create the route config
+  return {
+    match,
+    action,
+    name: options.name || `Port Mapping Route for ${options.domains || 'all domains'}`,
+    priority: options.priority,
+    ...options
+  };
+}
+
+/**
+ * Create a simple offset port mapping route
+ * @param options Offset port mapping route options
+ * @returns Route configuration object
+ */
+export function createOffsetPortMappingRoute(options: {
+  ports: TPortRange;
+  targetHost: string | string[];
+  offset: number;
+  name?: string;
+  domains?: string | string[];
+  priority?: number;
+  [key: string]: any;
+}): IRouteConfig {
+  return createPortMappingRoute({
+    sourcePortRange: options.ports,
+    targetHost: options.targetHost,
+    portMapper: (context) => context.port + options.offset,
+    name: options.name || `Offset Mapping (${options.offset > 0 ? '+' : ''}${options.offset}) for ${options.domains || 'all domains'}`,
+    domains: options.domains,
+    priority: options.priority,
+    ...options
+  });
+}
+
+/**
+ * Create a dynamic route with context-based host and port mapping
+ * @param options Dynamic route options
+ * @returns Route configuration object
+ */
+export function createDynamicRoute(options: {
+  ports: TPortRange;
+  targetHost: (context: IRouteContext) => string | string[];
+  portMapper: (context: IRouteContext) => number;
+  name?: string;
+  domains?: string | string[];
+  path?: string;
+  clientIp?: string[];
+  priority?: number;
+  [key: string]: any;
+}): IRouteConfig {
+  // Create route match
+  const match: IRouteMatch = {
+    ports: options.ports,
+    domains: options.domains,
+    path: options.path,
+    clientIp: options.clientIp
+  };
+
+  // Create route action
+  const action: IRouteAction = {
+    type: 'forward',
+    target: {
+      host: options.targetHost,
+      port: options.portMapper
+    }
+  };
+
+  // Create the route config
+  return {
+    match,
+    action,
+    name: options.name || `Dynamic Route for ${options.domains || 'all domains'}`,
+    priority: options.priority,
+    ...options
+  };
+}
+
+/**
+ * Create a smart load balancer with dynamic domain-based backend selection
+ * @param options Smart load balancer options
+ * @returns Route configuration object
+ */
+export function createSmartLoadBalancer(options: {
+  ports: TPortRange;
+  domainTargets: Record<string, string | string[]>;
+  portMapper: (context: IRouteContext) => number;
+  name?: string;
+  defaultTarget?: string | string[];
+  priority?: number;
+  [key: string]: any;
+}): IRouteConfig {
+  // Extract all domain keys to create the match criteria
+  const domains = Object.keys(options.domainTargets);
+
+  // Create the smart host selector function
+  const hostSelector = (context: IRouteContext) => {
+    const domain = context.domain || '';
+    return options.domainTargets[domain] || options.defaultTarget || 'localhost';
+  };
+
+  // Create route match
+  const match: IRouteMatch = {
+    ports: options.ports,
+    domains
+  };
+
+  // Create route action
+  const action: IRouteAction = {
+    type: 'forward',
+    target: {
+      host: hostSelector,
+      port: options.portMapper
+    }
+  };
+
+  // Create the route config
+  return {
+    match,
+    action,
+    name: options.name || `Smart Load Balancer for ${domains.join(', ')}`,
+    priority: options.priority,
+    ...options
+  };
 }

package/ts/proxies/smart-proxy/utils/route-validators.ts

@@ -9,14 +9,24 @@ import type { IRouteConfig, IRouteMatch, IRouteAction, TPortRange } from '../mod
 
 /**
  * Validates a port range or port number
- * @param port Port number or port range
+ * @param port Port number, port range, or port function
  * @returns True if valid, false otherwise
  */
-export function isValidPort(port: TPortRange): boolean {
+export function isValidPort(port: any): boolean {
   if (typeof port === 'number') {
     return port > 0 && port < 65536; // Valid port range is 1-65535
   } else if (Array.isArray(port)) {
-    return port.every(p => typeof p === 'number' && p > 0 && p < 65536);
+    return port.every(p =>
+      (typeof p === 'number' && p > 0 && p < 65536) ||
+      (typeof p === 'object' && 'from' in p && 'to' in p &&
+       p.from > 0 && p.from < 65536 && p.to > 0 && p.to < 65536)
+    );
+  } else if (typeof port === 'function') {
+    // For function-based ports, we can't validate the result at config time
+    // so we just check that it's a function
+    return true;
+  } else if (typeof port === 'object' && 'from' in port && 'to' in port) {
+    return port.from > 0 && port.from < 65536 && port.to > 0 && port.to < 65536;
   }
   return false;
 }
@@ -100,11 +110,20 @@ export function validateRouteAction(action: IRouteAction): { valid: boolean; err
       // Validate target host
       if (!action.target.host) {
         errors.push('Target host is required');
+      } else if (typeof action.target.host !== 'string' &&
+                !Array.isArray(action.target.host) &&
+                typeof action.target.host !== 'function') {
+        errors.push('Target host must be a string, array of strings, or function');
       }
 
       // Validate target port
-      if (!action.target.port || !isValidPort(action.target.port)) {
-        errors.push('Valid target port is required');
+      if (action.target.port === undefined) {
+        errors.push('Target port is required');
+      } else if (typeof action.target.port !== 'number' &&
+                typeof action.target.port !== 'function') {
+        errors.push('Target port must be a number or a function');
+      } else if (typeof action.target.port === 'number' && !isValidPort(action.target.port)) {
+        errors.push('Target port must be between 1 and 65535');
       }
     }