firebase-tools 14.12.1 → 14.14.0

package/lib/gcp/runv2.js

@@ -0,0 +1,178 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.serviceFromEndpoint = exports.endpointFromService = exports.FIREBASE_FUNCTION_METADTA_ANNOTATION = exports.FUNCTION_SIGNATURE_TYPE_ENV = exports.FUNCTION_TARGET_ENV = exports.FUNCTION_ID_ANNOTATION = exports.FUNCTION_TARGET_ANNOTATION = exports.TRIGGER_TYPE_ANNOTATION = exports.CLIENT_NAME_LABEL = exports.RUNTIME_LABEL = exports.updateService = exports.submitBuild = exports.API_VERSION = void 0;
+const apiv2_1 = require("../apiv2");
+const error_1 = require("../error");
+const api_1 = require("../api");
+const proto = require("./proto");
+const metaprogramming_1 = require("../metaprogramming");
+const operation_poller_1 = require("../operation-poller");
+const backend = require("../deploy/functions/backend");
+const constants_1 = require("../functions/constants");
+const k8s_1 = require("./k8s");
+const supported_1 = require("../deploy/functions/runtimes/supported");
+const __1 = require("..");
+const functional_1 = require("../functional");
+exports.API_VERSION = "v2";
+const client = new apiv2_1.Client({
+    urlPrefix: (0, api_1.runOrigin)(),
+    auth: true,
+    apiVersion: exports.API_VERSION,
+});
+(0, metaprogramming_1.assertImplements)();
+async function submitBuild(projectId, location, build) {
+    const res = await client.post(`/projects/${projectId}/locations/${location}/builds`, build);
+    if (res.status !== 200) {
+        throw new error_1.FirebaseError(`Failed to submit build: ${res.status} ${res.body}`);
+    }
+    await (0, operation_poller_1.pollOperation)({
+        apiOrigin: (0, api_1.cloudbuildOrigin)(),
+        apiVersion: "v1",
+        operationResourceName: res.body.buildOperation,
+    });
+}
+exports.submitBuild = submitBuild;
+async function updateService(service) {
+    const fieldMask = proto.fieldMasks(service, "labels", "annotations", "tags");
+    fieldMask.push("template.revision");
+    const res = await client.post(service.name, service, {
+        queryParams: {
+            updateMask: fieldMask.join(","),
+        },
+    });
+    const svc = await (0, operation_poller_1.pollOperation)({
+        apiOrigin: (0, api_1.runOrigin)(),
+        apiVersion: exports.API_VERSION,
+        operationResourceName: res.body.name,
+    });
+    return svc;
+}
+exports.updateService = updateService;
+function functionNameToServiceName(id) {
+    return id.toLowerCase().replace(/_/g, "-");
+}
+exports.RUNTIME_LABEL = "goog-cloudfunctions-runtime";
+exports.CLIENT_NAME_LABEL = "goog-managed-by";
+exports.TRIGGER_TYPE_ANNOTATION = "cloudfunctions.googleapis.com/trigger-type";
+exports.FUNCTION_TARGET_ANNOTATION = "run.googleapis.com/build-function-target";
+exports.FUNCTION_ID_ANNOTATION = "cloudfunctions.googleapis.com/function-id";
+exports.FUNCTION_TARGET_ENV = "FUNCTION_TARGET";
+exports.FUNCTION_SIGNATURE_TYPE_ENV = "FUNCTION_SIGNATURE_TYPE";
+exports.FIREBASE_FUNCTION_METADTA_ANNOTATION = "firebase-functions-metadata";
+function endpointFromService(service) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _j;
+    const [, project, , location, , svcId] = service.name.split("/");
+    const metadata = JSON.parse(((_a = service.annotations) === null || _a === void 0 ? void 0 : _a[exports.FIREBASE_FUNCTION_METADTA_ANNOTATION]) || "{}");
+    const [env, secretEnv] = (0, functional_1.partition)(service.template.containers[0].env || [], (e) => "value" in e);
+    const id = metadata.functionId ||
+        ((_b = service.annotations) === null || _b === void 0 ? void 0 : _b[exports.FUNCTION_ID_ANNOTATION]) ||
+        ((_c = service.annotations) === null || _c === void 0 ? void 0 : _c[exports.FUNCTION_TARGET_ANNOTATION]) ||
+        ((_d = env.find((e) => e.name === exports.FUNCTION_TARGET_ENV)) === null || _d === void 0 ? void 0 : _d.value) ||
+        svcId;
+    const memory = (0, k8s_1.mebibytes)(service.template.containers[0].resources.limits.memory);
+    if (!backend.isValidMemoryOption(memory)) {
+        __1.logger.debug("Converting a service to an endpoint with an invalid memory option", memory);
+    }
+    const cpu = Number(service.template.containers[0].resources.limits.cpu);
+    const endpoint = {
+        platform: ((_e = service.labels) === null || _e === void 0 ? void 0 : _e[exports.CLIENT_NAME_LABEL]) === "cloud-functions" ? "gcfv2" : "run",
+        id,
+        project,
+        labels: service.labels || {},
+        region: location,
+        runtime: ((_f = service.labels) === null || _f === void 0 ? void 0 : _f[exports.RUNTIME_LABEL]) || (0, supported_1.latest)("nodejs"),
+        availableMemoryMb: memory,
+        cpu: cpu,
+        entryPoint: ((_g = env.find((e) => e.name === exports.FUNCTION_TARGET_ENV)) === null || _g === void 0 ? void 0 : _g.value) ||
+            ((_h = service.annotations) === null || _h === void 0 ? void 0 : _h[exports.FUNCTION_TARGET_ANNOTATION]) ||
+            ((_j = service.annotations) === null || _j === void 0 ? void 0 : _j[exports.FUNCTION_ID_ANNOTATION]) ||
+            id,
+        httpsTrigger: {},
+    };
+    proto.renameIfPresent(endpoint, service.template, "concurrency", "containerConcurrency");
+    proto.renameIfPresent(endpoint, service.labels || {}, "codebase", constants_1.CODEBASE_LABEL);
+    proto.renameIfPresent(endpoint, service.scaling || {}, "minInstances", "minInstanceCount");
+    proto.renameIfPresent(endpoint, service.scaling || {}, "maxInstances", "maxInstanceCount");
+    endpoint.environmentVariables = env.reduce((acc, e) => {
+        acc[e.name] = e.value;
+        return acc;
+    }, {});
+    endpoint.secretEnvironmentVariables = secretEnv.map((e) => {
+        const [, projectId, , secret] = e.valueSource.secretKeyRef.secret.split("/");
+        return {
+            key: e.name,
+            projectId,
+            secret,
+            version: e.valueSource.secretKeyRef.version || "latest",
+        };
+    });
+    return endpoint;
+}
+exports.endpointFromService = endpointFromService;
+function serviceFromEndpoint(endpoint, image) {
+    const labels = Object.assign(Object.assign({}, endpoint.labels), { [exports.RUNTIME_LABEL]: endpoint.runtime, [exports.CLIENT_NAME_LABEL]: "firebase-functions" });
+    delete labels["deployment-tool"];
+    if (endpoint.codebase) {
+        labels[constants_1.CODEBASE_LABEL] = endpoint.codebase;
+    }
+    const annotations = {
+        [exports.FIREBASE_FUNCTION_METADTA_ANNOTATION]: JSON.stringify({
+            functionId: endpoint.id,
+        }),
+    };
+    const template = {
+        containers: [
+            {
+                name: "worker",
+                image,
+                env: [
+                    ...Object.entries(endpoint.environmentVariables || {}).map(([name, value]) => ({
+                        name,
+                        value,
+                    })),
+                    ...(endpoint.secretEnvironmentVariables || []).map((secret) => ({
+                        name: secret.key,
+                        valueSource: {
+                            secretKeyRef: {
+                                secret: secret.secret,
+                                version: secret.version,
+                            },
+                        },
+                    })),
+                    {
+                        name: exports.FUNCTION_TARGET_ENV,
+                        value: endpoint.entryPoint,
+                    },
+                    {
+                        name: exports.FUNCTION_SIGNATURE_TYPE_ENV,
+                        value: backend.isEventTriggered(endpoint) ? "cloudevent" : "http",
+                    },
+                ],
+                resources: {
+                    limits: {
+                        cpu: String(endpoint.cpu || 1),
+                        memory: `${endpoint.availableMemoryMb || 256}Mi`,
+                    },
+                    cpuIdle: true,
+                    startupCpuBoost: true,
+                },
+            },
+        ],
+        containerConcurrency: endpoint.concurrency || backend.DEFAULT_CONCURRENCY,
+    };
+    proto.renameIfPresent(template, endpoint, "containerConcurrency", "concurrency");
+    const service = {
+        name: `projects/${endpoint.project}/locations/${endpoint.region}/services/${functionNameToServiceName(endpoint.id)}`,
+        labels,
+        annotations,
+        template,
+        client: "cli-firebase",
+    };
+    if (endpoint.minInstances || endpoint.maxInstances) {
+        service.scaling = {};
+        proto.renameIfPresent(service.scaling, endpoint, "minInstanceCount", "minInstances");
+        proto.renameIfPresent(service.scaling, endpoint, "maxInstanceCount", "maxInstances");
+    }
+    return service;
+}
+exports.serviceFromEndpoint = serviceFromEndpoint;

