pub mod arc_cow;pub mod fs;pub mod paths;pub mod serde;#[cfg(any(test, feature = "test-support"))]pub mod test; use futures::Future;use rand::{seq::SliceRandom, Rng};use regex::Regex;use std::sync::OnceLock;use std::{    borrow::Cow,    cmp::{self, Ordering},    env,    ops::{AddAssign, Range, RangeInclusive},    panic::Location,    pin::Pin,    task::{Context, Poll},    time::Instant,};use unicase::UniCase; pub use take_until::*; #[macro_export]macro_rules! debug_panic {    ( $($fmt_arg:tt)* ) => {        if cfg!(debug_assertions) {            panic!( $($fmt_arg)* );        } else {            let backtrace = std::backtrace::Backtrace::capture();            log::error!("{}\n{:?}", format_args!($($fmt_arg)*), backtrace);        }    };} #[macro_export]macro_rules! with_clone {    ($i:ident, move ||$l:expr) => {{        let $i = $i.clone();        move || {            $l        }    }};    ($i:ident, move |$($k:pat_param),*|$l:expr) => {{        let $i = $i.clone();        move |$( $k ),*| {            $l        }    }};     (($($i:ident),+), move ||$l:expr) => {{        let ($($i),+) = ($($i.clone()),+);        move || {            $l        }    }};    (($($i:ident),+), move |$($k:pat_param),*|$l:expr) => {{        let ($($i),+) = ($($i.clone()),+);        move |$( $k ),*| {            $l        }    }};} mod test_with_clone {     // If this test compiles, it works    #[test]    fn test() {        let x = "String".to_string();        let y = std::sync::Arc::new(5);         fn no_arg(f: impl FnOnce()) {            f()        }         no_arg(with_clone!(x, move || {            drop(x);        }));         no_arg(with_clone!((x, y), move || {            drop(x);            drop(y);        }));         fn one_arg(f: impl FnOnce(usize)) {            f(1)        }         one_arg(with_clone!(x, move |_| {            drop(x);        }));        one_arg(with_clone!((x, y), move |b| {            drop(x);            drop(y);            println!("{}", b);        }));         fn two_arg(f: impl FnOnce(usize, bool)) {            f(5, true)        }         two_arg(with_clone!((x, y), move |a, b| {            drop(x);            drop(y);            println!("{}{}", a, b)        }));        two_arg(with_clone!((x, y), move |a, _| {            drop(x);            drop(y);            println!("{}", a)        }));        two_arg(with_clone!((x, y), move |_, b| {            drop(x);            drop(y);            println!("{}", b)        }));         struct Example {            z: usize,        }         fn destructuring_example(f: impl FnOnce(Example)) {            f(Example { z: 10 })        }         destructuring_example(with_clone!(x, move |Example { z }| {            drop(x);            println!("{}", z);        }));         let a_long_variable_1 = "".to_string();        let a_long_variable_2 = "".to_string();        let a_long_variable_3 = "".to_string();        let a_long_variable_4 = "".to_string();        two_arg(with_clone!(            (                x,                y,                a_long_variable_1,                a_long_variable_2,                a_long_variable_3,                a_long_variable_4            ),            move |a, b| {                drop(x);                drop(y);                drop(a_long_variable_1);                drop(a_long_variable_2);                drop(a_long_variable_3);                drop(a_long_variable_4);                println!("{}{}", a, b)            }        ));         fn single_expression_body(f: impl FnOnce(usize) -> usize) -> usize {            f(20)        }         let _result = single_expression_body(with_clone!(y, move |z| *y + z));         // Explicitly move all variables        drop(x);        drop(y);        drop(a_long_variable_1);        drop(a_long_variable_2);        drop(a_long_variable_3);        drop(a_long_variable_4);    }} pub fn truncate(s: &str, max_chars: usize) -> &str {    match s.char_indices().nth(max_chars) {        None => s,        Some((idx, _)) => &s[..idx],    }} /// Removes characters from the end of the string if its length is greater than `max_chars` and/// appends "..." to the string. Returns string unchanged if its length is smaller than max_chars.pub fn truncate_and_trailoff(s: &str, max_chars: usize) -> String {    debug_assert!(max_chars >= 5);     let truncation_ix = s.char_indices().map(|(i, _)| i).nth(max_chars);    match truncation_ix {        Some(length) => s[..length].to_string() + "…",        None => s.to_string(),    }} /// Removes characters from the front of the string if its length is greater than `max_chars` and/// prepends the string with "...". Returns string unchanged if its length is smaller than max_chars.pub fn truncate_and_remove_front(s: &str, max_chars: usize) -> String {    debug_assert!(max_chars >= 5);     let truncation_ix = s.char_indices().map(|(i, _)| i).nth_back(max_chars);    match truncation_ix {        Some(length) => "…".to_string() + &s[length..],        None => s.to_string(),    }} /// Takes only `max_lines` from the string and, if there were more than `max_lines-1`, appends a/// a newline and "..." to the string, so that `max_lines` are returned./// Returns string unchanged if its length is smaller than max_lines.pub fn truncate_lines_and_trailoff(s: &str, max_lines: usize) -> String {    let mut lines = s.lines().take(max_lines).collect::<Vec<_>>();    if lines.len() > max_lines - 1 {        lines.pop();        lines.join("\n") + "\n…"    } else {        lines.join("\n")    }} pub fn post_inc<T: From<u8> + AddAssign<T> + Copy>(value: &mut T) -> T {    let prev = *value;    *value += T::from(1);    prev} /// Extend a sorted vector with a sorted sequence of items, maintaining the vector's sort order and/// enforcing a maximum length. This also de-duplicates items. Sort the items according to the given callback. Before calling this,/// both `vec` and `new_items` should already be sorted according to the `cmp` comparator.pub fn extend_sorted<T, I, F>(vec: &mut Vec<T>, new_items: I, limit: usize, mut cmp: F)where    I: IntoIterator<Item = T>,    F: FnMut(&T, &T) -> Ordering,{    let mut start_index = 0;    for new_item in new_items {        if let Err(i) = vec[start_index..].binary_search_by(|m| cmp(m, &new_item)) {            let index = start_index + i;            if vec.len() < limit {                vec.insert(index, new_item);            } else if index < vec.len() {                vec.pop();                vec.insert(index, new_item);            }            start_index = index;        }    }} /// Parse the result of calling `usr/bin/env` with no argumentspub fn parse_env_output(env: &str, mut f: impl FnMut(String, String)) {    let mut current_key: Option<String> = None;    let mut current_value: Option<String> = None;     for line in env.split_terminator('\n') {        if let Some(separator_index) = line.find('=') {            if &line[..separator_index] != "" {                if let Some((key, value)) = Option::zip(current_key.take(), current_value.take()) {                    f(key, value)                }                current_key = Some(line[..separator_index].to_string());                current_value = Some(line[separator_index + 1..].to_string());                continue;            };        }        if let Some(value) = current_value.as_mut() {            value.push('\n');            value.push_str(line);        }    }    if let Some((key, value)) = Option::zip(current_key.take(), current_value.take()) {        f(key, value)    }} pub fn merge_json_value_into(source: serde_json::Value, target: &mut serde_json::Value) {    use serde_json::Value;     match (source, target) {        (Value::Object(source), Value::Object(target)) => {            for (key, value) in source {                if let Some(target) = target.get_mut(&key) {                    merge_json_value_into(value, target);                } else {                    target.insert(key.clone(), value);                }            }        }         (source, target) => *target = source,    }} pub fn merge_non_null_json_value_into(source: serde_json::Value, target: &mut serde_json::Value) {    use serde_json::Value;    if let Value::Object(source_object) = source {        let target_object = if let Value::Object(target) = target {            target        } else {            *target = Value::Object(Default::default());            target.as_object_mut().unwrap()        };        for (key, value) in source_object {            if let Some(target) = target_object.get_mut(&key) {                merge_non_null_json_value_into(value, target);            } else if !value.is_null() {                target_object.insert(key.clone(), value);            }        }    } else if !source.is_null() {        *target = source    }} pub fn measure<R>(label: &str, f: impl FnOnce() -> R) -> R {    static ZED_MEASUREMENTS: OnceLock<bool> = OnceLock::new();    let zed_measurements = ZED_MEASUREMENTS.get_or_init(|| {        env::var("ZED_MEASUREMENTS")            .map(|measurements| measurements == "1" || measurements == "true")            .unwrap_or(false)    });     if *zed_measurements {        let start = Instant::now();        let result = f();        let elapsed = start.elapsed();        eprintln!("{}: {:?}", label, elapsed);        result    } else {        f()    }} pub trait ResultExt<E> {    type Ok;     fn log_err(self) -> Option<Self::Ok>;    /// Assert that this result should never be an error in development or tests.    fn debug_assert_ok(self, reason: &str) -> Self;    fn warn_on_err(self) -> Option<Self::Ok>;    fn inspect_error(self, func: impl FnOnce(&E)) -> Self;} impl<T, E> ResultExt<E> for Result<T, E>where    E: std::fmt::Debug,{    type Ok = T;     #[track_caller]    fn log_err(self) -> Option<T> {        match self {            Ok(value) => Some(value),            Err(error) => {                let caller = Location::caller();                log::error!("{}:{}: {:?}", caller.file(), caller.line(), error);                None            }        }    }     #[track_caller]    fn debug_assert_ok(self, reason: &str) -> Self {        if let Err(error) = &self {            debug_panic!("{reason} - {error:?}");        }        self    }     fn warn_on_err(self) -> Option<T> {        match self {            Ok(value) => Some(value),            Err(error) => {                log::warn!("{:?}", error);                None            }        }    }     /// https://doc.rust-lang.org/std/result/enum.Result.html#method.inspect_err    fn inspect_error(self, func: impl FnOnce(&E)) -> Self {        if let Err(err) = &self {            func(err);        }         self    }} pub trait TryFutureExt {    fn log_err(self) -> LogErrorFuture<Self>    where        Self: Sized;     fn log_tracked_err(self, location: core::panic::Location<'static>) -> LogErrorFuture<Self>    where        Self: Sized;     fn warn_on_err(self) -> LogErrorFuture<Self>    where        Self: Sized;    fn unwrap(self) -> UnwrapFuture<Self>    where        Self: Sized;} impl<F, T, E> TryFutureExt for Fwhere    F: Future<Output = Result<T, E>>,    E: std::fmt::Debug,{    #[track_caller]    fn log_err(self) -> LogErrorFuture<Self>    where        Self: Sized,    {        let location = Location::caller();        LogErrorFuture(self, log::Level::Error, *location)    }     fn log_tracked_err(self, location: core::panic::Location<'static>) -> LogErrorFuture<Self>    where        Self: Sized,    {        LogErrorFuture(self, log::Level::Error, location)    }     #[track_caller]    fn warn_on_err(self) -> LogErrorFuture<Self>    where        Self: Sized,    {        let location = Location::caller();        LogErrorFuture(self, log::Level::Warn, *location)    }     fn unwrap(self) -> UnwrapFuture<Self>    where        Self: Sized,    {        UnwrapFuture(self)    }} #[must_use]pub struct LogErrorFuture<F>(F, log::Level, core::panic::Location<'static>); impl<F, T, E> Future for LogErrorFuture<F>where    F: Future<Output = Result<T, E>>,    E: std::fmt::Debug,{    type Output = Option<T>;     fn poll(self: Pin<&mut Self>, cx: &mut Context) -> Poll<Self::Output> {        let level = self.1;        let location = self.2;        let inner = unsafe { Pin::new_unchecked(&mut self.get_unchecked_mut().0) };        match inner.poll(cx) {            Poll::Ready(output) => Poll::Ready(match output {                Ok(output) => Some(output),                Err(error) => {                    log::log!(                        level,                        "{}:{}: {:?}",                        location.file(),                        location.line(),                        error                    );                    None                }            }),            Poll::Pending => Poll::Pending,        }    }} pub struct UnwrapFuture<F>(F); impl<F, T, E> Future for UnwrapFuture<F>where    F: Future<Output = Result<T, E>>,    E: std::fmt::Debug,{    type Output = T;     fn poll(self: Pin<&mut Self>, cx: &mut Context) -> Poll<Self::Output> {        let inner = unsafe { Pin::new_unchecked(&mut self.get_unchecked_mut().0) };        match inner.poll(cx) {            Poll::Ready(result) => Poll::Ready(result.unwrap()),            Poll::Pending => Poll::Pending,        }    }} pub struct Deferred<F: FnOnce()>(Option<F>); impl<F: FnOnce()> Deferred<F> {    /// Drop without running the deferred function.    pub fn abort(mut self) {        self.0.take();    }}

Lines 1–500 of 839. Load more