refactor: split project into multiple files
This commit is contained in:
parent
f8ba7f3f42
commit
c8e61a9cf8
18
git_utils.go
Normal file
18
git_utils.go
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"os/exec"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func gitUserData(key string) (string, error) {
|
||||||
|
cmd := exec.Command("git", "config", "--get", key)
|
||||||
|
out, err := cmd.Output()
|
||||||
|
if err != nil {
|
||||||
|
return "", errors.New("Can't read Git config")
|
||||||
|
}
|
||||||
|
|
||||||
|
value := strings.TrimSpace(string(out))
|
||||||
|
return value, nil
|
||||||
|
}
|
30
license_context.go
Normal file
30
license_context.go
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type LicenseContext struct {
|
||||||
|
AuthorName string
|
||||||
|
AuthorEmail string
|
||||||
|
Year int
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewLicenseContext(authorName string, authorEmail string) (LicenseContext, error) {
|
||||||
|
var err error
|
||||||
|
if authorName == "" {
|
||||||
|
authorName, err = gitUserData("user.name")
|
||||||
|
}
|
||||||
|
if authorEmail == "" {
|
||||||
|
authorEmail, err = gitUserData("user.email")
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return LicenseContext{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return LicenseContext{
|
||||||
|
AuthorName: authorName,
|
||||||
|
AuthorEmail: authorEmail,
|
||||||
|
Year: time.Now().Year(),
|
||||||
|
}, nil
|
||||||
|
}
|
65
licenser.go
Normal file
65
licenser.go
Normal file
@ -0,0 +1,65 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Licenser struct {
|
||||||
|
LicenseID string
|
||||||
|
LicenseContext LicenseContext
|
||||||
|
OutputFile string
|
||||||
|
licenseBody string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Licenser) ParseTemplate() (LicenseTemplate, error) {
|
||||||
|
licenseID := strings.ToUpper(l.LicenseID)
|
||||||
|
tmplPath := "templates/" + licenseID + ".tmpl"
|
||||||
|
data, err := TemplatesDir.ReadFile(tmplPath)
|
||||||
|
if err != nil {
|
||||||
|
return LicenseTemplate{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
parts := strings.SplitN(string(data), "---", 3)
|
||||||
|
|
||||||
|
var licenseTmpl LicenseTemplate
|
||||||
|
yaml.Unmarshal([]byte(parts[1]), &licenseTmpl)
|
||||||
|
licenseTmpl.Body = strings.TrimSpace(parts[2])
|
||||||
|
|
||||||
|
return licenseTmpl, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Licenser) Generate() error {
|
||||||
|
license, err := l.ParseTemplate()
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("Not supported license")
|
||||||
|
}
|
||||||
|
|
||||||
|
tmpl, _ := template.New(l.LicenseID).Parse(license.Body)
|
||||||
|
|
||||||
|
var output bytes.Buffer
|
||||||
|
tmpl.Execute(&output, l.LicenseContext)
|
||||||
|
|
||||||
|
l.licenseBody = output.String()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Licenser) WriteFile() error {
|
||||||
|
outFile, err := os.Create(l.OutputFile)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer outFile.Close()
|
||||||
|
|
||||||
|
if _, err := outFile.WriteString(l.licenseBody); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
132
main.go
132
main.go
@ -1,143 +1,11 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"embed"
|
|
||||||
"errors"
|
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io/fs"
|
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
|
||||||
"strings"
|
|
||||||
"text/template"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"gopkg.in/yaml.v3"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
//go:embed all:templates
|
|
||||||
var TemplatesDir embed.FS
|
|
||||||
|
|
||||||
type LicenseTemplate struct {
|
|
||||||
Title string `yaml:"title"`
|
|
||||||
ID string `yaml:"spdx-id"`
|
|
||||||
Description string `yaml:"description"` // TODO
|
|
||||||
Permissions []string `yaml:"permissions"` // TODO
|
|
||||||
Limitations []string `yaml:"limitations"` // TODO
|
|
||||||
Conditions []string `yaml:"conditions"` // TODO
|
|
||||||
Body string
|
|
||||||
}
|
|
||||||
|
|
||||||
type LicenseContext struct {
|
|
||||||
AuthorName string
|
|
||||||
AuthorEmail string
|
|
||||||
Year int
|
|
||||||
}
|
|
||||||
|
|
||||||
type Licenser struct {
|
|
||||||
LicenseID string
|
|
||||||
LicenseContext LicenseContext
|
|
||||||
OutputFile string
|
|
||||||
licenseBody string
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewLicenseContext(authorName string, authorEmail string) (LicenseContext, error) {
|
|
||||||
var err error
|
|
||||||
if authorName == "" {
|
|
||||||
authorName, err = gitUserData("user.name")
|
|
||||||
}
|
|
||||||
if authorEmail == "" {
|
|
||||||
authorEmail, err = gitUserData("user.email")
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return LicenseContext{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return LicenseContext{
|
|
||||||
AuthorName: authorName,
|
|
||||||
AuthorEmail: authorEmail,
|
|
||||||
Year: time.Now().Year(),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *Licenser) ParseTemplate() (LicenseTemplate, error) {
|
|
||||||
licenseID := strings.ToUpper(l.LicenseID)
|
|
||||||
tmplPath := "templates/" + licenseID + ".tmpl"
|
|
||||||
data, err := TemplatesDir.ReadFile(tmplPath)
|
|
||||||
if err != nil {
|
|
||||||
return LicenseTemplate{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
parts := strings.SplitN(string(data), "---", 3)
|
|
||||||
|
|
||||||
var licenseTmpl LicenseTemplate
|
|
||||||
yaml.Unmarshal([]byte(parts[1]), &licenseTmpl)
|
|
||||||
licenseTmpl.Body = strings.TrimSpace(parts[2])
|
|
||||||
|
|
||||||
return licenseTmpl, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *Licenser) Generate() error {
|
|
||||||
license, err := l.ParseTemplate()
|
|
||||||
if err != nil {
|
|
||||||
return errors.New("Not supported license")
|
|
||||||
}
|
|
||||||
|
|
||||||
tmpl, _ := template.New(l.LicenseID).Parse(license.Body)
|
|
||||||
|
|
||||||
var output bytes.Buffer
|
|
||||||
tmpl.Execute(&output, l.LicenseContext)
|
|
||||||
|
|
||||||
l.licenseBody = output.String()
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *Licenser) WriteFile() error {
|
|
||||||
outFile, err := os.Create(l.OutputFile)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer outFile.Close()
|
|
||||||
|
|
||||||
if _, err := outFile.WriteString(l.licenseBody); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func gitUserData(key string) (string, error) {
|
|
||||||
cmd := exec.Command("git", "config", "--get", key)
|
|
||||||
out, err := cmd.Output()
|
|
||||||
if err != nil {
|
|
||||||
return "", errors.New("Can't read Git config")
|
|
||||||
}
|
|
||||||
|
|
||||||
value := strings.TrimSpace(string(out))
|
|
||||||
return value, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func templateList() []string {
|
|
||||||
files, err := fs.ReadDir(TemplatesDir, "templates")
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var tmplList []string
|
|
||||||
for _, v := range files {
|
|
||||||
tmplList = append(tmplList, strings.Replace(v.Name(), ".tmpl", "", 1))
|
|
||||||
}
|
|
||||||
|
|
||||||
return tmplList
|
|
||||||
}
|
|
||||||
|
|
||||||
func listLicenses() {
|
|
||||||
tmplList := templateList()
|
|
||||||
fmt.Println(strings.Join(tmplList, ", "))
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
OutputFile := flag.String("output", "LICENSE", "Specify different output file")
|
OutputFile := flag.String("output", "LICENSE", "Specify different output file")
|
||||||
LicenseID := flag.String("license", "", "Specify license by SPDX ID (e.g. BSD-3-Clause)")
|
LicenseID := flag.String("license", "", "Specify license by SPDX ID (e.g. BSD-3-Clause)")
|
||||||
|
39
templates.go
Normal file
39
templates.go
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"embed"
|
||||||
|
"io/fs"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
//go:embed all:templates
|
||||||
|
var TemplatesDir embed.FS
|
||||||
|
|
||||||
|
type LicenseTemplate struct {
|
||||||
|
Title string `yaml:"title"`
|
||||||
|
ID string `yaml:"spdx-id"`
|
||||||
|
Description string `yaml:"description"` // TODO
|
||||||
|
Permissions []string `yaml:"permissions"` // TODO
|
||||||
|
Limitations []string `yaml:"limitations"` // TODO
|
||||||
|
Conditions []string `yaml:"conditions"` // TODO
|
||||||
|
Body string
|
||||||
|
}
|
||||||
|
|
||||||
|
func listTemplates() []string {
|
||||||
|
files, err := fs.ReadDir(TemplatesDir, "templates")
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var tmplList []string
|
||||||
|
for _, v := range files {
|
||||||
|
tmplList = append(tmplList, strings.Replace(v.Name(), ".tmpl", "", 1))
|
||||||
|
}
|
||||||
|
|
||||||
|
return tmplList
|
||||||
|
}
|
||||||
|
|
||||||
|
func listLicenses() {
|
||||||
|
tmplList := listTemplates()
|
||||||
|
println(strings.Join(tmplList, ", "))
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user