From 05c417da51da78b9ab9911c6e3b8987e9ff69141 Mon Sep 17 00:00:00 2001
From: Zach Lowry <zach.lowry@quantitech.com>
Date: Mon, 24 Jul 2023 10:13:34 -0500
Subject: [PATCH] add --builder argument, removed HTML-centrism

---
 pre_commit_sphinx/build_docs.py | 22 +++++++++++++---------
 setup.cfg                       |  2 +-
 2 files changed, 14 insertions(+), 10 deletions(-)

diff --git a/pre_commit_sphinx/build_docs.py b/pre_commit_sphinx/build_docs.py
index 82ec388..4189efd 100755
--- a/pre_commit_sphinx/build_docs.py
+++ b/pre_commit_sphinx/build_docs.py
@@ -1,7 +1,6 @@
 import argparse
 import os
-from typing import Optional
-from typing import Sequence
+from typing import Optional, Sequence
 
 
 def requires_build(filenames: Sequence[str], always_build: bool) -> bool:
@@ -11,15 +10,16 @@ def requires_build(filenames: Sequence[str], always_build: bool) -> bool:
         # Always return true for now (we rebuild with breathe/exhale so source code changes also require doc build)
         # In future allow rebuild e.g. only if files in the docs directory change
         pass
+
     return True
 
 
-def build(cache_dir: str, html_dir: str, src_dir: str):
+def build(builder: str, cache_dir: str, output_dir: str, src_dir: str):
     """ Invokes sphinx-build to build the docs
     """
 
     # Run Sphinx to build the documentation
-    ret = os.system(f'sphinx-build -b html -d {cache_dir} {src_dir} {html_dir}')
+    ret = os.system(f'sphinx-build -b {builder} -d {cache_dir} {src_dir} {output_dir}')
 
     # It's very weird that pre-commit marks this as 'PASSED' if I return an error code 512...! Workaround:
     return 0 if ret == 0 else 1
@@ -40,18 +40,22 @@ def main(argv: Optional[Sequence[str]] = None) -> int:
         '--cache-dir', type=str, default='docs/doctrees',
         help='Directory to cache doctrees for fast rebuild where there are few changes',
     )
-    parser.add_argument(
-        '--html-dir', type=str, default='docs/html',
-        help='Directory to output the build html',
-    )
     parser.add_argument(
         '--source-dir', type=str, default='docs/source',
         help='Directory containing documentation sources (where the conf.py file exists)',
     )
+    parser.add_argument(
+        '--output-dir', '--html-dir', type=str, default='docs/html',
+        help='Directory to output the documentation to',
+    )
+    parser.add_argument(
+        '--builder', type=str, default='html',
+        help='Builder to use to generate the documentation',
+    )
 
     args = parser.parse_args(argv)
     if requires_build(args.filenames, args.always_build):
-        return build(args.cache_dir, args.html_dir, args.source_dir)
+        return build(args.builder, args.cache_dir, args.output_dir, args.source_dir)
 
     return 0
 
diff --git a/setup.cfg b/setup.cfg
index 81c6e08..7984b09 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = pre_commit_sphinx
-version = 0.0.1
+version = 0.0.2
 description = Pre-commit hooks to check that sphinx docs build correctly
 long_description = file: README.md
 long_description_content_type = text/markdown