Add: leetcode

internal/base/base.go

@@ -0,0 +1,45 @@
+package base
+
+import (
+	"fmt"
+	"os"
+	"strings"
+)
+
+const CmdName = "leetcode"
+
+var Commands []*Command
+
+type Command struct {
+	Run       func(cmd *Command, args []string)
+	UsageLine string
+	Short     string
+	Long      string
+}
+
+func (c *Command) Name() string {
+	name := c.UsageLine
+	if i := strings.Index(name, " "); i > 0 {
+		name = name[0 : i+1]
+	}
+	return name
+}
+
+func (c *Command) Usage() {
+	fmt.Printf("usage: %s %s\n", CmdName, c.UsageLine)
+	fmt.Printf("Run '%s help %s' for details.\n", CmdName, c.Name())
+	os.Exit(0)
+}
+
+func Usage() {
+	fmt.Printf("%s is a tool for managing leetcode source code.\n\n", CmdName)
+	fmt.Println("Usage:")
+	fmt.Printf("\t%s <command> [arguments]\n", CmdName)
+	fmt.Println("The commands are:")
+	for _, cmd := range Commands {
+		fmt.Printf("\t%s \t%s\n", cmd.Name(), cmd.Short)
+	}
+	fmt.Println()
+	fmt.Printf(`Use "%s help <command>" for more information about a command.`, CmdName)
+	fmt.Println()
+}

internal/help/help.go

@@ -0,0 +1 @@
+package help

internal/login/login.go

@@ -0,0 +1 @@
+package login

internal/version/version.go

@@ -0,0 +1,25 @@
+package version
+
+import (
+	"fmt"
+	"runtime"
+
+	"github.com/openset/leetcode/internal/base"
+)
+
+const version = "0.0.1"
+
+var CmdVersion = &base.Command{
+	Run:       runVersion,
+	UsageLine: "version",
+	Short:     fmt.Sprintf("print %s version", base.CmdName),
+	Long:      fmt.Sprintf("Version prints the %s version.", base.CmdName),
+}
+
+func runVersion(cmd *base.Command, args []string) {
+	if len(args) != 0 {
+		cmd.Usage()
+	}
+
+	fmt.Printf("%s version %s %s/%s\n", base.CmdName, version, runtime.GOOS, runtime.GOARCH)
+}

main.go

@@ -4,23 +4,24 @@ import (
 	"flag"
 	"fmt"
 	"log"
-	"os"
+
+	"github.com/openset/leetcode/internal/base"
+	"github.com/openset/leetcode/internal/version"
 )
 
+func init() {
+	base.Commands = []*base.Command{
+		version.CmdVersion,
+	}
+}
+
 func main() {
 	log.SetFlags(0)
-	log.SetPrefix("leetcode: ")
+	log.SetPrefix(fmt.Sprintf("%s: ", base.CmdName))
 
-	flag.Usage = usage
+	flag.Usage = base.Usage
 	flag.Parse()
-
-	args := flag.Args()
-	if len(args) < 1 {
+	if flag.NArg() < 1 {
 		flag.Usage()
 	}
 }
-
-func usage() {
-	fmt.Fprintln(os.Stderr, "usage: ")
-	flag.PrintDefaults()
-}