Merge pull request #60 from rahulsalvi/better-foldtext

Author: kristijanhusak

indent/org.vim

@@ -6,9 +6,13 @@ function! OrgmodeIndentExpr()
   return luaeval('require("orgmode.org.indent").indentexpr()')
 endfunction
 
+function! OrgmodeFoldText()
+  return luaeval('require("orgmode.org.indent").foldtext()')
+endfunction
+
 setlocal foldmethod=expr
 setlocal foldexpr=OrgmodeFoldExpr()
-setlocal foldtext=getline(v:foldstart)
+setlocal foldtext=OrgmodeFoldText()
 setlocal indentexpr=OrgmodeIndentExpr()
 setlocal foldlevel=0
 setlocal nolisp

lua/orgmode/org/indent.lua

@@ -63,7 +63,16 @@ local function indentexpr()
   return vim.fn.indent(prev_line)
 end
 
+local function foldtext()
+  local line = vim.fn.getline(vim.v.foldstart)
+  if config.org_hide_leading_stars then
+    return vim.fn.substitute(line, '\\(^\\**\\)', '\\=repeat(" ", len(submatch(0))-1) . "*"', '')
+  end
+  return line
+end
+
 return {
   foldexpr = foldexpr,
   indentexpr = indentexpr,
+  foldtext = foldtext,
 }