wgsl-parser 0.5.0

use gramatika::{Parse, ParseStreamer, Span, Spanned, Substr, Token as _};

use crate::{
	token::{keyword, punct, Token, TokenKind},
	ParseStream,
};

#[derive(Clone, DebugLisp)]
pub struct ExtensionDecl {
	pub keyword: Token,
	pub name: Token,
}

#[derive(Clone, DebugLisp)]
pub struct ModuleDecl {
	pub import_kw: Token,
	pub name: Token,
	pub from_kw: Token,
	pub path: Token,
	pub semicolon: Token,
}

impl Parse for ExtensionDecl {
	type Stream = ParseStream;

	fn parse(input: &mut Self::Stream) -> gramatika::Result<Self> {
		let keyword = input.consume(keyword![enable])?;
		let name = input.consume_as(TokenKind::Ident, Token::Module)?;
		input.consume(punct![;])?;

		Ok(Self { keyword, name })
	}
}

impl Spanned for ExtensionDecl {
	fn span(&self) -> Span {
		self.keyword.span().through(self.name.span())
	}
}

impl ModuleDecl {
	pub fn path(&self) -> Substr {
		let path = self.path.lexeme();

		path.substr(1..path.len() - 1)
	}
}

impl Parse for ModuleDecl {
	type Stream = ParseStream;

	fn parse(input: &mut Self::Stream) -> gramatika::Result<Self> {
		let import_kw = input.consume(keyword![import])?;
		let name = input.consume_as(TokenKind::Ident, Token::Module)?;
		let from_kw = input.consume(keyword![from])?;
		let path = input.consume_kind(TokenKind::Path)?;
		let semicolon = input.consume(punct![;])?;

		Ok(Self {
			import_kw,
			name,
			from_kw,
			path,
			semicolon,
		})
	}
}

impl Spanned for ModuleDecl {
	fn span(&self) -> Span {
		self.import_kw.span().through(self.semicolon.span())
	}
}