mike_bin.py

# Copyright 2023 Aurora Operations, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os
import re
import subprocess
import sys

from unittest import mock
import mike.driver
import mike.mkdocs_utils
from mike.mkdocs_utils import docs_version_var


# We need to monkey patch any function in mike which calls `mkdocs` via its
# command line interface.  Fortunately, there are only two: `version` and
# `build`.


def override_version():
    """Override mike's native logic for retrieving the version."""
    # Original code:
    #    output = subprocess.run(
    #        ['mkdocs', '--version'],
    #        check=True, stdout=subprocess.PIPE, universal_newlines=True
    #    ).stdout.rstrip()
    #    m = re.search('^mkdocs, version (\\S*)', output)
    #    return m.group(1)
    # Changed version (reformatted with Black):
    output = subprocess.run(
        ["update_docs", "--version"],
        check=True,
        stdout=subprocess.PIPE,
        universal_newlines=True,
    ).stdout.rstrip()
    m = re.search("^\\S+, version (\\S*)", output)
    return m.group(1)


def override_build(config_file, version, *, quiet=False, output=None):
    """Override mike's native logic for building the docs."""
    # Original line:
    #    command = (
    #        ['mkdocs'] + (['--quiet'] if quiet else []) + ['build', '--clean'] +
    #        (['--config-file', config_file] if config_file else [])
    #    )
    # Changed version (reformatted with Black):
    command = (
        ["update_docs"]
        + (["--quiet"] if quiet else [])
        + ["build", "--clean"]
        + (["--config-file", config_file] if config_file else [])
    )

    env = os.environ.copy()
    env[docs_version_var] = version

    subprocess.run(command, check=True, env=env, stdout=output, stderr=output)


if __name__ == "__main__":
    # `update_docs` is our name for `mkdocs`.  It will be found in the same folder as this script.
    # We need to add this folder to the PATH.
    os.environ["PATH"] = f"{os.path.join(os.getcwd())}:{os.environ['PATH']}"

    with mock.patch("mike.mkdocs_utils.version", override_version):
        with mock.patch("mike.mkdocs_utils.build", override_build):
            sys.exit(mike.driver.main())