unkinben/dotfiles
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add puppet export template, extend puppet functions

Browse Source
This commit is contained in:
Ben Vincent 2023-11-10 23:31:43 +11:00
parent 455a4a9550
commit 5507303d92
2 changed files with 26 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
.config/nvim/lua/func/puppet.lua
View File

@ -126,27 +126,27 @@ endfunction
vim.cmd([[ vim.cmd([[
function! GeneratePuppetClassName() function! GeneratePuppetClassName()
" Get the full file path " get the full file path
let l:filepath = expand('%:p') let l:filepath = expand('%:p')
" Check if the file path ends with the desired patterns " check if the file path ends with the desired patterns
if l:filepath =~ 'site/roles/manifests/.\+\.pp$' || l:filepath =~ 'site/profiles/manifests/.\+\.pp$' if l:filepath =~ 'site/roles/manifests/.\+\.pp$' || l:filepath =~ 'site/profiles/manifests/.\+\.pp$'
" Extract the part of the path after 'manifests/' " extract the part of the path after 'manifests/'
let l:manifest_path = matchstr(l:filepath, 'manifests/.\+\.pp$') let l:manifest_path = matchstr(l:filepath, 'manifests/.\+\.pp$')
" Remove 'manifests/' and '.pp' to get the class path " remove 'manifests/' and '.pp' to get the class path
let l:class_path = substitute(l:manifest_path, 'manifests/\(.\+\)\.pp', '\1', '') let l:class_path = substitute(l:manifest_path, 'manifests/\(.\+\)\.pp', '\1', '')
" Replace slashes with double colons to form the class name " replace slashes with double colons to form the class name
let l:class_name = substitute(l:class_path, '/', '::', 'g') let l:class_name = substitute(l:class_path, '/', '::', 'g')
" Detect if it is a role or profile based on the path " detect if it is a role or profile based on the path
let l:type = l:filepath =~ 'site/roles/manifests/' ? 'roles' : 'profiles' let l:type = l:filepath =~ 'site/roles/manifests/' ? 'roles' : 'profiles'
" Return the generated class name with the type prefix " return the generated class name with the type prefix
return l:type . '::' . l:class_name return l:type . '::' . l:class_name
endif endif
" If the file path doesn't match the pattern, return an empty string " if the file path doesn't match the pattern, return an empty string
return '' return ''
endfunction endfunction
]]) ]])
@ -156,11 +156,19 @@ function! ApplyPuppetTemplate()
" Generate the class name based on the file path " Generate the class name based on the file path
let l:class_name = GeneratePuppetClassName() let l:class_name = GeneratePuppetClassName()
" If a class name was successfully generated, substitute the placeholder " If a class name was successfully generated
if l:class_name != '' if l:class_name != ''
" Read the template content " Determine the template based on class name pattern
execute '0r $HOME/.config/nvim/templates/puppet/newclass.pp' if l:class_name =~ '^profiles::exports::'
let l:template = '$HOME/.config/nvim/templates/puppet/export.pp'
else
let l:template = '$HOME/.config/nvim/templates/puppet/newclass.pp'
endif
" Read the appropriate template content
execute '0r ' . l:template
" Substitute the placeholder with the class name
execute "1,\$s/%CLASS_NAME%/" . l:class_name . "/g" execute "1,\$s/%CLASS_NAME%/" . l:class_name . "/g"
endif endif
endfunction endfunction

7
.config/nvim/templates/puppet/export.pp Normal file
View File

@ -0,0 +1,7 @@
# %CLASS_NAME%
define %CLASS_NAME% (
# add parameters to be exported from here
){
# exported resource, do not place anything in here
}