wrkmon 1.0.0__py3-none-any.whl

wrkmon/cli.py

@@ -0,0 +1,289 @@
+"""CLI entry point for wrkmon using Typer."""
+
+import asyncio
+from typing import Optional
+
+import typer
+from rich.console import Console
+from rich.table import Table
+
+from wrkmon import __version__
+
+app = typer.Typer(
+    name="wrkmon",
+    help="Work Monitor - A developer productivity tool",
+    no_args_is_help=False,
+)
+console = Console()
+
+
+def version_callback(value: bool) -> None:
+    """Show version and exit."""
+    if value:
+        console.print(f"wrkmon version {__version__}")
+        raise typer.Exit()
+
+
+@app.callback(invoke_without_command=True)
+def main(
+    ctx: typer.Context,
+    version: bool = typer.Option(
+        False,
+        "--version",
+        "-v",
+        callback=version_callback,
+        is_eager=True,
+        help="Show version and exit",
+    ),
+) -> None:
+    """
+    Work Monitor - Monitor and manage system processes.
+
+    Run without arguments to launch the interactive monitor.
+    """
+    if ctx.invoked_subcommand is None:
+        # Launch TUI
+        from wrkmon.app import run_app
+        run_app()
+
+
+@app.command()
+def search(
+    query: str = typer.Argument(..., help="Search query"),
+    limit: int = typer.Option(10, "--limit", "-l", help="Maximum results"),
+) -> None:
+    """Search for processes to monitor."""
+    from wrkmon.core.youtube import YouTubeClient
+    from wrkmon.utils.stealth import get_stealth
+
+    stealth = get_stealth()
+
+    async def do_search():
+        client = YouTubeClient()
+        results = await client.search(query, max_results=limit)
+        return results
+
+    console.print(f"[dim]Searching for: {query}[/dim]\n")
+
+    results = asyncio.run(do_search())
+
+    if not results:
+        console.print("[yellow]No results found[/yellow]")
+        return
+
+    table = Table(title="Search Results")
+    table.add_column("#", style="dim")
+    table.add_column("Process Name")
+    table.add_column("PID", style="dim")
+    table.add_column("Duration")
+    table.add_column("Status", style="green")
+
+    for i, result in enumerate(results, 1):
+        process_name = stealth.get_fake_process_name(result.title)
+        fake_pid = str(stealth.get_fake_pid())
+        duration = result.duration_str
+        table.add_row(str(i), process_name[:50], fake_pid, duration, "READY")
+
+    console.print(table)
+    console.print(f"\n[dim]Use 'wrkmon play <id>' to start a process[/dim]")
+
+
+@app.command()
+def play(
+    video_id: str = typer.Argument(..., help="Video ID or URL to play"),
+) -> None:
+    """Play a specific process by ID or URL."""
+    import re
+
+    # Extract video ID from URL if needed
+    if "youtube.com" in video_id or "youtu.be" in video_id:
+        # Extract ID from URL
+        patterns = [
+            r"(?:v=|/)([a-zA-Z0-9_-]{11})",
+            r"youtu\.be/([a-zA-Z0-9_-]{11})",
+        ]
+        for pattern in patterns:
+            match = re.search(pattern, video_id)
+            if match:
+                video_id = match.group(1)
+                break
+
+    from wrkmon.core.youtube import YouTubeClient
+    from wrkmon.core.player import AudioPlayer
+    from wrkmon.core.cache import Cache
+    from wrkmon.utils.stealth import get_stealth
+
+    stealth = get_stealth()
+    cache = Cache()
+
+    async def do_play():
+        client = YouTubeClient()
+        player = AudioPlayer()
+
+        # Get stream info
+        console.print("[dim]Getting process info...[/dim]")
+
+        # Check cache
+        cached = cache.get(video_id)
+        if cached:
+            audio_url = cached.audio_url
+            title = cached.title
+        else:
+            stream_info = await client.get_stream_url(video_id)
+            if not stream_info:
+                console.print("[red]Failed to get process info[/red]")
+                return False
+
+            audio_url = stream_info.audio_url
+            title = stream_info.title
+
+            # Cache it
+            cache.set(
+                video_id=video_id,
+                title=stream_info.title,
+                channel=stream_info.channel,
+                duration=stream_info.duration,
+                audio_url=audio_url,
+            )
+
+        # Start player
+        console.print("[dim]Starting process...[/dim]")
+        if not await player.start():
+            console.print("[red]Failed to start player[/red]")
+            return False
+
+        # Play
+        if await player.play(audio_url):
+            process_name = stealth.get_fake_process_name(title)
+            console.print(f"[green]Running:[/green] {process_name}")
+            console.print("[dim]Press Ctrl+C to stop[/dim]")
+
+            # Keep running until interrupted
+            try:
+                while True:
+                    await asyncio.sleep(1)
+            except KeyboardInterrupt:
+                console.print("\n[yellow]Stopping process...[/yellow]")
+            finally:
+                await player.shutdown()
+
+            return True
+        else:
+            console.print("[red]Failed to start process[/red]")
+            return False
+
+    asyncio.run(do_play())
+
+
+@app.command()
+def queue() -> None:
+    """Show the current process queue."""
+    # For CLI, we show an empty queue message since queue is session-based
+    console.print("[dim]No processes in queue[/dim]")
+    console.print("[dim]Run 'wrkmon' to use the interactive monitor[/dim]")
+
+
+@app.command()
+def history(
+    limit: int = typer.Option(20, "--limit", "-l", help="Maximum entries to show"),
+) -> None:
+    """Show process history."""
+    from wrkmon.data.database import Database
+    from wrkmon.utils.stealth import get_stealth
+
+    stealth = get_stealth()
+    db = Database()
+
+    entries = db.get_history(limit=limit)
+
+    if not entries:
+        console.print("[dim]No history yet[/dim]")
+        return
+
+    table = Table(title="Process History")
+    table.add_column("#", style="dim")
+    table.add_column("Process Name")
+    table.add_column("Duration")
+    table.add_column("Runs", style="cyan")
+    table.add_column("Last Run", style="dim")
+
+    for i, entry in enumerate(entries, 1):
+        process_name = stealth.get_fake_process_name(entry.track.title)
+        duration = entry.track.duration_str
+        runs = str(entry.play_count)
+        last_run = entry.played_at.strftime("%Y-%m-%d %H:%M")
+        table.add_row(str(i), process_name[:40], duration, runs, last_run)
+
+    console.print(table)
+    db.close()
+
+
+@app.command()
+def playlists() -> None:
+    """List all playlists."""
+    from wrkmon.data.database import Database
+
+    db = Database()
+    playlists = db.get_all_playlists()
+
+    if not playlists:
+        console.print("[dim]No playlists yet[/dim]")
+        console.print("[dim]Run 'wrkmon' to create playlists[/dim]")
+        return
+
+    table = Table(title="Playlists")
+    table.add_column("#", style="dim")
+    table.add_column("Name")
+    table.add_column("Tracks", style="cyan")
+    table.add_column("Created", style="dim")
+
+    for i, playlist in enumerate(playlists, 1):
+        created = playlist.created_at.strftime("%Y-%m-%d") if playlist.created_at else "N/A"
+        table.add_row(str(i), playlist.name, str(playlist.track_count), created)
+
+    console.print(table)
+    db.close()
+
+
+@app.command()
+def clear_cache() -> None:
+    """Clear the URL cache."""
+    from wrkmon.core.cache import Cache
+
+    cache = Cache()
+    stats = cache.get_stats()
+    count = cache.clear()
+
+    console.print(f"[green]Cleared {count} cached entries[/green]")
+
+
+@app.command()
+def clear_history() -> None:
+    """Clear play history."""
+    from wrkmon.data.database import Database
+
+    db = Database()
+    count = db.clear_history()
+    db.close()
+
+    console.print(f"[green]Cleared {count} history entries[/green]")
+
+
+@app.command()
+def config() -> None:
+    """Show configuration info."""
+    from wrkmon.utils.config import get_config
+
+    cfg = get_config()
+
+    console.print("[bold]Configuration[/bold]\n")
+    console.print(f"Config directory: {cfg.config_dir}")
+    console.print(f"Data directory: {cfg.data_dir}")
+    console.print(f"Database: {cfg.database_path}")
+    console.print(f"Cache: {cfg.cache_path}")
+    console.print(f"\nVolume: {cfg.volume}%")
+    console.print(f"Cache TTL: {cfg.url_ttl_hours} hours")
+
+
+if __name__ == "__main__":
+    app()

