David Snelling
f5e4b8b93e
- Removed obsolete scripts: `test-all-environments.js`, `test-fallback-function.js`, `test-fallback-simple.js`, `test-fix.js`, `test-tensorflow-textencoder.js`, `test-unified-encoding.js`, and `test-worker-utils.js`. - Introduced `scripts/comprehensive-test.js` as a unified testing script covering all environments: Browser, Node.js, and CLI. - Added `examples/cli-wrapper-example.js` to demonstrate a proper CLI implementation with TensorFlow.js initialization. This refactor simplifies the testing structure by consolidating redundant scripts into a single comprehensive script while ensuring robust cross-environment coverage.
113 lines
3.7 KiB
JavaScript
Executable file
113 lines
3.7 KiB
JavaScript
Executable file
#!/usr/bin/env node
|
|
|
|
/**
|
|
* CLI Wrapper Script for @soulcraft/brainy-cli
|
|
*
|
|
* This script serves as a wrapper for the Brainy CLI, ensuring that command-line arguments
|
|
* are properly passed to the CLI when invoked through the globally installed package.
|
|
*/
|
|
|
|
// CRITICAL: Apply TensorFlow.js environment patch before importing any other modules
|
|
// This prevents the "TextEncoder is not a constructor" error in Node.js environments
|
|
// by ensuring the global.PlatformNode class is defined before TensorFlow.js loads
|
|
function applyTensorFlowPatch() {
|
|
try {
|
|
// Define a custom Platform class that works in Node.js environments
|
|
class Platform {
|
|
constructor() {
|
|
// Create a util object with necessary methods and constructors
|
|
this.util = {
|
|
// Use native TextEncoder and TextDecoder constructors
|
|
TextEncoder: global.TextEncoder || TextEncoder,
|
|
TextDecoder: global.TextDecoder || TextDecoder
|
|
}
|
|
|
|
// Initialize using native constructors directly
|
|
this.textEncoder = new TextEncoder()
|
|
this.textDecoder = new TextDecoder()
|
|
}
|
|
|
|
// Define isTypedArray directly on the instance
|
|
isTypedArray(arr) {
|
|
return !!(ArrayBuffer.isView(arr) && !(arr instanceof DataView))
|
|
}
|
|
}
|
|
|
|
// Assign the Platform class to the global object as PlatformNode
|
|
global.PlatformNode = Platform
|
|
// Also create an instance and assign it to global.platformNode (lowercase p)
|
|
global.platformNode = new Platform()
|
|
|
|
console.log('Applied TensorFlow.js platform patch in CLI wrapper')
|
|
} catch (error) {
|
|
console.warn('Failed to apply TensorFlow.js platform patch:', error)
|
|
}
|
|
}
|
|
|
|
// Apply the patch immediately
|
|
applyTensorFlowPatch()
|
|
|
|
import { spawn } from 'child_process'
|
|
import { fileURLToPath } from 'url'
|
|
import { dirname, join } from 'path'
|
|
import fs from 'fs'
|
|
|
|
// Node.js v24+ compatibility patches are now applied above,
|
|
// before any imports, to ensure TensorFlow.js can correctly
|
|
// detect and use the TextEncoder/TextDecoder in the environment.
|
|
|
|
// Get the directory of the current module
|
|
const __filename = fileURLToPath(import.meta.url)
|
|
const __dirname = dirname(__filename)
|
|
|
|
// Find the main package
|
|
const mainPackagePath = join(__dirname, 'node_modules', '@soulcraft', 'brainy')
|
|
|
|
// Path to the actual CLI script in this package
|
|
const cliPath = join(__dirname, 'dist', 'cli.js')
|
|
|
|
// Check if the CLI script exists
|
|
if (!fs.existsSync(cliPath)) {
|
|
console.error(`Error: CLI script not found at ${cliPath}`)
|
|
console.error(
|
|
'This is likely because the CLI was not built during package installation.'
|
|
)
|
|
console.error('Please reinstall the package with:')
|
|
console.error('npm uninstall -g @soulcraft/brainy-cli')
|
|
console.error('npm install -g @soulcraft/brainy-cli')
|
|
process.exit(1)
|
|
}
|
|
|
|
// Special handling for version flags
|
|
if (process.argv.includes('--version') || process.argv.includes('-V')) {
|
|
// Read version directly from package.json to ensure it's always correct
|
|
try {
|
|
const packageJsonPath = join(__dirname, 'package.json')
|
|
const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, 'utf8'))
|
|
console.log(packageJson.version)
|
|
process.exit(0)
|
|
} catch (error) {
|
|
console.error('Error loading version information:', error.message)
|
|
process.exit(1)
|
|
}
|
|
}
|
|
|
|
// Forward all arguments to the CLI script
|
|
const args = process.argv.slice(2)
|
|
|
|
// Check if npm is passing --force flag
|
|
// When npm runs with --force, it sets the npm_config_force environment variable
|
|
if (
|
|
process.env.npm_config_force === 'true' &&
|
|
args.includes('clear') &&
|
|
!args.includes('--force') &&
|
|
!args.includes('-f')
|
|
) {
|
|
args.push('--force')
|
|
}
|
|
|
|
const cli = spawn('node', [cliPath, ...args], { stdio: 'inherit' })
|
|
|
|
cli.on('close', (code) => {
|
|
process.exit(code)
|
|
})
|