initial import
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
func (util *Util) AddFileCmds() {
|
||||
var subCmd = &cobra.Command{
|
||||
Use: "file",
|
||||
Short: "File operation",
|
||||
}
|
||||
var putFileCmd = &cobra.Command{
|
||||
Use: "put",
|
||||
Short: "Put file to storage",
|
||||
Run: util.PutFile,
|
||||
}
|
||||
putFileCmd.Flags().StringVarP(&util.putFileParams.Username, "username", "u", "", "Username")
|
||||
putFileCmd.Flags().StringVarP(&util.putFileParams.Password, "password", "p", "", "Password")
|
||||
putFileCmd.Flags().StringVarP(&util.putFileParams.Source, "source", "s", "", "Source path")
|
||||
putFileCmd.Flags().StringVarP(&util.putFileParams.Dest, "dest", "d", "", "Desctination path")
|
||||
|
||||
subCmd.AddCommand(putFileCmd)
|
||||
util.rootCmd.AddCommand(subCmd)
|
||||
|
||||
}
|
||||
|
||||
type PutFileParams struct {
|
||||
Source string
|
||||
Dest string
|
||||
Username string
|
||||
Password string
|
||||
}
|
||||
|
||||
func (util *Util) PutFile(cmd *cobra.Command, args []string) {}
|
||||
|
||||
type GetFileParams struct {
|
||||
Source string
|
||||
Dest string
|
||||
Username string
|
||||
Password string
|
||||
}
|
||||
|
||||
func (util *Util) GetFile(cmd *cobra.Command, args []string) {}
|
||||
|
||||
func (util *Util) DeleteFile(cmd *cobra.Command, args []string) {}
|
||||
@@ -0,0 +1,55 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
util := NewUtil()
|
||||
err = util.Build()
|
||||
if err != nil {
|
||||
os.Exit(1)
|
||||
}
|
||||
err = util.Exec(os.Args[1:])
|
||||
if err != nil {
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
type Util struct {
|
||||
rootCmd *cobra.Command
|
||||
putFileParams PutFileParams
|
||||
}
|
||||
|
||||
func NewUtil() *Util {
|
||||
return &Util{}
|
||||
}
|
||||
|
||||
func (util *Util) Build() error {
|
||||
var err error
|
||||
rootCmd := &cobra.Command{
|
||||
Use: "cmd",
|
||||
Short: "A brief description the command",
|
||||
}
|
||||
rootCmd.CompletionOptions.DisableDefaultCmd = true
|
||||
util.rootCmd = rootCmd
|
||||
|
||||
util.AddFileCmds()
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (util *Util) Exec(args []string) error {
|
||||
var err error
|
||||
util.rootCmd.SetArgs(args)
|
||||
err = util.rootCmd.Execute()
|
||||
return err
|
||||
}
|
||||
|
||||
func (util *Util) Hello(cmd *cobra.Command, args []string) {
|
||||
fmt.Println("hello, world!")
|
||||
}
|
||||
Reference in New Issue
Block a user