glenux-archive/spf13--cobra
1
0
Fork 0
mirror of https://github.com/spf13/cobra synced 2025-05-06 13:27:26 +00:00
Code Issues Projects Releases Packages Wiki Activity

adding --force flag to initCmd

Browse source
This commit is contained in:
Aaron Paz 2018-12-26 22:47:12 -08:00
parent d2d81d9a96
commit 22a124de4a
1 changed files with 29 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
cobra/cmd/init.go
View file

@ -23,6 +23,12 @@ import (
"github.com/spf13/viper" "github.com/spf13/viper"
) )
func init() {
initCmd.Flags().BoolVarP(&force, "force", "f", false, "force init to work in non-empty directories")
}
var force bool
var initCmd = &cobra.Command{ var initCmd = &cobra.Command{
Use: "init [name]", Use: "init [name]",
Aliases: []string{"initialize", "initialise", "create"}, Aliases: []string{"initialize", "initialise", "create"},
@ -36,8 +42,9 @@ and the appropriate structure for a Cobra-based CLI application.
(e.g. github.com/spf13/hugo); (e.g. github.com/spf13/hugo);
* If an absolute path is provided, it will be created; * If an absolute path is provided, it will be created;
* If the directory already exists but is empty, it will be used. * If the directory already exists but is empty, it will be used.
* If the directory already exists, the force flag is used, and it is not empty, existing files will not be overwritten
Init will not use an existing directory with contents.`, Init will not use an existing directory with contents unless specified via --force or -f`,
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
wd, err := os.Getwd() wd, err := os.Getwd()
@ -78,15 +85,30 @@ func initializeProject(project *Project) {
if err != nil { if err != nil {
er(err) er(err)
} }
} else if !isEmpty(project.AbsPath()) { // If path exists and is not empty don't use it } else if !isEmpty(project.AbsPath()) && !force { // If path exists and is not empty don't use it
er("Cobra will not create a new project in a non empty directory: " + project.AbsPath()) er("Cobra will not create a new project in a non empty directory: " + project.AbsPath())
} else if !isEmpty(project.AbsPath()) && force { // If path exists and force flag is true, use it but check to see if files exist first
if !exists(project.AbsPath() + "/" + project.License().Name) {
fmt.Println(project.AbsPath() + "/LICENSE.txt exists... Skipping")
createLicenseFile(project.License(), project.AbsPath())
} }
if !exists(project.AbsPath() + "/" + "main.go") {
fmt.Println(project.AbsPath() + "/main.go exists... Skipping")
createMainFile(project)
}
if !exists(project.AbsPath() + "/" + "cmd/root.go") {
fmt.Println(project.AbsPath() + "/cmd/root.go exists... Skipping")
createRootCmdFile(project)
}
} else {
// We have a directory and it's empty. Time to initialize it. // We have a directory and it's empty. Time to initialize it.
createLicenseFile(project.License(), project.AbsPath()) createLicenseFile(project.License(), project.AbsPath())
createMainFile(project) createMainFile(project)
createRootCmdFile(project) createRootCmdFile(project)
} }
}
func createLicenseFile(license License, path string) { func createLicenseFile(license License, path string) {
data := make(map[string]interface{}) data := make(map[string]interface{})