asockslib 0.1.0__py3-none-any.whl

asockslib/geo_picker/__init__.py

@@ -0,0 +1,30 @@
+"""Interactive geo-data selection via fuzzy search.
+
+Re-exports all public types and the :class:`GeoPicker` class
+so that ``from asockslib.geo_picker import GeoPicker`` continues
+to work after the module was split into a package.
+"""
+
+from __future__ import annotations
+
+from asockslib.geo_picker._types import (
+    PickedASN,
+    PickedCity,
+    PickedConnectionType,
+    PickedCountry,
+    PickedProxyType,
+    PickedServerPortType,
+    PickedState,
+)
+from asockslib.geo_picker.picker import GeoPicker
+
+__all__ = [
+    "GeoPicker",
+    "PickedASN",
+    "PickedCity",
+    "PickedConnectionType",
+    "PickedCountry",
+    "PickedProxyType",
+    "PickedServerPortType",
+    "PickedState",
+]

asockslib/geo_picker/_completer.py

@@ -0,0 +1,90 @@
+"""Fuzzy-match completer and validators for prompt_toolkit.
+
+Provides the autocompletion and input validation used by
+:class:`~asockslib.geo_picker.GeoPicker` interactive prompts.
+"""
+
+from __future__ import annotations
+
+import re
+from typing import TYPE_CHECKING
+
+from prompt_toolkit.completion import Completer, Completion
+from prompt_toolkit.styles import Style as PTStyle
+from prompt_toolkit.validation import ValidationError, Validator
+
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+
+    from prompt_toolkit.completion import CompleteEvent
+    from prompt_toolkit.document import Document
+
+_RE_GEO_NAME = re.compile(r"^[\w\s\-.'()/]+$", re.UNICODE)
+
+
+class FuzzyMatchCompleter(Completer):
+    """Completer that shows all choices on empty input.
+
+    On non-empty input performs case-insensitive substring matching.
+    """
+
+    def __init__(self, choices: list[str], ignore_case: bool = True) -> None:
+        self.choices = choices
+        self.ignore_case = ignore_case
+
+    def get_completions(
+        self, document: Document, complete_event: CompleteEvent
+    ) -> Iterable[Completion]:
+        """Yield completions matching current input."""
+        text = document.text_before_cursor
+        if self.ignore_case:
+            text = text.lower()
+
+        for choice in self.choices:
+            compare = choice.lower() if self.ignore_case else choice
+            if text in compare:
+                yield Completion(
+                    choice,
+                    start_position=-len(document.text_before_cursor),
+                )
+
+
+def make_choice_validator(
+    choices: list[str],
+    error_msg: str,
+    error_chars: str,
+) -> Validator:
+    """Create a choice validator for prompt_toolkit.
+
+    Only values present in *choices* (case-insensitive) are accepted.
+    """
+    lower_lookup: dict[str, str] = {c.strip().lower(): c for c in choices}
+
+    class _ChoiceValidator(Validator):
+        def validate(self, document: Document) -> None:
+            t = document.text.strip()
+            if t in choices:
+                return
+            if t.lower() in lower_lookup:
+                return
+            if t and not _RE_GEO_NAME.match(t):
+                raise ValidationError(message=error_chars)
+            raise ValidationError(message=error_msg)
+
+    return _ChoiceValidator()
+
+
+def pt_style() -> PTStyle:
+    """Color scheme for prompt_toolkit prompts."""
+    return PTStyle.from_dict(
+        {
+            "qmark": "fg:ansicyan bold",
+            "question": "fg:ansiwhite bold",
+            "answer": "fg:ansigreen bold",
+            "completion-menu": "bg:ansiblack fg:ansiwhite",
+            "completion-menu.completion": "",
+            "completion-menu.completion.current": "bg:ansicyan fg:ansiblack",
+            "scrollbar.background": "bg:ansibrightblack",
+            "scrollbar.button": "bg:ansicyan",
+        }
+    )

asockslib/geo_picker/_messages.py

