arn-browser 0.0.2 → 0.0.4

package/src/utility/proxy-utility/proxy-helper.js

@@ -0,0 +1,245 @@
+import { setTimeout as sleep } from "timers/promises";
+import randomstring from "randomstring";
+import fs from "fs";
+import path from "path";
+import { pathToFileURL } from "url";
+import { getMultiloginToken } from "../multilogin_token_manager.js";
+// Assuming fetchAwsProxy is in the same directory or adjust path accordingly
+import { getActiveProxyWithStartStop as fetchAwsProxy } from "./custom-proxy.js";
+
+// Global configuration cache
+let config = null;
+
+// ==========================================
+// SECTION 1: CONFIGURATION & UTILITIES
+// ==========================================
+
+/**
+ * Dynamically loads 'browser_automation_env.js' from the user's project root.
+ */
+async function loadUserConfig() {
+	if (config) return config;
+
+	try {
+		const projectRoot = process.cwd();
+		const envPath = path.join(projectRoot, "utility", "browser_automation_env.js");
+
+		if (!fs.existsSync(envPath)) {
+			throw new Error(`Could not find configuration file at: ${envPath}`);
+		}
+
+		const envUrl = pathToFileURL(envPath).href;
+		const userEnv = await import(envUrl);
+
+		// Validate required exports
+		const requiredKeys = [
+			"arn",
+			"query",
+			"PACKETSTREAM_USER",
+			"PACKETSTREAM_PASS_KEY",
+			"PACKETSTREAM_HOST",
+			"PACKETSTREAM_PORT",
+		];
+
+		const missing = requiredKeys.filter((key) => !userEnv[key]);
+
+		if (missing.length > 0) {
+			throw new Error(`[Proxy Manager] 'browser_automation_env.js' is missing exports: ${missing.join(", ")}`);
+		}
+
+		config = userEnv;
+		return config;
+	} catch (error) {
+		console.error("❌ Config Error:", error.message);
+		throw error;
+	}
+}
+
+// ==========================================
+// SECTION 2: MULTILOGIN PROXY
+// ==========================================
+
+/**
+ * Fetches and parses a Multilogin proxy using Native Fetch.
+ */
+export async function get_multilogin_proxy({
+	country = "us",
+	sessionType = "sticky",
+	protocol = "http",
+	region = "",
+	city = "",
+	IPTTL = 1800,
+	strictMode = false,
+} = {}) {
+	try {
+		// 1. Get Token
+		const token = await getMultiloginToken();
+
+		// 2. Prepare Data
+		const data = { country, sessionType, protocol, region, city, IPTTL, count: 1 };
+		Object.keys(data).forEach((k) => (data[k] === "" || data[k] === 0 || data[k] == null) && delete data[k]);
+
+		// 3. Setup Timeout
+		const controller = new AbortController();
+		const timeoutId = setTimeout(() => controller.abort(), 20000);
+
+		try {
+			const response = await fetch("https://profile-proxy.multilogin.com/v1/proxy/connection_url", {
+				method: "POST",
+				headers: {
+					Authorization: `Bearer ${token}`,
+					"X-Strict-Mode": strictMode ? "true" : "false",
+					"Content-Type": "application/json",
+				},
+				body: JSON.stringify(data),
+				signal: controller.signal,
+			});
+
+			if (!response.ok) {
+				const errText = await response.text();
+				throw new Error(`HTTP ${response.status}: ${errText}`);
+			}
+
+			const json = await response.json();
+
+			let proxyStr = json?.data;
+			if (Array.isArray(proxyStr)) proxyStr = proxyStr[0];
+
+			if (!proxyStr || typeof proxyStr !== "string") {
+				console.error("Multilogin proxy API: Unexpected format", json);
+				return null;
+			}
+
+			// Format: host:port:user:pass
+			const parts = proxyStr.split(":");
+			if (parts.length < 4) {
+				console.error("Unexpected Multilogin proxy string:", proxyStr);
+				return null;
+			}
+
+			const [host, port, user, ...passParts] = parts;
+			const pass = passParts.join(":");
+
+			const temp_proxy = {
+				type: protocol,
+				host,
+				port: parseInt(port, 10),
+				user,
+				pass,
+			};
+
+			console.log("Multilogin Proxy:", temp_proxy);
+			return temp_proxy;
+		} finally {
+			clearTimeout(timeoutId);
+		}
+	} catch (error) {
+		console.error("Failed to get Multilogin proxy:", error.message);
+		return null;
+	}
+}
+
+// ==========================================
+// SECTION 3: PACKETSTREAM PROXY
+// ==========================================
+
+/**
+ * Generates a PacketStream proxy object with a random session ID using credentials from config.
+ */
+export async function get_packetstream_proxy() {
+	// 1. Load config dynamically
+	const { PACKETSTREAM_USER, PACKETSTREAM_PASS_KEY, PACKETSTREAM_HOST, PACKETSTREAM_PORT } = await loadUserConfig();
+
+	// 2. Generate Random Session ID
+	const proxy_session_id = randomstring.generate({
+		length: 8,
+		charset: "alphanumeric",
+		capitalization: "both",
+		readable: false,
+	});
+
+	// 3. Construct the proxy object
+	const temp_proxy = {
+		type: "http",
+		host: PACKETSTREAM_HOST,
+		port: Number(PACKETSTREAM_PORT),
+		user: PACKETSTREAM_USER,
+		pass: `${PACKETSTREAM_PASS_KEY}-${proxy_session_id}`,
+	};
+
+	console.log("PacketStream Proxy:", temp_proxy);
+	return temp_proxy;
+}
+
+// ==========================================
+// SECTION 4: DATABASE PROXY (X_ACCOUNTS)
+// ==========================================
+
+export async function get_x_proxy({ provider }) {
+	const { arn, query } = await loadUserConfig();
+
+	const { data: [proxyInstance] = [], error: getProxy_error } = await arn.single(
+		query("x_accounts_proxy")
+			.update({
+				total_used: query.raw("total_used + 1"),
+				last_used_time: new Date().toISOString(),
+			})
+			.where("id", function () {
+				let subquery = this.select("id").from("x_accounts_proxy").where("status", 1);
+
+				if (provider) {
+					subquery = subquery.where("provider", provider);
+				}
+
+				subquery = subquery
+					.andWhere(
+						query.raw("COALESCE(last_used_time, '1970-01-01T00:00:00.000Z') < ?", [
+							new Date(Date.now() - 60 * 60 * 1000).toISOString(),
+						])
+					)
+					.orderBy("total_used", "asc")
+					.limit(1);
+				return subquery;
+			})
+			.returning("*")
+	);
+
+	if (getProxy_error || !proxyInstance) {
+		console.error(getProxy_error || "No proxy instances found");
+		await sleep(30000);
+		return null;
+	}
+
+	return proxyInstance;
+}
+
+// ==========================================
+// SECTION 5: AWS PROXY ROTATION
+// ==========================================
+
+export function fetchNextProxy(proxies) {
+	if (proxies.isFetching) return;
+
+	proxies.isFetching = true;
+
+	fetchAwsProxy({
+		instance_name: proxies.proxyInstance,
+	})
+		.then((newProxyIp) => {
+			if (newProxyIp) {
+				proxies.proxy = {
+					host: newProxyIp,
+					port: 9002,
+				};
+				proxies.proxyUsed = 0;
+				console.log(`Successfully prepared next proxy: ${newProxyIp}`);
+			} else {
+				console.log("Next proxy fetch returned null");
+			}
+			proxies.isFetching = false;
+		})
+		.catch((error) => {
+			console.error(`Error fetching next proxy ${proxies.proxyInstance}:`, error.message);
+			proxies.isFetching = false;
+		});
+}