sections.mq

# Extracts sections from markdown
# Returns an array where each element is a dict with:
# - title: the heading text
# - level: the heading level (1-6)
# - content: array of markdown nodes in that section (including the heading)

import "section"

|

def extract_sections(md_nodes):
  let section_list = section::sections(md_nodes)
  | map(section_list, fn(s):
    let heading = first(s[:header])
    | let title = to_text(heading)
    | let level = heading.level
    | {"title": title, "level": level, "content": s[:children]}
  end)
end

# Extracts code blocks from a section
def extract_code_blocks(section_content):
  let code_blocks = filter(section_content, is_code)
  | map(code_blocks, fn(code):
    let lang = attr(code, "lang")
    | let text = to_text(code)
    | {"lang": lang, "code": text}
  end)
end

# Extracts text from a section
def extract_description(section_content):
  filter(section_content, is_text)
  | first()
  | if (is_none()): None else: to_text()
end

# Main query to extract sections with their code blocks
def sections_with_code(md_nodes):
  let section_list = extract_sections(md_nodes)
  | map(section_list, fn(section):
    let title = section["title"]
    | let level = section["level"]
    | let content = section["content"]
    | let codes = extract_code_blocks(content)
    | let description = extract_description(content)
    | {"title": title, "level": level, "codes": codes, "description": description}
  end)
end