Files
leon-ai--leon/scripts/setup/setup-tools-dependencies.js
2026-07-13 12:44:08 +08:00

81 lines
2.0 KiB
JavaScript

import fs from 'node:fs'
import path from 'node:path'
import { PROFILE_TOOLS_PATH, TOOLS_PATH } from '@/constants'
import { createSetupStatus } from './setup-status'
import {
syncNodejsSourceDependencies,
syncPythonSourceDependencies
} from './sync-source-dependencies'
const NODEJS_SOURCE_PATH = path.join('src', 'nodejs')
const PYTHON_SOURCE_PATH = path.join('src', 'python')
const getToolSourcePaths = async (toolsPath) => {
if (!fs.existsSync(toolsPath)) {
return []
}
const toolkitEntries = await fs.promises.readdir(toolsPath, {
withFileTypes: true
})
const sourcePaths = []
for (const toolkitEntry of toolkitEntries) {
if (!toolkitEntry.isDirectory()) {
continue
}
const toolkitPath = path.join(toolsPath, toolkitEntry.name)
const toolEntries = await fs.promises.readdir(toolkitPath, {
withFileTypes: true
})
for (const toolEntry of toolEntries) {
if (!toolEntry.isDirectory()) {
continue
}
const toolPath = path.join(toolkitPath, toolEntry.name)
sourcePaths.push({
bridge: 'nodejs',
path: path.join(toolPath, NODEJS_SOURCE_PATH)
})
sourcePaths.push({
bridge: 'python',
path: path.join(toolPath, PYTHON_SOURCE_PATH)
})
}
}
return sourcePaths
}
/**
* Sync tool dependencies next to each tool source folder.
*/
export default async function setupToolsDependencies() {
const status = createSetupStatus('Setting up tool dependencies...').start()
try {
const sourcePaths = [
...(await getToolSourcePaths(TOOLS_PATH)),
...(await getToolSourcePaths(PROFILE_TOOLS_PATH))
]
for (const sourcePath of sourcePaths) {
if (sourcePath.bridge === 'nodejs') {
await syncNodejsSourceDependencies(sourcePath.path)
} else {
await syncPythonSourceDependencies(sourcePath.path)
}
}
status.succeed('Tool dependencies: ready')
} catch (e) {
status.fail('Failed to set up tool dependencies')
throw e
}
}