92 lines
2.5 KiB
Rust
92 lines
2.5 KiB
Rust
|
//! A higher level attributes based on TokenTree, with also some shortcuts.
|
||
|
|
|
||
|
|
use std::sync::Arc;
|
||
|
|
|
||
|
|
use hir_expand::db::AstDatabase;
|
||
|
|
use mbe::ast_to_token_tree;
|
||
|
|
use ra_cfg::CfgOptions;
|
||
|
|
use ra_syntax::{
|
||
|
|
ast::{self, AstNode, AttrsOwner},
|
||
|
|
SmolStr,
|
||
|
|
};
|
||
|
|
use tt::Subtree;
|
||
|
|
|
||
|
|
use crate::{path::Path, HirFileId, Source};
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
||
|
|
pub struct Attr {
|
||
|
|
pub(crate) path: Path,
|
||
|
|
pub(crate) input: Option<AttrInput>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
||
|
|
pub enum AttrInput {
|
||
|
|
Literal(SmolStr),
|
||
|
|
TokenTree(Subtree),
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Attr {
|
||
|
|
pub(crate) fn from_src(
|
||
|
|
Source { file_id, ast }: Source<ast::Attr>,
|
||
|
|
db: &impl AstDatabase,
|
||
|
|
) -> Option<Attr> {
|
||
|
|
let path = Path::from_src(Source { file_id, ast: ast.path()? }, db)?;
|
||
|
|
let input = match ast.input() {
|
||
|
|
None => None,
|
||
|
|
Some(ast::AttrInput::Literal(lit)) => {
|
||
|
|
// FIXME: escape? raw string?
|
||
|
|
let value = lit.syntax().first_token()?.text().trim_matches('"').into();
|
||
|
|
Some(AttrInput::Literal(value))
|
||
|
|
}
|
||
|
|
Some(ast::AttrInput::TokenTree(tt)) => {
|
||
|
|
Some(AttrInput::TokenTree(ast_to_token_tree(&tt)?.0))
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
Some(Attr { path, input })
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn from_attrs_owner(
|
||
|
|
file_id: HirFileId,
|
||
|
|
owner: &dyn AttrsOwner,
|
||
|
|
db: &impl AstDatabase,
|
||
|
|
) -> Option<Arc<[Attr]>> {
|
||
|
|
let mut attrs = owner.attrs().peekable();
|
||
|
|
if attrs.peek().is_none() {
|
||
|
|
// Avoid heap allocation
|
||
|
|
return None;
|
||
|
|
}
|
||
|
|
Some(attrs.flat_map(|ast| Attr::from_src(Source { file_id, ast }, db)).collect())
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn is_simple_atom(&self, name: &str) -> bool {
|
||
|
|
// FIXME: Avoid cloning
|
||
|
|
self.path.as_ident().map_or(false, |s| s.to_string() == name)
|
||
|
|
}
|
||
|
|
|
||
|
|
// FIXME: handle cfg_attr :-)
|
||
|
|
pub fn as_cfg(&self) -> Option<&Subtree> {
|
||
|
|
if !self.is_simple_atom("cfg") {
|
||
|
|
return None;
|
||
|
|
}
|
||
|
|
match &self.input {
|
||
|
|
Some(AttrInput::TokenTree(subtree)) => Some(subtree),
|
||
|
|
_ => None,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn as_path(&self) -> Option<&SmolStr> {
|
||
|
|
if !self.is_simple_atom("path") {
|
||
|
|
return None;
|
||
|
|
}
|
||
|
|
match &self.input {
|
||
|
|
Some(AttrInput::Literal(it)) => Some(it),
|
||
|
|
_ => None,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn is_cfg_enabled(&self, cfg_options: &CfgOptions) -> Option<bool> {
|
||
|
|
cfg_options.is_cfg_enabled(self.as_cfg()?)
|
||
|
|
}
|
||
|
|
}
|