yoshie
/
gofimports
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: split local, company, third party imports

main
sudo pacman -Syu 2022-11-26 16:26:41 +07:00
parent a030fec81f
commit 8edb24314f
No known key found for this signature in database
GPG Key ID: D6CB5C6C567C47B0
1 changed files with 25 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
internal/imports/formatter.go
View File

@ -150,7 +150,7 @@ func (ft *Formatter) formatFile(path string) error {
}
ft.log("moduleName: %+v\n", moduleName)
groupImports, err := ft.groupImports(importsAST)
groupImports, err := ft.groupImports(importsAST, moduleName)
if err != nil {
return err
}
@ -222,7 +222,10 @@ func (ft *Formatter) parseImports(pathASTFile *ast.File) (map[string]*ast.Import
// Copy from goimports-reviser
// Group imports to std, third-party, company if exist, local
func (ft *Formatter) groupImports(importsAST map[string]*ast.ImportSpec) (map[string][]string, error) {
func (ft *Formatter) groupImports(
importsAST map[string]*ast.ImportSpec,
moduleName string,
) (map[string][]string, error) {
result := make(map[string][]string)
result[stdImport] = make([]string, 0, 8)
result[thirdPartyImport] = make([]string, 0, 8)
@ -239,8 +242,28 @@ func (ft *Formatter) groupImports(importsAST map[string]*ast.ImportSpec) (map[st
result[stdImport] = append(result[stdImport], importNameAndPath)
continue
}
if strings.HasPrefix(importPath, moduleName) {
result[localImport] = append(result[localImport], importNameAndPath)
continue
}
if ft.companyPrefix != "" &&
strings.HasPrefix(importPath, ft.companyPrefix) {
result[companyImport] = append(result[companyImport], importNameAndPath)
continue
}
result[thirdPartyImport] = append(result[thirdPartyImport], importNameAndPath)
}
ft.log("std %+v\n", result[stdImport])
ft.log("third-party %+v\n", result[thirdPartyImport])
if ft.companyPrefix != "" {
ft.log("company %+v\n", result[companyImport])
}
ft.log("local %+v\n", result[localImport])
// TODO: not sure if this match gofumpt output, but at lease it is sorted
sort.Strings(result[stdImport])
sort.Strings(result[thirdPartyImport])