WIP: setup patterns to use with gf command for autoloading another file

Author: lelutin

autoload/puppet/include.vim

@@ -0,0 +1,19 @@
+function! puppet#include#IncludeExpr(fname) abort
+    " Remove puppet 3.x style leading :: since they mean nothing for the path
+    let l:fname = substitute(a:fname, '^::', '', '')
+
+    " If we can't find a second element, then we're referring to the init.pp
+    " file
+    if match(l:fname, '::') < 0
+        return 'init'
+    endif
+
+    " Remove first element. It'll make finding files easier since the module
+    " name might not be in the find path.
+    let l:fname = substitute(l:fname,'^[^:]\+::','','')
+
+    " Replace all subsequent :: in order to get a real file path
+    let l:fname = substitute(l:fname,'::','/','g')
+
+    return l:fname
+endfunction

ftplugin/puppet.vim

@@ -21,8 +21,13 @@ setlocal commentstring=#\ %s
 setlocal formatoptions-=t formatoptions+=croql
 setlocal formatexpr=puppet#format#Format()
 
+setlocal suffixesadd=.pp
+setlocal include=\\v^\\s*(include\|contain\|class\\s+\\\{\|Class\\s+\\\[)\\s+[\"\']?\\zs(\\f\|::)+
+setlocal includeexpr=puppet#include#IncludeExpr(v:fname)
+
 let b:undo_ftplugin = "
     \ setlocal tabstop< tabstop< softtabstop< shiftwidth< expandtab<
     \| setlocal keywordprg< iskeyword< comments< commentstring<
     \| setlocal formatoptions< formatexpr<
+    \| setlocal suffixesadd< include< includeexpr<
     \"