aria-code 0.167.0__py3-none-win_amd64.whl

aria_code/apps/cli/auto_update.py

@@ -0,0 +1,125 @@
+"""Install a verified update in the background, the way Claude Code does.
+
+The session keeps running on the version it started with. When a newer native
+release is out, it is downloaded, checked and switched in behind the scenes,
+and the header says "restart to use it". Nothing is interrupted: the new
+version only takes over the next time Aria starts.
+
+Only managed native installs update themselves. pip, npm and source checkouts
+belong to their own tools, so they keep the plain "update available" notice.
+Turn it off with `auto_update: false` in the config or `ARIA_AUTO_UPDATE=0`.
+A version that failed to install is not retried for a day, and two sessions
+never install at once.
+"""
+
+from __future__ import annotations
+
+import json
+import os
+import subprocess
+import time
+from pathlib import Path
+from typing import Callable, Optional
+
+RETRY_AFTER_FAILURE_S = 86_400
+LOCK_STALE_S = 900
+
+
+def enabled(config: dict) -> bool:
+    if os.environ.get("ARIA_AUTO_UPDATE", "").lower() in {"0", "false", "no", "off"}:
+        return False
+    if os.environ.get("ARIA_OFFLINE", "").lower() in {"1", "true", "yes", "on"}:
+        return False
+    return bool(config.get("auto_update", True))
+
+
+def _state_path(app_root: Path) -> Path:
+    return app_root / "auto-update.json"
+
+
+def _recently_failed(app_root: Path, version: str, now: float) -> bool:
+    try:
+        data = json.loads(_state_path(app_root).read_text())
+    except (OSError, ValueError):
+        return False
+    return data.get("failed") == version and now - float(data.get("at", 0)) < RETRY_AFTER_FAILURE_S
+
+
+def _remember_failure(app_root: Path, version: str, now: float) -> None:
+    try:
+        _state_path(app_root).write_text(json.dumps({"failed": version, "at": now}))
+    except OSError:
+        pass
+
+
+def _take_lock(app_root: Path, now: float) -> Optional[Path]:
+    lock = app_root / "auto-update.lock"
+    try:
+        if lock.exists() and now - lock.stat().st_mtime > LOCK_STALE_S:
+            lock.unlink()
+        fd = os.open(lock, os.O_CREAT | os.O_EXCL | os.O_WRONLY, 0o600)
+    except OSError:
+        return None
+    os.close(fd)
+    return lock
+
+
+def install_if_newer(
+    current: str,
+    channel: str,
+    config: dict,
+    *,
+    check: Callable[..., tuple],
+    install: Callable[[str, dict], Path],
+    newer: Callable[[str, str], bool],
+    app_root: Path,
+    now: Optional[float] = None,
+) -> Optional[str]:
+    """Install the latest release when there is one; return its version, else None.
+
+    Blocking: run it off the event loop. Any failure leaves the installed
+    version untouched and returns None.
+    """
+    if channel != "native" or not enabled(config):
+        return None
+    try:
+        # New managed packages let Rust own retry state, update locking and
+        # install orchestration. Keep the Python implementation below for
+        # older bundles and source/wheel installs without the native helper.
+        from aria_code.apps.cli.updater import _native_update_helper
+
+        helper = _native_update_helper()
+        if helper is not None:
+            result = subprocess.run(
+                [str(helper), "update", "auto-install", "--current", current],
+                capture_output=True,
+                text=True,
+                timeout=600,
+                check=False,
+            )
+            if result.returncode == 0:
+                payload = json.loads(result.stdout)
+                if payload.get("status") in {"installed", "skipped", "failed"}:
+                    version = payload.get("version")
+                    return version if payload["status"] == "installed" and isinstance(version, str) else None
+    except Exception:
+        pass
+    stamp = time.time() if now is None else now
+    try:
+        app_root.mkdir(parents=True, exist_ok=True)
+        latest, release = check("native")
+        if not newer(latest, current) or _recently_failed(app_root, latest, stamp):
+            return None
+        lock = _take_lock(app_root, stamp)
+        if lock is None:
+            return None
+        try:
+            install(latest, release)
+        except Exception:
+            _remember_failure(app_root, latest, stamp)
+            return None
+        finally:
+            lock.unlink(missing_ok=True)
+        return latest
+    except Exception:
+        return None

