Folding Expression for Fountain Files

fountain.vim

function! FountainFolds()
  let thisline = getline(v:lnum)
  let laterline = getline(v:lnum + 2)
  if match(thisline, '^EXT') >= 0
    return ">1"
  elseif match(thisline, '^INT') >= 0
    return ">1"
	elseif match(thisline, '^Title:') >= 0
    return ">1"
  elseif match(thisline, '^\.[^.]') >= 0
    return ">1"
  elseif match(laterline, '^EXT') >= 0
    return "<1"
  elseif match(laterline, '^INT') >= 0
    return "<1"
  elseif match(laterline, '^Title:') >= 0
    return "<1"
  elseif match(laterline, '^\.[^.]') >= 0
    return "<1"
  else
    return "="
  endif
endfunction

setlocal foldmethod=expr
setlocal foldexpr=FountainFolds()

function! FountainFoldText()
  let foldsize = (v:foldend-v:foldstart)
  return getline(v:foldstart).' ('.foldsize.' lines)'
endfunction
setlocal foldtext=FountainFoldText()