use crate::parse::text_block::TextBlock;
use crate::parse::text_block::find_text_block;
use crate::syntax::SyntaxKind;
use crate::syntax::SyntaxNode;
#[derive(Debug)]
pub struct PlainDocstring<'a>(pub(crate) &'a SyntaxNode);
impl<'a> PlainDocstring<'a> {
pub fn cast(node: &'a SyntaxNode) -> Option<Self> {
(node.kind() == SyntaxKind::PLAIN_DOCSTRING).then_some(Self(node))
}
pub fn syntax(&self) -> &'a SyntaxNode {
self.0
}
pub fn summary(&self) -> Option<TextBlock<'a>> {
find_text_block(self.0, SyntaxKind::SUMMARY)
}
pub fn extended_summary(&self) -> Option<TextBlock<'a>> {
find_text_block(self.0, SyntaxKind::EXTENDED_SUMMARY)
}
}