From bc86b80ea3d3778b243a70bf35fec1ea178eab73 Mon Sep 17 00:00:00 2001
From: Sviatoslav Sydorenko <wk@sydorenko.org.ua>
Date: Mon, 17 Dec 2018 16:07:44 +0100
Subject: [PATCH] Enable alabaster render Travis CI badges for .com

Fixes #145
---
 alabaster/__init__.py | 44 +++++++++++++++++++++++++++++++++++++++++++
 alabaster/about.html  | 11 +++--------
 alabaster/theme.conf  |  1 +
 3 files changed, 48 insertions(+), 8 deletions(-)

diff --git a/alabaster/__init__.py b/alabaster/__init__.py
index c88b2b5..96804ca 100644
--- a/alabaster/__init__.py
+++ b/alabaster/__init__.py
@@ -1,5 +1,10 @@
 import os
 
+try:
+    import requests
+except ImportError:
+    pass
+
 from alabaster import _version as version
 
 
@@ -14,6 +19,45 @@ def get_path():
 def update_context(app, pagename, templatename, context, doctree):
     context["alabaster_version"] = version.__version__
 
+    set_up_travis_context(context)
+
+
+def set_up_travis_context(context):
+    """Add complete Travis URLs to Jinja2 context."""
+    github_slug = "/".join((context["theme_github_user"], context["theme_github_repo"]))
+
+    travis_button = str(context["theme_travis_button"]).lower()
+    travis_button_enabled = travis_button == "true"
+
+    travis_slug = github_slug if travis_button_enabled else travis_button
+
+    travis_tld = context["theme_travis_tld"].lower()
+    if travis_button_enabled and travis_tld == "auto":
+        try:
+            travis_api_response = requests.get(
+                "https://api.travis-ci.com/repo/{}".format(
+                    travis_slug.replace("/", "%2F")
+                ),
+                headers={
+                    "Travis-API-Version": "3",
+                    "User-Agent": "Sphinx-Alabaster-Theme/{version} "
+                    "(+https://github.com/bitprophet/alabaster)".format(
+                        version=version.__version__
+                    ),
+                },
+            )
+            is_travis_com_repo = 200 <= travis_api_response.status_code < 300
+            travis_tld = "com" if is_travis_com_repo else "org"
+        except NameError:
+            travis_tld = "com"
+    elif travis_tld != "com":
+        travis_tld = "org"
+    travis_base_uri = "travis-ci.{}/{}".format(travis_tld, travis_slug)
+    context["theme_travis_build_url"] = "https://{}".format(travis_base_uri)
+    context["theme_travis_badge_url"] = "https://api.{}.svg?branch={}".format(
+        travis_base_uri, context["theme_badge_branch"]
+    )
+
 
 def setup(app):
     # add_html_theme is new in Sphinx 1.6+
diff --git a/alabaster/about.html b/alabaster/about.html
index dbb4262..b264ed4 100644
--- a/alabaster/about.html
+++ b/alabaster/about.html
@@ -25,16 +25,11 @@ <h1 class="logo"><a href="{{ pathto(master_doc) }}">{{ project }}</a></h1>
 {% endif %}
 
 {% if theme_travis_button|lower != 'false' %}
-{% if theme_travis_button|lower == 'true' %}
-    {% set path = theme_github_user + '/' + theme_github_repo %}
-{% else %}
-    {% set path = theme_travis_button %}
-{% endif %}
 <p>
-<a class="badge" href="https://travis-ci.org/{{ path }}">
+<a class="badge" href="{{ theme_travis_build_url }}">
     <img
-        alt="https://secure.travis-ci.org/{{ path }}.svg?branch={{ theme_badge_branch }}"
-        src="https://secure.travis-ci.org/{{ path }}.svg?branch={{ theme_badge_branch }}"
+        alt="{{ theme_travis_badge_url }}"
+        src="{{ theme_travis_badge_url }}"
     />
 </a>
 </p>
diff --git a/alabaster/theme.conf b/alabaster/theme.conf
index b6cdc8d..c810c69 100644
--- a/alabaster/theme.conf
+++ b/alabaster/theme.conf
@@ -39,6 +39,7 @@ sidebar_width = 220px
 tidelift_url =
 touch_icon =
 travis_button = false
+travis_tld = org
 
 gray_1 = #444
 gray_2 = #EEE