@@ -0,0 +1,89 @@
+"""UI messages and proxy output templates for the geo-picker."""
+
+from __future__ import annotations
+
+# ── Messages ──────────────────────────────────────────────────────────────── #
+
+MESSAGES: dict[str, str] = {
+    "pick_country": "🌍 Select country",
+    "pick_state": "📍 Select state/region",
+    "pick_city": "🏙  Select city",
+    "pick_asn": "🔗 Select ASN provider",
+    "skip": "⏭  (skip)",
+    "no_states": "No states available for this country.",
+    "no_cities": "No cities available.",
+    "no_asns": "No ASN providers available for this country.",
+    "cities_suffix": "cities",
+    "params": "Parameters",
+    "invalid_choice": "⚠ Value not in the list. Use arrows or type to filter.",
+    "invalid_country_code": "⚠ Invalid country code (must be 2 letters, e.g. US).",
+    "invalid_chars": "⚠ Input contains invalid characters.",
+    "pick_connection_type": "🔌 Connection type",
+    "pick_proxy_type": "🛡  Proxy type",
+    "pick_server_port_type": "🖥  Server port type",
+    "pick_traffic_limit": "📊 Traffic limit (GB)",
+    "pick_action": "🚀 What do you want to do?",
+    "action_create": "Create proxies — get working proxies quickly",
+    "action_best": "Find best proxies — create, benchmark, keep the fastest",
+    "pick_count": "🔢 How many proxies?",
+    "pick_keep": "🏆 How many best to keep?",
+    "pick_name": "📝 Port name (optional, Enter to skip)",
+    "pick_format": "📄 Export format",
+    "pick_output": "💾 Output file (optional, Enter to skip)",
+    "pick_timeout": "⏱  Benchmark timeout (seconds)",
+    "pick_concurrency": "⚡ Benchmark concurrency",
+    "pick_proxy_template": "📋 Output template (proxy format)",
+}
+
+
+def get_message(key: str) -> str:
+    """Get a message string by key."""
+    return MESSAGES.get(key, key)
+
+
+# ── Proxy output templates ────────────────────────────────────────────────── #
+
+PROXY_TEMPLATES: list[tuple[str, str]] = [
+    ("{protocol}://{login}:{password}@{ip}:{port}", "Standard URL — protocol://login:pass@ip:port"),
+    ("http://{login}:{password}@{ip}:{port}", "HTTP URL — http://login:pass@ip:port"),
+    ("socks5://{login}:{password}@{ip}:{port}", "SOCKS5 URL — socks5://login:pass@ip:port"),
+    ("{ip}:{port}:{login}:{password}", "ip:port:login:password"),
+    ("{protocol}://{ip}:{port}:{login}:{password}", "protocol://ip:port:login:password"),
+    (
+        "{protocol}://{login}:{password}@{ip}:{port}[{refresh_link}]",
+        "URL with refresh link — protocol://login:pass@ip:port[refresh]",
+    ),
+    (
+        "{protocol}://{login}:{password}@{ip}:{port}:{name}[{refresh_link}]",
+        "Full URL — login:pass@ip:port:name[refresh]",
+    ),
+    (
+        "http://{login}:{password}@{ip}:{port}:{name}[{refresh_link}]",
+        "HTTP full — http://login:pass@ip:port:name[refresh]",
+    ),
+    (
+        "socks5://{login}:{password}@{ip}:{port}:{name}[{refresh_link}]",
+        "SOCKS5 full — socks5://login:pass@ip:port:name[refresh]",
+    ),
+    ("{ip}:{port}:{login}:{password}:{refresh_link}", "ip:port:login:password:refresh_link"),
+    ("{ip}:{port}:{login}:{password}:{name}", "ip:port:login:password:name"),
+    ("{name}:{ip}:{port}:{login}:{password}", "name:ip:port:login:password"),
+    ("{ip}:{port}:{login}:{password}|{refresh_link}", "ip:port:login:password|refresh_link"),
+    ("HTTP|{ip}|{port}|{login}|{password}", "HTTP|ip|port|login|password (pipe separated)"),
+    (
+        "{protocol}://{ip}:{port}:{login}:{password}:{refresh_link}",
+        "protocol://ip:port:login:password:refresh",
+    ),
+    (
+        "{protocol}://{ip}:{port}:{login}:{password}[{refresh_link}]{{name}}",
+        "protocol://ip:port:login:password[refresh]{name}",
+    ),
+    (
+        "{protocol}:{ip}:{port}:{login}:{password}",
+        "protocol:ip:port:login:password (no slashes)",
+    ),
+    (
+        "curl -x http://{login}:{password}@{ip}:{port} https://i.pn",
+        "curl command — ready-to-use curl",
+    ),
+]

asockslib/geo_picker/_types.py

@@ -0,0 +1,89 @@
+"""Data types for the geo-picker module.
+
+Frozen dataclasses representing user selections, plus static
+choice data for connection types, proxy types, and server port types.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+# ── Picked result types ───────────────────────────────────────────────────── #
+
+
+@dataclass(frozen=True, slots=True)
+class PickedCountry:
+    """Selected country."""
+
+    code: str
+    name: str
+    id: int
+
+
+@dataclass(frozen=True, slots=True)
+class PickedState:
+    """Selected state/region."""
+
+    name: str
+    id: int
+
+
+@dataclass(frozen=True, slots=True)
+class PickedCity:
+    """Selected city."""
+
+    name: str
+    id: int
+
+
+@dataclass(frozen=True, slots=True)
+class PickedASN:
+    """Selected ASN."""
+
+    number: int
+    name: str
+
+
+@dataclass(frozen=True, slots=True)
+class PickedConnectionType:
+    """Selected connection type."""
+
+    id: int
+    label: str
+
+
+@dataclass(frozen=True, slots=True)
+class PickedProxyType:
+    """Selected proxy type."""
+
+    id: int
+    label: str
+
+
+@dataclass(frozen=True, slots=True)
+class PickedServerPortType:
+    """Selected server port type."""
+
+    id: int
+    label: str
+
+
+# ── Static choice data ───────────────────────────────────────────────────── #
+
+CONNECTION_TYPES: list[tuple[int, str]] = [
+    (1, "Keep Proxy — fixed proxy, highest trust"),
+    (2, "Keep Connection — fixed connection, high trust"),
+    (3, "Rotate Connection — rotation on each request"),
+]
+
+PROXY_TYPES: list[tuple[int, str]] = [
+    (1, "Residential — home IP addresses"),
+    (2, "All — all proxy types"),
+    (3, "Mobile — mobile operator IPs"),
+    (4, "Corporate — corporate network IPs"),
+]
+
+SERVER_PORT_TYPES: list[tuple[int, str]] = [
+    (0, "Shared — shared port (free)"),
+    (1, "Dedicated — dedicated port (paid, traffic limit)"),
+]