feat: move copy-src command to postinstall.js

fix #12

Author: HerringtonDarkholme

package.json

@@ -8,8 +8,7 @@
   "version": "0.0.1",
   "description": "Monorepo for `@ast-grep/lang-*` packages",
   "scripts": {
-    "postinstall": "pnpm -r compile && pnpm -r copy-src",
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "postinstall": "pnpm -r compile"
   },
   "keywords": [],
   "author": "Herrington Darkholme",

packages/toml/nursery.js

@@ -4,7 +4,6 @@ const assert = require('node:assert')
 
 setup({
   name: 'toml',
-  packageName: '@tree-sitter-grammars/tree-sitter-toml',
   languageRegistration: toml,
   testRunner: (parse) => {
     const sg = parse('name = "Tom"')

packages/toml/package.json

@@ -4,7 +4,6 @@
   "description": "",
   "main": "index.js",
   "scripts": {
-    "copy-src": "node nursery.js copy",
     "build": "tree-sitter build -o parser.so",
     "postinstall": "node postinstall.js",
     "test": "node nursery.js test"

packages/toml/postinstall.js

@@ -1,2 +1,5 @@
 const { postinstall } = require('@ast-grep/setup-lang')
-postinstall(__dirname)
+postinstall({
+  directory: __dirname,
+  treeSitterPackage: '@tree-sitter-grammars/tree-sitter-toml',
+})

scripts/nursery/index.ts

@@ -1,19 +1,11 @@
-import { cp } from 'fs/promises'
 import { parse, registerDynamicLanguage, SgRoot, DynamicLangRegistrations } from '@ast-grep/napi'
-import path from 'path'
-
-async function copySrc(packageName: string) {
-  const src = path.join(process.cwd(), 'node_modules', packageName,  'src')
-  await cp(src, 'src', { recursive: true })
-}
 
 /** Setup ast-grep/lang package's pre-release */
 interface SetupConfig {
   /** Name of the language. e.g. toml */
   name: string
   /** Language registration object, usually the export of index.js */
   languageRegistration: DynamicLangRegistrations[string]
-  /** Package name of tree-sitter package. e.g. tree-sitter-toml */
   packageName: string
   /** Test cases running in CI */
   testRunner: (parse: (c: string) => SgRoot) => void
@@ -28,9 +20,7 @@ function test(setupConfig: SetupConfig) {
 /** Setup ast-grep/lang package's pre-release build and test */
 export function setup(setupConfig: SetupConfig) {
   const arg = process.argv[2]
-  if (arg === 'copy') {
-    copySrc(setupConfig.packageName)
-  } else if (arg === 'test') {
+  if (arg === 'test') {
     test(setupConfig)
   }
 }

scripts/nursery/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ast-grep/nursery",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "private": false,
   "description": "",
   "main": "index.js",

scripts/setup/index.ts

@@ -9,10 +9,18 @@ function log(...args: unknown[]) {
   console.debug(`@ast-grep/lang:`, ...args)
 }
 
+interface SetupConfig {
+  /** Directory of the lang package. e.g. __dirname */
+  directory: string
+  /** Package name of tree-sitter package. e.g. tree-sitter-css */
+  treeSitterPackage: string
+}
+
 /**
  * Move prebuild or build parser
  */
-async function postinstall(dir: string) {
+function postinstall(config: SetupConfig) {
+  const dir = config.directory
   const parser = path.join(dir, 'parser.so')
   if (fs.existsSync(parser)) {
     log('parser already exists, skipping build')
@@ -24,15 +32,32 @@ async function postinstall(dir: string) {
     fs.copyFileSync(prebuild, parser)
     return
   }
+  buildDynamicLib(config)
+}
+
+function buildDynamicLib(config: SetupConfig) {
   log('building parser')
   try {
+    copySrcIfNeeded(config)
     execSync('npm run build')
   } catch (e: unknown) {
     log('build failed, please ensure tree-sitter-cli is installed as peer dependency')
     log(e)
   }
 }
 
+function copySrcIfNeeded(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')
+    return
+  }
+  log('copying tree-sitter src')
+  fs.cpSync(src, 'src', { recursive: true })
+}
+
 const PLATFORM_MAP: Record<string, string> = {
   darwin: 'macOS',
   linux: 'Linux',

scripts/setup/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ast-grep/setup-lang",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "private": false,
   "description": "",
   "main": "index.js",