kosame_dsl 0.3.0

Macro-based Rust ORM focused on developer ergonomics
Documentation
use crate::inferred_type::InferredType;
use crate::pretty::{BreakMode, Delim, PrettyPrint, Printer};
use crate::scopes::ScopeId;

use super::Expr;
use super::Visit;
use proc_macro2::Span;
use quote::{ToTokens, quote};
use syn::Ident;
use syn::spanned::Spanned;
use syn::{
    parenthesized,
    parse::{Parse, ParseStream},
};

pub struct Paren {
    pub paren: syn::token::Paren,
    pub expr: Box<Expr>,
}

impl Paren {
    #[must_use]
    pub fn span(&self) -> Span {
        self.paren.span.span()
    }

    #[must_use]
    pub fn infer_name(&self) -> Option<&Ident> {
        self.expr.infer_name()
    }

    #[must_use]
    pub fn infer_type(&self, scope_id: ScopeId) -> Option<InferredType<'_>> {
        self.expr.infer_type(scope_id)
    }
}

pub fn visit_paren<'a>(visit: &mut (impl Visit<'a> + ?Sized), paren: &'a Paren) {
    visit.visit_expr(&paren.expr);
}

impl Parse for Paren {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        let content;
        Ok(Self {
            paren: parenthesized!(content in input),
            expr: content.parse()?,
        })
    }
}

impl ToTokens for Paren {
    fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {
        let expr = &self.expr;
        quote! {
            ::kosame::repr::expr::Paren::new(&#expr)
        }
        .to_tokens(tokens);
    }
}

impl PrettyPrint for Paren {
    fn pretty_print(&self, printer: &mut Printer) {
        self.paren
            .pretty_print(printer, Some(BreakMode::Inconsistent), |printer| {
                self.expr.pretty_print(printer);
            });
    }
}