Support multiple import blocks (#106)

* support multiple import blocks

Signed-off-by: Loong <[email protected]>

* bump up version

Signed-off-by: Loong <[email protected]>

Author: daixiang0

Makefile

@@ -12,7 +12,7 @@ build: clean
 	@go build -v -trimpath -o ${BIN_OUTPUT} .
 
 test: clean
-	@go test -v -cover ./...
+	@go test -v -count=1 -cover ./...
 
 generate:
 	@go generate ./...

main.go

@@ -6,7 +6,7 @@ import (
 	"github.com/daixiang0/gci/cmd/gci"
 )
 
-var version = "0.6.3"
+var version = "0.7.0"
 
 func main() {
 	e := gci.NewExecutor(version)

pkg/gci/gci.go

@@ -138,8 +138,22 @@ func LoadFormatGoFile(file io.FileObj, cfg config.Config) (src, dist []byte, err
 		return nil, nil, err
 	}
 
-	head := src[:headEnd]
+	var head []byte
+	if src[headEnd-1] == '\t' {
+		head = src[:headEnd]
+	} else {
+		// handle multiple import blocks
+		// cover `import ` to `import (`
+		head = make([]byte, headEnd)
+		copy(head, src[:headEnd])
+		head = append(head, []byte{40, 10, 9}...)
+	}
+
 	tail := src[tailStart:]
+	// for test
+	if len(tail) == 0 {
+		tail = []byte(")\n")
+	}
 
 	firstWithIndex := true
 

pkg/gci/internal/testdata/multiple-imports.cfg.yaml

@@ -0,0 +1,4 @@
+sections:
+  - Standard
+  - Default
+  - Prefix(github.com/daixiang0)

pkg/gci/internal/testdata/multiple-imports.in.go

@@ -0,0 +1,11 @@
+package main
+
+import "fmt"
+
+import "context"
+
+import (
+	"os"
+
+	"github.com/daixiang0/test"
+)

pkg/gci/internal/testdata/multiple-imports.out.go

@@ -0,0 +1,9 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/daixiang0/test"
+)

pkg/gci/internal/testdata/one-line-import-after-import.cfg.yaml

@@ -0,0 +1,4 @@
+sections:
+  - Standard
+  - Default
+  - Prefix(github.com/daixiang0)

pkg/gci/internal/testdata/one-line-import-after-import.in.go

@@ -0,0 +1,10 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/daixiang0/test"
+)
+
+import "context"

pkg/gci/internal/testdata/one-line-import-after-import.out.go

@@ -0,0 +1,9 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/daixiang0/test"
+)