mirror of
https://github.com/astral-sh/uv.git
synced 2025-07-07 21:35:00 +00:00
cba270f750
## Summary The `--index-strategy` is linked to the index locations, which we propagate to source distribution builds; so it makes sense to pass the `--index-strategy` too. While I was here, I made `exclude_newer` a required argument so that we don't forget to set it via the `with_options` builder. Closes https://github.com/astral-sh/uv/issues/4465.
114 lines
3.5 KiB
Rust
114 lines
3.5 KiB
Rust
use std::env;
|
|
use std::path::PathBuf;
|
|
|
|
use anyhow::{Context, Result};
|
|
use clap::Parser;
|
|
use fs_err as fs;
|
|
|
|
use distribution_types::IndexLocations;
|
|
use rustc_hash::FxHashMap;
|
|
use uv_build::{SourceBuild, SourceBuildContext};
|
|
use uv_cache::{Cache, CacheArgs};
|
|
use uv_client::RegistryClientBuilder;
|
|
use uv_configuration::{
|
|
BuildKind, BuildOptions, Concurrency, ConfigSettings, IndexStrategy, PreviewMode,
|
|
SetupPyStrategy,
|
|
};
|
|
use uv_dispatch::BuildDispatch;
|
|
use uv_git::GitResolver;
|
|
use uv_resolver::{FlatIndex, InMemoryIndex};
|
|
use uv_toolchain::{EnvironmentPreference, PythonEnvironment, ToolchainRequest};
|
|
use uv_types::{BuildContext, BuildIsolation, InFlight};
|
|
|
|
#[derive(Parser)]
|
|
pub(crate) struct BuildArgs {
|
|
/// Base python in a way that can be found with `which`
|
|
/// TODO(konstin): Also use proper python parsing here
|
|
#[clap(short, long)]
|
|
python: Option<PathBuf>,
|
|
/// Directory to story the built wheel in
|
|
#[clap(short, long)]
|
|
wheels: Option<PathBuf>,
|
|
/// The source distribution to build, as a directory.
|
|
sdist: PathBuf,
|
|
/// The subdirectory to build within the source distribution.
|
|
subdirectory: Option<PathBuf>,
|
|
/// You can edit the python sources of an editable install and the changes will be used without
|
|
/// the need to reinstall it.
|
|
#[clap(short, long)]
|
|
editable: bool,
|
|
#[command(flatten)]
|
|
cache_args: CacheArgs,
|
|
}
|
|
|
|
/// Build a source distribution to a wheel
|
|
pub(crate) async fn build(args: BuildArgs) -> Result<PathBuf> {
|
|
let wheel_dir = if let Some(wheel_dir) = args.wheels {
|
|
fs::create_dir_all(&wheel_dir).context("Invalid wheel directory")?;
|
|
wheel_dir
|
|
} else {
|
|
env::current_dir()?
|
|
};
|
|
let build_kind = if args.editable {
|
|
BuildKind::Editable
|
|
} else {
|
|
BuildKind::Wheel
|
|
};
|
|
|
|
let cache = Cache::try_from(args.cache_args)?.init()?;
|
|
|
|
let client = RegistryClientBuilder::new(cache.clone()).build();
|
|
let concurrency = Concurrency::default();
|
|
let config_settings = ConfigSettings::default();
|
|
let exclude_newer = None;
|
|
let flat_index = FlatIndex::default();
|
|
let git = GitResolver::default();
|
|
let in_flight = InFlight::default();
|
|
let index = InMemoryIndex::default();
|
|
let index_urls = IndexLocations::default();
|
|
let index_strategy = IndexStrategy::default();
|
|
let setup_py = SetupPyStrategy::default();
|
|
let toolchain = PythonEnvironment::find(
|
|
&ToolchainRequest::default(),
|
|
EnvironmentPreference::OnlyVirtual,
|
|
&cache,
|
|
)?;
|
|
let build_options = BuildOptions::default();
|
|
|
|
let build_dispatch = BuildDispatch::new(
|
|
&client,
|
|
&cache,
|
|
toolchain.interpreter(),
|
|
&index_urls,
|
|
&flat_index,
|
|
&index,
|
|
&git,
|
|
&in_flight,
|
|
index_strategy,
|
|
setup_py,
|
|
&config_settings,
|
|
BuildIsolation::Isolated,
|
|
install_wheel_rs::linker::LinkMode::default(),
|
|
&build_options,
|
|
exclude_newer,
|
|
concurrency,
|
|
PreviewMode::Enabled,
|
|
);
|
|
|
|
let builder = SourceBuild::setup(
|
|
&args.sdist,
|
|
args.subdirectory.as_deref(),
|
|
build_dispatch.interpreter(),
|
|
&build_dispatch,
|
|
SourceBuildContext::default(),
|
|
args.sdist.display().to_string(),
|
|
setup_py,
|
|
config_settings.clone(),
|
|
BuildIsolation::Isolated,
|
|
build_kind,
|
|
FxHashMap::default(),
|
|
concurrency.builds,
|
|
)
|
|
.await?;
|
|
Ok(wheel_dir.join(builder.build_wheel(&wheel_dir).await?))
|
|
}
|