Add warmer

Author: gaetanlhf

functions/.env.example

@@ -1,2 +1,3 @@
 ACCESS_TOKEN=
-GITHUB_WEBHOOK_SECRET=
+GITHUB_WEBHOOK_SECRET=
+VERIFY_GITHUB_USER_AND_FETCH_SSH_KEYS_URL=

functions/index.js

@@ -5,12 +5,16 @@ const admin = require("firebase-admin");
 const {
   onRequest,
 } = require("firebase-functions/v2/https");
+const {
+  onSchedule,
+} = require("firebase-functions/v2/scheduler");
 
 admin.initializeApp();
 
 const handleGitHubStarsWebhookFunction = require("./src/handleGitHubStarsWebhook.js");
 const handleGitHubSponsorsWebhookFunction = require("./src/handleGitHubSponsorsWebhook.js");
 const verifyGitHubUserAndFetchSSHKeysFunction = require("./src/verifyGitHubUserAndFetchSSHKeys.js");
+const warmerFunction = require("./src/warmer.js");
 
 exports.handleGitHubStarsWebhook = onRequest({
   region: "europe-west9",
@@ -30,3 +34,9 @@ exports.verifyGitHubUserAndFetchSSHKeysFunction = onRequest({
   verifyGitHubUserAndFetchSSHKeysFunction.verifyGitHubUserAndFetchSSHKeys(req, res);
 }));
 
+exports.warmer = onSchedule({
+  region: "europe-west3",
+  schedule: "* * * * *",
+}, async (event) => {
+  warmerFunction.warmer(event);
+});

functions/src/verifyGitHubUserAndFetchSSHKeys.js

@@ -10,6 +10,11 @@ const {
 const logger = require("firebase-functions/logger");
 
 exports.verifyGitHubUserAndFetchSSHKeys = async (req, res) => {
+  if (req.query.isWarming === "true") {
+    logger.info("Warming request handled successfully");
+    return res.status(200).send("Warming request handled successfully");
+  }
+
   const username = req.query.username;
 
   try {

functions/src/warmer.js

@@ -0,0 +1,24 @@
+const axios = require("axios");
+
+const functionUrls = [
+  process.env.VERIFY_GITHUB_USER_AND_FETCH_SSH_KEYS_URL,
+];
+
+exports.warmer = async () => {
+  console.log("Starting to warm up functions...");
+  for (const url of functionUrls) {
+    try {
+      const response = await axios.get(url, {
+        headers: {
+          Authorization: process.env.ACCESS_TOKEN,
+        },
+        params: {
+          isWarming: "true",
+        },
+      });
+      console.log(`Warming up ${url}: Success with status ${response.status}`);
+    } catch (error) {
+      console.error(`Error warming up ${url}: ${error.message}`);
+    }
+  }
+};