86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
"""Refresh all stale documentation — relationships and repo summaries."""
|
|
|
|
import sys
|
|
import os
|
|
from pathlib import Path
|
|
|
|
# Load .env if present
|
|
_env_file = Path(__file__).parent / ".env"
|
|
if _env_file.exists():
|
|
for line in _env_file.read_text().splitlines():
|
|
line = line.strip()
|
|
if line and not line.startswith("#") and "=" in line:
|
|
key, _, val = line.partition("=")
|
|
os.environ.setdefault(key.strip(), val.strip())
|
|
|
|
sys.path.insert(0, os.path.dirname(__file__))
|
|
|
|
from db import GraphDB
|
|
from docgen import generate_relationship_doc, generate_repo_doc
|
|
|
|
REPO_DIR = os.environ.get("REPO_DIR", os.path.join(os.path.dirname(__file__), "repos", "target"))
|
|
|
|
|
|
def refresh_stale():
|
|
db = GraphDB()
|
|
|
|
print(f"\n{'='*60}")
|
|
print("Refreshing stale documentation")
|
|
print(f"{'='*60}")
|
|
|
|
stats_before = db.get_stats()
|
|
print(f"\nBefore: {stats_before['stale_relationships']} stale relationships")
|
|
|
|
# Refresh stale relationship docs
|
|
stale_rels = db.get_stale_relationships()
|
|
if stale_rels:
|
|
print(f"\n[1/2] Regenerating {len(stale_rels)} stale relationship docs...")
|
|
for i, rel in enumerate(stale_rels):
|
|
doc = generate_relationship_doc(
|
|
rel["from_file"], rel["from_doc"] or "",
|
|
rel["to_file"], rel["to_doc"] or "",
|
|
)
|
|
db.update_relationship_doc(rel["from_file"], rel["to_file"], doc)
|
|
if (i + 1) % 5 == 0 or (i + 1) == len(stale_rels):
|
|
print(f" Refreshed {i+1}/{len(stale_rels)}")
|
|
else:
|
|
print("\n[1/2] No stale relationships.")
|
|
|
|
# Refresh stale repo docs
|
|
stale_repos = db.get_stale_repos()
|
|
if stale_repos:
|
|
print(f"\n[2/2] Regenerating {len(stale_repos)} stale repo docs...")
|
|
for repo in stale_repos:
|
|
file_docs = db.get_repo_files_docs(repo["name"])
|
|
priority_names = ["echo.go", "router.go", "context.go", "group.go", "main.go", "server.go"]
|
|
entry_files = []
|
|
for name in priority_names:
|
|
for path, doc in file_docs:
|
|
if path.endswith(name) and doc:
|
|
entry_files.append((path, doc))
|
|
break
|
|
|
|
readme_path = os.path.join(REPO_DIR, "README.md")
|
|
readme = ""
|
|
if os.path.exists(readme_path):
|
|
with open(readme_path) as f:
|
|
readme = f.read()
|
|
|
|
repo_doc = generate_repo_doc(readme, entry_files)
|
|
db.update_repo_doc(repo["name"], repo_doc)
|
|
print(f" Refreshed repo: {repo['name']}")
|
|
else:
|
|
print("\n[2/2] No stale repos.")
|
|
|
|
stats_after = db.get_stats()
|
|
print(f"\n{'='*60}")
|
|
print("Refresh complete!")
|
|
print(f" Stale rels: {stats_before['stale_relationships']} → {stats_after['stale_relationships']}")
|
|
print(f"{'='*60}")
|
|
|
|
db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
refresh_stale()
|