Incorporate the spirit of changes from https://github.com/actions/con…

…figure-pages/pull/10
actions · Aug 5, 2022 · bce6391 · bce6391
1 parent 0a94d4c
commit bce6391
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 3 deletions.
diff --git a/src/api-client.js b/src/api-client.js
@@ -1,8 +1,12 @@
 const axios = require('axios')
 const core = require('@actions/core')
 
+function getApiBaseUrl() {
+  return process.env.GITHUB_API_URL || 'https://api.github.com'
+}
+
 async function enablePagesSite({ repositoryNwo, githubToken }) {
-  const pagesEndpoint = `https://api.github.com/repos/${repositoryNwo}/pages`
+  const pagesEndpoint = `${getApiBaseUrl()}/repos/${repositoryNwo}/pages`
 
   try {
     const response = await axios.post(
@@ -30,7 +34,7 @@ async function enablePagesSite({ repositoryNwo, githubToken }) {
 
 async function getPagesSite({ repositoryNwo, githubToken }) {
   try {
-    const pagesEndpoint = `https://api.github.com/repos/${repositoryNwo}/pages`
+    const pagesEndpoint = `${getApiBaseUrl()}/repos/${repositoryNwo}/pages`
 
     const response = await axios.get(pagesEndpoint, {
       headers: {
@@ -84,4 +88,4 @@ async function findOrCreatePagesSite({ repositoryNwo, githubToken, enablement =
   return pageObject
 }
 
-module.exports = { findOrCreatePagesSite, enablePagesSite, getPagesSite }
+module.exports = { findOrCreatePagesSite, enablePagesSite, getPagesSite, getApiBaseUrl }
diff --git a/src/api-client.test.js b/src/api-client.test.js
@@ -18,6 +18,27 @@ describe('apiClient', () => {
     jest.spyOn(core, 'debug').mockImplementation(jest.fn())
   })
 
+  describe('getApiBaseUrl', () => {
+    it('returns GITHUB_API_URL environment variable when set', async () => {
+      const expectedBaseUrl = 'https://api.ghe.com'
+      process.env.GITHUB_API_URL = expectedBaseUrl
+      const result = apiClient.getApiBaseUrl()
+      delete process.env.GITHUB_API_URL
+      expect(result).toEqual(expectedBaseUrl)
+    })
+
+    it('defaults to GitHub API if GITHUB_API_URL environment variable is empty', async () => {
+      process.env.GITHUB_API_URL = ''
+      const result = apiClient.getApiBaseUrl()
+      delete process.env.GITHUB_API_URL
+      expect(result).toEqual('https://api.github.com')
+    })
+
+    it('defaults to GitHub API if GITHUB_API_URL environment variable is not set', async () => {
+      const result = apiClient.getApiBaseUrl()
+      expect(result).toEqual('https://api.github.com')
+    })
+  })
 
   describe('enablePagesSite', () => {
     it('makes a request to create a page', async () => {