register jsonnetfmt using buildifier as a template

autoload/codefmt/jsonnetfmt.vim

@@ -0,0 +1,47 @@
+let s:plugin = maktaba#plugin#Get('codefmt')
+
+
+""
+" @private
+"
+" Formatter provider for jsonnet files using jsonnetfmt
+function! codefmt#jsonnetfmt#GetFormatter() abort
+  let l:formatter = {
+      \ 'name': 'jsonnetfmt',
+      \ 'setup_instructions': 'Install jsonnet. ' .
+          \ '(https://jsonnet.org/).'}
+
+  function l:formatter.IsAvailable() abort
+    return executable(s:plugin.Flag('jsonnetfmt_executable'))
+  endfunction
+
+  let l:supported_filetypes = ['json','jsonnet']
+
+  function l:formatter.AppliesToBuffer() abort
+    return index(l:supported_filetypes, &filetype) >= 0
+  endfunction
+
+  ""
+  " Reformat the current buffer with jsonnetfmt or the binary named in
+  " @flag(jsonnetfmt_executable)
+  " @throws ShellError
+  function l:formatter.Format() abort
+    let l:cmd = [ s:plugin.Flag('jsonnetfmt_executable') ]
+    let l:fname = expand('%:p')
+    if !empty(l:fname)
+      let l:cmd += ['-path', l:fname]
+    endif
+
+    try
+      " NOTE: Ignores any line ranges given and formats entire buffer.
+      " jsonnetfmt does not support range formatting.
+      call codefmt#formatterhelpers#Format(l:cmd)
+    catch
+      " TODO: Parse all the errors and stick them in the quickfix list.
+      " currently just echoes the errors.
+      call maktaba#error#Shout('Error formatting file: %s', v:exception)
+    endtry
+  endfunction
+
+  return l:formatter
+endfunction

instant/flags.vim

@@ -131,6 +131,10 @@ call s:plugin.Flag('buildifier_lint_mode', '')
 " - "+some-warning": Add 'some-warning' to the warning set.
 call s:plugin.Flag('buildifier_warnings', '')
 
+""
+" The path to the jsonnetfmt executable.
+call s:plugin.Flag('jsonnetfmt_executable', 'jsonnetfmt')
+
 ""
 " The path to the google-java executable.  Generally, this should have the
 " form:

plugin/register.vim

@@ -69,6 +69,7 @@ call s:registry.AddExtension(codefmt#fish_indent#GetFormatter())
 call s:registry.AddExtension(codefmt#gn#GetFormatter())
 call s:registry.AddExtension(codefmt#gofmt#GetFormatter())
 call s:registry.AddExtension(codefmt#googlejava#GetFormatter())
+call s.registry.AddExtension(codefmt#jsonnetfmt#GetFormatter())
 call s:registry.AddExtension(codefmt#jsbeautify#GetFormatter())
 call s:registry.AddExtension(codefmt#prettier#GetFormatter())
 call s:registry.AddExtension(codefmt#ktfmt#GetFormatter())

vroom/jsonnetfmt.vroom

@@ -0,0 +1,33 @@
+
+  :source $VROOMDIR/setupvroom.vim
+
+  :let g:repeat_calls = []
+  :function FakeRepeat(...)<CR>
+  |  call add(g:repeat_calls, a:000)<CR>
+  :endfunction
+  :call maktaba#test#Override('repeat#set', 'FakeRepeat')
+
+  :call codefmt#SetWhetherToPerformIsAvailableChecksForTesting(0)
+
+
+The jsonnetfmt formatter expects the jsonnetfmt executable to be installed on
+your system.
+
+  % {foo:'bar'}
+  :FormatCode jsonnetfmt
+  ! jsonnetfmt .*
+  $ { foo: 'foo' }
+
+
+The name or path of the jsonnetfmt executable can be configured via the
+jsonnetfmt_executable flag if the default of "jsonnetfmt" doesn't work.
+
+  :Glaive codefmt jsonnetfmt_executable='myjsonnetfmt'
+  :FormatCode jsonnetfmt
+  ! myjsonnetfmt .*
+  $ foo_library(
+  $     name = "foo",
+  $     srcs = ["bar.js"],
+  $ )
+  :Glaive codefmt jsonnetfmt_executable='jsonnetfmt'
+