UNPKG

firebase-functions

Version:
156 lines (154 loc) 5.27 kB
import { __require } from "../_virtual/rolldown_runtime.mjs"; import { globalManifest } from "./manifest.mjs"; import { requiresAPI } from "../common/api.mjs"; import * as path from "path"; import * as url from "url"; //#region src/runtime/loader.ts /** * Dynamically load import function to prevent TypeScript from * transpiling into a require. * See https://github.com/microsoft/TypeScript/issues/43329. */ const dynamicImport = new Function("modulePath", "return import(modulePath)"); async function loadModule(functionsDir) { const absolutePath = path.resolve(functionsDir); try { return __require(path.resolve(absolutePath)); } catch (e) { if (e.code === "ERR_REQUIRE_ESM" || e.code === "ERR_REQUIRE_ASYNC_MODULE") { const modulePath = __require.resolve(absolutePath); const moduleURL = url.pathToFileURL(modulePath).href; return await dynamicImport(moduleURL); } throw e; } } function extractStack(module, endpoints, requiredAPIs, extensions, prefix = "") { for (const [name, valAsUnknown] of Object.entries(module)) { const val = valAsUnknown; if (typeof val === "function" && val.__endpoint && typeof val.__endpoint === "object") { const funcName = prefix + name; endpoints[funcName] = { ...val.__endpoint, entryPoint: funcName.replace(/-/g, ".") }; if (val.__requiredAPIs && Array.isArray(val.__requiredAPIs)) { requiredAPIs.push(...val.__requiredAPIs); } } else if (isFirebaseRefExtension(val)) { extensions[val.instanceId] = { params: convertExtensionParams(val.params), ref: val.FIREBASE_EXTENSION_REFERENCE, events: val.events || [] }; } else if (isFirebaseLocalExtension(val)) { extensions[val.instanceId] = { params: convertExtensionParams(val.params), localPath: val.FIREBASE_EXTENSION_LOCAL_PATH, events: val.events || [] }; } else if (isObject(val)) { extractStack(val, endpoints, requiredAPIs, extensions, prefix + name + "-"); } } } function toTitleCase(txt) { return txt.charAt(0).toUpperCase() + txt.substring(1).toLowerCase(); } function snakeToCamelCase(txt) { let ret = txt.toLowerCase(); ret = ret.replace(/_/g, " "); ret = ret.replace(/\w\S*/g, toTitleCase); ret = ret.charAt(0).toLowerCase() + ret.substring(1); return ret; } function convertExtensionParams(params) { const systemPrefixes = { FUNCTION: "firebaseextensions.v1beta.function", V2FUNCTION: "firebaseextensions.v1beta.v2function" }; const converted = {}; for (const [rawKey, paramVal] of Object.entries(params)) { let key = rawKey; if (rawKey.startsWith("_") && rawKey !== "_EVENT_ARC_REGION") { const prefix = rawKey.substring(1).split("_")[0]; const suffix = rawKey.substring(2 + prefix.length); key = `${systemPrefixes[prefix]}/${snakeToCamelCase(suffix)}`; } if (Array.isArray(paramVal)) { converted[key] = paramVal.join(","); } else { converted[key] = paramVal; } } return converted; } function isObject(value) { return typeof value === "object" && value !== null; } const isFirebaseLocalExtension = (val) => { return isObject(val) && typeof val.FIREBASE_EXTENSION_LOCAL_PATH === "string" && typeof val.instanceId === "string" && isObject(val.params) && (!val.events || Array.isArray(val.events)); }; const isFirebaseRefExtension = (val) => { return isObject(val) && typeof val.FIREBASE_EXTENSION_REFERENCE === "string" && typeof val.instanceId === "string" && isObject(val.params) && (!val.events || Array.isArray(val.events)); }; function mergeRequiredAPIs(requiredAPIs) { const apiToReasons = {}; for (const { api, reason } of requiredAPIs) { const reasons = apiToReasons[api] || new Set(); reasons.add(reason); apiToReasons[api] = reasons; } const merged = []; for (const [api, reasons] of Object.entries(apiToReasons)) { merged.push({ api, reason: Array.from(reasons).join(" ") }); } return merged; } async function loadStack(functionsDir) { const endpoints = {}; const requiredAPIs = []; const extensions = {}; const mod = await loadModule(functionsDir); extractStack(mod, endpoints, requiredAPIs, extensions); for (const req of requiredAPIs) { requiresAPI(req.api, req.reason); } const existingExtensions = globalManifest.extensions && typeof globalManifest.extensions === "object" ? globalManifest.extensions : {}; globalManifest.extensions = { ...extensions, ...existingExtensions }; if (!globalManifest.requiredAPIs) { globalManifest.requiredAPIs = []; } const legacySymbols = Object.getOwnPropertySymbols(globalThis).filter((sym) => (sym.description || sym.toString()).includes("firebase-functions:params:declaredParams:v")); for (const sym of legacySymbols) { const legacyParams = globalThis[sym]; if (Array.isArray(legacyParams)) { for (const p of legacyParams) { if (typeof p?.toSpec === "function") { const spec = p.toSpec(); if (!Array.isArray(globalManifest.params)) { globalManifest.params = []; } const manifestParams = globalManifest.params; if (!manifestParams.some((m) => m.name === spec.name)) { manifestParams.push(spec); } } } } } const stack = { specVersion: "v1alpha1", endpoints, ...globalManifest }; return stack; } //#endregion export { extractStack, loadStack, mergeRequiredAPIs };