package/lib/gemini/fdcExperience.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.extractCodeBlock = exports.generateOperation = exports.chatWithFirebase = exports.generateSchema = void 0;
+exports.extractCodeBlock = exports.generateOperation = exports.chatWithFirebase = exports.generateSchema = exports.PROMPT_GENERATE_SEED_DATA = exports.PROMPT_GENERATE_CONNECTOR = void 0;
 const apiv2_1 = require("../apiv2");
 const api_1 = require("../api");
 const error_1 = require("../error");
@@ -9,6 +9,8 @@ const SCHEMA_GENERATOR_EXPERIENCE = "/appeco/firebase/fdc-schema-generator";
 const GEMINI_IN_FIREBASE_EXPERIENCE = "/appeco/firebase/firebase-chat/free";
 const OPERATION_GENERATION_EXPERIENCE = "/appeco/firebase/fdc-query-generator";
 const FIREBASE_CHAT_REQUEST_CONTEXT_TYPE_NAME = "type.googleapis.com/google.cloud.cloudaicompanion.v1main.FirebaseChatRequestContext";
+exports.PROMPT_GENERATE_CONNECTOR = "Create 4 operations for an app using the instance schema with proper authentication.";
+exports.PROMPT_GENERATE_SEED_DATA = "Create a mutation to populate the database with some seed data.";
 async function generateSchema(prompt, project, chatHistory = []) {
     const res = await apiClient.post(`/v1beta/projects/${project}/locations/global/instances/default:completeTask`, {
         input: { messages: [...chatHistory, { content: prompt, author: "USER" }] },
@@ -16,7 +18,7 @@ async function generateSchema(prompt, project, chatHistory = []) {
             experience: SCHEMA_GENERATOR_EXPERIENCE,
         },
     });
-    return res.body.output.messages[0].content;
+    return extractCodeBlock(res.body.output.messages[0].content);
 }
 exports.generateSchema = generateSchema;
 async function chatWithFirebase(prompt, project, chatHistory = []) {
@@ -42,7 +44,7 @@ async function generateOperation(prompt, service, project, chatHistory = []) {
             },
         },
     });
-    return res.body.output.messages[0].content;
+    return extractCodeBlock(res.body.output.messages[0].content);
 }
 exports.generateOperation = generateOperation;
 function extractCodeBlock(text) {