firebase-tools 13.7.5 → 13.8.0

package/lib/emulator/controller.js

@@ -12,14 +12,8 @@ const registry_1 = require("./registry");
 const types_1 = require("./types");
 const constants_1 = require("./constants");
 const functionsEmulator_1 = require("./functionsEmulator");
-const auth_1 = require("./auth");
-const databaseEmulator_1 = require("./databaseEmulator");
-const firestoreEmulator_1 = require("./firestoreEmulator");
-const hostingEmulator_1 = require("./hostingEmulator");
-const eventarcEmulator_1 = require("./eventarcEmulator");
 const error_1 = require("../error");
 const projectUtils_1 = require("../projectUtils");
-const pubsubEmulator_1 = require("./pubsubEmulator");
 const commandUtils = require("./commandUtils");
 const hub_1 = require("./hub");
 const hubExport_1 = require("./hubExport");
@@ -31,10 +25,9 @@ const hubClient_1 = require("./hubClient");
 const prompt_1 = require("../prompt");
 const commandUtils_1 = require("./commandUtils");
 const fsutils_1 = require("../fsutils");
-const storage_1 = require("./storage");
 const config_1 = require("./storage/rules/config");
 const getDefaultDatabaseInstance_1 = require("../getDefaultDatabaseInstance");
-const auth_2 = require("../auth");
+const auth_1 = require("../auth");
 const extensionsEmulator_1 = require("./extensionsEmulator");
 const projectConfig_1 = require("../functions/projectConfig");
 const downloadableEmulators_1 = require("./downloadableEmulators");
@@ -42,6 +35,15 @@ const frameworks_1 = require("../frameworks");
 const experiments = require("../experiments");
 const portUtils_1 = require("./portUtils");
 const supported_1 = require("../deploy/functions/runtimes/supported");
