devpulse-tui 0.1.0__py3-none-any.whl

devpulse/activity.py

@@ -0,0 +1,50 @@
+import subprocess
+import time
+
+
+CODING_KEYWORDS = [
+    "code",
+    "nvim",
+    "vim",
+    "pycharm",
+    "idea",
+    "terminal",
+    "konsole",
+    "kitty",
+    "github",
+]
+
+
+session_start = time.time()
+
+
+def get_active_window():
+    try:
+        window_name = subprocess.check_output(
+            ["xdotool", "getactivewindow", "getwindowname"],
+            text=True
+        ).strip()
+
+        return window_name
+
+    except Exception:
+        return "Unknown"
+
+
+def is_coding(window_name):
+    window_name = window_name.lower()
+
+    return any(
+        keyword in window_name
+        for keyword in CODING_KEYWORDS
+    )
+
+
+def get_session_duration():
+    seconds = int(time.time() - session_start)
+
+    hours = seconds // 3600
+    minutes = (seconds % 3600) // 60
+    secs = seconds % 60
+
+    return f"{hours:02}:{minutes:02}:{secs:02}"

devpulse/database.py

@@ -0,0 +1,75 @@
+from datetime import datetime
+from sqlalchemy import (
+    create_engine,
+    Column,
+    Integer,
+    Float,
+    String,
+)
+
+from sqlalchemy.orm import (
+    declarative_base,
+    sessionmaker,
+)
+
+Base = declarative_base()
+
+
+class SystemStat(Base):
+    __tablename__ = "system_stats"
+
+    id = Column(Integer, primary_key=True)
+    timestamp = Column(
+        String,
+        default=lambda: datetime.now().isoformat()
+    )
+    cpu = Column(Float)
+    ram = Column(Float)
+    disk = Column(Float)
+
+    gpu = Column(String)
+
+    upload = Column(Float)
+    download = Column(Float)
+
+    repo = Column(String)
+    branch = Column(String)
+
+
+engine = create_engine(
+    "sqlite:///devpulse.db"
+)
+
+SessionLocal = sessionmaker(bind=engine)
+
+
+def init_db():
+    Base.metadata.create_all(engine)
+
+def log_system_stats(
+    cpu,
+    ram,
+    disk,
+    gpu,
+    upload,
+    download,
+    repo,
+    branch,
+):
+    session = SessionLocal()
+
+    stat = SystemStat(
+        cpu=cpu,
+        ram=ram,
+        disk=disk,
+        gpu=gpu,
+        upload=upload,
+        download=download,
+        repo=repo,
+        branch=branch,
+    )
+
+    session.add(stat)
+    session.commit()
+
+    session.close()

devpulse/devinfo.py

