fix: update file

Author: HerringtonDarkholme

package.json

@@ -8,7 +8,7 @@
   "version": "0.0.1",
   "description": "Monorepo for `@ast-grep/lang-*` packages",
   "scripts": {
-    "postinstall": "pnpm -r compile"
+    "postinstall": "pnpm -r compile-ts"
   },
   "keywords": [],
   "author": "Herrington Darkholme",

scripts/nursery/index.ts

@@ -1,8 +1,14 @@
-
-import { readFileSync, writeFileSync } from 'node:fs'
+import fs from 'node:fs'
 import path from 'node:path'
 import { parse, registerDynamicLanguage, SgRoot, DynamicLangRegistrations } from '@ast-grep/napi'
 
+/**
+ * Log to console
+ */
+function log(...args: unknown[]) {
+  console.debug(`@ast-grep/lang:`, ...args)
+}
+
 /** Setup ast-grep/lang package's pre-release */
 interface SetupConfig {
   /** the root directory of the package */
@@ -29,10 +35,23 @@ export function setup(setupConfig: SetupConfig) {
   if (arg === 'test') {
     test(setupConfig)
   } else if (arg === 'source') {
+    copySrcIfNeeded(setupConfig)
     generateLangNodeTypes(setupConfig)
   }
 }
 
+function copySrcIfNeeded(config: SetupConfig) {
+  const { dirname, treeSitterPackage } = config
+  const existing = path.join(dirname, 'src')
+  const src = path.join(dirname, 'node_modules', treeSitterPackage,  'src')
+  if (fs.existsSync(existing)) {
+    log('src exists, skipping copy')
+    return
+  }
+  log('copying tree-sitter src')
+  fs.cpSync(src, 'src', { recursive: true })
+}
+
 interface NodeBasicInfo {
   type: string
   named: boolean
@@ -83,7 +102,7 @@ function processNodeTypes(nodeTypes: NodeType[]): Record<string, NodeType> {
 
 function readLangNodeTypes(dirname: string): NodeType[] {
   const staticNodePath = path.join(dirname, 'src', 'node-types.json')
-  const content = readFileSync(staticNodePath, 'utf-8')
+  const content = fs.readFileSync(staticNodePath, 'utf-8')
   return JSON.parse(content)
 }
 
@@ -98,7 +117,7 @@ function generateLangNodeTypes(setupConfig: SetupConfig) {
       `type ${lang}Types = ${JSON.stringify(nodeTypeMap, null, 2)};` +
       '\n' +
       `export default ${lang}Types;`
-    writeFileSync(path.join(dirname, `type.d.ts`), fileContent)
+    fs.writeFileSync(path.join(dirname, `type.d.ts`), fileContent)
   } catch (e) {
     console.error(`Error while generating node types for ${lang}`)
     throw e

scripts/nursery/package.json

@@ -8,7 +8,7 @@
   "author": "",
   "license": "ISC",
   "scripts": {
-    "compile": "tsc",
+    "compile-ts": "tsc",
     "prepublishOnly": "tsc"
   },
   "dependencies": {

scripts/setup/index.ts

@@ -1,6 +1,6 @@
-import { execSync } from 'child_process'
-import fs from 'fs'
-import path from 'path'
+import { execSync } from 'node:child_process'
+import fs from 'node:fs'
+import path from 'node:path'
 
 /**
  * Log to console
@@ -32,30 +32,23 @@ function postinstall(config: SetupConfig) {
     fs.copyFileSync(prebuild, parser)
     return
   }
-  buildDynamicLib(config)
-}
-
-function buildDynamicLib(config: SetupConfig) {
-  log('building parser')
   try {
-    copySrcIfNeeded(config)
-    execSync('npm run build')
+    buildSrc(config)
   } catch (e: unknown) {
     log('build failed, please ensure tree-sitter-cli is installed as peer dependency')
     log(e)
   }
 }
 
-function copySrcIfNeeded(config: SetupConfig) {
+function buildSrc(config: SetupConfig) {
   const { directory, treeSitterPackage } = config
   const existing = path.join(directory, 'src')
-  const src = path.join(directory, 'node_modules', treeSitterPackage,  'src')
-  if (fs.existsSync(existing)) {
-    log('src exists, skipping copy')
+  if (!fs.existsSync(existing)) {
+    log('tree-sitter src not found. If you are making a lang package, please run `pnpm source`.')
     return
   }
-  log('copying tree-sitter src')
-  fs.cpSync(src, 'src', { recursive: true })
+  log('building parser from source')
+  execSync('npm run build')
 }
 
 const PLATFORM_MAP: Record<string, string> = {

scripts/setup/package.json

@@ -8,7 +8,7 @@
   "author": "",
   "license": "ISC",
   "scripts": {
-    "compile": "tsc",
+    "compile-ts": "tsc",
     "prepublishOnly": "tsc"
   },
   "publishConfig": {