add proxy support for aws lambda

Author: decyjphr

handler.js

@@ -2,15 +2,16 @@ const {
   createLambdaFunction,
   createProbot
 } = require('@probot/adapter-aws-lambda-serverless')
+const { getProbotOctoKit } = require('lib/proxyAwareProbotOctokit')
 
 const appFn = require('./')
 
 module.exports.webhooks = createLambdaFunction(appFn, {
-  probot: createProbot()
+  probot: createProbot({ octokit: getProbotOctoKit() })
 })
 
 module.exports.scheduler = function () {
-  const probot = createProbot()
+  const probot = createProbot({ octokit: getProbotOctoKit() })
   const app = appFn(probot, {})
   return app.syncInstallation()
 }

lib/plugins/proxyAwareProbotOctokit.js

@@ -0,0 +1,49 @@
+// This custom plugin overrides the default ProbotOctokit plugin and support Http Proxy.
+const { Octokit } = require('@octokit/core')
+const { createProbotAuth } = require('octokit-auth-probot')
+const getProxiedFetch = require('../proxiedFetch')
+
+const ProbotOctokit = Octokit.plugin().defaults((instanceOptions) => {
+  const defaultOptions = {
+    authStrategy: createProbotAuth,
+    throttle: {
+      onSecondaryRateLimit: (
+        retryAfter,
+        options,
+        octokit
+      ) => {
+        octokit.log.warn(
+          `SecondaryRateLimit hit with "${options.method} ${options.url}", retrying in ${retryAfter} seconds.`
+        )
+        return true
+      },
+      onRateLimit: (
+        retryAfter,
+        options,
+        octokit
+      ) => {
+        octokit.log.warn(
+          `Rate limit hit with "${options.method} ${options.url}", retrying in ${retryAfter} seconds.`
+        )
+        return true
+      }
+    },
+    userAgent: 'probot',
+    request: {
+      fetch: getProxiedFetch(instanceOptions.baseUrl) || fetch
+    }
+  }
+  // merge options deeply
+  const options = Object.assign({}, defaultOptions, instanceOptions, {
+    request: Object.assign({}, defaultOptions.request, instanceOptions.request),
+    throttle: instanceOptions.throttle
+      ? Object.assign({}, defaultOptions.throttle, instanceOptions.throttle)
+      : defaultOptions.throttle
+  })
+  return options
+})
+
+function getProbotOctoKit () {
+  return ProbotOctokit
+}
+module.exports = { ProbotOctokit, getProbotOctoKit }

lib/proxiedFetch.js

@@ -0,0 +1,16 @@
+const { getProxyForUrl } = require('proxy-from-env')
+const { ProxyAgent, fetch: undiciFetch } = require('undici')
+
+export function getProxiedFetch (url) {
+  const proxyUrl = getProxyForUrl(url)
+  if (proxyUrl) {
+    // LOG.debug('Setting up proxy agent for ', url)
+    return (url, options) => {
+      return undiciFetch(url, {
+        ...options,
+        dispatcher: new ProxyAgent(proxyUrl)
+      })
+    }
+  }
+  return null
+}

package-lock.json

@@ -33,7 +33,9 @@
     "lodash": "^4.17.21",
     "node-cron": "^3.0.2",
     "octokit": "^4.1.2",
-    "probot": "^13.4.4"
+    "probot": "^13.4.4",
+    "proxy-from-env": "^1.1.0",
+    "undici": "^7.7.0"
   },
   "devDependencies": {
     "@eslint/eslintrc": "^3.1.0",