mirror of
https://github.com/spf13/cobra
synced 2025-05-05 12:57:22 +00:00
b45934648f
When creating new commands via `cobra add`, it is not possible to add commands with the same name to different parents. They will just overwrite each other. With this patch, the command's object identifier is templated `parentCommandCmd`, the `Use` field `command` and the file name `parent_command.go`. Fixes #1059 Fixes #1131 Signed-off-by: Carlos Marx <me@carlosmarx.com>
100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"text/template"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/cobra/cobra/tpl"
|
|
)
|
|
|
|
// Project contains name, license and paths to projects.
|
|
type Project struct {
|
|
// v2
|
|
PkgName string
|
|
Copyright string
|
|
AbsolutePath string
|
|
Legal License
|
|
Viper bool
|
|
AppName string
|
|
}
|
|
|
|
type Command struct {
|
|
CmdName string
|
|
CmdUse string
|
|
CmdFileName string
|
|
CmdParent string
|
|
*Project
|
|
}
|
|
|
|
func (p *Project) Create() error {
|
|
// check if AbsolutePath exists
|
|
if _, err := os.Stat(p.AbsolutePath); os.IsNotExist(err) {
|
|
// create directory
|
|
if err := os.Mkdir(p.AbsolutePath, 0754); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// create main.go
|
|
mainFile, err := os.Create(fmt.Sprintf("%s/main.go", p.AbsolutePath))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer mainFile.Close()
|
|
|
|
mainTemplate := template.Must(template.New("main").Parse(string(tpl.MainTemplate())))
|
|
err = mainTemplate.Execute(mainFile, p)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// create cmd/root.go
|
|
if _, err = os.Stat(fmt.Sprintf("%s/cmd", p.AbsolutePath)); os.IsNotExist(err) {
|
|
cobra.CheckErr(os.Mkdir(fmt.Sprintf("%s/cmd", p.AbsolutePath), 0751))
|
|
}
|
|
rootFile, err := os.Create(fmt.Sprintf("%s/cmd/root.go", p.AbsolutePath))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer rootFile.Close()
|
|
|
|
rootTemplate := template.Must(template.New("root").Parse(string(tpl.RootTemplate())))
|
|
err = rootTemplate.Execute(rootFile, p)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// create license
|
|
return p.createLicenseFile()
|
|
}
|
|
|
|
func (p *Project) createLicenseFile() error {
|
|
data := map[string]interface{}{
|
|
"copyright": copyrightLine(),
|
|
}
|
|
licenseFile, err := os.Create(fmt.Sprintf("%s/LICENSE", p.AbsolutePath))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer licenseFile.Close()
|
|
|
|
licenseTemplate := template.Must(template.New("license").Parse(p.Legal.Text))
|
|
return licenseTemplate.Execute(licenseFile, data)
|
|
}
|
|
|
|
func (c *Command) Create() error {
|
|
cmdFile, err := os.Create(fmt.Sprintf("%s/cmd/%s.go", c.AbsolutePath, c.CmdFileName))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer cmdFile.Close()
|
|
|
|
commandTemplate := template.Must(template.New("sub").Parse(string(tpl.AddCommandTemplate())))
|
|
err = commandTemplate.Execute(cmdFile, c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|