@@ -0,0 +1,240 @@
+import os
+import subprocess
+from pathlib import Path
+from datetime import datetime
+from git import Repo
+
+
+def find_git_repo():
+    current = Path.cwd()
+
+    for parent in [current] + list(current.parents):
+        if (parent / ".git").exists():
+            return parent
+
+    return None
+
+
+def get_git_info():
+    try:
+        repo = subprocess.check_output(
+            ["git", "rev-parse", "--show-toplevel"],
+            stderr=subprocess.DEVNULL
+        ).decode().strip().split("/")[-1]
+
+        branch = subprocess.check_output(
+            ["git", "branch", "--show-current"],
+            stderr=subprocess.DEVNULL
+        ).decode().strip()
+
+        status_output = subprocess.check_output(
+            ["git", "status", "--short"],
+            stderr=subprocess.DEVNULL
+        ).decode().splitlines()
+
+        commit_msg = subprocess.check_output(
+            ["git", "log", "-1", "--pretty=%s"],
+            stderr=subprocess.DEVNULL
+        ).decode().strip()
+
+        author = subprocess.check_output(
+            ["git", "log", "-1", "--pretty=%an"],
+            stderr=subprocess.DEVNULL
+        ).decode().strip()
+
+        commit_age = subprocess.check_output(
+            ["git", "log", "-1", "--pretty=%cr"],
+            stderr=subprocess.DEVNULL
+        ).decode().strip()
+
+        insertions = 0
+        deletions = 0
+        untracked = 0
+
+        for line in status_output:
+            if line.startswith("??"):
+                untracked += 1
+
+        diff_stats = subprocess.check_output(
+            ["git", "diff", "--shortstat"],
+            stderr=subprocess.DEVNULL
+        ).decode().strip()
+
+        if "insertion" in diff_stats:
+            insertions = diff_stats.split("insertion")[0].split()[-1]
+
+        if "deletion" in diff_stats:
+            deletions = diff_stats.split("deletion")[0].split()[-1]
+
+        return {
+            "repo": repo,
+            "branch": branch,
+            "status": "DIRTY" if status_output else "CLEAN",
+            "untracked": str(untracked),
+            "insertions": f"+{insertions}",
+            "deletions": f"-{deletions}",
+            "author": author,
+            "commit_age": commit_age,
+            "last_commit": commit_msg,
+            "files_changed": str(len(status_output))
+        }
+
+    except Exception:
+        return {
+            "repo": "Unknown",
+            "branch": "Unknown",
+            "status": "Unknown",
+            "untracked": "0",
+            "insertions": "0",
+            "deletions": "0",
+            "author": "Unknown",
+            "commit_age": "Unknown",
+            "last_commit": "Unknown",
+            "files_changed": "0"
+        }
+
+def get_shell():
+    return os.environ.get("SHELL", "Unknown")
+
+
+def get_terminal():
+    return os.environ.get("TERM", "Unknown")
+
+
+def get_current_directory():
+    return str(Path.cwd())
+
+def get_last_commit():
+    repo_path = find_git_repo()
+
+    if not repo_path:
+        return "No Repo"
+
+    try:
+        repo = Repo(repo_path)
+
+        commit = repo.head.commit.message.strip()
+
+        return commit[:40]
+
+    except Exception:
+        return "Unknown"
+
+
+def get_changed_files_count():
+    repo_path = find_git_repo()
+
+    if not repo_path:
+        return 0
+
+    try:
+        repo = Repo(repo_path)
+
+        changed_files = repo.index.diff(None)
+
+        return len(changed_files)
+
+    except Exception:
+        return 0
+
+
+def get_repo_status():
+    repo_path = find_git_repo()
+
+    if not repo_path:
+        return "NO REPO"
+
+    try:
+        repo = Repo(repo_path)
+
+        return "DIRTY" if repo.is_dirty() else "CLEAN"
+
+    except Exception:
+        return "UNKNOWN"
+
+
+def get_untracked_files_count():
+    repo_path = find_git_repo()
+
+    if not repo_path:
+        return 0
+
+    try:
+        repo = Repo(repo_path)
+
+        return len(repo.untracked_files)
+
+    except Exception:
+        return 0
+
+
+def get_insertions_deletions():
+    repo_path = find_git_repo()
+
+    if not repo_path:
+        return 0, 0
+
+    try:
+        repo = Repo(repo_path)
+
+        diff = repo.git.diff("--shortstat")
+
+        insertions = 0
+        deletions = 0
+
+        import re
+
+        insert_match = re.search(r'(\d+) insertion', diff)
+        delete_match = re.search(r'(\d+) deletion', diff)
+
+        if insert_match:
+            insertions = int(insert_match.group(1))
+
+        if delete_match:
+            deletions = int(delete_match.group(1))
+
+        return insertions, deletions
+
+    except Exception:
+        return 0, 0
+
+
+def get_last_commit_author():
+    repo_path = find_git_repo()
+
+    if not repo_path:
+        return "Unknown"
+
+    try:
+        repo = Repo(repo_path)
+
+        return repo.head.commit.author.name
+
+    except Exception:
+        return "Unknown"
+
+
+def get_last_commit_age():
+    repo_path = find_git_repo()
+
+    if not repo_path:
+        return "Unknown"
+
+    try:
+        repo = Repo(repo_path)
+
+        commit_time = datetime.fromtimestamp(
+            repo.head.commit.committed_date
+        )
+
+        delta = datetime.now() - commit_time
+
+        hours = delta.seconds // 3600
+
+        if delta.days > 0:
+            return f"{delta.days}d ago"
+
+        return f"{hours}h ago"
+
+    except Exception:
+        return "Unknown"

devpulse/doctor.py

@@ -0,0 +1,71 @@
+import os
+import sqlite3
+import importlib.util
+
+from rich.console import Console
+from rich.table import Table
+
+
+console = Console()
+
+
+def check_package(package_name):
+    return importlib.util.find_spec(package_name) is not None
+
+
+def run_doctor():
+    table = Table(title="DevPulse Doctor")
+
+    table.add_column("Check", style="cyan")
+    table.add_column("Status", style="green")
+
+    # Database check
+    db_exists = os.path.exists("devpulse.db")
+
+    table.add_row(
+        "SQLite Database",
+        "OK" if db_exists else "MISSING"
+    )
+
+    # SQLite readable
+    try:
+        conn = sqlite3.connect("devpulse.db")
+        conn.execute("SELECT 1")
+        conn.close()
+
+        table.add_row(
+            "Database Access",
+            "OK"
+        )
+
+    except Exception:
+        table.add_row(
+            "Database Access",
+            "FAILED"
+        )
+
+    # Dependencies
+    dependencies = [
+        "psutil",
+        "rich",
+        "sqlalchemy",
+        "git",
+    ]
+
+    for dep in dependencies:
+        installed = check_package(dep)
+
+        table.add_row(
+            f"Dependency: {dep}",
+            "OK" if installed else "MISSING"
+        )
+
+    # NVIDIA support
+    nvidia_available = check_package("py3nvml")
+
+    table.add_row(
+        "NVIDIA Support",
+        "ENABLED" if nvidia_available else "DISABLED"
+    )
+
+    console.print(table)