use crate::{
inferred_type::InferredType,
pretty::{PrettyPrint, Printer},
scopes::ScopeId,
};
use super::Visit;
use proc_macro2::{Span, TokenStream};
use quote::{ToTokens, quote};
use syn::{
Ident, LitStr, Token,
parse::{Parse, ParseStream},
};
pub struct Raw {
pub dollar_token: Token![$],
pub string: LitStr,
}
impl Raw {
#[must_use]
pub fn infer_name(&self) -> Option<&Ident> {
None
}
#[must_use]
pub fn infer_type(&self, _scope_id: ScopeId) -> Option<InferredType<'_>> {
None
}
pub fn peek(input: ParseStream) -> bool {
input.peek(Token![$])
}
#[must_use]
pub fn span(&self) -> Span {
self.dollar_token
.span
.join(self.string.span())
.unwrap_or(self.string.span())
}
}
pub fn visit_raw<'a>(_visit: &mut (impl Visit<'a> + ?Sized), _raw: &'a Raw) {}
impl Parse for Raw {
fn parse(input: ParseStream) -> syn::Result<Self> {
Ok(Self {
dollar_token: input.parse()?,
string: input.parse()?,
})
}
}
impl ToTokens for Raw {
fn to_tokens(&self, tokens: &mut TokenStream) {
let string = &self.string;
quote! { ::kosame::repr::expr::Raw::new(#string) }.to_tokens(tokens);
}
}
impl PrettyPrint for Raw {
fn pretty_print(&self, printer: &mut Printer<'_>) {
self.dollar_token.pretty_print(printer);
self.string.pretty_print(printer);
}
}