repr-cli 0.2.16__py3-none-any.whl → 0.2.18__py3-none-any.whl

repr/storage.py

@@ -557,3 +557,331 @@ def get_storage_stats() -> dict[str, Any]:
         },
         "total_size_bytes": stories_size + profiles_size + cache_size + config_size,
     }
+
+
+# ============================================================================
+# ReprStore - Project-level storage for .repr/ directory
+# ============================================================================
+
+REPR_DIR_NAME = ".repr"
+STORE_FILE = "store.json"
+
+
+def get_repr_store_path(project_path: Path) -> Path:
+    """Get the .repr/store.json path for a project."""
+    return project_path / REPR_DIR_NAME / STORE_FILE
+
+
+def save_repr_store(store: "ReprStore", project_path: Path) -> Path:
+    """
+    Save ReprStore to SQLite.
+
+    Note: This function now writes to SQLite only. The .repr/store.json
+    format is deprecated. Use `repr data migrate-db` to import old stores.
+
+    Args:
+        store: ReprStore to save
+        project_path: Project root path
+
+    Returns:
+        Path to the .repr directory
+    """
+    repr_dir = project_path / REPR_DIR_NAME
+    repr_dir.mkdir(parents=True, exist_ok=True)
+
+    # Write to SQLite
+    if store.stories:
+        save_stories_to_db(store.stories, project_path)
+
+    return repr_dir
+
+
+def load_repr_store(project_path: Path) -> "ReprStore | None":
+    """
+    Load ReprStore from .repr/store.json.
+    
+    Args:
+        project_path: Project root path
+    
+    Returns:
+        ReprStore or None if not found
+    """
+    from .models import ReprStore as ReprStoreModel
+    
+    store_path = get_repr_store_path(project_path)
+    
+    if not store_path.exists():
+        return None
+    
+    try:
+        store_json = store_path.read_text()
+        return ReprStoreModel.model_validate_json(store_json)
+    except Exception:
+        return None
+
+
+def create_repr_store(project_path: Path) -> "ReprStore":
+    """
+    Create a new empty ReprStore for a project.
+    
+    Args:
+        project_path: Project root path
+    
+    Returns:
+        New ReprStore instance
+    """
+    from datetime import datetime, timezone
+    from .models import ReprStore as ReprStoreModel, ContentIndex
+    
+    return ReprStoreModel(
+        project_path=str(project_path),
+        initialized_at=datetime.now(timezone.utc),
+        last_updated=datetime.now(timezone.utc),
+        commits=[],
+        sessions=[],
+        stories=[],
+        index=ContentIndex(),
+    )
+
+
+def update_repr_store_index(store: "ReprStore") -> None:
+    """
+    Rebuild the ContentIndex from stories in the store.
+    
+    Args:
+        store: ReprStore to update (mutates in place)
+    """
+    from datetime import datetime, timezone
+    from .models import ContentIndex, StoryDigest
+    
+    index = ContentIndex(
+        last_updated=datetime.now(timezone.utc),
+        story_count=len(store.stories),
+    )
+    
+    for story in store.stories:
+        # File → story mapping
+        for f in story.files:
+            if f not in index.files_to_stories:
+                index.files_to_stories[f] = []
+            index.files_to_stories[f].append(story.id)
+        
+        # Keyword extraction and mapping
+        import re
+        text = f"{story.title} {story.problem}".lower()
+        words = re.findall(r'\b[a-z]+\b', text)
+        stopwords = {'the', 'a', 'an', 'is', 'are', 'was', 'were', 'for', 'to', 'in', 'on', 'of', 'and', 'or', 'with', 'from'}
+        keywords = [w for w in words if len(w) > 2 and w not in stopwords]
+        
+        for kw in set(keywords):
+            if kw not in index.keywords_to_stories:
+                index.keywords_to_stories[kw] = []
+            index.keywords_to_stories[kw].append(story.id)
+        
+        # Weekly index
+        if story.started_at:
+            week = story.started_at.strftime("%Y-W%W")
+            if week not in index.by_week:
+                index.by_week[week] = []
+            index.by_week[week].append(story.id)
+        
+        # Story digest
+        index.story_digests.append(StoryDigest(
+            story_id=story.id,
+            title=story.title,
+            problem_keywords=list(set(keywords))[:10],
+            files=story.files[:5],
+            tech_stack=_detect_tech_stack(story.files),
+            category=story.category,
+            timestamp=story.started_at or story.created_at,
+        ))
+    
+    store.index = index
+
+
+def _detect_tech_stack(files: list[str]) -> list[str]:
+    """Detect technologies from file extensions."""
+    tech = set()
+
+    ext_map = {
+        '.py': 'Python', '.ts': 'TypeScript', '.tsx': 'React',
+        '.js': 'JavaScript', '.jsx': 'React', '.go': 'Go',
+        '.rs': 'Rust', '.vue': 'Vue', '.sql': 'SQL',
+    }
+
+    for f in files:
+        for ext, name in ext_map.items():
+            if f.endswith(ext):
+                tech.add(name)
+
+    return sorted(tech)
+
+
+# ============================================================================
+# SQLite Storage Functions
+# ============================================================================
+
+def save_stories_to_db(stories: list, project_path: Path) -> int:
+    """
+    Save stories to central SQLite database.
+
+    Args:
+        stories: List of Story objects
+        project_path: Path to the project
+
+    Returns:
+        Number of stories saved
+    """
+    from .db import get_db
+
+    db = get_db()
+    project = db.get_project_by_path(project_path)
+
+    if not project:
+        project_id = db.register_project(project_path, project_path.name)
+    else:
+        project_id = project["id"]
+
+    for story in stories:
+        db.save_story(story, project_id)
+
+    return len(stories)
+
+
+def load_stories_from_db(
+    project_path: Path | None = None,
+    category: str | None = None,
+    since: "datetime | None" = None,
+    limit: int = 100,
+) -> list:
+    """
+    Load stories from central SQLite database.
+
+    Args:
+        project_path: Optional project path filter
+        category: Optional category filter
+        since: Optional date filter
+        limit: Maximum stories to return
+
+    Returns:
+        List of Story objects
+    """
+    from .db import get_db
+
+    db = get_db()
+
+    project_id = None
+    if project_path:
+        project = db.get_project_by_path(project_path)
+        if project:
+            project_id = project["id"]
+
+    return db.list_stories(
+        project_id=project_id,
+        category=category,
+        since=since,
+        limit=limit,
+    )
+
+
+def search_stories_in_db(
+    query: str,
+    files: list[str] | None = None,
+    limit: int = 20,
+) -> list:
+    """
+    Search stories using FTS5.
+
+    Args:
+        query: Search query
+        files: Optional file paths to filter by
+        limit: Maximum results
+
+    Returns:
+        List of Story objects
+    """
+    from .db import get_db
+
+    db = get_db()
+    return db.search_stories(query, files=files, limit=limit)
+
+
+def get_story_from_db(story_id: str):
+    """
+    Get a story by ID from SQLite.
+
+    Args:
+        story_id: Story UUID
+
+    Returns:
+        Story object or None
+    """
+    from .db import get_db
+
+    db = get_db()
+    return db.get_story(story_id)
+
+
+def migrate_stores_to_db(
+    project_paths: list[Path] | None = None,
+    dry_run: bool = False,
+) -> dict:
+    """
+    Migrate existing store.json files to SQLite.
+
+    Args:
+        project_paths: List of project paths to migrate (None = all tracked)
+        dry_run: If True, don't actually write to DB
+
+    Returns:
+        Migration statistics
+    """
+    from .db import get_db
+    from .config import get_tracked_repos
+
+    db = get_db()
+
+    if project_paths is None:
+        # Get all tracked repos
+        tracked = get_tracked_repos()
+        project_paths = [Path(r["path"]) for r in tracked if Path(r["path"]).exists()]
+
+    stats = {
+        "projects_scanned": 0,
+        "projects_migrated": 0,
+        "stories_imported": 0,
+        "errors": [],
+    }
+
+    for project_path in project_paths:
+        stats["projects_scanned"] += 1
+
+        store = load_repr_store(project_path)
+        if not store:
+            continue
+
+        if not store.stories:
+            continue
+
+        if dry_run:
+            stats["projects_migrated"] += 1
+            stats["stories_imported"] += len(store.stories)
+            continue
+
+        try:
+            imported = db.import_from_store(store, project_path)
+            stats["projects_migrated"] += 1
+            stats["stories_imported"] += imported
+        except Exception as e:
+            stats["errors"].append(f"{project_path.name}: {str(e)}")
+
+    return stats
+
+
+def get_db_stats() -> dict:
+    """Get SQLite database statistics."""
+    from .db import get_db
+
+    db = get_db()
+    return db.get_stats()
+