pomsky-macro 0.12.0

//! This crate provides the [`pomsky!`] macro to compile [pomsky] expressions at
//! compile time.
//!
//! [pomsky]: https://pomsky-lang.org
#![cfg_attr(feature = "diagnostics", feature(proc_macro_span))]

extern crate proc_macro;

use std::iter::Peekable;

use proc_macro::{Delimiter, Group, Literal, Span, TokenStream, TokenTree};

use pomsky::{
    Expr,
    options::{CompileOptions, RegexFlavor},
};

mod diagnostic;

/// Macro to compile a [pomsky] expression at compile time.
///
/// ### Example
///
/// ```
/// use pomsky_macro::pomsky;
///
/// const REGEX: &str = pomsky! {
///     let number = '-'? [digit]+;
///     let op = ["+-*/"];
///     number (op number)*
/// };
/// ```
///
/// **NOTE**: Code points (e.g. `U+FFEF`) should be written without the `+`
/// (i.e., `UFFEF`), because rustfmt surrounds `+` with spaces by default, which
/// would break parsing.
///
/// The regex flavor defaults to `Rust`, so it can be used with the [regex]
/// crate:
///
/// ```no_run
/// # mod regex {
/// #     pub(super) struct Regex;
/// #     impl Regex {
/// #         pub(super) fn new(s: &str) -> Result<Self, ()> { Ok(Regex) }
/// #     }
/// # }
/// # use pomsky_macro::pomsky;
/// use regex::Regex;
///
/// fn get_regex() -> Regex {
///     Regex::new(pomsky! { "hello world" }).unwrap()
/// }
/// ```
///
/// If you want to use a different flavor, you can specify it in the first line,
/// like so:
///
/// ```rust
/// # use pomsky_macro::pomsky;
/// pomsky! {
///     #flavor = Pcre
///     // your pomsky expression goes here
/// }
/// # ;
/// ```
///
/// Available flavors are
///
/// - **DotNet** (C#, F#)
/// - **Java**
/// - **JavaScript** (ECMAScript, Dart)
/// - **Pcre** (Crystal, Delphi, Elixir, Erlang, Hack, Julia, PHP, R, Vala, ...)
/// - **Python** (`re` module)
/// - **Ruby**
/// - **Rust** (`regex` crate)
///
/// [pomsky]: https://pomsky-lang.org
/// [regex]: https://docs.rs/regex
#[proc_macro]
pub fn pomsky(items: TokenStream) -> TokenStream {
    let group = Group::new(Delimiter::None, items);
    let global_span = group.span();

    match pomsky_impl(group.stream().into_iter()) {
        Ok(lit) => TokenTree::Literal(lit).into(),
        Err(Error { msg, span }) => {
            let span = span.unwrap_or(global_span);
            diagnostic::error(&msg, span, span)
        }
    }
}

struct Error {
    msg: String,
    span: Option<Span>,
}

impl Error {
    fn new(msg: String, span: Span) -> Self {
        Error { msg, span: Some(span) }
    }

    fn from_msg(msg: String) -> Self {
        Error { msg, span: None }
    }
}

macro_rules! bail {
    ($l:literal) => {
        return Err(Error::from_msg(format!($l)))
    };
    ($l:literal, $e:expr) => {
        return Err(Error::new(format!($l), $e))
    };
    ($e1:expr) => {
        return Err(Error::from_msg($e1))
    };
    ($e1:expr, $e2:expr) => {
        return Err(Error::new($e1, $e2))
    };
}

fn expect(
    iter: &mut Peekable<impl Iterator<Item = TokenTree>>,
    pred: fn(&TokenTree) -> bool,
    error_msg: impl Into<String>,
) -> Result<(), Error> {
    match iter.peek() {
        Some(tt) if pred(tt) => {
            iter.next();
            Ok(())
        }
        Some(tt) => bail!(error_msg.into(), tt.span()),
        None => bail!(error_msg.into()),
    }
}

fn pomsky_impl(items: impl Iterator<Item = TokenTree>) -> Result<Literal, Error> {
    let mut iter = items.peekable();

    let found_hashtag =
        expect(&mut iter, |t| matches!(t, TokenTree::Punct(p) if p.as_char() == '#'), "");

    let flavor = if found_hashtag.is_ok() {
        expect(
            &mut iter,
            |t| matches!(t, TokenTree::Ident(id) if &id.to_string() == "flavor"),
            "expected `flavor`",
        )?;
        expect(
            &mut iter,
            |t| matches!(t, TokenTree::Punct(p) if p.as_char() == '='),
            "expected `=`",
        )?;

        get_flavor(iter.next())?
    } else {
        RegexFlavor::Rust
    };

    let group = Group::new(Delimiter::None, iter.collect());

    #[cfg(not(feature = "diagnostics"))]
    let (span, input) = (group.span(), group.to_string());

    #[cfg(feature = "diagnostics")]
    let (span, input) = {
        if let (Some(first), Some(last)) =
            (group.stream().into_iter().next(), group.stream().into_iter().last())
        {
            let span = first.span().join(last.span()).unwrap();
            (span, span.source_text().unwrap())
        } else {
            (group.span_close(), String::new())
        }
    };

    let input = input.trim_start_matches("/*«*/").trim_end_matches("/*»*/");

    match Expr::parse_and_compile(input, CompileOptions { flavor, ..Default::default() }) {
        (Some(compiled), _warnings, _tests) => Ok(Literal::string(&compiled)),

        (None, errors, _) => {
            let errors =
                errors.into_iter().map(|d| diagnostic::fmt(d, &group, input)).collect::<Vec<_>>();
            bail!(errors.join("\n\n"), span)
        }
    }
}

fn get_flavor(item: Option<TokenTree>) -> Result<RegexFlavor, Error> {
    Ok(match item {
        Some(TokenTree::Ident(id)) => match id.to_string().as_str() {
            "DotNet" => RegexFlavor::DotNet,
            "Java" => RegexFlavor::Java,
            "JavaScript" => RegexFlavor::JavaScript,
            "Pcre" => RegexFlavor::Pcre,
            "Python" => RegexFlavor::Python,
            "Ruby" => RegexFlavor::Ruby,
            "Rust" => RegexFlavor::Rust,
            s => bail!(
                "Expected one of: DotNet, Java, JavaScript, Pcre, Python, Ruby, Rust\nGot: {s}",
                id.span()
            ),
        },
        Some(tt) => bail!("Unexpected token `{tt}`", tt.span()),
        None => bail!("Expected identifier"),
    })
}