feat(cli): add flag ai removal command

apps/angular-example/CHANGELOG.md

@@ -1,5 +1,12 @@
 # angular-example
 
+## 0.0.19
+
+### Patch Changes
+
+- @tryabby/angular@2.0.9
+- @tryabby/devtools@5.0.1
+
 ## 0.0.18
 
 ### Patch Changes

apps/angular-example/package.json

@@ -1,6 +1,6 @@
 {
   "name": "angular-example",
-  "version": "0.0.18",
+  "version": "0.0.19",
   "private": true,
   "scripts": {
     "ng": "ng",

apps/web/CHANGELOG.md

@@ -1,5 +1,14 @@
 # web
 
+## 0.2.37
+
+### Patch Changes
+
+- Updated dependencies [d05cb9a]
+  - @tryabby/core@5.3.1
+  - @tryabby/devtools@5.0.1
+  - @tryabby/next@5.1.2
+
 ## 0.2.36
 
 ### Patch Changes

apps/web/package.json

@@ -1,6 +1,6 @@
 {
   "name": "web",
-  "version": "0.2.36",
+  "version": "0.2.37",
   "private": true,
   "scripts": {
     "build": "next build",

packages/angular/CHANGELOG.md

@@ -1,5 +1,12 @@
 # @tryabby/angular
 
+## 2.0.9
+
+### Patch Changes
+
+- Updated dependencies [d05cb9a]
+  - @tryabby/core@5.3.1
+
 ## 2.0.8
 
 ### Patch Changes

packages/angular/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryabby/angular",
-  "version": "2.0.8",
+  "version": "2.0.9",
   "scripts": {
     "ng": "ng",
     "start": "ng serve",

packages/cli/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @tryabby/cli
 
+## 1.1.1
+
+### Patch Changes
+
+- d05cb9a: add feature flag removal command
+
 ## 1.1.0
 
 ### Minor Changes

packages/cli/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryabby/cli",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "private": false,
   "main": "./dist/index.js",
   "bin": {
@@ -27,6 +27,7 @@
     "dotenv": "^16.0.3",
     "esbuild": "0.18.17",
     "figlet": "^1.6.0",
+    "globby": "^14.0.2",
     "magicast": "^0.3.2",
     "msw": "^1.2.2",
     "node-fetch": "^3.3.1",

packages/cli/src/ai.ts

@@ -0,0 +1,58 @@
+import { loadLocalConfig } from "./util";
+import { globby } from "globby";
+import { getUseFeatureFlagRegex } from "@tryabby/core";
+import { readFile } from "fs/promises";
+import chalk from "chalk";
+import { HttpService } from "./http";
+
+export async function removeFlagInstance(options: {
+  flagName: string;
+  apiKey: string;
+  path: string;
+  host?: string;
+  configPath?: string;
+}) {
+  const files = await globby("**/*.tsx", {
+    cwd: options.path ?? process.cwd(),
+    absolute: true,
+    gitignore: true,
+    onlyFiles: true,
+  });
+
+  const regex = getUseFeatureFlagRegex(options.flagName);
+
+  const filesToUse = (
+    await Promise.all(
+      files.flatMap(async (filePath) => {
+        const content = await readFile(filePath, "utf-8").then((content) => {
+          const matches = content.match(regex);
+          return matches ? content : null;
+        });
+        if (!content) return [];
+
+        return {
+          filePath,
+          fileContent: content,
+        };
+      })
+    )
+  ).flat();
+
+  await HttpService.getFilesWithFlagsRemoved({
+    apiKey: options.apiKey,
+    files: filesToUse,
+    flagName: options.flagName,
+    apiUrl: options.host,
+  });
+
+  try {
+    const { mutableConfig, saveMutableConfig } = await loadLocalConfig(
+      options.configPath
+    );
+    mutableConfig.flags = mutableConfig.flags.filter((flag: string) => flag !== options.flagName);
+    await saveMutableConfig();
+  } catch (e) {
+    // fail silently
+  }
+  console.log(chalk.green("Flag removed successfully"));
+}

packages/cli/src/http.ts

@@ -3,6 +3,7 @@ import { ABBY_BASE_URL } from "./consts";
 import fetch from "node-fetch";
 import { multiLineLog } from "./util";
 import chalk from "chalk";
+import { writeFile } from "fs/promises";
 
 export abstract class HttpService {
   static async getConfigFromServer({
@@ -69,4 +70,53 @@ export abstract class HttpService {
       throw e;
     }
   }
+
+  static async getFilesWithFlagsRemoved({
+    apiKey,
+    files,
+    flagName,
+    apiUrl,
+  }: {
+    apiKey: string;
+    files: Array<{ filePath: string; fileContent: string }>;
+    flagName: string;
+    apiUrl?: string;
+  }) {
+    const url = apiUrl ?? ABBY_BASE_URL;
+
+    try {
+      const response = await fetch(`${url}/api/ee/v1/abby-ai/flag-removal`, {
+        method: "POST",
+        headers: {
+          Authorization: "Bearer " + apiKey,
+          "Content-Type": "application/json",
+        },
+        body: JSON.stringify({
+          flagName,
+          files,
+        }),
+      });
+
+      const status = response.status;
+
+      if (status === 200) {
+        const res = await response.json();
+        if (!Array.isArray(res)) {
+          throw new Error("Invalid response from server");
+        }
+        console.log({ res, files });
+        await Promise.all(res.map((file) => writeFile(file.filePath, file.fileContent)));
+        console.log(chalk.green("All files have been updated successfully"));
+      } else if (status === 500) {
+        throw new Error("Internal server error trying to update files");
+      } else if (status === 401) {
+        throw new Error("Invalid API Key");
+      } else {
+        throw new Error("Unable to update files");
+      }
+    } catch (e) {
+      console.log(chalk.red(multiLineLog("Error: " + e)));
+      throw e;
+    }
+  }
 }

packages/cli/src/index.ts

@@ -13,6 +13,7 @@ import { initAbbyConfig } from "./init";
 import { addCommandTypeSchema } from "./schemas";
 import { addFlag } from "./add-flag";
 import { addRemoteConfig } from "./add-remote-config";
+import { removeFlagInstance } from "./ai";
 
 const program = new Command();
 
@@ -184,4 +185,24 @@ program
     }
   });
 
+const aiCommand = program.command("ai").description("Abby AI helpers");
+
+aiCommand
+  .command("remove")
+  .description("remove a flag from your code")
+  .argument("<dir>", "The directory to scan for")
+  .argument("<flag>", "The flag name to remove")
+  .addOption(ConfigOption)
+  .addOption(HostOption)
+  .action(async (dir: string, flagName: string, options: { config?: string; host?: string }) => {
+    const files = await removeFlagInstance({
+      apiKey: await getToken(),
+      flagName,
+      path: dir,
+      configPath: options.config,
+      host: options.host,
+    });
+    console.log(files);
+  });
+
 program.parse(process.argv);

packages/core/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @tryabby/core
 
+## 5.3.1
+
+### Patch Changes
+
+- d05cb9a: add feature flag removal command
+
 ## 5.3.0
 
 ### Minor Changes

packages/core/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryabby/core",
-  "version": "5.3.0",
+  "version": "5.3.1",
   "description": "",
   "main": "dist/index.js",
   "files": [

packages/core/src/shared/helpers.ts

@@ -67,3 +67,6 @@ export function stringifyRemoteConfigValue(value: RemoteConfigValue) {
       assertUnreachable(value);
   }
 }
+
+export const getUseFeatureFlagRegex = (flagName: string) =>
+  new RegExp(`useFeatureFlag\\s*\\(\\s*['"\`]${flagName}['"\`]\\s*\\)`);

packages/next/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @tryabby/next
 
+## 5.1.2
+
+### Patch Changes
+
+- @tryabby/react@5.2.1
+
 ## 5.1.1
 
 ### Patch Changes

packages/next/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryabby/next",
-  "version": "5.1.1",
+  "version": "5.1.2",
   "description": "",
   "main": "dist/index.js",
   "files": [

packages/node/CHANGELOG.md

@@ -1,5 +1,12 @@
 # @tryabby/node
 
+## 5.1.7
+
+### Patch Changes
+
+- Updated dependencies [d05cb9a]
+  - @tryabby/core@5.3.1
+
 ## 5.1.6
 
 ### Patch Changes

packages/node/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryabby/node",
-  "version": "5.1.6",
+  "version": "5.1.7",
   "description": "",
   "main": "dist/index.js",
   "files": [

packages/react/CHANGELOG.md

@@ -1,5 +1,12 @@
 # @tryabby/react
 
+## 5.2.1
+
+### Patch Changes
+
+- Updated dependencies [d05cb9a]
+  - @tryabby/core@5.3.1
+
 ## 5.2.0
 
 ### Minor Changes

packages/react/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryabby/react",
-  "version": "5.2.0",
+  "version": "5.2.1",
   "description": "",
   "main": "dist/index.js",
   "files": [

packages/remix/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @tryabby/next
 
+## 1.0.3
+
+### Patch Changes
+
+- @tryabby/react@5.2.1
+
 ## 1.0.2
 
 ### Patch Changes

packages/remix/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryabby/remix",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "",
   "main": "dist/index.js",
   "files": [

packages/svelte/CHANGELOG.md

@@ -1,5 +1,12 @@
 # @tryabby/svelte
 
+## 2.2.1
+
+### Patch Changes
+
+- Updated dependencies [d05cb9a]
+  - @tryabby/core@5.3.1
+
 ## 2.2.0
 
 ### Minor Changes

packages/svelte/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryabby/svelte",
-  "version": "2.2.0",
+  "version": "2.2.1",
   "main": "dist/index.umd.cjs",
   "homepage": "https://docs.tryabby.dev",
   "type": "module",