Skip to content

Commit

Permalink
feat(cli): add support for building binaries
Browse files Browse the repository at this point in the history
  • Loading branch information
messense authored and Brooooooklyn committed Apr 1, 2022
1 parent 1a2198e commit 20b1edc
Show file tree
Hide file tree
Showing 8 changed files with 167 additions and 93 deletions.
1 change: 1 addition & 0 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ members = [
"./crates/sys",
"./examples/napi",
"./examples/napi-compat-mode",
"./examples/binary",
"./bench",
"./memory-testing",
]
Expand Down
222 changes: 130 additions & 92 deletions cli/src/build.ts
Original file line number Diff line number Diff line change
Expand Up @@ -100,6 +100,10 @@ export class BuildCommand extends Command {
description: `Bypass to ${chalk.green('cargo build --features')}`,
})

bin?: string = Option.String('--bin', {
description: `Bypass to ${chalk.green('cargo build --bin')}`,
})

dts?: string = Option.String('--dts', 'index.d.ts', {
description: `The filename and path of ${chalk.green(
'.d.ts',
Expand Down Expand Up @@ -180,12 +184,54 @@ export class BuildCommand extends Command {
const cwd = this.cargoCwd
? join(process.cwd(), this.cargoCwd)
: process.cwd()

let tomlContentString: string
let tomlContent: any
try {
debug('Start read toml')
tomlContentString = await readFileAsync(join(cwd, 'Cargo.toml'), 'utf-8')
} catch {
throw new TypeError(`Could not find Cargo.toml in ${cwd}`)
}

try {
debug('Start parse toml')
tomlContent = toml.parse(tomlContentString)
} catch {
throw new TypeError('Could not parse the Cargo.toml')
}

let cargoPackageName: string
if (tomlContent.package?.name) {
cargoPackageName = tomlContent.package.name
} else {
throw new TypeError('No package.name field in Cargo.toml')
}

const cargoMetadata = JSON.parse(
execSync('cargo metadata --format-version 1', {
stdio: 'pipe',
}).toString('utf8'),
)
const packages = cargoMetadata.packages
const cargoPackage = packages.find(
(p: { name: string }) => p.name === cargoPackageName,
)
if (
!this.bin &&
cargoPackage?.targets?.length === 1 &&
cargoPackage?.targets[0].kind.length === 1 &&
cargoPackage?.targets[0].kind[0] === 'bin'
) {
this.bin = cargoPackageName
}
const releaseFlag = this.isRelease ? `--release` : ''

const targetFlag = this.targetTripleDir
? `--target ${this.targetTripleDir}`
: ''
const featuresFlag = this.features ? `--features ${this.features}` : ''
const binFlag = this.bin ? `--bin ${this.bin}` : ''
const triple = this.targetTripleDir
? parseTriple(this.targetTripleDir)
: getDefaultTargetTriple(
Expand All @@ -199,6 +245,7 @@ export class BuildCommand extends Command {
releaseFlag,
targetFlag,
featuresFlag,
binFlag,
pFlag,
this.cargoFlags,
]
Expand All @@ -224,7 +271,7 @@ export class BuildCommand extends Command {
const rustflags = process.env.RUSTFLAGS
? process.env.RUSTFLAGS.split(' ')
: []
if (triple.raw.includes('musl')) {
if (triple.raw.includes('musl') && !this.bin) {
if (!rustflags.includes('target-feature=-crt-static')) {
rustflags.push('-C target-feature=-crt-static')
}
Expand Down Expand Up @@ -317,34 +364,15 @@ export class BuildCommand extends Command {
cwd,
})
const { binaryName, packageName } = getNapiConfig(this.configFileName)
let dylibName = this.cargoName
if (!dylibName) {
let tomlContentString: string
let tomlContent: any
try {
debug('Start read toml')
tomlContentString = await readFileAsync(
join(cwd, 'Cargo.toml'),
'utf-8',
)
} catch {
throw new TypeError(`Could not find Cargo.toml in ${cwd}`)
}

try {
debug('Start parse toml')
tomlContent = toml.parse(tomlContentString)
} catch {
throw new TypeError('Could not parse the Cargo.toml')
}

if (tomlContent.package?.name) {
dylibName = tomlContent.package.name.replace(/-/g, '_')
let cargoArtifactName = this.cargoName
if (!cargoArtifactName) {
if (this.bin) {
cargoArtifactName = cargoPackageName
} else {
throw new TypeError('No package.name field in Cargo.toml')
cargoArtifactName = cargoPackageName.replace(/-/g, '_')
}

if (!tomlContent.lib?.['crate-type']?.includes?.('cdylib')) {
if (!this.bin && !tomlContent.lib?.['crate-type']?.includes?.('cdylib')) {
throw new TypeError(
`Missing ${chalk.green('crate-type = ["cdylib"]')} in ${chalk.green(
'[lib]',
Expand All @@ -353,34 +381,40 @@ export class BuildCommand extends Command {
}
}

debug(`Dylib name: ${chalk.greenBright(dylibName)}`)
if (this.bin) {
debug(`Binary name: ${chalk.greenBright(cargoArtifactName)}`)
} else {
debug(`Dylib name: ${chalk.greenBright(cargoArtifactName)}`)
}

const platform = triple.platform
let libExt
let libExt = ''

debug(`Platform: ${chalk.greenBright(platform)}`)

// Platform based massaging for build commands
switch (platform) {
case 'darwin':
libExt = '.dylib'
dylibName = `lib${dylibName}`
break
case 'win32':
libExt = '.dll'
break
case 'linux':
case 'freebsd':
case 'openbsd':
case 'android':
case 'sunos':
dylibName = `lib${dylibName}`
libExt = '.so'
break
default:
throw new TypeError(
'Operating system not currently supported or recognized by the build script',
)
if (!this.bin) {
switch (platform) {
case 'darwin':
libExt = '.dylib'
cargoArtifactName = `lib${cargoArtifactName}`
break
case 'win32':
libExt = '.dll'
break
case 'linux':
case 'freebsd':
case 'openbsd':
case 'android':
case 'sunos':
cargoArtifactName = `lib${cargoArtifactName}`
libExt = '.so'
break
default:
throw new TypeError(
'Operating system not currently supported or recognized by the build script',
)
}
}

const targetRootDir = await findUp(cwd)
Expand All @@ -399,7 +433,9 @@ export class BuildCommand extends Command {
: ''

debug(`Platform name: ${platformName || chalk.green('[Empty]')}`)
const distFileName = `${binaryName}${platformName}.node`
const distFileName = this.bin
? cargoArtifactName!
: `${binaryName}${platformName}.node`

const distModulePath = join(this.destDir ?? '.', distFileName)

Expand All @@ -419,7 +455,7 @@ export class BuildCommand extends Command {
targetRootDir,
'target',
targetDir,
`${dylibName}${libExt}`,
`${cargoArtifactName}${libExt}`,
)

if (existsSync(distModulePath)) {
Expand All @@ -430,50 +466,52 @@ export class BuildCommand extends Command {
debug(`Write binary content to [${chalk.yellowBright(distModulePath)}]`)
await copyFileAsync(sourcePath, distModulePath)

const dtsFilePath = join(
process.cwd(),
this.destDir ?? '.',
this.dts ?? 'index.d.ts',
)
if (!this.bin) {
const dtsFilePath = join(
process.cwd(),
this.destDir ?? '.',
this.dts ?? 'index.d.ts',
)

const idents = await processIntermediateTypeFile(
intermediateTypeFile,
dtsFilePath,
)
if (this.pipe) {
const pipeCommand = `${this.pipe} ${dtsFilePath}`
console.info(`Run ${chalk.green(pipeCommand)}`)
try {
execSync(pipeCommand, { stdio: 'inherit', env: process.env })
} catch (e) {
console.warn(
chalk.bgYellowBright('Pipe the dts file to command failed'),
e,
)
if (this.pipe) {
const pipeCommand = `${this.pipe} ${dtsFilePath}`
console.info(`Run ${chalk.green(pipeCommand)}`)
try {
execSync(pipeCommand, { stdio: 'inherit', env: process.env })
} catch (e) {
console.warn(
chalk.bgYellowBright('Pipe the dts file to command failed'),
e,
)
}
}
}
const jsBindingFilePath =
this.jsBinding &&
this.jsBinding !== 'false' &&
this.appendPlatformToFilename
? join(process.cwd(), this.jsBinding)
: null
await writeJsBinding(
binaryName,
this.jsPackageName ?? packageName,
jsBindingFilePath,
idents,
)
if (this.pipe && jsBindingFilePath) {
const pipeCommand = `${this.pipe} ${jsBindingFilePath}`
console.info(`Run ${chalk.green(pipeCommand)}`)
try {
execSync(pipeCommand, { stdio: 'inherit', env: process.env })
} catch (e) {
console.warn(
chalk.bgYellowBright('Pipe the js binding file to command failed'),
e,
)
const jsBindingFilePath =
this.jsBinding &&
this.jsBinding !== 'false' &&
this.appendPlatformToFilename
? join(process.cwd(), this.jsBinding)
: null
const idents = await processIntermediateTypeFile(
intermediateTypeFile,
dtsFilePath,
)
await writeJsBinding(
binaryName,
this.jsPackageName ?? packageName,
jsBindingFilePath,
idents,
)
if (this.pipe && jsBindingFilePath) {
const pipeCommand = `${this.pipe} ${jsBindingFilePath}`
console.info(`Run ${chalk.green(pipeCommand)}`)
try {
execSync(pipeCommand, { stdio: 'inherit', env: process.env })
} catch (e) {
console.warn(
chalk.bgYellowBright('Pipe the js binding file to command failed'),
e,
)
}
}
}
}
Expand Down
2 changes: 2 additions & 0 deletions examples/binary/.gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
*.node
napi-examples-binary
7 changes: 7 additions & 0 deletions examples/binary/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
[package]
edition = "2021"
name = "napi-examples-binary"
publish = false
version = "0.1.0"

[dependencies]
14 changes: 14 additions & 0 deletions examples/binary/package.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
{
"name": "binary",
"private": true,
"version": "0.0.0",
"bin": "napi-examples-binary",
"scripts": {
"build": "node ../../cli/scripts/index.js build --js false",
"build-aarch64": "node ../../cli/scripts/index.js build --js false --target aarch64-unknown-linux-gnu",
"build-armv7": "node ../../cli/scripts/index.js build --js false --target armv7-unknown-linux-gnueabihf",
"build-i686": "node ../../cli/scripts/index.js build --js false --target i686-pc-windows-msvc",
"build-i686-release": "node ../../cli/scripts/index.js build --js false --release --target i686-pc-windows-msvc",
"build-release": "node ../../cli/scripts/index.js build --js false --release"
}
}
3 changes: 3 additions & 0 deletions examples/binary/src/main.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
fn main() {
println!("Hello World!");
}
3 changes: 2 additions & 1 deletion package.json
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,8 @@
"triples",
"memory-testing",
"examples/napi",
"examples/napi-compat-mode"
"examples/napi-compat-mode",
"examples/binary"
],
"repository": {
"type": "git",
Expand Down
8 changes: 8 additions & 0 deletions yarn.lock
Original file line number Diff line number Diff line change
Expand Up @@ -2217,6 +2217,14 @@ __metadata:
languageName: node
linkType: hard

"binary@workspace:examples/binary":
version: 0.0.0-use.local
resolution: "binary@workspace:examples/binary"
bin:
binary: napi-examples-binary
languageName: unknown
linkType: soft

"bl@npm:^4.0.3, bl@npm:^4.1.0":
version: 4.1.0
resolution: "bl@npm:4.1.0"
Expand Down

0 comments on commit 20b1edc

Please sign in to comment.