abi_stable_derive/sabi_trait/
replace_self_path.rs

Help
//! Contains the `replace_self_path` function,and the `ReplaceWith` enum.

use as_derive_utils::spanned_err;

use syn::visit_mut::VisitMut;
use syn::{Ident, TraitItemType, TypePath};

use std::mem;

use crate::utils::{LinearResult, SynResultExt};

/// What to do with the a path component when it's found.
#[derive(Debug, Clone)]
pub(crate) enum ReplaceWith {
    /// Replaces the identifier of the path component with another.
    Ident(Ident),
    /// Removes the path component.
    Remove,
    /// Keeps the path component.
    Keep,
}

// This is only pub(crate) because it appears as a bound of `replace_self_path`.
pub(crate) trait VisitMutWith {
    fn visit_mut_with<F>(&mut self, other: &mut SelfReplacer<F>)
    where
        F: FnMut(&Ident) -> Option<ReplaceWith>;
}

macro_rules! impl_visit_mut_with {
    (
        $( ($self_:ty,$method:path) ),*
        $(,)*
    ) => (
        $(
            impl VisitMutWith for $self_{
                #[inline]
                fn visit_mut_with<F>(&mut self,other:&mut SelfReplacer<F>)
                where
                    F: FnMut(&Ident) -> Option<ReplaceWith>,
                {
                    $method(other,self);
                }
            }
        )*
    )
}

impl_visit_mut_with! {
    (syn::WherePredicate,VisitMut::visit_where_predicate_mut),
    (TraitItemType,VisitMut::visit_trait_item_type_mut),
    (syn::Type,VisitMut::visit_type_mut),
}

/// Replaces all associated types of `Self` from `value`.
///
/// `replace_with` determines what happens to `Self::` when `Some()` is
/// returned from `is_assoc_type`.
///
/// `is_assoc_type` is used to find the associated types to replace
/// (when the function returns Some(_)),
/// as well as what to replace them with.
///
///
pub(crate) fn replace_self_path<V, F>(
    value: &mut V,
    replace_with: ReplaceWith,
    is_assoc_type: F,
) -> Result<(), syn::Error>
where
    V: VisitMutWith,
    F: FnMut(&Ident) -> Option<ReplaceWith>,
{
    let mut replacer = SelfReplacer {
        is_assoc_type,
        buffer: Vec::with_capacity(2),
        replace_with,
        errors: LinearResult::ok(()),
    };
    value.visit_mut_with(&mut replacer);
    replacer.errors.into()
}

// This is only pub(crate) because it is used within the VisitMutWith trait.
pub(crate) struct SelfReplacer<F> {
    is_assoc_type: F,
    buffer: Vec<ReplaceWith>,
    replace_with: ReplaceWith,
    errors: LinearResult<()>,
}

impl<F> VisitMut for SelfReplacer<F>
where
    F: FnMut(&Ident) -> Option<ReplaceWith>,
{
    fn visit_type_path_mut(&mut self, i: &mut TypePath) {
        if let Some(qself) = i.qself.as_mut() {
            self.visit_type_mut(&mut qself.ty);
        }

        let segments = &mut i.path.segments;

        for segment in &mut *segments {
            self.visit_path_arguments_mut(&mut segment.arguments);
        }

        // println!("\nbefore:{}",(&*segments).into_token_stream() );
        // println!("segments[1]:{}",segments.iter().nth(1).into_token_stream() );

        let is_self = segments[0].ident == "Self";

        match (segments.len(), is_self) {
            (0, true) | (1, true) => {
                self.errors.push_err(spanned_err!(
                    segments,
                    "Self can't be used in a parameter,return type,or associated type.",
                ));
                return;
            }
            (2, true) => {}
            (_, true) => {
                self.errors.push_err(spanned_err!(
                    segments,
                    "Paths with 3 or more components are currently unsupported",
                ));
                return;
            }
            (_, false) => return,
        }

        let is_replaced = (self.is_assoc_type)(&segments[1].ident);
        // println!("is_replaced:{:?}",is_replaced );
        if let Some(replace_assoc_with) = is_replaced {
            let mut prev_segments = mem::take(segments).into_iter();

            self.buffer.clear();
            self.buffer.push(self.replace_with.clone());
            self.buffer.push(replace_assoc_with);
            for replace_with in self.buffer.drain(..) {
                let prev_segment = prev_segments.next();
                match replace_with {
                    ReplaceWith::Ident(ident) => {
                        segments.push(ident.into());
                    }
                    ReplaceWith::Remove => {}
                    ReplaceWith::Keep => {
                        segments.extend(prev_segment);
                    }
                }
            }
        }
        // println!("after:{}",(&*i).into_token_stream() );
    }
}
abi_stable_derive/sabi_trait/replace_self_path.rs

abi_stable_derive/sabi_trait/
replace_self_path.rs
