nia-sync 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

{nia_sync-0.1.6.dist-info → nia_sync-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nia-sync
-Version: 0.1.6
+Version: 0.1.8
 Summary: Keep your local files in sync with Nia Cloud
 Requires-Python: >=3.10
 Requires-Dist: typer>=0.9.0

nia_sync-0.1.8.dist-info/RECORD

@@ -0,0 +1,13 @@
+api_client.py,sha256=wg3oRixlyzR_GmJDrmpVemJb0GI1ixd7UceSyHBR52s,2712
+auth.py,sha256=-DeD3azCHlqqI7zySKEhsxeeT8kJYsfzpuqAsGKyA1s,5720
+config.py,sha256=_zeDLBggazFGlR1LUitMNTiMpqHMxZBfcUJ-C6tXgdY,7233
+extractor.py,sha256=KhLTRJQAgbsZ_qqXzBjU5Qh2R83r2vmNMF-bB4Pzbqw,30279
+main.py,sha256=owUa1k_f3BWlARv7erfxhhvl-aIHWbjTPhD0Ln3T9sc,52672
+sync.py,sha256=gHxA0pn2_I7habRwWJ33tSit0aRDxEZHA2Gdyt6Fe20,12859
+ui.py,sha256=yxlRz2VIkiaMTzV95QhsV3Uojt3Ns6_QXLCSlUWJPr0,3456
+watcher.py,sha256=JmsN9uR7Ss1mDC-kApXL6Hg_wQZWTsO7rRIFkQu8GbM,9978
+nia_sync-0.1.8.dist-info/METADATA,sha256=Phtagx-S9wDy-ARK6C3ZkN-NPlgjoSe3V_R5qKk5hMo,246
+nia_sync-0.1.8.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+nia_sync-0.1.8.dist-info/entry_points.txt,sha256=Fx8TIOgXqWdZzZEkEateDtcNfgnwuPW4jZTqlEUrHVs,33
+nia_sync-0.1.8.dist-info/top_level.txt,sha256=nHlEpqudkMdi7d4wpt_qUI-vjVay1X5CaB6foOQ85Dg,54
+nia_sync-0.1.8.dist-info/RECORD,,

{nia_sync-0.1.6.dist-info → nia_sync-0.1.8.dist-info}/top_level.txt

@@ -1,6 +1,8 @@
+api_client
 auth
 config
 extractor
 main
 sync
+ui
 watcher

sync.py

@@ -14,8 +14,8 @@ from pathlib import Path
 from typing import Any
 import httpx
 
-from config import API_BASE_URL, get_api_key
-from extractor import extract_incremental, detect_source_type
+from config import get_api_base_url, get_api_key
+from extractor import extract_incremental, detect_source_type, TYPE_FOLDER
 
 logger = logging.getLogger(__name__)
 
@@ -26,10 +26,31 @@ MAX_BATCH_SIZE_BYTES = 4 * 1024 * 1024  # 4MB max payload per batch
 MAX_RETRIES = 4
 RETRY_BASE_DELAY = 1.5
 RETRY_MAX_DELAY = 15.0
+FOLDER_CURSOR_VERSION = 1
 
 # Reusable client for connection pooling
 _http_client: httpx.Client | None = None
 
+def _normalize_folder_cursor(
+    path: str,
+    cursor: dict[str, Any] | None,
+) -> tuple[dict[str, Any], str | None]:
+    if not cursor:
+        return {}, "missing"
+
+    cursor_version = cursor.get("cursor_version")
+    root_path = cursor.get("root_path")
+    normalized_root = os.path.abspath(os.path.expanduser(root_path)) if root_path else None
+
+    if cursor_version != FOLDER_CURSOR_VERSION:
+        return {}, "version_mismatch"
+    if not normalized_root:
+        return {}, "missing_root_path"
+    if normalized_root != path:
+        return {}, "root_path_changed"
+
+    return cursor, None
+
 def get_http_client() -> httpx.Client:
     """Get or create HTTP client with connection pooling."""
     global _http_client
@@ -88,8 +109,9 @@ def sync_source(source: dict[str, Any]) -> dict[str, Any]:
             "message": "No local path configured",
         }
 
-    # Expand ~ in path
+    # Expand ~ and normalize path
     path = os.path.expanduser(path)
