overleaf-pull 0.1.0__py3-none-any.whl

overleaf_sync/scheduler.py

@@ -0,0 +1,132 @@
+import os
+import platform
+import subprocess
+from .config import get_app_paths, get_config_path
+
+LAUNCHAGENTS_DIR = os.path.expanduser("~/Library/LaunchAgents")
+SYSTEMD_USER_DIR = os.path.expanduser("~/.config/systemd/user")
+
+PLIST_LABEL = "com.overleaf.sync"
+SERVICE_NAME = "overleaf-sync.service"
+TIMER_NAME = "overleaf-sync.timer"
+
+
+def _run(cmd: list[str]) -> subprocess.CompletedProcess:
+    return subprocess.run(cmd, check=False, capture_output=True, text=True)
+
+
+def _python_exec() -> str:
+    return os.environ.get("PYTHON", "python3")
+
+
+def _cli_entry() -> list[str]:
+    # Use module invocation to avoid packaging complexities
+    return [_python_exec(), "-m", "overleaf_sync.cli", "run-once"]
+
+
+def install_macos_launchagent(interval: str):
+    os.makedirs(LAUNCHAGENTS_DIR, exist_ok=True)
+    start_interval = {"1h": 3600, "12h": 43200, "24h": 86400}.get(interval, 3600)
+    support, logs_dir, _ = get_app_paths()
+    os.makedirs(logs_dir, exist_ok=True)
+    stdout = os.path.join(logs_dir, "runner.log")
+    stderr = os.path.join(logs_dir, "runner.err.log")
+
+    args = _cli_entry()
+    program_arguments_xml = "\n".join([f"\t\t<string>{a}</string>" for a in args])
+
+    plist = f"""
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+  <dict>
+    <key>Label</key>
+    <string>{PLIST_LABEL}</string>
+    <key>ProgramArguments</key>
+    <array>
+{program_arguments_xml}
+    </array>
+    <key>RunAtLoad</key>
+    <true/>
+    <key>StartInterval</key>
+    <integer>{start_interval}</integer>
+    <key>StandardOutPath</key>
+    <string>{stdout}</string>
+    <key>StandardErrorPath</key>
+    <string>{stderr}</string>
+  </dict>
+</plist>
+"""
+    plist_path = os.path.join(LAUNCHAGENTS_DIR, f"{PLIST_LABEL}.plist")
+    with open(plist_path, "w", encoding="utf-8") as f:
+        f.write(plist)
+    _run(["launchctl", "unload", "-w", plist_path])
+    res = _run(["launchctl", "load", "-w", plist_path])
+    if res.returncode == 0:
+        print(f"Installed LaunchAgent at {plist_path}")
+    else:
+        print(f"Failed to load LaunchAgent: {res.stderr}")
+
+
+def uninstall_macos_launchagent():
+    plist_path = os.path.join(LAUNCHAGENTS_DIR, f"{PLIST_LABEL}.plist")
+    _run(["launchctl", "unload", "-w", plist_path])
+    if os.path.exists(plist_path):
+        os.remove(plist_path)
+        print(f"Removed {plist_path}")
+
+
+def install_systemd_user(interval: str):
+    os.makedirs(SYSTEMD_USER_DIR, exist_ok=True)
+    args = " ".join(_cli_entry())
+    service = f"""
+[Unit]
+Description=Overleaf Sync pull-only job
+
+[Service]
+Type=oneshot
+ExecStart={args}
+"""
+    if interval == "1h":
+        on_calendar = "hourly"
+    elif interval == "12h":
+        on_calendar = "*-*-* 00,12:00:00"
+    else:
+        on_calendar = "daily"
+
+    timer = f"""
+[Unit]
+Description=Run Overleaf Sync periodically ({interval})
+
+[Timer]
+OnCalendar={on_calendar}
+Persistent=true
+RandomizedDelaySec=300
+
+[Install]
+WantedBy=timers.target
+"""
+    service_path = os.path.join(SYSTEMD_USER_DIR, SERVICE_NAME)
+    timer_path = os.path.join(SYSTEMD_USER_DIR, TIMER_NAME)
+    with open(service_path, "w", encoding="utf-8") as f:
+        f.write(service)
+    with open(timer_path, "w", encoding="utf-8") as f:
+        f.write(timer)
+
+    _run(["systemctl", "--user", "daemon-reload"])
+    _run(["systemctl", "--user", "disable", "--now", TIMER_NAME])
+    res = _run(["systemctl", "--user", "enable", "--now", TIMER_NAME])
+    if res.returncode == 0:
+        print(f"Installed systemd user timer at {timer_path}")
+    else:
+        print(f"Failed to enable timer: {res.stderr}")
+
+
+def uninstall_systemd_user():
+    _run(["systemctl", "--user", "disable", "--now", TIMER_NAME])
+    service_path = os.path.join(SYSTEMD_USER_DIR, SERVICE_NAME)
+    timer_path = os.path.join(SYSTEMD_USER_DIR, TIMER_NAME)
+    for p in (service_path, timer_path):
+        if os.path.exists(p):
+            os.remove(p)
+            print(f"Removed {p}")

