copy go's function because it may diverge

AndrewRadev · Feb 22, 2024 · cf178b0 · cf178b0
1 parent 43e42ee
commit cf178b0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/autoload/sj/cue.vim b/autoload/sj/cue.vim
@@ -1,3 +1,15 @@
+function! sj#cue#SplitImports()
+  let pattern = '^import\s\+\(\%(\k\+\s\+\)\=\%(".*"\)\)$'
+
+  if getline('.') =~ pattern
+    call sj#Keeppatterns('s/' . pattern . '/import (\r\1\r)/')
+    normal! k==
+    return 1
+  else
+    return 0
+  endif
+endfunction
+
 function! sj#cue#SplitStructLiteral()
   let [from, to] = sj#LocateBracesOnLine('{', '}')
 

diff --git a/ftplugin/cue/splitjoin.vim b/ftplugin/cue/splitjoin.vim
@@ -3,7 +3,7 @@ if !exists('b:splitjoin_split_callbacks')
             \ 'sj#cue#SplitStructLiteral',
             \ 'sj#cue#SplitArray',
             \ 'sj#cue#SplitArgs',
-            \ 'sj#go#SplitImports',
+            \ 'sj#cue#SplitImports',
             \ ]
 endif