mirror of https://github.com/astral-sh/ruff
52 lines
1.4 KiB
Rust
52 lines
1.4 KiB
Rust
use std::iter::FusedIterator;
|
|
|
|
pub use list::{all_modules, list_modules};
|
|
pub(crate) use module::KnownModule;
|
|
pub use module::Module;
|
|
pub use path::{SearchPath, SearchPathValidationError};
|
|
pub use resolver::SearchPaths;
|
|
pub(crate) use resolver::file_to_module;
|
|
pub use resolver::{resolve_module, resolve_real_module};
|
|
use ruff_db::system::SystemPath;
|
|
|
|
use crate::Db;
|
|
pub(crate) use resolver::{ModuleResolveMode, SearchPathIterator, search_paths};
|
|
|
|
mod list;
|
|
mod module;
|
|
mod path;
|
|
mod resolver;
|
|
mod typeshed;
|
|
|
|
#[cfg(test)]
|
|
mod testing;
|
|
|
|
/// Returns an iterator over all search paths pointing to a system path
|
|
pub fn system_module_search_paths(db: &dyn Db) -> SystemModuleSearchPathsIter<'_> {
|
|
SystemModuleSearchPathsIter {
|
|
// Always run in `StubsAllowed` mode because we want to include as much as possible
|
|
// and we don't care about the "real" stdlib
|
|
inner: search_paths(db, ModuleResolveMode::StubsAllowed),
|
|
}
|
|
}
|
|
|
|
pub struct SystemModuleSearchPathsIter<'db> {
|
|
inner: SearchPathIterator<'db>,
|
|
}
|
|
|
|
impl<'db> Iterator for SystemModuleSearchPathsIter<'db> {
|
|
type Item = &'db SystemPath;
|
|
|
|
fn next(&mut self) -> Option<Self::Item> {
|
|
loop {
|
|
let next = self.inner.next()?;
|
|
|
|
if let Some(system_path) = next.as_system_path() {
|
|
return Some(system_path);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FusedIterator for SystemModuleSearchPathsIter<'_> {}
|