@catladder/cli 3.37.0 → 3.39.0

package/package.json

@@ -53,7 +53,7 @@
     }
   ],
   "license": "MIT",
-  "version": "3.37.0",
+  "version": "3.39.0",
   "scripts": {
     "lint": "eslint \"src/**/*.ts\"",
     "lint:fix": "eslint \"src/**/*.ts\" --fix",

package/src/apps/cli/commands/project/cloudSql/commandProjectRestoreDb.ts

@@ -1,14 +1,27 @@
 import type Vorpal from "vorpal";
 import {
+  getAllPipelineContexts,
   getEnvVarsResolved,
   getPipelineContextByChoice,
 } from "../../../../../config/getProjectConfig";
 import type { CloudSqlBackgroundProxy } from "../../../../../gcloud/cloudSql/startProxy";
 import { startCloudSqlProxyInBackground } from "../../../../../gcloud/cloudSql/startProxy";
-import { envAndComponents } from "../utils/autocompletions";
 
 import { parseConnectionString } from "../../../../../gcloud/cloudSql/parseConnectionString";
 import { spawnCopyDb } from "../../../../../gcloud/cloudSql/copyDb";
+import type { ComponentContext } from "@catladder/pipeline";
+import { isOfDeployType } from "@catladder/pipeline";
+
+const hasCloudSql = (context: ComponentContext) => {
+  const deployConfig = context.deploy?.config;
+  if (isOfDeployType(deployConfig, "google-cloudrun")) {
+    return !!deployConfig.cloudSql;
+  }
+  if (isOfDeployType(deployConfig, "kubernetes")) {
+    return !!deployConfig.values?.cloudsql?.enabled;
+  }
+  return false;
+};
 
 export default async (vorpal: Vorpal) =>
   vorpal
@@ -17,7 +30,29 @@ export default async (vorpal: Vorpal) =>
       "restores a project db from one source to another target",
     )
     .action(async function restoreDb() {
-      const envs = await envAndComponents();
+      const allContexts = await getAllPipelineContexts(undefined, {
+        includeLocal: true,
+      });
+      const filteredContexts = allContexts.filter(hasCloudSql);
+      const sortByEnv = (a: string, b: string) => {
+        const aLocal = a.startsWith("local:");
+        const bLocal = b.startsWith("local:");
+        if (aLocal !== bLocal) return aLocal ? 1 : -1;
+        return a.localeCompare(b);
+      };
+      const envs = filteredContexts
+        .map((c) => `${c.env}:${c.name}`)
+        .sort(sortByEnv);
+      const targetEnvChoices = filteredContexts
+        .map((c) => {
+          const value = `${c.env}:${c.name}`;
+          const isProd = c.environment.envType === "prod";
+          return {
+            name: isProd ? `⚠️ ${value}` : value,
+            value,
+          };
+        })
+        .sort((a, b) => sortByEnv(a.value, b.value));
 
       const { sourceEnvAndComponent } = await this.prompt({
         type: "list",
@@ -83,7 +118,7 @@ export default async (vorpal: Vorpal) =>
       const { targetEnvAndComponent } = await this.prompt({
         type: "list",
         name: "targetEnvAndComponent",
-        choices: envs,
+        choices: targetEnvChoices,
 
         message: "target env? 🤔 ",
       });

package/src/config/getProjectConfig.ts

@@ -94,6 +94,7 @@ export const getAllComponentsWithAllEnvsHierarchical = async (): Promise<{
 
 export const getAllPipelineContexts = async (
   onlyComponent?: string | string[],
+  { includeLocal = false }: { includeLocal?: boolean } = {},
 ) => {
   const onlyComponentsArray = onlyComponent
     ? Array.isArray(onlyComponent)
@@ -102,7 +103,7 @@ export const getAllPipelineContexts = async (
     : null;
   return Promise.all(
     (await getAllComponentsWithAllEnvsFlat())
-      .filter((c) => c.env !== "local")
+      .filter((c) => includeLocal || c.env !== "local")
       .filter(
         (c) =>
           !onlyComponentsArray || onlyComponentsArray.includes(c.componentName),