andrey
f9a0411192
All checks were successful
Go Action / goaction (push) Successful in 29s
Reviewed-on: #12
107 lines
1.8 KiB
Go
107 lines
1.8 KiB
Go
package pkg
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func NewImports(pkg string) *Imports {
|
|
imp := Imports{
|
|
data: make(map[string]string),
|
|
pkg: pkg,
|
|
}
|
|
|
|
return &imp
|
|
}
|
|
|
|
type Imports struct {
|
|
data map[string]string
|
|
pkg string
|
|
}
|
|
|
|
func (i *Imports) Imports() []Import {
|
|
imports := make([]Import, 0, len(i.data))
|
|
for name, alias := range i.data {
|
|
imports = append(imports, Import{
|
|
Package: name,
|
|
Alias: alias,
|
|
})
|
|
}
|
|
|
|
return imports
|
|
}
|
|
|
|
func (i *Imports) Short(fullType string) (string, error) {
|
|
idx := strings.LastIndexByte(fullType, '.')
|
|
if idx == -1 {
|
|
return "", fmt.Errorf("%w: expect package.Type", ErrWrongFormat)
|
|
}
|
|
|
|
if alias, ok := i.data[fullType[:idx]]; ok {
|
|
return alias + fullType[idx:], nil
|
|
}
|
|
|
|
return "", fmt.Errorf("%w alias for pkg %v", ErrNotFound, fullType[:idx])
|
|
}
|
|
|
|
func (i *Imports) AddType(fullType string) (string, error) {
|
|
idx := strings.LastIndexByte(fullType, '.')
|
|
if idx == -1 {
|
|
return "", fmt.Errorf("%w: expect pckage.Type got %v", ErrWrongFormat, fullType)
|
|
}
|
|
|
|
imp := i.Add(fullType[:idx])
|
|
|
|
if imp.Alias == "" {
|
|
return fullType[idx+1:], nil
|
|
}
|
|
|
|
return imp.Alias + fullType[idx:], nil
|
|
}
|
|
|
|
func (i *Imports) Adds(pkgs ...string) *Imports {
|
|
for _, pkg := range pkgs {
|
|
i.Add(pkg)
|
|
}
|
|
|
|
return i
|
|
}
|
|
|
|
func (i *Imports) Add(pkg string) Import {
|
|
if pkg == i.pkg {
|
|
return Import{
|
|
Alias: "",
|
|
Package: pkg,
|
|
}
|
|
}
|
|
|
|
alias := pkg
|
|
|
|
if idx := strings.LastIndexByte(pkg, '/'); idx != -1 {
|
|
alias = AliasName(pkg[idx+1:])
|
|
}
|
|
|
|
if al, ok := i.data[pkg]; ok {
|
|
return Import{Package: pkg, Alias: al}
|
|
}
|
|
|
|
for _, al := range i.data {
|
|
if al == alias {
|
|
alias += strconv.Itoa(len(i.data))
|
|
}
|
|
}
|
|
|
|
i.data[pkg] = alias
|
|
|
|
return Import{
|
|
Alias: alias,
|
|
Package: pkg,
|
|
}
|
|
}
|
|
|
|
type Import struct {
|
|
Alias string
|
|
Package string
|
|
}
|