docwright 0.1.4__tar.gz → 0.1.6__tar.gz

{docwright-0.1.4 → docwright-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docwright
-Version: 0.1.4
+Version: 0.1.6
 Summary: AI-powered documentation agent: auto-generates and maintains README & wiki on every commit
 License: MIT
 License-File: LICENSE

{docwright-0.1.4 → docwright-0.1.6}/docwright/cli.py

@@ -29,10 +29,96 @@ def cli() -> None:
     """AI-powered documentation agent."""
 
 
+def run_interactive_setup(repo_root: Path) -> None:
+    """Run interactive setup wizard and write .ai-docgen/ai-docgen.yml."""
+    provider = click.prompt(
+        "AI Provider",
+        type=click.Choice(["claude", "openai", "ollama"]),
+        default="openai",
+        show_choices=True,
+    )
+    model_defaults = {"claude": "claude-opus-4-5", "openai": "gpt-4o", "ollama": "llama3"}
+    model = click.prompt("Model", default=model_defaults.get(provider, "gpt-4o"))
+    api_key_env_defaults = {
+        "claude": "ANTHROPIC_API_KEY",
+        "openai": "OPENAI_API_KEY",
+        "ollama": "",
+    }
+    api_key_env = click.prompt(
+        "API key environment variable",
+        default=api_key_env_defaults.get(provider, "OPENAI_API_KEY"),
+    )
+    language = click.prompt(
+        "Language",
+        type=click.Choice(["en", "ru"]),
+        default="en",
+        show_choices=True,
+    )
+    output_mode = click.prompt(
+        "Output mode",
+        type=click.Choice(["direct", "pull_request", "gitlab_wiki"]),
+        default="direct",
+        show_choices=True,
+    )
+
+    wiki_url: str | None = None
+    token_env: str | None = None
+    if output_mode == "gitlab_wiki":
+        wiki_url = click.prompt("GitLab Wiki URL")
+        token_env = click.prompt("GitLab token env variable", default="CI_JOB_TOKEN")
+
+    config_dir = repo_root / ".ai-docgen"
+    config_dir.mkdir(parents=True, exist_ok=True)
+    config_path = config_dir / "ai-docgen.yml"
+
+    output_section: dict[str, object] = {"mode": output_mode}
+    if wiki_url:
+        output_section["wiki_url"] = wiki_url
+    if token_env:
+        output_section["token_env"] = token_env
+
+    config_data: dict[str, object] = {
+        "provider": {
+            "type": provider,
+            "model": model,
+            "api_key_env": api_key_env,
+        },
+        "output": output_section,
+        "language": language,
+    }
+
+    import yaml
+
+    config_path.write_text(yaml.dump(config_data, default_flow_style=False, allow_unicode=True))
+
+    click.echo(f"\n✓ Config saved to {config_path.relative_to(repo_root)}\n")
+    click.echo("Set these environment variables before running:")
+    if api_key_env:
+        click.echo(f"  export {api_key_env}=...        # AI provider key")
+    if output_mode == "gitlab_wiki" and token_env:
+        click.echo(f"  export {token_env}=...          # GitLab token (if gitlab_wiki mode)")
+        click.echo("\nIn GitLab CI — add them to:")
+        click.echo("  Group → Settings → CI/CD → Variables")
+        click.echo("  (CI_JOB_TOKEN is provided automatically for GitLab token)")
+    else:
+        click.echo(
+            f"\n  export {api_key_env or 'OPENAI_API_KEY'}=...   # platform.openai.com → API Keys"
+        )
+        click.echo("\nIn GitLab CI — add to Group → Settings → CI/CD → Variables")
+
+    click.echo("\nProceeding with initialization...")
+
+
 @cli.command()
 def init() -> None:
     """Generate documentation from scratch."""
-    engine = build_engine(get_repo_root())
+    repo_root = get_repo_root()
+    from docwright.config import CONFIG_DIR, CONFIG_FILE
+
+    config_exists = (repo_root / CONFIG_DIR / CONFIG_FILE).exists()
+    if not config_exists:
+        run_interactive_setup(repo_root)
+    engine = build_engine(repo_root)
     asyncio.run(engine.init())
     click.echo("Documentation initialized.")
 

{docwright-0.1.4 → docwright-0.1.6}/docwright/config.py

@@ -19,9 +19,11 @@ class ProviderConfig(BaseModel):
 
 
 class OutputConfig(BaseModel):
-    mode: Literal["direct", "pr"] = "pr"
+    mode: str = "direct"
     pr_title: str = "docs: auto-update documentation"
     branch_prefix: str = "docs/auto-"
+    wiki_url: str | None = None
+    token_env: str | None = None
 
 
 class TemplatesConfig(BaseModel):

{docwright-0.1.4 → docwright-0.1.6}/docwright/engine.py

@@ -72,7 +72,7 @@ class DocsEngine:
         for doc_config in self.config.documents:
             target = self.repo_root / doc_config.target
             template_text = self.loader.load(doc_config.template)
-            document = target.read_text() if target.exists() else template_text
+            document = template_text
             section_names = self.renderer.auto_section_names(template_text)
 
             for section_name in section_names:

docwright-0.1.6/docwright/outputs/factory.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+import os
+from pathlib import Path
+
+from docwright.config import OutputConfig
+from docwright.outputs.base import Output
+from docwright.outputs.direct import DirectOutput
+from docwright.outputs.pull_request import PullRequestOutput
+
+
+def build_output(config: OutputConfig, repo_root: Path) -> Output:
+    if config.mode == "direct":
+        return DirectOutput(repo_root=repo_root)
+    elif config.mode == "gitlab_wiki":
+        token = os.getenv(config.token_env or "CI_JOB_TOKEN", "")
+        if not token:
+            raise OSError("GitLab token not found. Set CI_JOB_TOKEN or configure token_env.")
+        if not config.wiki_url:
+            raise ValueError("output.wiki_url is required for gitlab_wiki mode.")
+        from docwright.outputs.gitlab_wiki import GitLabWikiOutput
+
+        return GitLabWikiOutput(wiki_url=config.wiki_url, token=token, repo_root=repo_root)
+    return PullRequestOutput(repo_root=repo_root, config=config)

docwright-0.1.6/docwright/outputs/gitlab_wiki.py

@@ -0,0 +1,36 @@
+from __future__ import annotations
+
+import shutil
+import tempfile
+from pathlib import Path
+from urllib.parse import urlparse, urlunparse
+
+from git import Repo
+
+from docwright.outputs.base import Output
+
+
+class GitLabWikiOutput(Output):
+    def __init__(self, wiki_url: str, token: str, repo_root: Path) -> None:
+        self.wiki_url = wiki_url
+        self.token = token
+        self.repo_root = repo_root
+
+    def build_auth_url(self) -> str:
+        parsed = urlparse(self.wiki_url)
+        auth_netloc = f"oauth2:{self.token}@{parsed.netloc}"
+        return urlunparse(parsed._replace(netloc=auth_netloc))
+
+    def apply(self, changed_files: list[Path], message: str) -> None:
+        auth_url = self.build_auth_url()
+        tmp_dir = tempfile.mkdtemp()
+        try:
+            repo = Repo.clone_from(auth_url, tmp_dir)
+            for file in changed_files:
+                dest = Path(tmp_dir) / file.name
+                dest.write_bytes(file.read_bytes())
+            repo.index.add("*")
+            repo.index.commit(message)
+            repo.remotes.origin.push()
+        finally:
+            shutil.rmtree(tmp_dir, ignore_errors=True)

{docwright-0.1.4 → docwright-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docwright"
-version = "0.1.4"
+version = "0.1.6"
 description = "AI-powered documentation agent: auto-generates and maintains README & wiki on every commit"
 authors = ["Artem Gotlib <gotlib.artem.m@gmail.com>"]
 license = "MIT"

docwright-0.1.4/docwright/outputs/factory.py

@@ -1,12 +0,0 @@
-from pathlib import Path
-
-from docwright.config import OutputConfig
-from docwright.outputs.base import Output
-from docwright.outputs.direct import DirectOutput
-from docwright.outputs.pull_request import PullRequestOutput
-
-
-def build_output(config: OutputConfig, repo_root: Path) -> Output:
-    if config.mode == "direct":
-        return DirectOutput(repo_root=repo_root)
-    return PullRequestOutput(repo_root=repo_root, config=config)