wrkmon/core/__init__.py

@@ -0,0 +1,8 @@
+"""Core functionality for wrkmon."""
+
+from wrkmon.core.youtube import YouTubeClient
+from wrkmon.core.player import AudioPlayer
+from wrkmon.core.queue import PlayQueue
+from wrkmon.core.cache import Cache
+
+__all__ = ["YouTubeClient", "AudioPlayer", "PlayQueue", "Cache"]

wrkmon/core/cache.py

@@ -0,0 +1,208 @@
+"""Audio URL and metadata caching."""
+
+import json
+import sqlite3
+import time
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Optional
+
+from wrkmon.utils.config import get_config
+
+
+@dataclass
+class CachedStream:
+    """Cached stream information."""
+
+    video_id: str
+    title: str
+    channel: str
+    duration: int
+    audio_url: str
+    thumbnail_url: Optional[str]
+    cached_at: float
+    expires_at: float
+
+    @property
+    def is_expired(self) -> bool:
+        """Check if cache entry is expired."""
+        return time.time() > self.expires_at
+
+
+class Cache:
+    """Manages caching of audio URLs and metadata."""
+
+    def __init__(self, db_path: Optional[Path] = None):
+        config = get_config()
+        self._db_path = db_path or config.cache_path
+        self._ttl_hours = config.url_ttl_hours
+        self._max_entries = config.get("cache", "max_entries", 1000)
+        self._init_db()
+
+    def _init_db(self) -> None:
+        """Initialize the cache database."""
+        self._db_path.parent.mkdir(parents=True, exist_ok=True)
+
+        with sqlite3.connect(self._db_path) as conn:
+            conn.execute("""
+                CREATE TABLE IF NOT EXISTS stream_cache (
+                    video_id TEXT PRIMARY KEY,
+                    title TEXT NOT NULL,
+                    channel TEXT NOT NULL,
+                    duration INTEGER NOT NULL,
+                    audio_url TEXT NOT NULL,
+                    thumbnail_url TEXT,
+                    cached_at REAL NOT NULL,
+                    expires_at REAL NOT NULL
+                )
+            """)
+            conn.execute("""
+                CREATE INDEX IF NOT EXISTS idx_expires_at ON stream_cache(expires_at)
+            """)
+            conn.commit()
+
+    def get(self, video_id: str) -> Optional[CachedStream]:
+        """Get a cached stream by video ID."""
+        with sqlite3.connect(self._db_path) as conn:
+            conn.row_factory = sqlite3.Row
+            cursor = conn.execute(
+                """
+                SELECT * FROM stream_cache
+                WHERE video_id = ? AND expires_at > ?
+                """,
+                (video_id, time.time()),
+            )
+            row = cursor.fetchone()
+
+            if row:
+                return CachedStream(
+                    video_id=row["video_id"],
+                    title=row["title"],
+                    channel=row["channel"],
+                    duration=row["duration"],
+                    audio_url=row["audio_url"],
+                    thumbnail_url=row["thumbnail_url"],
+                    cached_at=row["cached_at"],
+                    expires_at=row["expires_at"],
+                )
+
+        return None
+
+    def set(
+        self,
+        video_id: str,
+        title: str,
+        channel: str,
+        duration: int,
+        audio_url: str,
+        thumbnail_url: Optional[str] = None,
+    ) -> CachedStream:
+        """Cache a stream URL."""
+        now = time.time()
+        expires_at = now + (self._ttl_hours * 3600)
+
+        with sqlite3.connect(self._db_path) as conn:
+            conn.execute(
+                """
+                INSERT OR REPLACE INTO stream_cache
+                (video_id, title, channel, duration, audio_url, thumbnail_url, cached_at, expires_at)
+                VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+                """,
+                (video_id, title, channel, duration, audio_url, thumbnail_url, now, expires_at),
+            )
+            conn.commit()
+
+        # Cleanup old entries if we're over the limit
+        self._cleanup()
+
+        return CachedStream(
+            video_id=video_id,
+            title=title,
+            channel=channel,
+            duration=duration,
+            audio_url=audio_url,
+            thumbnail_url=thumbnail_url,
+            cached_at=now,
+            expires_at=expires_at,
+        )
+
+    def delete(self, video_id: str) -> bool:
+        """Delete a cache entry."""
+        with sqlite3.connect(self._db_path) as conn:
+            cursor = conn.execute(
+                "DELETE FROM stream_cache WHERE video_id = ?",
+                (video_id,),
+            )
+            conn.commit()
+            return cursor.rowcount > 0
+
+    def clear(self) -> int:
+        """Clear all cache entries. Returns count deleted."""
+        with sqlite3.connect(self._db_path) as conn:
+            cursor = conn.execute("DELETE FROM stream_cache")
+            conn.commit()
+            return cursor.rowcount
+
+    def clear_expired(self) -> int:
+        """Clear expired cache entries. Returns count deleted."""
+        with sqlite3.connect(self._db_path) as conn:
+            cursor = conn.execute(
+                "DELETE FROM stream_cache WHERE expires_at <= ?",
+                (time.time(),),
+            )
+            conn.commit()
+            return cursor.rowcount
+
+    def _cleanup(self) -> None:
+        """Cleanup old entries if over the limit."""
+        with sqlite3.connect(self._db_path) as conn:
+            # First clear expired
+            conn.execute(
+                "DELETE FROM stream_cache WHERE expires_at <= ?",
+                (time.time(),),
+            )
+
+            # Then check count
+            cursor = conn.execute("SELECT COUNT(*) FROM stream_cache")
+            count = cursor.fetchone()[0]
+
+            if count > self._max_entries:
+                # Delete oldest entries
+                excess = count - self._max_entries
+                conn.execute(
+                    """
+                    DELETE FROM stream_cache
+                    WHERE video_id IN (
+                        SELECT video_id FROM stream_cache
+                        ORDER BY cached_at ASC
+                        LIMIT ?
+                    )
+                    """,
+                    (excess,),
+                )
+
+            conn.commit()
+
+    def get_stats(self) -> dict:
+        """Get cache statistics."""
+        with sqlite3.connect(self._db_path) as conn:
+            cursor = conn.execute("SELECT COUNT(*) FROM stream_cache")
+            total = cursor.fetchone()[0]
+
+            cursor = conn.execute(
+                "SELECT COUNT(*) FROM stream_cache WHERE expires_at > ?",
+                (time.time(),),
+            )
+            valid = cursor.fetchone()[0]
+
+            cursor = conn.execute(
+                "SELECT SUM(LENGTH(audio_url)) FROM stream_cache"
+            )
+            size = cursor.fetchone()[0] or 0
+
+        return {
+            "total_entries": total,
+            "valid_entries": valid,
+            "expired_entries": total - valid,
+            "approximate_size_bytes": size,
+        }