Merge pull request #19 from goshlanguage/configurable-github-api-1

Additional changes for Github enterprise support

Author: tchelovilar

README.md

@@ -12,6 +12,8 @@ Prometheus metrics exporter for github actions self-hosted runners.
 | REFRESH_INTERVAL     | No       | Internval time in seconds betwen api requests (Default: 20)
 | LOG_LEVEL            | No       | Log level: DEBUG, INFO, WARNING or ERROR (Default: INFO)
 | API_URL              | No       | URL to your github API (Default: https://api.github.com)
+| GITHUB_APP_ID        | No       | The Github app id to login as
+| GITHUB_PRIVATE_KEY   | No*      | The Github app private key generated from the app settings, required if GITHUB_APP_ID is set
 
 
 ## How to deploy

requirements.txt

@@ -1,4 +1,5 @@
+cryptography==42.0.5
 prometheus_client==0.15.0
 requests==2.28.2
-PyJWT==2.6.0
+PyJWT==2.8.0
 python-dateutil==2.8.1

runner_exporter/github_api.py

@@ -38,6 +38,7 @@ def __init__(
         self.github_app_id = github_app_id
         self.private_key = private_key
         self.github_owner = github_owner
+        self.api_url = api_url
         self.logger = logger
 
     def app_jwt_header(self):
@@ -152,7 +153,7 @@ def list_runners(self) -> list:
         headers = self.get_headers()
 
         per_page = 100
-        url = f"https://api.github.com/orgs/{self.github_owner}/actions/runners?per_page={per_page}"
+        url = f"{self.api_url}/orgs/{self.github_owner}/actions/runners?per_page={per_page}"
 
         while True:
             try:

runner_exporter/runner_exporter.py

@@ -156,17 +156,14 @@ def main():
     GITHUB_APP_ID = os.getenv("GITHUB_APP_ID")
     GITHUB_PRIVATE_KEY = os.getenv("GITHUB_PRIVATE_KEY")
     OWNER = os.getenv("OWNER")
-    API_URL = os.getenv("API_URL")
+    API_URL = os.getenv("API_URL", "https://api.github.com")
 
     # Start prometheus metrics
     logger.info("Starting metrics server")
     start_http_server(8000)
 
     runner_exports = runnerExports()
 
-    if API_URL == None or API_URL == "":
-        API_URL = "https://api.github.com"
-
     github = githubApi(
         OWNER,
         logger,