haskell - 如何使用 megaparsec 获取 AST 节点的源范围？

Question

我正在尝试为正在解析的某些源文件生成源映射，并且我想获取每个节点的范围。 getSourcePos只给出 start position一个节点(src:line:column)。如何获得它的结束位置？

Answer 1

如果你想为每个词位构造一个这样的源跨度:

data Span = Span SourcePos SourcePos

data Spanned a = Spanned Span a

您可以调用 getSourcePos两次，一次在标记的开头，一次在结尾，在消耗任何空格之前，假设您处于词法分析阶段。我过去曾使用过这样的结构来使其更方便:

-- Augment a parser with a source span.
spanned :: Parser (a, SourcePos) -> Parser (Spanned a)
spanned parser = do
  start <- getSourcePos
  (x, end) <- parser
  pure (Spanned (Span start end) x)

-- Consume whitespace following a lexeme, but record
-- its endpoint as being before the whitespace.
lexeme :: Parser a -> Parser (a, SourcePos)
lexeme parser = (,) <$> parser <*> (getSourcePos <* whitespace)

请记住 getSourcePos根据文档，这有点贵，如果我没记错的话，这取决于源文件的大小。

如果 AST 使用 span 进行注释，您可以通过将其与 Span 的 monoid 实例折叠来计算树的任何部分的 span。这需要他们的联合(或更具体地说是他们的边界框)，即 a <> b是 (beginRow a, beginCol a) `min` (beginRow b, beginCol b) 的跨度至 (endRow a, endCol a) `max` (endRow b, endCol b) .