spec(bundle): Add gem autocomplete and binaries autocomplete

src/_utils/spec.ts

@@ -0,0 +1,7 @@
+// eslint-disable-next-line @withfig/fig-linter/no-missing-default-export
+export const specToSuggestions = (spec: Fig.Spec): Fig.Suggestion => {
+  return {
+    name: spec.name,
+    ...("icon" in spec && { icon: spec.icon }),
+  };
+};

src/bundle.ts

@@ -1,3 +1,19 @@
+import rails from "./rails";
+import rubocop from "./rubocop";
+import rspec from "./rspec";
+import rake from "./rake";
+import { gemsGenerator } from "./gem";
+import pry from "./pry";
+import { specToSuggestions } from "./_utils/spec";
+
+const EXEC_SUGGESTIONS: Fig.Suggestion[] = [
+  specToSuggestions(rails),
+  specToSuggestions(rubocop),
+  specToSuggestions(rspec),
+  specToSuggestions(rake),
+  specToSuggestions(pry),
+];
+
 const gemfileGemsGenerator: Fig.Generator = {
   script: ["bundle", "list", "--name-only"],
   postProcess: (out) => {
@@ -175,7 +191,30 @@ const completionSpec: Fig.Spec = {
           description: "Pass all file descriptors to the new process",
         },
       ],
-      args: { isCommand: true },
+      args: {
+        isCommand: true,
+        generators: {
+          script: ["bundle", "list", "--name-only"],
+          cache: {
+            cacheByDirectory: true,
+            strategy: "stale-while-revalidate",
+            ttl: 60 * 60,
+          },
+          postProcess: (out) => {
+            const gems = out.split("\n");
+
+            return EXEC_SUGGESTIONS.filter((spec) => {
+              if (spec.name === "rspec") {
+                return gems.includes("rspec-core");
+              }
+
+              return gems.includes(
+                typeof spec.name === "string" ? spec.name : spec.name[0]
+              );
+            });
+          },
+        },
+      },
     },
     { name: "config", args: {} },
     { name: "help" },
@@ -184,7 +223,10 @@ const completionSpec: Fig.Spec = {
     {
       name: "add",
       description: "Add gem to the Gemfile and run bundle install",
-      args: {},
+      args: {
+        name: "gem",
+        generators: gemsGenerator,
+      },
       options: [
         {
           name: ["--version", "-v"],

src/gem.ts

@@ -1,4 +1,4 @@
-const gems: Fig.Generator = {
+export const gemsGenerator: Fig.Generator = {
   trigger: () => true,
   custom: async (tokens, executeShellCommand) => {
     const searchTerm = tokens[tokens.length - 1];
@@ -178,7 +178,7 @@ const completionSpec: Fig.Spec = {
       description: "Install a gem into the local repository",
       args: {
         name: "GEMNAME",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
       },
       options: [
@@ -288,7 +288,7 @@ const completionSpec: Fig.Spec = {
       description: "Check a gem repository for added or missing files",
       args: {
         name: "GEMNAME",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
         isOptional: true,
       },
@@ -351,7 +351,7 @@ const completionSpec: Fig.Spec = {
       args: {
         name: "GEMNAME",
         description: "Name of gem to cleanup",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
         isOptional: true,
       },
@@ -390,7 +390,7 @@ const completionSpec: Fig.Spec = {
       args: {
         name: "GEMNAME",
         description: "Name of gem to list contents for",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
       },
       options: [
@@ -539,7 +539,7 @@ const completionSpec: Fig.Spec = {
       args: {
         name: "GEMNAME",
         description: "Name of gem to download",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
       },
       options: [
@@ -640,7 +640,7 @@ const completionSpec: Fig.Spec = {
       args: {
         name: "GEMNAME",
         description: "Name of the gem to print information about",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
       },
       options: [
@@ -684,7 +684,7 @@ const completionSpec: Fig.Spec = {
       args: {
         name: "GEMNAME",
         description: "Name of the gem to print information about",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
       },
       options: [
@@ -714,7 +714,7 @@ const completionSpec: Fig.Spec = {
       args: {
         name: "GEMNAME",
         description: "Name of the gem to print information about",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
         isOptional: true,
       },
@@ -729,7 +729,7 @@ const completionSpec: Fig.Spec = {
           args: {
             name: "GEMNAME",
             description: "Name of the gem to print information about",
-            generators: gems,
+            generators: gemsGenerator,
             debounce: true,
           },
           requiresSeparator: true,
@@ -834,7 +834,7 @@ const completionSpec: Fig.Spec = {
       args: {
         name: "GEMNAME",
         description: "Gem to generate documentation for (unless –all)",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
         isOptional: true,
       },
@@ -997,7 +997,7 @@ const completionSpec: Fig.Spec = {
         {
           name: "GEMFILE",
           description: "Name of gem to show the gemspec for",
-          generators: gems,
+          generators: gemsGenerator,
           debounce: true,
         },
         {
@@ -1060,7 +1060,7 @@ const completionSpec: Fig.Spec = {
       args: {
         name: "GEMNAME",
         description: "Name of gem to unpack",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
       },
       options: [
@@ -1094,7 +1094,7 @@ const completionSpec: Fig.Spec = {
       args: {
         name: "GEM",
         description: "Name of gem",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
       },
       options: [
@@ -1122,7 +1122,7 @@ const completionSpec: Fig.Spec = {
       description: "Uninstall gems from the local repository",
       args: {
         name: "GEMNAME",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
       },
       options: [
@@ -1303,7 +1303,7 @@ const completionSpec: Fig.Spec = {
       description: "Manage gem owners of a gem on the push server",
       args: {
         name: "GEMNAME",
-        generators: gems,
+        generators: gemsGenerator,
         debounce: true,
       },
       options: [

src/rspec.ts

@@ -0,0 +1,16 @@
+const completionSpec: Fig.Spec = {
+  name: "rspec",
+  icon: "https://rspec.info/images/logo.png",
+  description:
+    "Behaviour Driven Development for Ruby. Making TDD Productive and Fun",
+  args: {
+    name: "file",
+    description: "The file to run",
+    default: "./spec",
+    generators: {
+      template: ["filepaths", "folders"],
+    },
+  },
+};
+
+export default completionSpec;

src/rubocop.ts

@@ -0,0 +1,23 @@
+const completionSpec: Fig.Spec = {
+  name: "rubocop",
+  description:
+    "A Ruby static code analyzer and formatter, based on the community Ruby style guide",
+  icon: "https://github.com/rubocop/rubocop/blob/master/assets/logo.png?raw=true",
+  args: {
+    name: "file",
+    template: "filepaths",
+    isVariadic: true,
+  },
+  options: [
+    {
+      name: ["-a", "--autocorrect"],
+      description: "Autocorrect offenses (only when it's safe)",
+    },
+    {
+      name: ["-A", "--autocorrect-all"],
+      description: "Autocorrect offenses (safe and unsafe)",
+    },
+  ],
+};
+
+export default completionSpec;