+const auth_2 = require("./auth");
+const databaseEmulator_1 = require("./databaseEmulator");
+const eventarcEmulator_1 = require("./eventarcEmulator");
+const dataconnectEmulator_1 = require("./dataconnectEmulator");
+const firestoreEmulator_1 = require("./firestoreEmulator");
+const hostingEmulator_1 = require("./hostingEmulator");
+const pubsubEmulator_1 = require("./pubsubEmulator");
+const storage_1 = require("./storage");
+const fileUtils_1 = require("../dataconnect/fileUtils");
 const START_LOGGING_EMULATOR = utils.envOverride("START_LOGGING_EMULATOR", "false", (val) => val === "true");
 async function exportOnExit(options) {
     const exportOnExitDir = options.exportOnExit;
@@ -346,7 +348,7 @@ async function startAll(options, showUI = true, runningTestScript = false) {
         if (emulatorsNotRunning.length > 0 && !constants_1.Constants.isDemoProject(projectId)) {
             functionsLogger.logLabeled("WARN", "functions", `The following emulators are not running, calls to these services from the Functions emulator will affect production: ${clc.bold(emulatorsNotRunning.join(", "))}`);
         }
-        const account = (0, auth_2.getProjectDefaultAccount)(options.projectRoot);
+        const account = (0, auth_1.getProjectDefaultAccount)(options.projectRoot);
         const functionsEmulator = new functionsEmulator_1.FunctionsEmulator({
             projectId,
             projectDir,
@@ -486,13 +488,13 @@ async function startAll(options, showUI = true, runningTestScript = false) {
             throw new error_1.FirebaseError(`Cannot start the ${constants_1.Constants.description(types_1.Emulators.AUTH)} without a project: run 'firebase init' or provide the --project flag`);
         }
         const authAddr = legacyGetFirstAddr(types_1.Emulators.AUTH);
-        const authEmulator = new auth_1.AuthEmulator({
+        const authEmulator = new auth_2.AuthEmulator({
             host: authAddr.host,
             port: authAddr.port,
             projectId,
             singleProjectMode: singleProjectModeEnabled
-                ? auth_1.SingleProjectMode.WARNING
-                : auth_1.SingleProjectMode.NO_WARNING,
+                ? auth_2.SingleProjectMode.WARNING
+                : auth_2.SingleProjectMode.NO_WARNING,
         });
         await startEmulator(authEmulator);
         if (exportMetadata.auth) {
@@ -515,6 +517,30 @@ async function startAll(options, showUI = true, runningTestScript = false) {
         });
         await startEmulator(pubsubEmulator);
     }
+    if (listenForEmulator.dataconnect) {
+        const dataConnectAddr = legacyGetFirstAddr(types_1.Emulators.DATACONNECT);
+        const config = (0, fileUtils_1.readFirebaseJson)(options.config);
+        if (!config.length) {
+            throw new error_1.FirebaseError("No Data Connect service found in firebase.json");
+        }
+        else if (config.length > 1) {
+            logger_1.logger.warn(`TODO: Add support for multiple services in the Data Connect emulator. Currently emulating first service ${config[0].source}`);
+        }
+        let configDir = config[0].source;
+        if (!path.isAbsolute(configDir)) {
+            const cwd = options.cwd || process.cwd();
+            configDir = path.resolve(path.join(cwd), configDir);
+        }
+        const dataConnectEmulator = new dataconnectEmulator_1.DataConnectEmulator({
+            host: dataConnectAddr.host,
+            port: dataConnectAddr.port,
+            projectId,
+            auto_download: true,
+            configDir,
+            rc: options.rc,
+        });
+        await startEmulator(dataConnectEmulator);
+    }
     if (listenForEmulator.storage) {
         const storageAddr = legacyGetFirstAddr(types_1.Emulators.STORAGE);
         const storageEmulator = new storage_1.StorageEmulator({

package/lib/emulator/dataconnectEmulator.js

@@ -0,0 +1,86 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DataConnectEmulator = void 0;
+const childProcess = require("child_process");
+const api_1 = require("../api");
+const constants_1 = require("./constants");
+const downloadableEmulators_1 = require("./downloadableEmulators");
+const types_1 = require("./types");
+const error_1 = require("../error");
+const emulatorLogger_1 = require("./emulatorLogger");
+const types_2 = require("../dataconnect/types");
+class DataConnectEmulator {
+    constructor(args) {
+        this.args = args;
+        this.logger = emulatorLogger_1.EmulatorLogger.forEmulator(types_1.Emulators.DATACONNECT);
+    }
+    async start() {
+        const port = this.args.port || constants_1.Constants.getDefaultPort(types_1.Emulators.DATACONNECT);
+        this.logger.log("DEBUG", `Using Postgres connection string: ${this.getLocalConectionString()}`);
+        const info = await this.build();
+        if ((0, types_2.requiresVector)(info.metadata)) {
+            if (constants_1.Constants.isDemoProject(this.args.projectId)) {
+                this.logger.logLabeled("WARN", "Data Connect", "Detected a 'demo-' project, but vector embeddings require a real project. Operations that use vector_embed will fail.");
+            }
+            else {
+                this.logger.logLabeled("WARN", "Data Connect", "Operations that use vector_embed will make calls to production Vertex AI");
+            }
+        }
+        return (0, downloadableEmulators_1.start)(types_1.Emulators.DATACONNECT, Object.assign(Object.assign({}, this.args), { http_port: port, grpc_port: port + 1, config_dir: this.args.configDir, local_connection_string: this.getLocalConectionString(), project_id: this.args.projectId }));
+    }
+    connect() {
+        return Promise.resolve();
+    }
+    stop() {
+        return (0, downloadableEmulators_1.stop)(types_1.Emulators.DATACONNECT);
+    }
+    getInfo() {
+        const host = this.args.host || constants_1.Constants.getDefaultHost();
+        const port = this.args.port || constants_1.Constants.getDefaultPort(types_1.Emulators.DATACONNECT);
+        return {
+            name: this.getName(),
+            host,
+            port,
+            pid: (0, downloadableEmulators_1.getPID)(types_1.Emulators.DATACONNECT),
+        };
+    }
+    getName() {
+        return types_1.Emulators.DATACONNECT;
+    }
+    async generate(connectorId) {
+        const commandInfo = await (0, downloadableEmulators_1.downloadIfNecessary)(types_1.Emulators.DATACONNECT);
+        const cmd = [
+            "generate",
+            `--config_dir=${this.args.configDir}`,
+            `--connector_id=${connectorId}`,
+        ];
+        const res = childProcess.spawnSync(commandInfo.binary, cmd, { encoding: "utf-8" });
+        if (res.error) {
+            throw new error_1.FirebaseError(`Error starting up Data Connect emulator: ${res.error}`);
+        }
+        return res.stdout;
+    }
+    async build() {
+        var _a;
+        const commandInfo = await (0, downloadableEmulators_1.downloadIfNecessary)(types_1.Emulators.DATACONNECT);
+        const cmd = ["build", `--config_dir=${this.args.configDir}`];
+        const res = childProcess.spawnSync(commandInfo.binary, cmd, { encoding: "utf-8" });
+        if (res.stderr) {
+            throw new error_1.FirebaseError(`Unable to build your Data Connect schema and connectors: ${res.stderr}`);
+        }
+        try {
+            return JSON.parse(res.stdout);
+        }
+        catch (err) {
+            throw new error_1.FirebaseError(`Unable to parse 'fdc build' output: ${(_a = res.stdout) !== null && _a !== void 0 ? _a : res.stderr}`);
+        }
+    }
+    getLocalConectionString() {
+        var _a, _b;
+        if ((0, api_1.dataConnectLocalConnString)()) {
+            return (0, api_1.dataConnectLocalConnString)();
+        }
+        return (_b = (_a = this.args.rc.getDataconnect()) === null || _a === void 0 ? void 0 : _a.postgres) === null || _b === void 0 ? void 0 : _b.localConnectionString;
+    }
+}
+exports.DataConnectEmulator = DataConnectEmulator;

package/lib/emulator/download.js

@@ -15,7 +15,7 @@ async function downloadEmulator(name) {
     const emulator = downloadableEmulators.getDownloadDetails(name);
     emulatorLogger_1.EmulatorLogger.forEmulator(name).logLabeled("BULLET", name, `downloading ${path.basename(emulator.downloadPath)}...`);
     fs.ensureDirSync(emulator.opts.cacheDir);
-    const tmpfile = await downloadUtils.downloadToTmp(emulator.opts.remoteUrl);
+    const tmpfile = await downloadUtils.downloadToTmp(emulator.opts.remoteUrl, !!emulator.opts.auth);
     if (!emulator.opts.skipChecksumAndSize) {
         await validateSize(tmpfile, emulator.opts.expectedSize);
         await validateChecksum(tmpfile, emulator.opts.expectedChecksum);

package/lib/emulator/downloadableEmulators.js

@@ -44,6 +44,17 @@ const EMULATOR_UPDATE_DETAILS = {
         expectedSize: 65611398,
         expectedChecksum: "70bb840321423e6ae621a3ae2f314903",
     },
+    dataconnect: process.platform === "darwin"
+        ? {
+            version: "1.1.12",
+            expectedSize: 25542352,
+            expectedChecksum: "b523fe9d5506cc07480d1a571c72de9a",
+        }
+        : {
+            version: "1.1.12",
+            expectedSize: 23004624,
+            expectedChecksum: "e69f79755263b7cd77ec7cd15ef0d825",
+        },
 };
 exports.DownloadDetails = {
     database: {
@@ -107,6 +118,22 @@ exports.DownloadDetails = {
             namePrefix: "pubsub-emulator",
         },
     },
+    dataconnect: {
+        downloadPath: path.join(CACHE_DIR, `dataconnect-emulator-${EMULATOR_UPDATE_DETAILS.dataconnect.version}`),
+        version: EMULATOR_UPDATE_DETAILS.dataconnect.version,
+        binaryPath: path.join(CACHE_DIR, `dataconnect-emulator-${EMULATOR_UPDATE_DETAILS.dataconnect.version}`),
+        opts: {
+            cacheDir: CACHE_DIR,
+            remoteUrl: process.platform === "darwin"
+                ? `https://storage.googleapis.com/firemat-preview-drop/emulator/dataconnect-emulator-macos-v${EMULATOR_UPDATE_DETAILS.dataconnect.version}`
+                : `https://storage.googleapis.com/firemat-preview-drop/emulator/dataconnect-emulator-linux-v${EMULATOR_UPDATE_DETAILS.dataconnect.version}`,
+            expectedSize: EMULATOR_UPDATE_DETAILS.dataconnect.expectedSize,
+            expectedChecksum: EMULATOR_UPDATE_DETAILS.dataconnect.expectedChecksum,
+            skipChecksumAndSize: false,
+            namePrefix: "dataconnect-emulator",
+            auth: true,
+        },
+    },
 };
 const EmulatorDetails = {
     database: {
@@ -134,6 +161,11 @@ const EmulatorDetails = {
         instance: null,
         stdout: null,
     },
+    dataconnect: {
+        name: types_1.Emulators.DATACONNECT,
+        instance: null,
+        stdout: null,
+    },
 };
 const Commands = {
     database: {
@@ -197,6 +229,13 @@ const Commands = {
         joinArgs: false,
         shell: false,
     },
+    dataconnect: {
+        binary: getExecPath(types_1.Emulators.DATACONNECT),
+        args: ["dev"],
+        optionalArgs: ["http_port", "grpc_port", "config_dir", "local_connection_string", "project_id"],
+        joinArgs: true,
+        shell: true,
+    },
 };
 function getExecPath(name) {
     const details = getDownloadDetails(name);
@@ -364,10 +403,10 @@ async function stop(targetName) {
 exports.stop = stop;
 async function downloadIfNecessary(targetName) {
     const hasEmulator = fs.existsSync(getExecPath(targetName));
-    if (hasEmulator) {
-        return;
+    if (!hasEmulator) {
+        await (0, download_1.downloadEmulator)(targetName);
     }
-    await (0, download_1.downloadEmulator)(targetName);
+    return Commands[targetName];
 }
 exports.downloadIfNecessary = downloadIfNecessary;
 async function start(targetName, args, extraEnv = {}) {

package/lib/emulator/portUtils.js

@@ -132,8 +132,8 @@ async function checkListenable(arg1, port) {
 }
 exports.checkListenable = checkListenable;
 async function waitForPortUsed(port, host) {
-    const interval = 250;
-    const timeout = 60000;
+    const interval = 200;
+    const timeout = 5000;
     try {
         await tcpport.waitUntilUsedOnHost(port, host, interval, timeout);
     }
@@ -147,6 +147,7 @@ const EMULATOR_CAN_LISTEN_ON_PRIMARY_ONLY = {
     firestore: true,
     "firestore.websocket": true,
     pubsub: true,
+    dataconnect: true,
     hub: false,
     ui: false,
     auth: true,

package/lib/emulator/registry.js

@@ -8,6 +8,7 @@ const constants_1 = require("./constants");
 const emulatorLogger_1 = require("./emulatorLogger");
 const utils_1 = require("../utils");
 const apiv2_1 = require("../apiv2");
+const downloadableEmulators_1 = require("./downloadableEmulators");
 class EmulatorRegistry {
     static async start(instance) {
         const description = constants_1.Constants.description(instance.getName());
@@ -47,6 +48,7 @@ class EmulatorRegistry {
             auth: 3.3,
             storage: 3.5,
             eventarc: 3.6,
+            dataconnect: 3.7,
             hub: 4,
             logging: 5,
         };
@@ -83,6 +85,9 @@ class EmulatorRegistry {
         }
         return Object.assign(Object.assign({}, info), { host: (0, utils_1.connectableHostname)(info.host) });
     }
+    static getDetails(emulator) {
+        return (0, downloadableEmulators_1.get)(emulator);
+    }
     static url(emulator, req) {
         const url = new URL("http://unknown/");
         if (req) {

package/lib/emulator/types.js

@@ -15,6 +15,7 @@ var Emulators;
     Emulators["STORAGE"] = "storage";
     Emulators["EXTENSIONS"] = "extensions";
     Emulators["EVENTARC"] = "eventarc";
+    Emulators["DATACONNECT"] = "dataconnect";
 })(Emulators = exports.Emulators || (exports.Emulators = {}));
 exports.DOWNLOADABLE_EMULATORS = [
     Emulators.FIRESTORE,
@@ -22,6 +23,7 @@ exports.DOWNLOADABLE_EMULATORS = [
     Emulators.PUBSUB,
     Emulators.UI,
     Emulators.STORAGE,
+    Emulators.DATACONNECT,
 ];
 exports.IMPORT_EXPORT_EMULATORS = [
     Emulators.FIRESTORE,
@@ -38,6 +40,7 @@ exports.ALL_SERVICE_EMULATORS = [
     Emulators.PUBSUB,
     Emulators.STORAGE,
     Emulators.EVENTARC,
+    Emulators.DATACONNECT,
 ].filter((v) => v);
 exports.EMULATORS_SUPPORTED_BY_FUNCTIONS = [
     Emulators.FIRESTORE,

package/lib/experiments.js

@@ -89,6 +89,11 @@ exports.ALL_EXPERIMENTS = experiments({
         default: false,
         public: false,
     },
+    dataconnect: {
+        shortDescription: "Enable Data Connect related features.",
+        fullDescription: "Enable Data Connect related features.",
+        public: false,
+    },
 });
 function isValidExperiment(name) {
     return Object.keys(exports.ALL_EXPERIMENTS).includes(name);

package/lib/extensions/emulator/specHelper.js

@@ -1,13 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getRuntime = exports.DEFAULT_RUNTIME = exports.getFunctionProperties = exports.getFunctionResourcesWithParamSubstitution = exports.readFileFromDirectory = exports.readPostinstall = exports.readExtensionYaml = void 0;
-const yaml = require("yaml");
-const path = require("path");
-const fs = require("fs-extra");
+exports.getRuntime = exports.DEFAULT_RUNTIME = exports.getFunctionProperties = exports.getFunctionResourcesWithParamSubstitution = exports.readPostinstall = exports.readExtensionYaml = void 0;
 const supported = require("../../deploy/functions/runtimes/supported");
 const error_1 = require("../../error");
 const extensionsHelper_1 = require("../extensionsHelper");
 const utils_1 = require("../utils");
+const utils_2 = require("../../utils");
 const SPEC_FILE = "extension.yaml";
 const POSTINSTALL_FILE = "POSTINSTALL.md";
 const validFunctionTypes = [
@@ -15,22 +13,11 @@ const validFunctionTypes = [
     "firebaseextensions.v1beta.v2function",
     "firebaseextensions.v1beta.scheduledFunction",
 ];
-function wrappedSafeLoad(source) {
-    try {
-        return yaml.parse(source);
-    }
-    catch (err) {
-        if (err instanceof yaml.YAMLParseError) {
-            throw new error_1.FirebaseError(`YAML Error: ${err.message}`, { original: err });
-        }
-        throw err;
-    }
-}
 async function readExtensionYaml(directory) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-    const extensionYaml = await readFileFromDirectory(directory, SPEC_FILE);
+    const extensionYaml = await (0, utils_2.readFileFromDirectory)(directory, SPEC_FILE);
     const source = extensionYaml.source;
-    const spec = wrappedSafeLoad(source);
+    const spec = (0, utils_2.wrappedSafeLoad)(source);
     spec.params = (_a = spec.params) !== null && _a !== void 0 ? _a : [];
     spec.systemParams = (_b = spec.systemParams) !== null && _b !== void 0 ? _b : [];
     spec.resources = (_c = spec.resources) !== null && _c !== void 0 ? _c : [];
@@ -44,31 +31,10 @@ async function readExtensionYaml(directory) {
 }
 exports.readExtensionYaml = readExtensionYaml;
 async function readPostinstall(directory) {
-    const content = await readFileFromDirectory(directory, POSTINSTALL_FILE);
+    const content = await (0, utils_2.readFileFromDirectory)(directory, POSTINSTALL_FILE);
     return content.source;
 }
 exports.readPostinstall = readPostinstall;
-function readFileFromDirectory(directory, file) {
-    return new Promise((resolve, reject) => {
-        fs.readFile(path.resolve(directory, file), "utf8", (err, data) => {
-            if (err) {
-                if (err.code === "ENOENT") {
-                    return reject(new error_1.FirebaseError(`Could not find "${file}" in "${directory}"`, { original: err }));
-                }
-                reject(new error_1.FirebaseError(`Failed to read file "${file}" in "${directory}"`, { original: err }));
-            }
-            else {
-                resolve(data);
-            }
-        });
-    }).then((source) => {
-        return {
-            source,
-            sourceDirectory: directory,
-        };
-    });
-}
-exports.readFileFromDirectory = readFileFromDirectory;
 function getFunctionResourcesWithParamSubstitution(extensionSpec, params) {
     const rawResources = extensionSpec.resources.filter((resource) => validFunctionTypes.includes(resource.type));
     return (0, extensionsHelper_1.substituteParams)(rawResources, params);

package/lib/gcp/apphosting.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getNextRolloutId = exports.ensureApiEnabled = exports.listLocations = exports.updateTraffic = exports.listRollouts = exports.createRollout = exports.createBuild = exports.listBuilds = exports.getBuild = exports.deleteBackend = exports.listBackends = exports.getBackend = exports.createBackend = exports.serviceAgentEmail = exports.client = exports.API_VERSION = void 0;
+exports.getNextRolloutId = exports.ensureApiEnabled = exports.listLocations = exports.updateTraffic = exports.listRollouts = exports.createRollout = exports.createBuild = exports.listBuilds = exports.getBuild = exports.deleteBackend = exports.listBackends = exports.getBackend = exports.createBackend = exports.parseBackendName = exports.serviceAgentEmail = exports.client = exports.API_VERSION = void 0;
 const proto = require("../gcp/proto");
 const apiv2_1 = require("../apiv2");
 const projectUtils_1 = require("../projectUtils");
@@ -24,6 +24,11 @@ function serviceAgentEmail(projectNumber) {
     return `service-${projectNumber}@${P4SA_DOMAIN}`;
 }
 exports.serviceAgentEmail = serviceAgentEmail;
+function parseBackendName(backendName) {
+    const [, projectName, , location, , id] = backendName.split("/");
+    return { projectName, location, id };
+}
+exports.parseBackendName = parseBackendName;
 async function createBackend(projectId, location, backendReqBoby, backendId) {
     const res = await exports.client.post(`projects/${projectId}/locations/${location}/backends`, Object.assign(Object.assign({}, backendReqBoby), { labels: Object.assign(Object.assign({}, backendReqBoby.labels), deploymentTool.labels()) }), { queryParams: { backendId } });
     return res.body;

package/lib/gcp/cloudsql/cloudsqladmin.js

@@ -0,0 +1,155 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.listUsers = exports.deleteUser = exports.getUser = exports.createUser = exports.createDatabase = exports.getDatabase = exports.listDatabases = exports.updateInstanceForDataConnect = exports.createInstance = exports.getInstance = exports.listInstances = void 0;
+const apiv2_1 = require("../../apiv2");
+const api_1 = require("../../api");
+const operationPoller = require("../../operation-poller");
+const API_VERSION = "v1";
+const client = new apiv2_1.Client({
+    urlPrefix: (0, api_1.cloudSQLAdminOrigin)(),
+    auth: true,
+    apiVersion: API_VERSION,
+});
+async function listInstances(projectId) {
+    var _a;
+    const res = await client.get(`projects/${projectId}/instances`);
+    return (_a = res.body.items) !== null && _a !== void 0 ? _a : [];
+}
+exports.listInstances = listInstances;
+async function getInstance(projectId, instanceId) {
+    const res = await client.get(`projects/${projectId}/instances/${instanceId}`);
+    return res.body;
+}
+exports.getInstance = getInstance;
+async function createInstance(projectId, location, instanceId, enableGoogleMlIntegration) {
+    const databaseFlags = [{ name: "cloudsql.iam_authentication", value: "on" }];
+    if (enableGoogleMlIntegration) {
+        databaseFlags.push({ name: "cloudsql.enable_google_ml_integration", value: "on" });
+    }
+    const op = await client.post(`projects/${projectId}/instances`, {
+        name: instanceId,
+        region: location,
+        databaseVersion: "POSTGRES_15",
+        settings: {
+            tier: "db-f1-micro",
+            edition: "ENTERPRISE",
+            ipConfiguration: {
+                authorizedNetworks: [],
+            },
+            enableGoogleMlIntegration,
+            databaseFlags,
+            storageAutoResize: false,
+            userLabels: { "firebase-data-connect": "ft" },
+            insightsConfig: {
+                queryInsightsEnabled: true,
+            },
+        },
+    });
+    const opName = `projects/${projectId}/operations/${op.body.name}`;
+    const pollRes = await operationPoller.pollOperation({
+        apiOrigin: (0, api_1.cloudSQLAdminOrigin)(),
+        apiVersion: API_VERSION,
+        operationResourceName: opName,
+        doneFn: (op) => op.status === "DONE",
+        masterTimeout: 1200000,
+    });
+    return pollRes;
+}
+exports.createInstance = createInstance;
+async function updateInstanceForDataConnect(instance, enableGoogleMlIntegration) {
+    let dbFlags = setDatabaseFlag({ name: "cloudsql.iam_authentication", value: "on" }, instance.settings.databaseFlags);
+    if (enableGoogleMlIntegration) {
+        dbFlags = setDatabaseFlag({ name: "cloudsql.enable_google_ml_integration", value: "on" }, dbFlags);
+    }
+    const op = await client.patch(`projects/${instance.project}/instances/${instance.name}`, {
+        settings: {
+            ipConfiguration: {
+                ipv4Enabled: true,
+            },
+            databaseFlags: dbFlags,
+            enableGoogleMlIntegration,
+        },
+    });
+    const opName = `projects/${instance.project}/operations/${op.body.name}`;
+    const pollRes = await operationPoller.pollOperation({
+        apiOrigin: (0, api_1.cloudSQLAdminOrigin)(),
+        apiVersion: API_VERSION,
+        operationResourceName: opName,
+        doneFn: (op) => op.status === "DONE",
+        masterTimeout: 1200000,
+    });
+    return pollRes;
+}
+exports.updateInstanceForDataConnect = updateInstanceForDataConnect;
+function setDatabaseFlag(flag, flags = []) {
+    const temp = flags.filter((f) => f.name !== flag.name);
+    temp.push(flag);
+    return temp;
+}
+async function listDatabases(projectId, instanceId) {
+    const res = await client.get(`projects/${projectId}/instances/${instanceId}/databases`);
+    return res.body.items;
+}
+exports.listDatabases = listDatabases;
+async function getDatabase(projectId, instanceId, databaseId) {
+    const res = await client.get(`projects/${projectId}/instances/${instanceId}/databases/${databaseId}`);
+    return res.body;
+}
+exports.getDatabase = getDatabase;
+async function createDatabase(projectId, instanceId, databaseId) {
+    const op = await client.post(`projects/${projectId}/instances/${instanceId}/databases`, {
+        project: projectId,
+        instance: instanceId,
+        name: databaseId,
+    });
+    const opName = `projects/${projectId}/operations/${op.body.name}`;
+    const pollRes = await operationPoller.pollOperation({
+        apiOrigin: (0, api_1.cloudSQLAdminOrigin)(),
+        apiVersion: API_VERSION,
+        operationResourceName: opName,
+        doneFn: (op) => op.status === "DONE",
+    });
+    return pollRes;
+}
+exports.createDatabase = createDatabase;
+async function createUser(projectId, instanceId, type, username, password) {
+    const op = await client.post(`projects/${projectId}/instances/${instanceId}/users`, {
+        name: username,
+        instance: instanceId,
+        project: projectId,
+        password: password,
+        sqlserverUserDetails: {
+            disabled: false,
+            serverRoles: ["cloudsqlsuperuser"],
+        },
+        type,
+    });
+    const opName = `projects/${projectId}/operations/${op.body.name}`;
+    const pollRes = await operationPoller.pollOperation({
+        apiOrigin: (0, api_1.cloudSQLAdminOrigin)(),
+        apiVersion: API_VERSION,
+        operationResourceName: opName,
+        doneFn: (op) => op.status === "DONE",
+    });
+    return pollRes;
+}
+exports.createUser = createUser;
+async function getUser(projectId, instanceId, username) {
+    const res = await client.get(`projects/${projectId}/instances/${instanceId}/users/${username}`);
+    return res.body;
+}
+exports.getUser = getUser;
+async function deleteUser(projectId, instanceId, username) {
+    const res = await client.delete(`projects/${projectId}/instances/${instanceId}/users`, {
+        queryParams: {
+            name: username,
+        },
+    });
+    return res.body;
+}
+exports.deleteUser = deleteUser;
+async function listUsers(projectId, instanceId) {
+    const res = await client.get(`projects/${projectId}/instances/${instanceId}/users`);
+    return res.body.items;
+}
+exports.listUsers = listUsers;