overleaf_sync/sync.py

@@ -0,0 +1,113 @@
+import platform
+import os
+from datetime import datetime
+from .config import load_config, prompt_first_run, Config, get_logs_dir
+from .cookies import load_overleaf_cookies
+from .overleaf_api import create_api, list_projects_sorted_by_last_updated
+from .projects import folder_name_for, ensure_dir
+from .git_ops import (
+    clone_if_missing,
+    ensure_remote,
+    detect_default_branch,
+    pull_remote,
+    enable_git_helper,
+    is_worktree_clean,
+    has_unpushed_commits,
+)
+
+
+def run_sync(cfg: Config):
+    # Require Git token for all sync operations to ensure non-interactive background runs
+    if not cfg.git_token:
+        raise RuntimeError("Git token is required. Run 'overleaf-pull set-git-token' and retry.")
+    if cfg.git_helper:
+        enable_git_helper(platform.system())
+
+    ensure_dir(cfg.base_dir)
+    # Prefer cookies from config if present
+    if cfg.cookies:
+        cookies = cfg.cookies
+    else:
+        cookies = load_overleaf_cookies(cfg.browser, cfg.profile)
+    api = create_api(cfg.host)
+
+    projects = list_projects_sorted_by_last_updated(api, cookies, cfg.count)
+
+    for p in projects:
+        pid = p["id"]
+        name = p["name"]
+        folder = folder_name_for(name, pid)
+        repo_dir = os.path.join(cfg.base_dir, folder)
+        needs_clone = not os.path.isdir(os.path.join(repo_dir, ".git"))
+        if needs_clone and not cfg.git_token:
+            raise RuntimeError(
+                "Missing Overleaf Git token for cloning. Run 'overleaf-pull set-git-token' and retry."
+            )
+        repo_path = clone_if_missing(cfg.base_dir, folder, pid, cfg.git_token)
+        ensure_remote(repo_path, pid, cfg.git_token)
+        branch = detect_default_branch(repo_path)
+        pull_remote(repo_path, branch)
+    # After successful sync of latest set, automatically prune old projects safely
+    expected = {folder_name_for(p.get("name"), p.get("id")) for p in projects}
+    pruned = 0
+    lingering = 0
+    for entry in os.listdir(cfg.base_dir):
+        path = os.path.join(cfg.base_dir, entry)
+        if os.path.isdir(os.path.join(path, ".git")) and entry not in expected:
+            # Remove only if clean and with no unpushed commits
+            try:
+                branch = detect_default_branch(path)
+                clean = is_worktree_clean(path)
+                ahead = has_unpushed_commits(path, branch)
+                if clean and ahead is False:
+                    import shutil
+                    shutil.rmtree(path)
+                    pruned += 1
+                else:
+                    lingering += 1
+            except Exception:
+                lingering += 1
+    msg = f"[{datetime.now().isoformat(timespec='seconds')}] Synced {len(projects)} projects into {cfg.base_dir}"
+    if pruned or lingering:
+        msg += f"; pruned {pruned} old, {lingering} lingering"
+    print(msg)
+    # Append to app log for status checks
+    try:
+        logs_dir = get_logs_dir()
+        with open(os.path.join(logs_dir, "app.log"), "a", encoding="utf-8") as lf:
+            lf.write(msg + "\n")
+    except Exception:
+        pass
+
+
+def run_sync_validate_first(cfg: Config):
+    if cfg.git_helper:
+        enable_git_helper(platform.system())
+
+    ensure_dir(cfg.base_dir)
+    if cfg.cookies:
+        cookies = cfg.cookies
+    else:
+        cookies = load_overleaf_cookies(cfg.browser, cfg.profile)
+    api = create_api(cfg.host)
+    projects = list_projects_sorted_by_last_updated(api, cookies, 1)
+    if not projects:
+        raise RuntimeError("No projects found for validation.")
+    p = projects[0]
+    pid = p["id"]
+    name = p["name"]
+    folder = folder_name_for(name, pid)
+    repo_dir = os.path.join(cfg.base_dir, folder)
+    needs_clone = not os.path.isdir(os.path.join(repo_dir, ".git"))
+    if needs_clone and not cfg.git_token:
+        raise RuntimeError("Missing Overleaf Git token for cloning. Run 'overleaf-pull set-git-token'.")
+    repo_path = clone_if_missing(cfg.base_dir, folder, pid, cfg.git_token)
+    ensure_remote(repo_path, pid, cfg.git_token)
+    branch = detect_default_branch(repo_path)
+    pull_remote(repo_path, branch)
+    print(f"Validation sync OK for '{name}' ({pid}).")
+
+
+def run_sync_once():
+    cfg = load_config() or prompt_first_run()
+    run_sync(cfg)