aria_code/apps/cli/bootstrap.py

@@ -0,0 +1,206 @@
+"""CLI bootstrap utilities shared by legacy and packaged entrypoints."""
+
+from __future__ import annotations
+
+import os
+import socket
+import ssl
+import sys
+import urllib.parse
+from pathlib import Path
+
+from aria_code.apps.cli.config_paths import resolve_paths
+
+
+def load_aria_env(env_file: Path | None = None) -> None:
+    """Load persisted CLI environment values without overriding the process."""
+    target = env_file or (Path.home() / ".aria" / ".env")
+    if not target.exists():
+        return
+    try:
+        for line in target.read_text(encoding="utf-8").splitlines():
+            line = line.strip()
+            if not line or line.startswith("#") or "=" not in line:
+                continue
+            key, _, value = line.partition("=")
+            key = key.strip()
+            value = value.strip()
+            if key and value:
+                os.environ.setdefault(key, value)
+    except Exception:
+        return
+
+
+def disable_broken_proxy(timeout: float = 1.5) -> None:
+    """Unset proxy variables when the configured proxy cannot be reached."""
+    proxy = (
+        os.environ.get("HTTPS_PROXY")
+        or os.environ.get("https_proxy")
+        or os.environ.get("HTTP_PROXY")
+        or os.environ.get("http_proxy")
+    )
+    if not proxy:
+        return
+    try:
+        parsed = urllib.parse.urlparse(proxy if "://" in proxy else f"http://{proxy}")
+        host = parsed.hostname
+        port = parsed.port or 80
+        if not host:
+            return
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        sock.settimeout(timeout)
+        alive = sock.connect_ex((host, port)) == 0
+        sock.close()
+    except Exception:
+        alive = False
+    if not alive:
+        for name in ("HTTP_PROXY", "HTTPS_PROXY", "http_proxy", "https_proxy", "ALL_PROXY", "all_proxy"):
+            os.environ.pop(name, None)
+
+
+_LOOPBACK = ("localhost", "127.0.0.1", "::1")
+
+
+def ensure_loopback_bypasses_proxy() -> None:
+    """With a proxy set, keep local services (Ollama, LM Studio, the API) off it.
+
+    HTTP clients honour HTTP(S)_PROXY, so a machine with a system proxy and no
+    NO_PROXY would send requests for localhost:11434 to the proxy and fail.
+    """
+    if not any(os.environ.get(name) for name in ("HTTP_PROXY", "HTTPS_PROXY", "http_proxy", "https_proxy",
+                                                 "ALL_PROXY", "all_proxy")):
+        return
+    current = os.environ.get("NO_PROXY") or os.environ.get("no_proxy") or ""
+    entries = [item.strip() for item in current.split(",") if item.strip()]
+    missing = [host for host in _LOOPBACK if host not in entries]
+    if missing:
+        value = ",".join(entries + missing)
+        os.environ["NO_PROXY"] = value
+        os.environ["no_proxy"] = value
+
+
+def prepare_network() -> None:
+    """Proxy settings every entry point needs before its first request."""
+    disable_broken_proxy()
+    ensure_loopback_bypasses_proxy()
+
+
+def use_system_trust_store() -> bool:
+    """Verify TLS against the OS trust store instead of certifi's bundle.
+
+    Corporate and local TLS-intercepting proxies present their own CA. That CA
+    is installed in the system keychain — which is why ``curl`` works on such a
+    machine — but Python verifies against certifi, which has never heard of it,
+    so every HTTPS call fails with CERTIFICATE_VERIFY_FAILED.
+
+    This was not hypothetical: on a machine behind an intercepting proxy,
+    ``oauth2.googleapis.com`` failed 5 times out of 5 from Python and returned
+    404 from curl, so roughly a third of Vertex turns died fetching an OAuth
+    token. Trusting the OS store fixed it without disabling verification —
+    which is the tempting wrong fix, and would accept any certificate at all.
+
+    Returns whether the injection happened, and never raises: an environment
+    without ``truststore`` keeps certifi's behaviour exactly as before.
+    """
+    try:
+        import truststore
+
+        truststore.inject_into_ssl()
+        return True
+    except Exception:
+        return False
+
+
+def use_macos_ca_bundle() -> bool:
+    """Use macOS's verified CA bundle when Python has no default CA file.
+
+    Some standalone Python runtimes have no configured OpenSSL CA path, while
+    curl and the system browser can verify the same HTTPS service. This keeps
+    verification enabled and respects an explicit SSL_CERT_FILE override.
+    """
+    if sys.platform != "darwin" or os.environ.get("SSL_CERT_FILE"):
+        return False
+    if ssl.get_default_verify_paths().cafile:
+        return False
+    bundle = Path("/etc/ssl/cert.pem")
+    if not bundle.is_file():
+        return False
+    os.environ["SSL_CERT_FILE"] = str(bundle)
+    return True
+
+
+def initialize_cli_environment() -> None:
+    os.environ.setdefault("TQDM_DISABLE", "1")
+    load_aria_env()
+    prepare_network()
+    if not use_system_trust_store():
+        use_macos_ca_bundle()
+
+
+# The model used when nothing has been configured yet.
+#
+# This exists so there is ONE place that answers "which model by default".
+# Call sites used to spell their own fallback — `config.get("model",
+# "qwen2.5:7b")`, ten times over — which quietly made Ollama the answer
+# whenever a config key was missing, no matter what the user had chosen.
+# Provider configuration is the user's to make (that is what `/model` and
+# `.ariarc` are for); the code's job is to have a sane default and then get
+# out of the way.
+DEFAULT_MODEL = os.getenv("ARIA_DEFAULT_MODEL", "google/gemini-3.8-flash")
+# A stable model (Google: available until at least May 2027), not a short-term
+# one: an install that is rarely updated should not lose its default within
+# months. gemini-2.5-pro, the previous default, retires in October 2026.
+
+
+def default_config() -> dict:
+    return {
+        # The Arthera backend and gateway. Not localhost: see endpoint_migration.
+        "api_url": os.getenv("ARTHERA_API_URL", "").strip() or "https://api.arthera.xyz",
+        "local_url": "http://localhost:8000",
+        "ollama_url": os.getenv("OLLAMA_URL") or os.getenv("OLLAMA_HOST") or "http://127.0.0.1:11434",
+        # Which backend serves a bare, unprefixed model name. Only consulted
+        # for names that do not name their own provider ("qwen2.5:7b"); a
+        # prefixed id like "google/gemini-3.8-flash" always wins.
+        "local_provider": os.getenv("ARIA_LLM_PROVIDER", "ollama"),
+        "provider_fallback": "configured",
+        "model": DEFAULT_MODEL,
+        "thinking_mode": "auto",
+        "watchlist": ["AAPL", "MSFT", "GOOGL", "TSLA", "NVDA"],
+        "auth_token": None,
+        "user_id": None,
+        "last_session_id": None,
+        "auto_save_sessions": True,
+        "auto_compact_context": True,
+        "auto_compact_threshold": 0.78,
+        "command_policy": "safe",
+        "permission_mode": "workspace-write",
+        # Approval by risk level (safety/risk.py): "manual" asks as before;
+        # "risk" runs actions at or below auto_approve_level (0-2) without a
+        # prompt. L4 always asks, even after "always allow".
+        "approval_mode": "manual",
+        "auto_approve_level": 1,
+        # Independent review of a turn's changes by a fresh-context reviewer
+        # before it ends (runtime/review.py). Costs one extra model call.
+        "review_gate": False,
+        # A different model for that review (provider/model); "" = the builder itself.
+        "review_model": "",
+        # The runtime's DONE/INCOMPLETE report under coding answers.
+        "delivery_report": True,
+        "network_enabled": True,
+        "data_sharing": False,
+        "feedback_upload": False,
+        "write_policy": "desktop_only",
+        "lsp_autocheck": False,
+        "input_style": "panel",
+        "input_theme": "auto",
+        "response_footer": "compact",
+        "report_agent_timeout": 40.0,
+        "report_synthesis_timeout": 20.0,
+        "local_mode": False,
+        "conversation_history": [],
+        "ui_lang": "",
+    }
+
+
+def runtime_paths():
+    return resolve_paths()

