feat: sort import (wip)

haunt98 · Nov 25, 2022 · c921e92 · c921e92
1 parent 1d3685f
commit c921e92
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/imports/formatter.go b/internal/imports/formatter.go
@@ -9,6 +9,7 @@ import (
 	"log"
 	"os"
 	"path/filepath"
+	"sort"
 	"strings"
 
 	"golang.org/x/tools/go/packages"
@@ -204,6 +205,14 @@ func (ft *Formatter) groupImports(importsAST map[string]*ast.ImportSpec) (map[st
 		}
 	}
 
+	// TODO: not sure if this match gofumpt output, but at lease it is sorted
+	sort.Strings(result[stdImport])
+	sort.Strings(result[thirdPartyImport])
+	if ft.companyPrefix != "" {
+		sort.Strings(result[companyImport])
+	}
+	sort.Strings(result[localImport])
+
 	return result, nil
 }