Extract code blocks

code-blocks.awk

# Usage:
#
# $ awk -f block.awk -v START="fn myfunc" foo.rs bar.rs | pygmentize -l rs

{
  if(acc == 0) {
    if($0 ~ START) {
      indent = match($0, /[^[:space:]]/)
      acc = $0
    }
  } else {
    acc = acc RS $0

    if(match(substr($0, 0, indent), /[^[:space:]]/)) {
      print acc
      acc = 0
    }
  }
}

END {
  if(acc != 0)
    print acc
}