+    path = os.path.abspath(path)
 
     # Validate path exists
     if not os.path.exists(path):
@@ -105,6 +127,12 @@ def sync_source(source: dict[str, Any]) -> dict[str, Any]:
     if not detected_type:
         detected_type = detect_source_type(path)
 
+    cursor_reset_reason = None
+    if detected_type == TYPE_FOLDER:
+        cursor, cursor_reset_reason = _normalize_folder_cursor(path, cursor)
+        if cursor_reset_reason:
+            logger.info(f"Resetting folder sync cursor for {path} ({cursor_reset_reason})")
+
     logger.info(f"Syncing {path} (type={detected_type})")
 
     try:
@@ -116,10 +144,38 @@ def sync_source(source: dict[str, Any]) -> dict[str, Any]:
         )
 
         files = extraction_result.get("files", [])
-        new_cursor = extraction_result.get("cursor", {})
+        new_cursor = dict(extraction_result.get("cursor", {}))
         stats = extraction_result.get("stats", {})
 
+        if detected_type == TYPE_FOLDER:
+            new_cursor["cursor_version"] = FOLDER_CURSOR_VERSION
+            new_cursor["root_path"] = path
+
         if not files:
+            if detected_type == TYPE_FOLDER and cursor_reset_reason:
+                cursor_update = new_cursor
+                upload_result = upload_sync_data(
+                    local_folder_id=local_folder_id,
+                    files=[],
+                    cursor=cursor_update,
+                    stats=stats,
+                    is_final_batch=True,
+                )
+                if upload_result.get("status") == "ok":
+                    source["cursor"] = cursor_update
+                    return {
+                        "path": path,
+                        "status": "success",
+                        "added": 0,
+                        "message": "No new data (cursor updated)",
+                    }
+                report_sync_error(local_folder_id, upload_result.get("message", "Upload failed"), path)
+                return {
+                    "path": path,
+                    "status": "error",
+                    "error": upload_result.get("message", "Upload failed"),
+                }
+
             logger.info(f"No new data to sync for {path}")
             return {
                 "path": path,
@@ -199,7 +255,7 @@ def upload_sync_data(
         client = get_http_client()
         response = _post_with_retries(
             client=client,
-            url=f"{API_BASE_URL}/v2/daemon/sync",
+            url=f"{get_api_base_url()}/v2/daemon/sync",
             headers={"Authorization": f"Bearer {api_key}"},
             payload={
                 "local_folder_id": local_folder_id,
@@ -278,7 +334,7 @@ def report_sync_error(local_folder_id: str | None, error: str, path: str | None
         client = get_http_client()
         _post_with_retries(
             client=client,
-            url=f"{API_BASE_URL}/v2/daemon/sources/{local_folder_id}/error",
+            url=f"{get_api_base_url()}/v2/daemon/sources/{local_folder_id}/error",
             headers={"Authorization": f"Bearer {api_key}"},
             payload={"error": error, "path": path},
         )

ui.py

@@ -0,0 +1,119 @@
+"""
+UI utilities for Nia CLI - ASCII art and spinners.
+"""
+from rich.console import Console
+from rich.text import Text
+from rich.panel import Panel
+from contextlib import contextmanager
+import random
+
+console = Console()
+
+NIA_LOGO_MINI = "[bold cyan]NIA[/bold cyan]"
+
+# Spinner styles that look good
+SPINNER_STYLES = [
+    "dots",
+    "dots2", 
+    "dots3",
+    "dots12",
+    "line",
+    "arc",
+    "bouncingBar",
+    "bouncingBall",
+    "moon",
+    "runner",
+    "aesthetic",
+]
+
+# Action-specific spinners and messages
+SPINNER_MESSAGES = {
+    "login": ["Authenticating...", "Connecting to Nia...", "Verifying..."],
+    "sync": ["Syncing...", "Uploading changes...", "Processing files..."],
+    "search": ["Searching...", "Finding matches...", "Querying knowledge..."],
+    "upgrade": ["Checking for updates...", "Upgrading...", "Installing..."],
+    "fetch": ["Fetching...", "Loading data...", "Retrieving..."],
+    "connect": ["Connecting...", "Establishing connection...", "Reaching server..."],
+    "process": ["Processing...", "Working...", "Computing..."],
+    "default": ["Working...", "Please wait...", "Processing..."],
+}
+
+
+def print_logo(subtitle: str | None = None, compact: bool = False):
+    """Print the Nia header (no logo)."""
+    pass
+
+
+def print_header(title: str, subtitle: str | None = None):
+    """Print a styled header with optional subtitle."""
+    console.print(f"\n{NIA_LOGO_MINI} [bold cyan]{title}[/bold cyan]")
+    if subtitle:
+        console.print(f"  [dim]{subtitle}[/dim]")
+    console.print()
+
+
+@contextmanager
+def spinner(action: str = "default", message: str | None = None, min_time: float = 0.4):
+    """
+    Context manager for showing a spinner during operations.
+    
+    Usage:
+        with spinner("sync", "Syncing files..."):
+            do_sync()
+    """
+    import time
+    from rich.status import Status
+    
+    # Pick message
+    if message is None:
+        messages = SPINNER_MESSAGES.get(action, SPINNER_MESSAGES["default"])
+        message = messages[0]
+    
+    # Pick a good spinner
+    spinner_name = "dots"
+    start_time = time.time()
+    
+    with Status(f"[bold cyan]{message}[/bold cyan]", spinner=spinner_name, console=console) as status:
+        # Expose update method through a simple interface
+        status._message = message
+        yield status
+        
+        # Ensure spinner is visible for at least min_time
+        elapsed = time.time() - start_time
+        if elapsed < min_time:
+            time.sleep(min_time - elapsed)
+
+
+def update_spinner(status, message: str):
+    """Update the spinner message."""
+    status.update(f"[bold cyan]{message}[/bold cyan]")
+
+
+def success(message: str, prefix: str = "✓"):
+    """Print a success message."""
+    console.print(f"[green]{prefix}[/green] {message}")
+
+
+def error(message: str, prefix: str = "✗"):
+    """Print an error message."""
+    console.print(f"[red]{prefix}[/red] {message}")
+
+
+def warning(message: str, prefix: str = "⚠"):
+    """Print a warning message."""
+    console.print(f"[yellow]{prefix}[/yellow] {message}")
+
+
+def info(message: str, prefix: str = "•"):
+    """Print an info message."""
+    console.print(f"[cyan]{prefix}[/cyan] {message}")
+
+
+def dim(message: str):
+    """Print a dimmed message."""
+    console.print(f"[dim]{message}[/dim]")
+
+
+def print_welcome():
+    """Print the welcome message."""
+    console.print("[bold cyan]Nia Sync Engine[/bold cyan] — Keep local folders in sync with Nia cloud\n")

nia_sync-0.1.6.dist-info/RECORD

@@ -1,11 +0,0 @@
-auth.py,sha256=n0ezRqIbz3kZYcyIjH47_MDKgwNgAaVr0Y2NEfRxa50,5704
-config.py,sha256=dbnkDihOiJL0-WBzTyxDXM2AIduwsvZycgQuhAoYglU,7637
-extractor.py,sha256=IxpFGOlklusmPs2Jz398DH1hVVYpciwQN_HDyxpmm_Q,29193
-main.py,sha256=f1e2q99QfI_U0YyYgsC75P1wHebKuweWa0pYEAQPMO8,26582
-sync.py,sha256=jdVPceViN5uxoRCr5ukCVumBhvl3t53xkJnvZwvKmdw,10710
-watcher.py,sha256=JmsN9uR7Ss1mDC-kApXL6Hg_wQZWTsO7rRIFkQu8GbM,9978
-nia_sync-0.1.6.dist-info/METADATA,sha256=BpVvABRhWNRG4dcuFh06zVr7e7Zba2jh95GcR7lZOts,246
-nia_sync-0.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-nia_sync-0.1.6.dist-info/entry_points.txt,sha256=Fx8TIOgXqWdZzZEkEateDtcNfgnwuPW4jZTqlEUrHVs,33
-nia_sync-0.1.6.dist-info/top_level.txt,sha256=_ZWBugSHWwSpLXYJAcF6TlWmzECu18k0y1-EX27jtBw,40
-nia_sync-0.1.6.dist-info/RECORD,,