ravmem-cli 0.1.0__py3-none-any.whl

ravmem/config_loader.py

@@ -0,0 +1,247 @@
+"""Structured YAML config loader with pydantic validation and env-var override.
+
+Resolution order (first found wins):
+  1. --config CLI flag  → passed as config_path argument
+  2. RAVMEM_CONFIG env var → path to a YAML file
+  3. $RAVMEM_DATA_DIR/ravmem.yaml
+  4. ~/.ravmem/ravmem.yaml
+  5. Built-in defaults (pydantic model defaults)
+
+Env vars always win over file values — existing env var names continue to work
+exactly as before; no breaking change for Docker Compose / systemd users.
+"""
+
+from __future__ import annotations
+
+import logging
+import os
+from pathlib import Path
+from typing import Optional
+
+import yaml
+from pydantic import BaseModel, field_validator, model_validator
+
+logger = logging.getLogger(__name__)
+
+_SENSITIVE_KEYS = {"admin_token", "password"}
+
+
+# ---------------------------------------------------------------------------
+# Pydantic config models
+# ---------------------------------------------------------------------------
+
+class TierLimits(BaseModel):
+    read: str = "1000/min"
+    mcp: str = "600/min"
+    write: str = "200/min"
+    index: str = "30/min"
+    auth: str = "60/min"
+
+    @field_validator("read", "mcp", "write", "index", "auth", mode="before")
+    @classmethod
+    def _validate_limit(cls, v: object) -> str:
+        if not isinstance(v, str):
+            v = str(v)
+        parts = v.split("/")
+        if len(parts) != 2 or not parts[0].strip().isdigit() or parts[1].strip() not in ("sec", "min", "hour"):
+            raise ValueError(f"Invalid rate limit format '{v}' — expected '<int>/sec|min|hour'")
+        return v
+
+
+class RateLimitTiers(BaseModel):
+    admin: TierLimits = TierLimits(
+        read="5000/min", mcp="3000/min", write="1000/min", index="300/min", auth="300/min"
+    )
+    high: TierLimits = TierLimits(
+        read="3000/min", mcp="1800/min", write="600/min", index="120/min", auth="180/min"
+    )
+    standard: TierLimits = TierLimits(
+        read="1000/min", mcp="600/min", write="200/min", index="30/min", auth="60/min"
+    )
+    restricted: TierLimits = TierLimits(
+        read="200/min", mcp="120/min", write="40/min", index="5/min", auth="20/min"
+    )
+
+
+class RateLimitConfig(BaseModel):
+    enabled: bool = True
+    limits: TierLimits = TierLimits()
+    tiers: RateLimitTiers = RateLimitTiers()
+    # Brute-force lockout (Phase 2)
+    lockout_threshold: int = 5
+    lockout_duration: int = 600  # seconds
+
+
+class ServerConfig(BaseModel):
+    host: str = "0.0.0.0"
+    port: int = 7432
+    max_concurrent_jobs: int = 2
+    log_level: str = "INFO"
+    data_dir: str = ""  # resolved from env/default at load time
+
+
+class ArcadeDBConfig(BaseModel):
+    host: str = "localhost"
+    port: int = 2480
+    password: str = "arcadedb"  # override via ARCADEDB_PASSWORD — never write to YAML
+
+
+class AuthConfig(BaseModel):
+    system_db: str = "ravmem_system"
+    audit_log_retention_days: int = 90
+    admin_token: str = ""  # env-only — warn if found in YAML
+
+
+class ClusteringConfig(BaseModel):
+    seed: int = 42
+    resolution: float = 1.0
+    max_cluster_size: int = 500
+    subcluster_threshold: int = 200
+
+
+class RavMemConfig(BaseModel):
+    server: ServerConfig = ServerConfig()
+    arcadedb: ArcadeDBConfig = ArcadeDBConfig()
+    auth: AuthConfig = AuthConfig()
+    clustering: ClusteringConfig = ClusteringConfig()
+    rate_limit: RateLimitConfig = RateLimitConfig()
+
+    @model_validator(mode="after")
+    def _warn_sensitive_in_file(self) -> "RavMemConfig":
+        if self.auth.admin_token:
+            logger.warning(
+                "SECURITY WARNING: 'auth.admin_token' found in config file. "
+                "Use the RAVMEM_ADMIN_TOKEN env var instead — config files may be world-readable."
+            )
+        if self.arcadedb.password not in ("arcadedb", ""):
+            pass  # password in file is acceptable for dev; no warning needed
+        return self
+
+
+# ---------------------------------------------------------------------------
+# Loader
+# ---------------------------------------------------------------------------
+
+def _find_config_file(explicit_path: str | Path | None = None) -> Path | None:
+    """Return the first config file that exists, following the resolution order."""
+    candidates: list[Path] = []
+
+    if explicit_path:
+        candidates.append(Path(explicit_path))
+    elif env_path := os.environ.get("RAVMEM_CONFIG"):
+        candidates.append(Path(env_path))
+
+    data_dir = os.environ.get("RAVMEM_DATA_DIR", str(Path.home() / ".ravmem" / "server"))
+    candidates.append(Path(data_dir) / "ravmem.yaml")
+    candidates.append(Path.home() / ".ravmem" / "ravmem.yaml")
+
+    for path in candidates:
+        if path.exists():
+            return path
+    return None
+
+
+def _deep_merge(base: dict, override: dict) -> dict:
+    """Recursively merge override into base. Override values take precedence."""
+    result = dict(base)
+    for key, val in override.items():
+        if key in result and isinstance(result[key], dict) and isinstance(val, dict):
+            result[key] = _deep_merge(result[key], val)
+        else:
+            result[key] = val
+    return result
+
+
+def _apply_env_overrides(data: dict) -> dict:
+    """Apply env vars on top of parsed YAML data. Env vars always win."""
+    env_map: dict[str, list[tuple]] = {
+        "RAVMEM_LOG_LEVEL": [("server", "log_level")],
+        "RAVMEM_MAX_CONCURRENT_JOBS": [("server", "max_concurrent_jobs")],
+        "RAVMEM_DATA_DIR": [("server", "data_dir")],
+        "ARCADEDB_HOST": [("arcadedb", "host")],
+        "ARCADEDB_PORT": [("arcadedb", "port")],
+        "ARCADEDB_PASSWORD": [("arcadedb", "password")],
+        "RAVMEM_SYSTEM_DB": [("auth", "system_db")],
+        "RAVMEM_AUDIT_LOG_RETENTION_DAYS": [("auth", "audit_log_retention_days")],
+        "RAVMEM_CLUSTER_SEED": [("clustering", "seed")],
+        "RAVMEM_CLUSTER_RESOLUTION": [("clustering", "resolution")],
+        "RAVMEM_CLUSTER_MAX_SIZE": [("clustering", "max_cluster_size")],
+        "RAVMEM_CLUSTER_SUBCLUSTER_THRESHOLD": [("clustering", "subcluster_threshold")],
+        "RAVMEM_RATELIMIT_ENABLED": [("rate_limit", "enabled")],
+        "RAVMEM_RATELIMIT_AUTH": [
+            ("rate_limit", "limits", "auth"),
+            ("rate_limit", "tiers", "standard", "auth"),
+        ],
+        "RAVMEM_RATELIMIT_WRITE": [
+            ("rate_limit", "limits", "write"),
+            ("rate_limit", "tiers", "standard", "write"),
+        ],
+        "RAVMEM_RATELIMIT_INDEX": [
+            ("rate_limit", "limits", "index"),
+            ("rate_limit", "tiers", "standard", "index"),
+        ],
+        "RAVMEM_RATELIMIT_READ": [
+            ("rate_limit", "limits", "read"),
+            ("rate_limit", "tiers", "standard", "read"),
+        ],
+        "RAVMEM_RATELIMIT_MCP": [
+            ("rate_limit", "limits", "mcp"),
+            ("rate_limit", "tiers", "standard", "mcp"),
+        ],
+        "RAVMEM_RATELIMIT_LOCKOUT_THRESHOLD": [("rate_limit", "lockout_threshold")],
+        "RAVMEM_RATELIMIT_LOCKOUT_DURATION": [("rate_limit", "lockout_duration")],
+    }
+    for env_key, paths in env_map.items():
+        val = os.environ.get(env_key)
+        if val is None:
+            continue
+        # Convert boolean strings
+        if isinstance(val, str) and val.lower() in ("true", "1", "yes"):
+            val = True
+        elif isinstance(val, str) and val.lower() in ("false", "0", "no"):
+            val = False
+        # Apply to all target paths
+        for path in paths:
+            node = data
+            for part in path[:-1]:
+                node = node.setdefault(part, {})
+            node[path[-1]] = val
+    return data
+
+
+def load_config(config_path: str | Path | None = None) -> RavMemConfig:
+    """Load, merge, validate and return the resolved RavMemConfig."""
+    raw: dict = {}
+
+    found = _find_config_file(config_path)
+    if found:
+        logger.debug("Loading config from %s", found)
+        with open(found) as fh:
+            file_data = yaml.safe_load(fh) or {}
+        raw = _deep_merge(raw, file_data)
+
+    raw = _apply_env_overrides(raw)
+
+    try:
+        return RavMemConfig.model_validate(raw)
+    except Exception as exc:
+        raise ValueError(f"Invalid ravmem config: {exc}") from exc
+
+
+# Module-level singleton — loaded lazily on first access
+_config: RavMemConfig | None = None
+
+
+def get_config(config_path: str | Path | None = None) -> RavMemConfig:
+    """Return the module-level config singleton, loading it if necessary."""
+    global _config
+    if _config is None:
+        _config = load_config(config_path)
+    return _config
+
+
+def reload_config(config_path: str | Path | None = None) -> RavMemConfig:
+    """Force-reload config (useful in tests or after serve --config flag)."""
+    global _config
+    _config = load_config(config_path)
+    return _config

