napi-rs/scripts/napi.js
2020-06-21 17:00:40 +08:00

90 lines
2 KiB
JavaScript
Executable file

#!/usr/bin/env node
const parseArgs = require('minimist')
const path = require('path')
const os = require('os')
const toml = require('toml')
const fs = require('fs')
let tomlContentString
let tomlContent
let moduleName
try {
tomlContentString = fs.readFileSync(
path.join(process.cwd(), 'Cargo.toml'),
'utf-8',
)
} catch {
throw new TypeError('Can not find Cargo.toml in process.cwd')
}
try {
tomlContent = toml.parse(tomlContentString)
} catch {
throw new TypeError('Can not parse the Cargo.toml')
}
if (tomlContent.package && tomlContent.package.name) {
moduleName = tomlContent.package.name.replace(/-/g, '_')
} else {
throw new TypeError('No package.name field in Cargo.toml')
}
const argv = parseArgs(process.argv.slice(2), {
boolean: ['release', 'platform', 'musl'],
})
const platform = os.platform()
let libExt
let dylibName = moduleName
// Platform based massaging for build commands
switch (platform) {
case 'darwin':
libExt = '.dylib'
dylibName = `lib${moduleName}`
break
case 'win32':
libExt = '.dll'
break
case 'linux':
dylibName = `lib${moduleName}`
libExt = '.so'
break
default:
console.error(
'Operating system not currently supported or recognized by the build script',
)
process.exit(1)
}
const targetDir = argv.release ? 'release' : 'debug'
const platformName = argv.musl ? '.musl' : argv.platform ? `.${platform}` : ''
let distModulePath =
argv._[0] ||
path.join('target', targetDir, `${moduleName}${platformName}.node`)
const parsedDist = path.parse(distModulePath)
if (!parsedDist.name || parsedDist.name === '.') {
distModulePath = moduleName
}
if (!parsedDist.ext) {
distModulePath = `${distModulePath}${platformName}.node`
}
const pos = __dirname.indexOf('node_modules')
const dylibContent = fs.readFileSync(
path.join(
__dirname.substring(0, pos),
'target',
targetDir,
`${dylibName}${libExt}`,
),
)
fs.writeFileSync(distModulePath, dylibContent)