feature: add support for ghe

README.md

@@ -45,6 +45,29 @@ jobs:
       ...
 ```
 
+## GHE Support
+
+For GHE support you just need to pass in `api-endpoint` as an input.
+
+```yml
+name: Publish
+
+on: [push]
+
+jobs:
+  publish:
+    name: Publish the package
+    runs-on: ubuntu-latest
+    steps:
+      - name: Wait for tests to succeed
+        uses: lewagon/wait-on-check-action@v1.0.0
+        with:
+          ref: ${{ github.ref }}
+          check-name: 'Run tests'
+          repo-token: ${{ secrets.GITHUB_TOKEN }}
+          api-endpoint: https://{YOU_GHE_URL}/api/v3/
+      ...
+```
 ## Alternatives
 
 If you can keep the dependent jobs in a single workflow:

action.yml

@@ -26,6 +26,10 @@ inputs:
     description: "Seconds to wait between Checks API requests"
     required: false
     default: "10"
+  api-endpoint:
+    description: "Github API Endpoint to use."
+    required: false
+    default: ""
   running-workflow-name:
     description: "Name of the workflow to be ignored (the one who is waiting for the rest)"
     required: false
@@ -68,6 +72,7 @@ runs:
         VERBOSE: ${{ inputs.verbose }}
         WAIT_INTERVAL: ${{ inputs.wait-interval }}
         RUNNING_WORKFLOW_NAME: ${{ inputs.running-workflow-name }}
+        API_ENDPOINT: ${{ inputs.api-endpoint }}
 branding:
   icon: "check-circle"
   color: "green"

entrypoint.rb

@@ -10,12 +10,13 @@
 verbose = ENV["VERBOSE"]
 wait = ENV["WAIT_INTERVAL"]
 workflow_name = ENV["RUNNING_WORKFLOW_NAME"]
+api_endpoint = ENV['API_ENDPOINT'] || 'https://api.github.com'
 
 GithubChecksVerifier.configure do |config|
   config.allowed_conclusions = allowed_conclusions.split(",").map(&:strip)
   config.check_name = check_name
   config.check_regexp = check_regexp
-  config.client = Octokit::Client.new(access_token: token)
+  config.client = Octokit::Client.new(access_token: token, api_endpoint: api_endpoint)
   config.ref = ref
   config.repo = ENV["GITHUB_REPOSITORY"]
   config.verbose = verbose