feat: add setup package

Author: HerringtonDarkholme

packages/toml/package.json

@@ -18,6 +18,9 @@
   "keywords": [],
   "author": "",
   "license": "ISC",
+  "dependencies": {
+    "@ast-grep/setup-lang": "workspace:*"
+  },
   "peerDependencies": {
     "tree-sitter-cli": "0.24.6"
   },

packages/toml/postinstall.js

@@ -1,28 +1,2 @@
-const dir = __dirname
-
-const { execSync } = require('child_process')
-const fs = require('fs')
-const path = require('path')
-
-async function postinstall() {
-  const parser = path.join(dir, 'parser.so')
-  if (fs.existsSync(parser)) {
-    return
-  }
-  // resolve parser path
-  const prebuild = resolvePrebuild()
-  if (prebuild) {
-    fs.copyFileSync(prebuild, parser)
-  } else {
-    // build parser
-    execSync('npm run build')
-  }
-}
-
-function resolvePrebuild() {
-  const os = process.platform
-  const arch = process.arch
-  throw new Error(`TODO: ${os} ${arch}`)
-}
-
-postinstall()
+const { postinstall } = require('@ast-grep/setup-lang')
+postinstall(__dirname)

pnpm-lock.yaml

@@ -0,0 +1,65 @@
+const { execSync } = require('child_process')
+const fs = require('fs')
+const path = require('path')
+
+/**
+ * Log to console
+ * @param  {...string} args
+ */
+function log(...args) {
+  console.debug(`@ast-grep/lang:`, ...args)
+}
+
+/**
+ * Move prebuild or build parser
+ * @param {string} dir
+ */
+async function postinstall(dir) {
+  const parser = path.join(dir, 'parser.so')
+  if (fs.existsSync(parser)) {
+    log('parser already exists, skipping build')
+    return
+  }
+  const prebuild = resolvePrebuild()
+  if (prebuild) {
+    log('copying prebuild parser')
+    fs.copyFileSync(prebuild, parser)
+    return
+  }
+  log('building parser')
+  try {
+    execSync('npm run build')
+  } catch (e) {
+    log('build failed, please ensure tree-sitter-cli is installed as peer dependency')
+    log(e)
+  }
+}
+
+const PLATFORM_MAP = {
+  darwin: 'macOS',
+  linux: 'Linux',
+  win32: 'Windows',
+}
+
+const ARCH_MAP = {
+  x64: 'x64',
+  arm64: 'ARM64',
+}
+
+/**
+ * Resolve prebuild path
+ * @param {string} dir
+ */
+function resolvePrebuild(dir) {
+  const os = PLATFORM_MAP[process.platform]
+  const arch = ARCH_MAP[process.arch]
+  const prebuild = path.join(dir, 'prebuilds', `prebuild-${os}-${arch}`, 'parser.so')
+  if (!os || !arch || !fs.existsSync(prebuild)) {
+    log(`no prebuild for ${os} ${arch}`)
+    return undefined
+  }
+  log(`found prebuild for ${os} ${arch}`)
+  return prebuild
+}
+
+exports.postinstall = postinstall

scripts/setup/index.js

@@ -0,0 +1,9 @@
+{
+  "name": "@ast-grep/setup-lang",
+  "version": "0.0.1",
+  "description": "",
+  "main": "index.js",
+  "keywords": [],
+  "author": "",
+  "license": "ISC"
+}