diff --git a/src/rocm_docs/projects.py b/src/rocm_docs/projects.py index 482c6afb..6c073135 100644 --- a/src/rocm_docs/projects.py +++ b/src/rocm_docs/projects.py @@ -324,11 +324,11 @@ def _update_theme_configs( ) -> None: """Update configurations for use in theme.py""" latest_version = requests.get( - "https://github.com/raw/RadeonOpenCompute/rocm-docs-core/header-versions/latest_version.txt" + "https://github.com/raw/ROCm/rocm-docs-core/header-versions/latest_version.txt" ).text.strip("\r\n") latest_version_string = f"docs-{latest_version}" release_candidate = requests.get( - "https://github.com/raw/RadeonOpenCompute/rocm-docs-core/header-versions/release_candidate.txt" + "https://github.com/raw/ROCm/rocm-docs-core/header-versions/release_candidate.txt" ).text.strip("\r\n") release_candidate_string = f"docs-{release_candidate}" diff --git a/src/rocm_docs/rocm_docs_theme/sections/header.html b/src/rocm_docs/rocm_docs_theme/sections/header.html index bdb862c7..7b5c0b31 100644 --- a/src/rocm_docs/rocm_docs_theme/sections/header.html +++ b/src/rocm_docs/rocm_docs_theme/sections/header.html @@ -19,7 +19,11 @@ AMD Logo
- {{ top_level_header(theme_repository_branch | replace("docs-", ""), theme_header_latest_version, theme_header_release_candidate_version) }} + {{ top_level_header( + theme_repository_branch | replace("docs-", ""), + header_latest_version, + header_release_candidate_version + ) }} {{ version_list() }}
diff --git a/src/rocm_docs/rocm_docs_theme/theme.conf b/src/rocm_docs/rocm_docs_theme/theme.conf index 92e883ce..13635ec0 100644 --- a/src/rocm_docs/rocm_docs_theme/theme.conf +++ b/src/rocm_docs/rocm_docs_theme/theme.conf @@ -9,6 +9,3 @@ show_toc_level = 1 flavor = rocm link_main_doc = True - -header_latest_version = 6.1.1 -header_release_candidate_version = 6.2.0 diff --git a/src/rocm_docs/theme.py b/src/rocm_docs/theme.py index 2c22bdab..f8ad196c 100644 --- a/src/rocm_docs/theme.py +++ b/src/rocm_docs/theme.py @@ -17,6 +17,36 @@ logger = sphinx.util.logging.getLogger(__name__) +def _get_version_from_url(url: str) -> str: + try: + response = requests.get(url) + return response.text.strip() + except requests.RequestException as e: + print(f"Error in rocm-docs-core _get_version_from_url: {e}") + return "" + + +def _add_custom_context( + app: Sphinx, # noqa: ARG001 + pagename: str, # noqa: ARG001 + templatename: str, # noqa: ARG001 + context: dict[str, str], + doctree: object, # noqa: ARG001 +) -> None: + header_latest_version = _get_version_from_url( + "https://github.com/raw/ROCm/rocm-docs-core/header-versions/latest_version.txt" + ) + + header_release_candidate_version = _get_version_from_url( + "https://github.com/raw/ROCm/rocm-docs-core/header-versions/release_candidate.txt" + ) + + context["header_latest_version"] = header_latest_version + context["header_release_candidate_version"] = ( + header_release_candidate_version + ) + + def _update_repo_opts(srcdir: str, theme_opts: dict[str, Any]) -> None: default_branch_options: dict[str, Any] = { "use_edit_page_button": False, @@ -88,12 +118,13 @@ def _update_theme_options(app: Sphinx) -> None: 0, "components/left-side-menu" ) - header_latest_version = requests.get( + header_latest_version = _get_version_from_url( "https://github.com/raw/RadeonOpenCompute/rocm-docs-core/header-versions/latest_version.txt" - ).text.strip("\r\n") - header_release_candidate_version = requests.get( + ) + + header_release_candidate_version = _get_version_from_url( "https://github.com/raw/RadeonOpenCompute/rocm-docs-core/header-versions/release_candidate.txt" - ).text.strip("\r\n") + ) default_config_opts = { "html_show_sphinx": False, @@ -101,8 +132,8 @@ def _update_theme_options(app: Sphinx) -> None: "notfound_context": {"title": "404 - Page Not Found"}, "notfound_template": "404.html", "html_context": { - "theme_header_latest_version": header_latest_version, - "theme_header_release_candidate_version": header_release_candidate_version, + "header_latest_version": header_latest_version, + "header_release_candidate_version": header_release_candidate_version, }, } for key, default in default_config_opts.items(): @@ -132,6 +163,7 @@ def setup(app: Sphinx) -> dict[str, Any]: ]: app.add_css_file(css) + app.connect("html-page-context", _add_custom_context) app.connect("builder-inited", _update_theme_options) return {