aria_code/apps/cli/broker_render.py

@@ -0,0 +1,206 @@
+"""券商账户 / 持仓 / 订单的终端渲染 —— 从 aria_cli.py 抽出的第二块。
+
+原本在 aria_cli.py 的 4399-4578 行。跟 football_reports 一样，这些函数依赖
+aria_cli 的模块级状态（``console`` / ``HAS_RICH`` / ``Panel`` / ``rich_box``），
+所以必须由 aria_cli 用 ``_rebind_module_function_globals`` 绑回它自己的
+globals，不能只是普通 import。
+
+调用方有两种形态，重绑后都成立：
+  - ``apps/cli/commands/broker_cmds.py`` 用显式 ``from aria_cli import
+    _print_broker_account`` —— 读的是 aria_cli 的命名空间；
+  - ``tests/test_responsive_rendering.py`` 用 ``aria_cli._print_broker_positions``
+    —— 同上。
+
+（这一组不像 football 那样被 mixin 裸名调用，但依赖 aria_cli 模块级 rich
+状态，所以约束是一样的。）
+"""
+
+from __future__ import annotations
+
+__all__ = [
+    "_print_broker_account",
+    "_print_broker_positions",
+    "_print_broker_orders",
+]
+
+
+def _print_broker_account(acct: "AccountInfo"):
+    """Render AccountInfo in a Rich Panel."""
+    if not HAS_RICH:
+        print(f"{acct.label}  总资产:{acct.total_assets:,.2f}  可用:{acct.cash:,.2f}  市值:{acct.market_value:,.2f}")
+        return
+    pnl_color = "green" if acct.pnl_today >= 0 else "red"
+    pnl_sign  = "+" if acct.pnl_today >= 0 else ""
+    body = (
+        f"[dim]账户:[/dim]  [bold]{acct.masked_account}[/bold]  [dim]({acct.broker_type})[/dim]\n\n"
+        f"  总资产       [bold]{acct.currency} {acct.total_assets:>14,.2f}[/bold]\n"
+        f"  持仓市值     [bold]{acct.market_value:>14,.2f}[/bold]\n"
+        f"  可用现金     [bold]{acct.cash:>14,.2f}[/bold]\n"
+        f"  冻结资金     [dim]{acct.frozen:>14,.2f}[/dim]\n"
+        f"  当日盈亏     [{pnl_color}]{pnl_sign}{acct.pnl_today:>14,.2f}[/{pnl_color}]\n"
+    )
+    if acct.pnl_total:
+        tp_color = "green" if acct.pnl_total >= 0 else "red"
+        body += f"  累计盈亏     [{tp_color}]{pnl_sign}{acct.pnl_total:>14,.2f}[/{tp_color}]\n"
+    console.print(Panel(body, title=f"[bold]{acct.label}[/bold]",
+                        border_style="dim", box=rich_box.ROUNDED, padding=(0, 1)))
+
+
+def _print_broker_positions(positions: list, broker_label: str, currency: str = "CNY"):
+    """Render Position list as a Rich Table."""
+    if not HAS_RICH:
+        for p in positions:
+            print(f"  {p.symbol:<8} {p.name:<12} 持仓:{p.quantity}  市值:{p.market_value:,.2f}  盈亏:{p.pnl:+,.2f} ({p.pnl_pct:+.2f}%)")
+        return
+    if not positions:
+        console.print(f"[dim]{broker_label} — 当前无持仓[/dim]")
+        return
+    from rich.markup import escape as _esc
+    from ui.render.responsive import StackedRecord, render_stacked_records, structured_layout
+
+    ordered = sorted(positions, key=lambda x: -abs(x.market_value))
+    total_mv = sum(p.market_value for p in positions)
+    total_pnl = sum(p.pnl for p in positions)
+    total_color = "green" if total_pnl >= 0 else "red"
+    layout = structured_layout(console)
+    if layout == "stacked":
+        records = []
+        for p in ordered:
+            pnl_color = "green" if p.pnl >= 0 else "red"
+            records.append(StackedRecord(
+                headline=f"{_esc(str(p.symbol))}  {_esc(str(p.name or '—'))}",
+                lines=(
+                    f"[dim]持仓[/dim] {p.quantity:,.0f}  ·  [dim]可卖[/dim] {p.available_qty:,.0f}",
+                    f"[dim]成本[/dim] {p.cost_price:.3f}  ·  [dim]现价[/dim] {p.current_price:.3f}",
+                    f"[dim]市值[/dim] {p.market_value:,.2f}  ·  "
+                    f"[dim]盈亏[/dim] [{pnl_color}]{p.pnl:+,.2f} ({p.pnl_pct:+.2f}%)[/{pnl_color}]",
+                ),
+            ))
+        render_stacked_records(
+            console,
+            title=f"{_esc(str(broker_label))} 持仓",
+            records=records,
+            footer=(
+                f"共 {len(positions)} 只 · 总市值 {currency} {total_mv:,.2f} · "
+                f"总盈亏 [{total_color}]{total_pnl:+,.2f}[/{total_color}]"
+            ),
+        )
+        return
+    from rich.table import Table
+    tbl = Table(title=f"[bold]{broker_label}[/bold] 持仓", show_header=True, header_style="bold")
+    tbl.add_column("代码",   style="bold", no_wrap=True)
+    tbl.add_column("名称",   max_width=12)
+    tbl.add_column("持仓",   justify="right")
+    if layout == "full":
+        tbl.add_column("可卖",   justify="right", style="dim")
+        tbl.add_column("成本",   justify="right", style="dim")
+        tbl.add_column("现价",   justify="right")
+    tbl.add_column("市值",   justify="right")
+    tbl.add_column("盈亏",   justify="right")
+    tbl.add_column("盈亏%",  justify="right")
+    for p in ordered:
+        pnl_color = "green" if p.pnl >= 0 else "red"
+        pnl_sign  = "+" if p.pnl >= 0 else ""
+        row = [p.symbol, p.name[:12] or "—", f"{p.quantity:,.0f}"]
+        if layout == "full":
+            row.extend([
+                f"{p.available_qty:,.0f}",
+                f"{p.cost_price:.3f}",
+                f"{p.current_price:.3f}",
+            ])
+        row.extend([
+            f"{p.market_value:,.2f}",
+            f"[{pnl_color}]{pnl_sign}{p.pnl:,.2f}[/{pnl_color}]",
+            f"[{pnl_color}]{pnl_sign}{p.pnl_pct:.2f}%[/{pnl_color}]",
+        ])
+        tbl.add_row(*row)
+    console.print(tbl)
+    console.print(
+        f"  [dim]共 {len(positions)} 只  总市值 {total_mv:,.2f}  "
+        f"总盈亏 [{total_color}]{'+' if total_pnl>=0 else ''}{total_pnl:,.2f}[/{total_color}][/dim]"
+    )
+
+
+def _print_broker_orders(orders: list, broker_label: str, status_filter: str = "all"):
+    """Render Order list as a Rich Table."""
+    if not HAS_RICH:
+        for o in orders:
+            print(f"  {o.order_id[:8]} {o.symbol:<8} {o.side:<4} {o.quantity:>8.0f} @ {o.price:.3f}  {o.status}")
+        return
+    if not orders:
+        console.print(f"[dim]{broker_label} — 无 {status_filter} 订单[/dim]")
+        return
+    from rich.markup import escape as _esc
+    from ui.render.responsive import StackedRecord, render_stacked_records, structured_layout
+
+    layout = structured_layout(console)
+    _STATUS_STYLE = {"filled":"[green]成交[/green]","partial":"[yellow]部成[/yellow]",
+                     "open":"[cyan]委托中[/cyan]","cancelled":"[dim]已撤[/dim]"}
+    _SIDE_STYLE   = {"buy":"[green]买入[/green]","sell":"[red]卖出[/red]"}
+    if layout == "stacked":
+        records = []
+        for o in orders:
+            avg_price = f"{o.avg_price:.3f}" if o.avg_price else "—"
+            records.append(StackedRecord(
+                headline=(
+                    f"{_esc(str(o.symbol))}  {_esc(str(o.name or '—'))}  "
+                    f"[dim]#{_esc(str(o.order_id[-8:]))}[/dim]"
+                ),
+                lines=(
+                    f"[dim]方向[/dim] {_SIDE_STYLE.get(o.side, _esc(str(o.side)))}  ·  "
+                    f"[dim]状态[/dim] {_STATUS_STYLE.get(o.status, _esc(str(o.status)))}",
+                    f"[dim]委托[/dim] {o.quantity:,.0f} @ {o.price:.3f}",
+                    f"[dim]成交[/dim] {o.filled_qty:,.0f} @ {avg_price}",
+                    f"[dim]时间[/dim] {_esc(str(o.created_at[:16] if o.created_at else '—'))}",
+                ),
+            ))
+        render_stacked_records(
+            console,
+            title=f"{_esc(str(broker_label))} 订单 · {_esc(str(status_filter))}",
+            records=records,
+            footer=f"共 {len(orders)} 笔",
+        )
+        return
+    from rich.table import Table
+    tbl = Table(title=f"[bold]{broker_label}[/bold] 订单 [dim]({status_filter})[/dim]",
+                show_header=True, header_style="bold")
+    if layout == "full":
+        tbl.add_column("订单号",  style="dim",   max_width=12)
+        tbl.add_column("代码",    style="bold",  no_wrap=True)
+        tbl.add_column("名称",    max_width=10)
+        tbl.add_column("方向",    justify="center")
+        tbl.add_column("类型",    style="dim")
+        tbl.add_column("委托量",  justify="right")
+        tbl.add_column("成交量",  justify="right")
+        tbl.add_column("委托价",  justify="right", style="dim")
+        tbl.add_column("均价",    justify="right")
+        tbl.add_column("状态")
+        tbl.add_column("时间",    style="dim", max_width=16)
+    else:
+        tbl.add_column("代码",    style="bold",  no_wrap=True)
+        tbl.add_column("方向",    justify="center")
+        tbl.add_column("委托量",  justify="right")
+        tbl.add_column("委托价",  justify="right", style="dim")
+        tbl.add_column("状态")
+        tbl.add_column("时间",    style="dim", max_width=16)
+    for o in orders:
+        if layout == "full":
+            row = [
+                o.order_id[-8:], o.symbol, o.name[:10] or "—",
+                _SIDE_STYLE.get(o.side, o.side), o.order_type,
+                f"{o.quantity:,.0f}", f"{o.filled_qty:,.0f}",
+                f"{o.price:.3f}", f"{o.avg_price:.3f}" if o.avg_price else "—",
+                _STATUS_STYLE.get(o.status, o.status),
+                o.created_at[:16] if o.created_at else "—",
+            ]
+        else:
+            row = [
+                o.symbol, _SIDE_STYLE.get(o.side, o.side),
+                f"{o.quantity:,.0f}", f"{o.price:.3f}",
+                _STATUS_STYLE.get(o.status, o.status),
+                o.created_at[:16] if o.created_at else "—",
+            ]
+        tbl.add_row(*row)
+    console.print(tbl)
+
+