From 58d72c4f1939ae1af4e853c19b9e1cd1be3de8db Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=A9r=C3=B4me=20BOULMIER?= <jerome.boulmier@outlook.fr>
Date: Sun, 4 Aug 2019 20:58:25 +0200
Subject: [PATCH] feat(github): add support for github enterprise (#1841)

---
 include/modules/github.hpp | 3 ++-
 src/modules/github.cpp     | 7 ++++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/include/modules/github.hpp b/include/modules/github.hpp
index 845569d1..099eb57a 100644
--- a/include/modules/github.hpp
+++ b/include/modules/github.hpp
@@ -18,12 +18,13 @@ namespace modules {
     bool build(builder* builder, const string& tag) const;
 
    private:
-    void update_label(const int);
+    void update_label(int);
     int get_number_of_notification();
     string request();
     static constexpr auto TAG_LABEL = "<label>";
 
     label_t m_label{};
+    string m_api_url;
     string m_accesstoken{};
     unique_ptr<http_downloader> m_http{};
     bool m_empty_notifications{false};
diff --git a/src/modules/github.cpp b/src/modules/github.cpp
index dab58c93..eecaae6e 100644
--- a/src/modules/github.cpp
+++ b/src/modules/github.cpp
@@ -17,6 +17,11 @@ namespace modules {
   github_module::github_module(const bar_settings& bar, string name_)
       : timer_module<github_module>(bar, move(name_)), m_http(http_util::make_downloader()) {
     m_accesstoken = m_conf.get(name(), "token");
+    m_api_url = m_conf.get(name(), "api-url", "https://api.github.com/"s);
+    if (m_api_url.back() != '/') {
+      m_api_url += '/';
+    }
+
     m_interval = m_conf.get<decltype(m_interval)>(name(), "interval", 60s);
     m_empty_notifications = m_conf.get(name(), "empty-notifications", m_empty_notifications);
 
@@ -41,7 +46,7 @@ namespace modules {
   }
 
   string github_module::request() {
-    string content{m_http->get("https://api.github.com/notifications?access_token=" + m_accesstoken)};
+    string content{m_http->get(m_api_url + "notifications?access_token=" + m_accesstoken)};
 
     long response_code{m_http->response_code()};
     switch (response_code) {