Charlie Marsh
2f8f126f2f
## Summary This PR uses a single `Index` that's shared between the top-level resolver and any sub-resolutions happen in the course of that top-level resolution (namely, to resolve build dependencies for any source distributions). In theory it's an optimization, since (e.g.) if we have two packages that both need the `flit-core` build system, and we attempt to build them both at once, we'll only fetch its metadata _once_, and share it across the two resolutions. In practice, I haven't been able to get this to show up in benchmarks. I suspect you'd need a _lot_ of source distributions for it to matter... Though it may still be worth doing, it strikes me as a cleaner design. Closes #200. Closes #541.
90 lines
2.7 KiB
Rust
90 lines
2.7 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 platform_host::Platform;
|
|
use puffin_build::{SourceBuild, SourceBuildContext};
|
|
use puffin_cache::{Cache, CacheArgs};
|
|
use puffin_client::{FlatIndex, RegistryClientBuilder};
|
|
use puffin_dispatch::BuildDispatch;
|
|
use puffin_interpreter::Virtualenv;
|
|
use puffin_resolver::InMemoryIndex;
|
|
use puffin_traits::{BuildContext, BuildKind, InFlight, SetupPyStrategy};
|
|
|
|
#[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, either a directory or a source archive.
|
|
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)?;
|
|
|
|
let platform = Platform::current()?;
|
|
let venv = Virtualenv::from_env(platform, &cache)?;
|
|
let client = RegistryClientBuilder::new(cache.clone()).build();
|
|
let index_urls = IndexLocations::default();
|
|
let flat_index = FlatIndex::default();
|
|
let index = InMemoryIndex::default();
|
|
let setup_py = SetupPyStrategy::default();
|
|
let in_flight = InFlight::default();
|
|
|
|
let build_dispatch = BuildDispatch::new(
|
|
&client,
|
|
&cache,
|
|
venv.interpreter(),
|
|
&index_urls,
|
|
&flat_index,
|
|
&index,
|
|
&in_flight,
|
|
venv.python_executable(),
|
|
setup_py,
|
|
false,
|
|
);
|
|
|
|
let builder = SourceBuild::setup(
|
|
&args.sdist,
|
|
args.subdirectory.as_deref(),
|
|
build_dispatch.interpreter(),
|
|
&build_dispatch,
|
|
SourceBuildContext::default(),
|
|
args.sdist.display().to_string(),
|
|
setup_py,
|
|
build_kind,
|
|
)
|
|
.await?;
|
|
Ok(wheel_dir.join(builder.build(&wheel_dir).await?))
|
|
}
|