use std::io;
use std::path::{Path, PathBuf};
use thiserror::Error;
use uv_python::{Interpreter, PythonEnvironment};
pub use virtualenv::{OnExisting, RemovalReason, remove_virtualenv};
mod virtualenv;
#[derive(Debug, Error)]
pub enum Error {
#[error(transparent)]
Io(#[from] io::Error),
#[error(
"Could not find a suitable Python executable for the virtual environment based on the interpreter: {0}"
)]
NotFound(String),
#[error(transparent)]
Python(#[from] uv_python::managed::Error),
#[error("A {name} already exists at `{}`. Use `--clear` to replace it", path.display())]
Exists {
name: &'static str,
path: PathBuf,
},
}
#[derive(Debug)]
pub enum Prompt {
CurrentDirectoryName,
Static(String),
None,
}
impl Prompt {
pub fn from_args(prompt: Option<String>) -> Self {
match prompt {
Some(prompt) if prompt == "." => Self::CurrentDirectoryName,
Some(prompt) => Self::Static(prompt),
None => Self::None,
}
}
}
#[expect(clippy::fn_params_excessive_bools)]
pub fn create_venv(
location: &Path,
interpreter: Interpreter,
prompt: Prompt,
system_site_packages: bool,
on_existing: OnExisting,
relocatable: bool,
seed: bool,
upgradeable: bool,
) -> Result<PythonEnvironment, Error> {
let virtualenv = virtualenv::create(
location,
&interpreter,
prompt,
system_site_packages,
on_existing,
relocatable,
seed,
upgradeable,
)?;
let interpreter = interpreter.with_virtualenv(virtualenv);
Ok(PythonEnvironment::from_interpreter(interpreter))
}