use std::env;
use std::path::{Path, PathBuf};
use std::process::Command;
use std::time::Duration;
use lean_rs::{LeanBuiltCapability, LeanCapabilityPreflight, LeanLoaderDiagnosticCode};
use serde::Deserialize;
use serde_json::Value;
use crate::pool::{LeanWorkerRestartPolicyClass, LeanWorkerSessionKey};
use crate::session::{
LeanWorkerCancellationToken, LeanWorkerCapabilityMetadata, LeanWorkerProgressSink, LeanWorkerRuntimeMetadata,
LeanWorkerSession, LeanWorkerSessionConfig,
};
use crate::supervisor::{
LEAN_WORKER_REQUEST_TIMEOUT_LONG_RUNNING, LeanWorker, LeanWorkerConfig, LeanWorkerError, LeanWorkerRestartPolicy,
};
const WORKER_CHILD_ENV: &str = "LEAN_RS_WORKER_CHILD";
#[derive(Clone, Debug)]
pub struct LeanWorkerCapabilityBuilder {
project_root: PathBuf,
package: String,
lib_name: String,
imports: Vec<String>,
built_dylib_path: Option<PathBuf>,
built_capability: Option<LeanBuiltCapability>,
worker_child: Option<LeanWorkerChild>,
startup_timeout: Option<Duration>,
request_timeout: Option<Duration>,
restart_policy: Option<LeanWorkerRestartPolicy>,
metadata_check: Option<CapabilityMetadataCheck>,
}
impl LeanWorkerCapabilityBuilder {
#[must_use]
pub fn new(
project_root: impl Into<PathBuf>,
package: impl Into<String>,
lib_name: impl Into<String>,
imports: impl IntoIterator<Item = impl Into<String>>,
) -> Self {
Self {
project_root: project_root.into(),
package: package.into(),
lib_name: lib_name.into(),
imports: imports.into_iter().map(Into::into).collect(),
built_dylib_path: None,
built_capability: None,
worker_child: None,
startup_timeout: None,
request_timeout: None,
restart_policy: None,
metadata_check: None,
}
}
pub fn from_built_capability(
spec: &LeanBuiltCapability,
imports: impl IntoIterator<Item = impl Into<String>>,
) -> Result<Self, LeanWorkerError> {
let artifact = WorkerCapabilityArtifact::from_built_capability(spec)?;
let project_root = infer_lake_project_root_from_dylib(&artifact.dylib_path)?;
Ok(Self {
project_root,
package: artifact.package,
lib_name: artifact.module,
imports: imports.into_iter().map(Into::into).collect(),
built_dylib_path: Some(artifact.dylib_path),
built_capability: Some(spec.clone()),
worker_child: None,
startup_timeout: None,
request_timeout: None,
restart_policy: None,
metadata_check: None,
})
}
#[must_use]
pub fn worker_executable(mut self, path: impl Into<PathBuf>) -> Self {
self.worker_child = Some(LeanWorkerChild::path(path));
self
}
#[must_use]
pub fn worker_child(mut self, child: LeanWorkerChild) -> Self {
self.worker_child = Some(child);
self
}
#[must_use]
pub fn startup_timeout(mut self, timeout: Duration) -> Self {
self.startup_timeout = Some(timeout);
self
}
#[must_use]
pub fn request_timeout(mut self, timeout: Duration) -> Self {
self.request_timeout = Some(timeout);
self
}
#[must_use]
pub fn long_running_requests(mut self) -> Self {
self.request_timeout = Some(LEAN_WORKER_REQUEST_TIMEOUT_LONG_RUNNING);
self
}
#[must_use]
pub fn restart_policy(mut self, policy: LeanWorkerRestartPolicy) -> Self {
self.restart_policy = Some(policy);
self
}
#[must_use]
pub fn validate_metadata(mut self, export: impl Into<String>, request: Value) -> Self {
self.metadata_check = Some(CapabilityMetadataCheck {
export: export.into(),
request,
expected: None,
});
self
}
#[must_use]
pub fn expect_metadata(
mut self,
export: impl Into<String>,
request: Value,
expected: LeanWorkerCapabilityMetadata,
) -> Self {
self.metadata_check = Some(CapabilityMetadataCheck {
export: export.into(),
request,
expected: Some(expected),
});
self
}
#[must_use]
pub fn session_key(&self) -> LeanWorkerSessionKey {
let restart_policy_class = match &self.restart_policy {
Some(policy) if policy == &LeanWorkerRestartPolicy::default() => LeanWorkerRestartPolicyClass::Default,
Some(_policy) => LeanWorkerRestartPolicyClass::Custom,
None => LeanWorkerRestartPolicyClass::Default,
};
let mut key = LeanWorkerSessionKey::new(
self.project_root.clone(),
self.package.clone(),
self.lib_name.clone(),
self.imports.clone(),
)
.restart_policy_class(restart_policy_class);
if let Some(check) = &self.metadata_check {
key = key.metadata_expectation(check.export.clone(), check.request.clone(), check.expected.clone());
}
key
}
pub(crate) fn pool_request_timeout(&self) -> Duration {
self.request_timeout
.unwrap_or(crate::supervisor::LEAN_WORKER_REQUEST_TIMEOUT_DEFAULT)
}
#[must_use]
pub fn check(&self) -> LeanWorkerBootstrapReport {
let mut checks = self.bootstrap_static_checks();
if checks.iter().any(LeanWorkerBootstrapCheck::is_error) {
return LeanWorkerBootstrapReport::new(checks);
}
match self.clone().open_unchecked() {
Ok(capability) => {
drop(capability.terminate());
}
Err(err) => checks.push(check_from_open_error(&err)),
}
LeanWorkerBootstrapReport::new(checks)
}
fn bootstrap_static_checks(&self) -> Vec<LeanWorkerBootstrapCheck> {
let mut checks = Vec::new();
match self
.worker_child
.as_ref()
.map_or_else(resolve_default_worker_executable, LeanWorkerChild::resolve)
{
Ok(path) => {
if let Err(err) = validate_worker_child_path(&path) {
checks.push(check_from_open_error(&err));
}
}
Err(err) => checks.push(check_from_open_error(&err)),
}
if let Some(spec) = &self.built_capability
&& spec.resolved_manifest_path().is_ok()
{
let report = LeanCapabilityPreflight::new(spec.clone()).check();
for check in report.errors() {
checks.push(LeanWorkerBootstrapCheck::error(
LeanWorkerBootstrapDiagnosticCode::CapabilityPreflight { code: check.code() },
check.subject().to_owned(),
check.message().to_owned(),
check.repair_hint().to_owned(),
));
}
}
checks
}
pub fn open(self) -> Result<LeanWorkerCapability, LeanWorkerError> {
let report = self.bootstrap_static_report();
if let Some(check) = report.first_error() {
return Err(LeanWorkerError::Bootstrap {
code: check.code(),
message: check.message().to_owned(),
});
}
self.open_unchecked()
}
fn bootstrap_static_report(&self) -> LeanWorkerBootstrapReport {
LeanWorkerBootstrapReport::new(self.bootstrap_static_checks())
}
fn open_unchecked(self) -> Result<LeanWorkerCapability, LeanWorkerError> {
let dylib_path = match self.built_dylib_path {
Some(path) => path,
None => lean_toolchain::build_lake_target_quiet(&self.project_root, &self.lib_name)
.map_err(|diagnostic| LeanWorkerError::CapabilityBuild { diagnostic })?,
};
let worker_executable = self
.worker_child
.map_or_else(resolve_default_worker_executable, |child| child.resolve())?;
validate_worker_child_path(&worker_executable)?;
let mut config = LeanWorkerConfig::new(worker_executable);
if let Some(timeout) = self.startup_timeout {
config = config.startup_timeout(timeout);
}
if let Some(timeout) = self.request_timeout {
config = config.request_timeout(timeout);
}
if let Some(policy) = self.restart_policy {
config = config.restart_policy(policy);
}
let mut worker = LeanWorker::spawn(&config)?;
worker.health()?;
let session_config = LeanWorkerSessionConfig::new(
self.project_root.clone(),
self.package.clone(),
self.lib_name.clone(),
self.imports.clone(),
);
let validated_metadata = {
let mut session = worker.open_session(&session_config, None, None)?;
match self.metadata_check {
Some(check) => {
let metadata = session.capability_metadata(&check.export, &check.request, None, None)?;
if let Some(expected) = check.expected
&& metadata != expected
{
return Err(LeanWorkerError::CapabilityMetadataMismatch {
export: check.export,
expected: Box::new(expected),
actual: Box::new(metadata),
});
}
Some(metadata)
}
None => None,
}
};
Ok(LeanWorkerCapability {
worker,
session_config,
dylib_path,
validated_metadata,
})
}
}
#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
#[non_exhaustive]
pub enum LeanWorkerBootstrapDiagnosticCode {
WorkerChildUnresolved,
WorkerChildNotExecutable,
CapabilityPreflight { code: LeanLoaderDiagnosticCode },
WorkerHandshakeFailed,
CapabilityMetadataMismatch,
WorkerStartupFailed,
}
impl LeanWorkerBootstrapDiagnosticCode {
#[must_use]
pub const fn as_str(self) -> &'static str {
match self {
Self::WorkerChildUnresolved => "lean_rs.worker.bootstrap.child_unresolved",
Self::WorkerChildNotExecutable => "lean_rs.worker.bootstrap.child_not_executable",
Self::CapabilityPreflight { code } => code.as_str(),
Self::WorkerHandshakeFailed => "lean_rs.worker.bootstrap.handshake_failed",
Self::CapabilityMetadataMismatch => "lean_rs.worker.bootstrap.metadata_mismatch",
Self::WorkerStartupFailed => "lean_rs.worker.bootstrap.startup_failed",
}
}
}
impl std::fmt::Display for LeanWorkerBootstrapDiagnosticCode {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.write_str(self.as_str())
}
}
#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
#[non_exhaustive]
pub enum LeanWorkerBootstrapSeverity {
Info,
Warning,
Error,
}
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct LeanWorkerBootstrapCheck {
code: LeanWorkerBootstrapDiagnosticCode,
severity: LeanWorkerBootstrapSeverity,
subject: String,
message: String,
repair_hint: String,
}
impl LeanWorkerBootstrapCheck {
fn error(
code: LeanWorkerBootstrapDiagnosticCode,
subject: impl Into<String>,
message: impl Into<String>,
repair_hint: impl Into<String>,
) -> Self {
Self {
code,
severity: LeanWorkerBootstrapSeverity::Error,
subject: bound_bootstrap_text(subject.into()),
message: bound_bootstrap_text(message.into()),
repair_hint: bound_bootstrap_text(repair_hint.into()),
}
}
#[must_use]
pub fn code(&self) -> LeanWorkerBootstrapDiagnosticCode {
self.code
}
#[must_use]
pub fn severity(&self) -> LeanWorkerBootstrapSeverity {
self.severity
}
#[must_use]
pub fn subject(&self) -> &str {
&self.subject
}
#[must_use]
pub fn message(&self) -> &str {
&self.message
}
#[must_use]
pub fn repair_hint(&self) -> &str {
&self.repair_hint
}
fn is_error(&self) -> bool {
self.severity == LeanWorkerBootstrapSeverity::Error
}
}
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct LeanWorkerBootstrapReport {
checks: Vec<LeanWorkerBootstrapCheck>,
}
impl LeanWorkerBootstrapReport {
fn new(checks: Vec<LeanWorkerBootstrapCheck>) -> Self {
Self { checks }
}
#[must_use]
pub fn checks(&self) -> &[LeanWorkerBootstrapCheck] {
&self.checks
}
pub fn errors(&self) -> impl Iterator<Item = &LeanWorkerBootstrapCheck> {
self.checks
.iter()
.filter(|check| check.severity == LeanWorkerBootstrapSeverity::Error)
}
#[must_use]
pub fn is_ok(&self) -> bool {
self.first_error().is_none()
}
#[must_use]
pub fn first_error(&self) -> Option<&LeanWorkerBootstrapCheck> {
self.errors().next()
}
}
#[derive(Debug)]
pub struct LeanWorkerCapability {
worker: LeanWorker,
session_config: LeanWorkerSessionConfig,
dylib_path: PathBuf,
validated_metadata: Option<LeanWorkerCapabilityMetadata>,
}
impl LeanWorkerCapability {
pub fn open_session(
&mut self,
cancellation: Option<&LeanWorkerCancellationToken>,
progress: Option<&dyn LeanWorkerProgressSink>,
) -> Result<LeanWorkerSession<'_>, LeanWorkerError> {
self.worker.open_session(&self.session_config, cancellation, progress)
}
#[must_use]
pub fn dylib_path(&self) -> &Path {
&self.dylib_path
}
#[must_use]
pub fn session_config(&self) -> &LeanWorkerSessionConfig {
&self.session_config
}
#[must_use]
pub fn validated_metadata(&self) -> Option<&LeanWorkerCapabilityMetadata> {
self.validated_metadata.as_ref()
}
#[must_use]
pub fn runtime_metadata(&self) -> LeanWorkerRuntimeMetadata {
self.worker.runtime_metadata()
}
#[must_use]
pub fn worker(&self) -> &LeanWorker {
&self.worker
}
#[must_use]
pub fn worker_mut(&mut self) -> &mut LeanWorker {
&mut self.worker
}
pub fn terminate(self) -> Result<crate::supervisor::LeanWorkerExit, LeanWorkerError> {
self.worker.terminate()
}
}
#[derive(Clone, Debug)]
struct CapabilityMetadataCheck {
export: String,
request: Value,
expected: Option<LeanWorkerCapabilityMetadata>,
}
#[derive(Debug)]
struct WorkerCapabilityArtifact {
dylib_path: PathBuf,
package: String,
module: String,
}
impl WorkerCapabilityArtifact {
fn from_built_capability(spec: &LeanBuiltCapability) -> Result<Self, LeanWorkerError> {
if let Ok(manifest_path) = spec.resolved_manifest_path() {
return Self::from_manifest(&manifest_path);
}
let dylib_path = spec.dylib_path().map_err(|err| LeanWorkerError::Setup {
message: err.to_string(),
})?;
let package = spec.package_name().ok_or_else(|| LeanWorkerError::Setup {
message: "LeanBuiltCapability is missing the Lake package name; call `.package(...)`".to_owned(),
})?;
let module = spec.module_name().ok_or_else(|| LeanWorkerError::Setup {
message: "LeanBuiltCapability is missing the root Lean module name; call `.module(...)`".to_owned(),
})?;
Ok(Self {
dylib_path,
package: package.to_owned(),
module: module.to_owned(),
})
}
fn from_manifest(manifest_path: &Path) -> Result<Self, LeanWorkerError> {
let bytes = std::fs::read(manifest_path).map_err(|err| LeanWorkerError::Bootstrap {
code: LeanWorkerBootstrapDiagnosticCode::CapabilityPreflight {
code: LeanLoaderDiagnosticCode::MissingManifest,
},
message: format!(
"could not read Lean capability manifest '{}': {err}",
manifest_path.display()
),
})?;
let manifest: WorkerCapabilityManifest =
serde_json::from_slice(&bytes).map_err(|err| LeanWorkerError::Bootstrap {
code: LeanWorkerBootstrapDiagnosticCode::CapabilityPreflight {
code: LeanLoaderDiagnosticCode::MalformedManifest,
},
message: format!(
"Lean capability manifest '{}' is malformed: {err}",
manifest_path.display()
),
})?;
if manifest.schema_version != u64::from(lean_toolchain::CAPABILITY_MANIFEST_SCHEMA_VERSION) {
return Err(LeanWorkerError::Bootstrap {
code: LeanWorkerBootstrapDiagnosticCode::CapabilityPreflight {
code: LeanLoaderDiagnosticCode::UnsupportedManifestSchema,
},
message: format!(
"unsupported Lean capability manifest schema {}; supported schema is {}",
manifest.schema_version,
lean_toolchain::CAPABILITY_MANIFEST_SCHEMA_VERSION
),
});
}
Ok(Self {
dylib_path: manifest.primary_dylib,
package: manifest.package,
module: manifest.module,
})
}
}
#[derive(Deserialize)]
struct WorkerCapabilityManifest {
schema_version: u64,
primary_dylib: PathBuf,
package: String,
module: String,
}
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct LeanWorkerChild {
executable_name: Option<String>,
explicit_path: Option<PathBuf>,
env_var: Option<String>,
}
impl LeanWorkerChild {
#[must_use]
pub fn sibling(executable_name: impl Into<String>) -> Self {
Self {
executable_name: Some(with_exe_suffix(executable_name.into())),
explicit_path: None,
env_var: None,
}
}
#[must_use]
pub fn path(path: impl Into<PathBuf>) -> Self {
Self {
executable_name: None,
explicit_path: Some(path.into()),
env_var: None,
}
}
#[must_use]
pub fn env_override(mut self, env_var: impl Into<String>) -> Self {
self.env_var = Some(env_var.into());
self
}
fn resolve(&self) -> Result<PathBuf, LeanWorkerError> {
let mut tried = Vec::new();
if let Some(env_var) = &self.env_var
&& let Some(value) = env::var_os(env_var)
{
let path = PathBuf::from(value);
if path.is_file() {
return Ok(path);
}
tried.push(path);
return Err(LeanWorkerError::WorkerChildUnresolved { tried });
}
if let Some(path) = &self.explicit_path {
return Ok(path.clone());
}
let executable_name = self
.executable_name
.clone()
.unwrap_or_else(|| with_exe_suffix("lean-rs-worker-child".to_owned()));
tried.extend(candidate_sibling_worker_paths(&executable_name));
if executable_name == with_exe_suffix("lean-rs-worker-child".to_owned())
&& let Some(path) = try_build_workspace_worker_child(&executable_name, &mut tried)
{
return Ok(path);
}
for path in dedup_paths(&tried) {
if path.is_file() {
return Ok(path);
}
}
Err(LeanWorkerError::WorkerChildUnresolved { tried })
}
}
impl Default for LeanWorkerChild {
fn default() -> Self {
Self::sibling("lean-rs-worker-child").env_override(WORKER_CHILD_ENV)
}
}
fn resolve_default_worker_executable() -> Result<PathBuf, LeanWorkerError> {
LeanWorkerChild::default().resolve()
}
fn validate_worker_child_path(path: &Path) -> Result<(), LeanWorkerError> {
if !path.is_file() {
return Err(LeanWorkerError::WorkerChildNotExecutable {
path: path.to_path_buf(),
reason: "path does not point to a file".to_owned(),
});
}
if !is_executable_file(path) {
return Err(LeanWorkerError::WorkerChildNotExecutable {
path: path.to_path_buf(),
reason: "file is not executable by this user".to_owned(),
});
}
Ok(())
}
#[cfg(unix)]
fn is_executable_file(path: &Path) -> bool {
use std::os::unix::fs::PermissionsExt as _;
std::fs::metadata(path).is_ok_and(|metadata| metadata.permissions().mode() & 0o111 != 0)
}
#[cfg(not(unix))]
fn is_executable_file(_path: &Path) -> bool {
true
}
fn check_from_open_error(err: &LeanWorkerError) -> LeanWorkerBootstrapCheck {
match err {
LeanWorkerError::WorkerChildUnresolved { tried } => LeanWorkerBootstrapCheck::error(
LeanWorkerBootstrapDiagnosticCode::WorkerChildUnresolved,
"worker child",
format!("could not resolve worker child; tried {}", format_paths(tried)),
"ship an app-owned worker child binary beside the app or configure LeanWorkerChild::env_override",
),
LeanWorkerError::WorkerChildNotExecutable { path, reason } => LeanWorkerBootstrapCheck::error(
LeanWorkerBootstrapDiagnosticCode::WorkerChildNotExecutable,
path.display().to_string(),
reason.clone(),
"ship an app-owned worker child binary and ensure it is executable",
),
LeanWorkerError::Bootstrap { code, message } => LeanWorkerBootstrapCheck::error(
*code,
code.as_str(),
message.clone(),
"fix the reported bootstrap input",
),
LeanWorkerError::Handshake { message } => LeanWorkerBootstrapCheck::error(
LeanWorkerBootstrapDiagnosticCode::WorkerHandshakeFailed,
"worker handshake",
message.clone(),
"ensure the worker child calls lean_rs_worker::run_worker_child_stdio and matches this crate version",
),
LeanWorkerError::Timeout {
operation: "startup", ..
} => LeanWorkerBootstrapCheck::error(
LeanWorkerBootstrapDiagnosticCode::WorkerHandshakeFailed,
"worker handshake",
err.to_string(),
"check that the worker child starts promptly and writes the lean-rs-worker handshake",
),
LeanWorkerError::CapabilityMetadataMismatch { export, .. } => LeanWorkerBootstrapCheck::error(
LeanWorkerBootstrapDiagnosticCode::CapabilityMetadataMismatch,
export.clone(),
"capability metadata did not match the requested expectation",
"rebuild or select a capability whose metadata matches the caller expectation",
),
other @ (LeanWorkerError::Spawn { .. }
| LeanWorkerError::CapabilityBuild { .. }
| LeanWorkerError::Setup { .. }
| LeanWorkerError::Protocol { .. }
| LeanWorkerError::Worker { .. }
| LeanWorkerError::ChildExited { .. }
| LeanWorkerError::ChildPanicOrAbort { .. }
| LeanWorkerError::Timeout { .. }
| LeanWorkerError::Cancelled { .. }
| LeanWorkerError::ProgressPanic { .. }
| LeanWorkerError::DataSinkPanic { .. }
| LeanWorkerError::DiagnosticSinkPanic { .. }
| LeanWorkerError::StreamExportFailed { .. }
| LeanWorkerError::StreamCallbackFailed { .. }
| LeanWorkerError::StreamRowMalformed { .. }
| LeanWorkerError::CapabilityMetadataMalformed { .. }
| LeanWorkerError::CapabilityDoctorMalformed { .. }
| LeanWorkerError::TypedCommandRequestEncode { .. }
| LeanWorkerError::TypedCommandResponseDecode { .. }
| LeanWorkerError::TypedCommandRowDecode { .. }
| LeanWorkerError::TypedCommandSummaryDecode { .. }
| LeanWorkerError::LeaseInvalidated { .. }
| LeanWorkerError::WorkerPoolExhausted { .. }
| LeanWorkerError::WorkerPoolMemoryBudgetExceeded { .. }
| LeanWorkerError::WorkerPoolQueueTimeout { .. }
| LeanWorkerError::UnsupportedRequest { .. }
| LeanWorkerError::Wait { .. }) => LeanWorkerBootstrapCheck::error(
LeanWorkerBootstrapDiagnosticCode::WorkerStartupFailed,
"worker bootstrap",
other.to_string(),
"run the bootstrap check in a deployment environment and rebuild the worker child or capability artifact",
),
}
}
fn format_paths(paths: &[PathBuf]) -> String {
if paths.is_empty() {
return "<none>".to_owned();
}
paths
.iter()
.map(|path| path.display().to_string())
.collect::<Vec<_>>()
.join(", ")
}
fn bound_bootstrap_text(mut text: String) -> String {
const LIMIT: usize = 1_024;
if text.len() <= LIMIT {
return text;
}
while !text.is_char_boundary(LIMIT) {
text.pop();
}
text.truncate(LIMIT);
text.push_str("...");
text
}
fn candidate_sibling_worker_paths(executable_name: &str) -> Vec<PathBuf> {
let mut tried = Vec::new();
if let Ok(current_exe) = env::current_exe() {
if let Some(dir) = current_exe.parent() {
tried.push(dir.join(executable_name));
}
if let Some(profile_dir) = current_exe.parent().and_then(Path::parent) {
tried.push(profile_dir.join(executable_name));
}
}
tried
}
fn with_exe_suffix(mut executable_name: String) -> String {
if !env::consts::EXE_SUFFIX.is_empty() && !executable_name.ends_with(env::consts::EXE_SUFFIX) {
executable_name.push_str(env::consts::EXE_SUFFIX);
}
executable_name
}
fn infer_lake_project_root_from_dylib(dylib_path: &Path) -> Result<PathBuf, LeanWorkerError> {
let lib_dir = dylib_path.parent();
let build_dir = lib_dir.and_then(Path::parent);
let lake_dir = build_dir.and_then(Path::parent);
let project_root = lake_dir.and_then(Path::parent);
match (lib_dir, build_dir, lake_dir, project_root) {
(Some(lib), Some(build), Some(lake), Some(root))
if lib.file_name().is_some_and(|name| name == "lib")
&& build.file_name().is_some_and(|name| name == "build")
&& lake.file_name().is_some_and(|name| name == ".lake") =>
{
Ok(root.to_path_buf())
}
_ => Err(LeanWorkerError::Setup {
message: format!(
"built capability dylib '{}' is not under a standard .lake/build/lib directory",
dylib_path.display()
),
}),
}
}
fn try_build_workspace_worker_child(executable_name: &str, tried: &mut Vec<PathBuf>) -> Option<PathBuf> {
let manifest_dir = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
let workspace = manifest_dir.parent()?.parent()?;
if !workspace
.join("crates")
.join("lean-rs-worker")
.join("Cargo.toml")
.is_file()
{
return None;
}
let debug = workspace.join("target").join("debug").join(executable_name);
let release = workspace.join("target").join("release").join(executable_name);
tried.push(debug.clone());
tried.push(release.clone());
if debug.is_file() {
return Some(debug);
}
if release.is_file() {
return Some(release);
}
let cargo = env::var_os("CARGO").unwrap_or_else(|| "cargo".into());
let status = Command::new(cargo)
.current_dir(workspace)
.args(["build", "-p", "lean-rs-worker", "--bin", "lean-rs-worker-child"])
.status()
.ok()?;
if !status.success() {
return None;
}
debug.is_file().then_some(debug)
}
fn dedup_paths(paths: &[PathBuf]) -> Vec<PathBuf> {
let mut unique = Vec::new();
for path in paths {
if !unique.iter().any(|existing| existing == path) {
unique.push(path.clone());
}
}
unique
}