import: fix adding import to empty imports

fixes #1534
fatih · Sep 2, 2018 · 9ab996c · 9ab996c
1 parent 0b21790
commit 9ab996c
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/autoload/go/import.vim b/autoload/go/import.vim
@@ -65,6 +65,9 @@ function! go#import#SwitchImport(enabled, localname, path, bang) abort
       let packageline = line
       let appendline = line
 
+    elseif linestr =~# '^import\s\+(\+)'
+      let appendline = line
+      let appendstr = qlocalpath
     elseif linestr =~# '^import\s\+('
       let appendstr = qlocalpath
       let indentstr = 1
@@ -161,8 +164,16 @@ function! go#import#SwitchImport(enabled, localname, path, bang) abort
         let linesdelta += 3
         let appendstr = qlocalpath
         let indentstr = 1
+      elseif getline(appendline) =~# '^import\s\+(\+)'
+        call setline(appendline, 'import (')
+        call append(appendline + 0, appendstr)
+        call append(appendline + 1, ')')
+        let linesdelta -= 1
+        let indentstr = 1
+      else
+        call append(appendline, appendstr)
       endif
-      call append(appendline, appendstr)
+
       execute appendline + 1
       if indentstr
         execute 'normal! >>'