use std::path::{Path, PathBuf};
use eyre::Result;
use crate::prepare::rule::PrepareProviderConfig;
use crate::prepare::{PrepareCommand, PrepareProvider};
use super::ProviderBase;
#[derive(Debug)]
pub struct PoetryPrepareProvider {
base: ProviderBase,
}
impl PoetryPrepareProvider {
pub fn new(project_root: &Path, config: PrepareProviderConfig) -> Self {
Self {
base: ProviderBase::new("poetry", project_root, config),
}
}
}
impl PrepareProvider for PoetryPrepareProvider {
fn base(&self) -> &ProviderBase {
&self.base
}
fn sources(&self) -> Vec<PathBuf> {
let root = self.base.config_root();
vec![root.join("poetry.lock"), root.join("pyproject.toml")]
}
fn outputs(&self) -> Vec<PathBuf> {
vec![self.base.config_root().join(".venv")]
}
fn prepare_command(&self) -> Result<PrepareCommand> {
if let Some(run) = &self.base.config.run {
return PrepareCommand::from_string(run, &self.base.project_root, &self.base.config);
}
Ok(PrepareCommand {
program: "poetry".to_string(),
args: vec!["install".to_string()],
env: self.base.config.env.clone(),
cwd: Some(self.base.config_root()),
description: self
.base
.config
.description
.clone()
.unwrap_or_else(|| "poetry install".to_string()),
})
}
fn is_applicable(&self) -> bool {
self.base.config_root().join("poetry.lock").exists()
}
}