Support specializing snippet definitions by containing scope

Author: pokey

cursorless-snippets/functionDeclaration.cursorless-snippets

@@ -8,6 +8,10 @@
             "typescriptreact",
             "javascript",
             "javascriptreact"
+          ],
+          "scopeTypes": [
+            "namedFunction",
+            "document"
           ]
         },
         "body": [
@@ -21,6 +25,29 @@
           }
         }
       },
+      {
+        "scope": {
+          "langIds": [
+            "typescript",
+            "typescriptreact",
+            "javascript",
+            "javascriptreact"
+          ],
+          "scopeTypes": [
+            "class"
+          ]
+        },
+        "body": [
+          "$name($parameterList) {",
+          "\t$body",
+          "}"
+        ],
+        "variables": {
+          "name": {
+            "formatter": "camelCase"
+          }
+        }
+      },
       {
         "scope": {
           "langIds": [

packages/cursorless-engine/src/actions/InsertSnippet.ts

@@ -94,6 +94,7 @@ export default class InsertSnippet implements Action {
       const snippet = this.snippets.getSnippetStrict(name);
 
       const definition = findMatchingSnippetDefinitionStrict(
+        this.modifierStageFactory,
         targets,
         snippet.definitions,
       );

packages/cursorless-engine/src/actions/WrapWithSnippet.ts

@@ -86,6 +86,7 @@ export default class WrapWithSnippet implements Action {
       const snippet = this.snippets.getSnippetStrict(name);
 
       const definition = findMatchingSnippetDefinitionStrict(
+        this.modifierStageFactory,
         targets,
         snippet.definitions,
       );

packages/cursorless-engine/src/core/Snippets.ts

@@ -1,9 +1,10 @@
 import { showError, Snippet, SnippetMap, walkFiles } from "@cursorless/common";
 import { readFile, stat } from "fs/promises";
-import { cloneDeep, max, merge } from "lodash";
+import { max } from "lodash";
 import { join } from "path";
 import { ide } from "../singletons/ide.singleton";
 import { mergeStrict } from "../util/object";
+import { mergeSnippets } from "./mergeSnippets";
 
 const CURSORLESS_SNIPPETS_SUFFIX = ".cursorless-snippets";
 const SNIPPET_DIR_REFRESH_INTERVAL_MS = 1000;
@@ -21,7 +22,7 @@ interface DirectoryErrorMessage {
 export class Snippets {
   private coreSnippets!: SnippetMap;
   private thirdPartySnippets: Record<string, SnippetMap> = {};
-  private userSnippets!: SnippetMap;
+  private userSnippets!: SnippetMap[];
 
   private mergedSnippets!: SnippetMap;
 
@@ -133,7 +134,7 @@ export class Snippets {
         };
       }
 
-      this.userSnippets = {};
+      this.userSnippets = [];
       this.mergeSnippets();
 
       return;
@@ -154,34 +155,32 @@ export class Snippets {
 
     this.maxSnippetMtimeMs = maxSnippetMtime;
 
-    this.userSnippets = mergeStrict(
-      ...(await Promise.all(
-        snippetFiles.map(async (path) => {
-          try {
-            const content = await readFile(path, "utf8");
-
-            if (content.length === 0) {
-              // Gracefully handle an empty file
-              return {};
-            }
-
-            return JSON.parse(content);
-          } catch (err) {
-            showError(
-              ide().messages,
-              "snippetsFileError",
-              `Error with cursorless snippets file "${path}": ${
-                (err as Error).message
-              }`,
-            );
-
-            // We don't want snippets from all files to stop working if there is
-            // a parse error in one file, so we just effectively ignore this file
-            // once we've shown an error message
+    this.userSnippets = await Promise.all(
+      snippetFiles.map(async (path) => {
+        try {
+          const content = await readFile(path, "utf8");
+
+          if (content.length === 0) {
+            // Gracefully handle an empty file
             return {};
           }
-        }),
-      )),
+
+          return JSON.parse(content);
+        } catch (err) {
+          showError(
+            ide().messages,
+            "snippetsFileError",
+            `Error with cursorless snippets file "${path}": ${
+              (err as Error).message
+            }`,
+          );
+
+          // We don't want snippets from all files to stop working if there is
+          // a parse error in one file, so we just effectively ignore this file
+          // once we've shown an error message
+          return {};
+        }
+      }),
     );
 
     this.mergeSnippets();
@@ -206,34 +205,11 @@ export class Snippets {
    * party > core.
    */
   private mergeSnippets() {
-    this.mergedSnippets = {};
-
-    // We make a list of all entries from all sources, in order of increasing
-    // precedence: user > third party > core.
-    const entries = [
-      ...Object.entries(cloneDeep(this.coreSnippets)),
-      ...Object.values(this.thirdPartySnippets).flatMap((snippets) =>
-        Object.entries(cloneDeep(snippets)),
-      ),
-      ...Object.entries(cloneDeep(this.userSnippets)),
-    ];
-
-    entries.forEach(([key, value]) => {
-      if (Object.prototype.hasOwnProperty.call(this.mergedSnippets, key)) {
-        const { definitions, ...rest } = value;
-        const mergedSnippet = this.mergedSnippets[key];
-
-        // NB: We make sure that the new definitions appear before the previous
-        // ones so that they take precedence
-        mergedSnippet.definitions = definitions.concat(
-          ...mergedSnippet.definitions,
-        );
-
-        merge(mergedSnippet, rest);
-      } else {
-        this.mergedSnippets[key] = value;
-      }
-    });
+    this.mergedSnippets = mergeSnippets(
+      this.coreSnippets,
+      this.thirdPartySnippets,
+      this.userSnippets,
+    );
   }
 
   /**

packages/cursorless-engine/src/core/compareSnippetDefinitions.ts

@@ -0,0 +1,93 @@
+import {
+  SimpleScopeTypeType,
+  SnippetDefinition,
+  SnippetScope,
+} from "@cursorless/common";
+import { SnippetOrigin } from "./mergeSnippets";
+
+/**
+ * Compares two snippet definitions by how specific their scope, breaking
+ * ties by origin.
+ * @param a One of the snippet definitions to compare
+ * @param b The other snippet definition to compare
+ * @returns A negative number if a should come before b, a positive number if b
+ */
+export function compareSnippetDefinitions(
+  a: SnippetDefinitionWithOrigin,
+  b: SnippetDefinitionWithOrigin,
+): number {
+  const scopeComparision = compareSnippetScopes(
+    a.definition.scope,
+    b.definition.scope,
+  );
+
+  if (scopeComparision !== 0) {
+    return scopeComparision;
+  }
+
+  return a.origin - b.origin;
+}
+
+function compareSnippetScopes(
+  a: SnippetScope | undefined,
+  b: SnippetScope | undefined,
+): number {
+  if (a == null && b == null) {
+    return 0;
+  }
+
+  if (a == null) {
+    return -1;
+  }
+
+  if (b == null) {
+    return 1;
+  }
+
+  const langIdsComparision = compareLangIds(a.langIds, b.langIds);
+
+  if (langIdsComparision !== 0) {
+    return langIdsComparision;
+  }
+
+  return compareScopeTypes(a.scopeTypes, b.scopeTypes);
+}
+
+function compareLangIds(
+  a: string[] | undefined,
+  b: string[] | undefined,
+): number {
+  if (a == null && b == null) {
+    return 0;
+  }
+
+  if (a == null) {
+    return -1;
+  }
+
+  if (b == null) {
+    return 1;
+  }
+
+  return b.length - a.length;
+}
+
+function compareScopeTypes(
+  a: SimpleScopeTypeType[] | undefined,
+  b: SimpleScopeTypeType[] | undefined,
+): number {
+  if (a == null && b != null) {
+    return -1;
+  }
+
+  if (b == null && a != null) {
+    return 1;
+  }
+
+  return 0;
+}
+
+interface SnippetDefinitionWithOrigin {
+  origin: SnippetOrigin;
+  definition: SnippetDefinition;
+}