fix(keep-sorted): support satisfies, close #25

antfu · Sep 7, 2024 · 8556eab · 8556eab
1 parent 3c02f93
commit 8556eab
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 1 deletion.
diff --git a/src/commands/keep-sorted.test.ts b/src/commands/keep-sorted.test.ts
@@ -356,4 +356,32 @@ run(
       ]
     `,
   },
+  {
+    description: 'With satisfies',
+    code: $`
+      // @keep-sorted
+      const a = {
+        foo,
+        bar,
+        apple
+      } satisfies Record<string, any>
+    `,
+    output: $`
+      // @keep-sorted
+      const a = {
+        apple,
+        bar,
+        foo,
+      } satisfies Record<string, any>
+    `,
+    errors: ['command-fix'],
+  },
+  {
+    description: 'With satisfies',
+    code: $`
+      // @keep-sorted
+      const a = bar satisfies Record<string, any>
+    `,
+    errors: ['command-error'],
+  },
 )
diff --git a/src/commands/keep-sorted.ts b/src/commands/keep-sorted.ts
@@ -22,15 +22,27 @@ export const keepSorted: Command = {
       return ctx.reportError(`Failed to parse options: ${optionsRaw}`)
     }
 
-    const node = ctx.findNodeBelow(
+    let node = ctx.findNodeBelow(
       'ObjectExpression',
       'ObjectPattern',
       'ArrayExpression',
       'TSInterfaceBody',
       'TSTypeLiteral',
+      'TSSatisfiesExpression',
     ) || ctx.findNodeBelow(
       'ExportNamedDeclaration',
     )
+
+    // Unwrap TSSatisfiesExpression
+    if (node?.type === 'TSSatisfiesExpression') {
+      if (node.expression.type !== 'ArrayExpression' && node.expression.type !== 'ObjectExpression') {
+        node = undefined
+      }
+      else {
+        node = node.expression
+      }
+    }
+
     if (!node)
       return ctx.reportError('Unable to find object/array/interface to sort')