41 lines
872 B
Go
41 lines
872 B
Go
package codegen
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"git.kmsign.com/royalcat/repogen/internal/code"
|
|
)
|
|
|
|
const baseTemplate = `// Code generated by {{.Program}}. DO NOT EDIT.
|
|
package {{.PackageName}}
|
|
|
|
import (
|
|
{{.GenImports}}
|
|
)
|
|
`
|
|
|
|
type baseTemplateData struct {
|
|
Program string
|
|
PackageName string
|
|
Imports [][]code.Import
|
|
}
|
|
|
|
func (data baseTemplateData) GenImports() string {
|
|
var sections []string
|
|
for _, importSection := range data.Imports {
|
|
var section []string
|
|
for _, imp := range importSection {
|
|
section = append(section, data.generateImportLine(imp))
|
|
}
|
|
sections = append(sections, strings.Join(section, "\n"))
|
|
}
|
|
return strings.Join(sections, "\n\n")
|
|
}
|
|
|
|
func (data baseTemplateData) generateImportLine(imp code.Import) string {
|
|
if imp.Name == "" {
|
|
return fmt.Sprintf("\t\"%s\"", imp.Path)
|
|
}
|
|
return fmt.Sprintf("\t%s \"%s\"", imp.Name, imp.Path)
|
|
}
|