ravmem_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,62 @@
+Metadata-Version: 2.4
+Name: ravmem-cli
+Version: 0.1.0
+Summary: RavMem CLI — lightweight client for the RavMem code knowledge graph server
+Author-email: RavMem <harpreet.singh@ravmem.com>
+License-Expression: MIT
+Project-URL: Homepage, https://www.ravmem.com
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Operating System :: OS Independent
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: typer[all]>=0.12.0
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: pydantic>=2.0
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.23.0; extra == "dev"
+Dynamic: license-file
+
+# ravmem-cli
+
+Lightweight CLI client for the [RavMem](https://github.com/ravmem/ravmem) code knowledge graph server.
+
+## Installation
+
+```bash
+pip install ravmem-cli
+```
+
+## Quick start
+
+```bash
+# Point the CLI at your RavMem server
+ravmem remote config --url http://localhost:7432
+
+# Register a repository
+ravmem remote add /path/to/repo --name my-project --default-branch main
+
+# Index a repository
+ravmem analyze /path/to/repo --project <ID>
+
+# Search symbols
+ravmem search "authentication handler" --project <ID>
+
+# Blast-radius analysis
+ravmem impact MyClass.my_method --project <ID>
+```
+
+## Requirements
+
+- Python 3.10+
+- A running [ravmem-server](https://pypi.org/project/ravmem-server/) instance
+
+## License
+
+MIT — see [LICENSE](LICENSE).

ravmem_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,25 @@
+ravmem/config.py,sha256=SkoObiRD-kyHWkI94l3wJv1bURiHTQ1a6YSlrlBCvuk,3284
+ravmem/config_loader.py,sha256=vvWJh3H0sUJgNMkaPnXLZNNPR_-xUozMdYurtpy6t2E,8757
+ravmem/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ravmem/cli/analyze.py,sha256=asBWizsgTLNEc3rV8zohSJn7Yh83vTX81qMF_MO_dQU,16322
+ravmem/cli/auth.py,sha256=avYt_tgM6JtgAAl5LHTCbfFoa-8_4IYu6zAeAnyn5pw,21887
+ravmem/cli/browse.py,sha256=9xYb0TBjUqmTMWKkd8a-MRCVi0TlgDak1Q2UrW2mlBA,4854
+ravmem/cli/config_cmd.py,sha256=Cl-dm5KdfLOuGYL12j-SZGjGdDl0fdoPMVhvepzK4nI,2210
+ravmem/cli/diff.py,sha256=m2Je-j9V1Yf4TSAvLUXBMB58Ff48IjD6Vm1qbI8NhDQ,3901
+ravmem/cli/git_local.py,sha256=WodcZZeEgfnQX9w8rw70o6W56pkNJUHEgkrtjHvBBzM,4117
+ravmem/cli/impact.py,sha256=JjTr_VQY9jGOFA1ZUZxXmn1mzv5ALmpNkuMhLw3VE4w,4021
+ravmem/cli/info.py,sha256=PY1WCEJoKYa1UQ7u0ZdUdJb_KKmgI-st46D7M0Cwmuk,2492
+ravmem/cli/kg.py,sha256=FnWEjCUq0Qw9TDRNk7zhvOrdl1nANNuh5LyUBoanwSU,30978
+ravmem/cli/main.py,sha256=L5WFFptXvzssuc388Az_e8HGMGIeEWlrX6lSxH01_X0,2952
+ravmem/cli/remote.py,sha256=l616Cs_txtGAP5udFR6JtZGo4tPpIf8ttNsrCh612ms,19622
+ravmem/cli/search.py,sha256=ht56_YvblfXBibUYyRIJKynQnJQVJI28HEx6AdqbtbQ,3081
+ravmem/cli/serve.py,sha256=HAsy3N4EVi7BPntg9220PowMLke65zgm7s31SjtT8bs,1503
+ravmem/client/__init__.py,sha256=kqQNEjY_MMB4U7V7jeQVvwpyAH8My_indn47BERBgxk,71
+ravmem/client/api.py,sha256=evBGJQOrmb6iBAHBrqYQVU1N637ltUPRj-aooXN-cAg,9417
+ravmem/client/config.py,sha256=fvpB9CYxjsI-lM5gmuLtat73xqQFUEWzJsXRUt_fJ90,1405
+ravmem_cli-0.1.0.dist-info/licenses/LICENSE,sha256=wI3Yj7pnbSHXvrmP3869k5Ft3ln5C83l-1w-RN0R14I,1063
+ravmem_cli-0.1.0.dist-info/METADATA,sha256=5RUE_oivL9LW-P14cr4X82CjIhs9t0HSY0kFOfEmpWg,1696
+ravmem_cli-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+ravmem_cli-0.1.0.dist-info/entry_points.txt,sha256=ldUYV83DRLplX6etlK-gvrs2MpBpWgk8KIGVMpNL354,47
+ravmem_cli-0.1.0.dist-info/top_level.txt,sha256=nSWRo9Gm3vH4guHC29V3rbB9smoQm3zYxFhbrqLH8jo,7
+ravmem_cli-0.1.0.dist-info/RECORD,,

ravmem_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

ravmem_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ravmem = ravmem.cli.main:app

ravmem_cli-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 RavMem
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

ravmem_cli-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+ravmem