add esquery, simplify numeric literals (updated 22:04)

Author: mmkal

package.json

@@ -45,13 +45,15 @@
     "@types/babel__generator": "^7.27.0",
     "@types/dedent": "0.7.0",
     "@types/eslint": "^8.44.7",
+    "@types/esquery": "^1.5.4",
     "@types/glob": "7.1.3",
     "@types/js-yaml": "3.12.5",
     "@types/lodash": "^4.14.202",
     "@types/node": "^20.0.0",
     "cheerio": "^1.0.0",
     "dedent": "^1.5.1",
     "eslint-plugin-markdown": "^4.0.1",
+    "esquery": "^1.6.0",
     "expect": "^29.7.0",
     "fp-ts": "^2.1.0",
     "glob": "^10.3.10",

pnpm-lock.yaml

@@ -5,6 +5,7 @@ import * as babelTypes from '@babel/types'
 import * as cheerio from 'cheerio'
 import * as child_process from 'child_process'
 import dedent from 'dedent'
+import esquery from 'esquery'
 // eslint-disable-next-line no-restricted-imports
 import * as fs from 'fs'
 import * as glob from 'glob'
@@ -33,6 +34,7 @@ export interface PresetDependencies {
   cheerio: typeof import('cheerio')
   child_process: typeof import('child_process')
   dedent: typeof import('dedent').default
+  esquery: typeof import('esquery')
   fetchSync: typeof import('./fetch-sync').fetchSync
   fs: typeof import('fs')
   glob: Pick<typeof import('glob'), 'globSync'>
@@ -44,6 +46,8 @@ export interface PresetDependencies {
   path: typeof import('path')
   readPkgUp: Pick<typeof import('read-pkg-up'), 'sync'>
   recast: typeof import('recast')
+  /** require an arbitrary module *from eslint-plugin-codegen* - this means you can require dependencies of the plugin even when they're not your dependencies */
+  require: <T = unknown>(path: string) => T
   simplify: typeof import('./simplify')
   unionfs: typeof import('unionfs')
   zx: typeof import('zx')
@@ -58,6 +62,7 @@ export const dependencies: PresetDependencies = {
   cheerio,
   child_process,
   dedent: Object.assign(dedent, {default: dedent}), // backcompat: accidentally used `import * as dedent from 'dedent'` previously
+  esquery,
   fetchSync,
   fs,
   glob,
@@ -69,6 +74,8 @@ export const dependencies: PresetDependencies = {
   path,
   readPkgUp,
   recast,
+  // eslint-disable-next-line @typescript-eslint/no-require-imports
+  require: <T>(modulePath: string) => require(modulePath) as T,
   simplify,
   unionfs,
   zx,

src/dependencies.ts

@@ -153,6 +153,12 @@ export const simplifyCodeTraverseOptions: import('@babel/traverse').TraverseOpti
       path.node.shorthand = true
     }
   },
+  // remove underscores from numbers
+  NumericLiteral(path) {
+    if (path.node.extra && typeof path.node.extra.raw === 'string' && path.node.extra.raw.includes('_')) {
+      path.node.extra.raw = path.node.extra.raw.replaceAll('_', '')
+    }
+  },
 }
 
 export function stripTypes(typeScriptCode: string) {

src/simplify.ts

@@ -41,3 +41,7 @@ test('simplifyCode', async () => {
 
   expect(simplifyCode(leftCode)).toBe(simplifyCode(rightCode))
 })
+
+test('snapshots', async () => {
+  expect(simplifyCode(`const a = 123_456`)).toMatchInlineSnapshot(`"const a = 123456;"`)
+})