Implement go command to print contest directory

README.md

@@ -17,6 +17,7 @@ Usage:
   hack [command]
 
 Available Commands:
+  go          Print path to the directory
   help        Help about any command
   init        Create directories and download samples
   open        Open current task page
@@ -38,6 +39,7 @@ Write configuration and save it as a `~/.config/hack` file.
 atcoder:
   pass: <password>
   user: <username>
+basedir: <path/to/directory>
 ```
 
 Initialize contest directory and download samples.

cmd/root.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/spf13/cobra"
 
+	"github.com/Ladicle/hack/pkg/cmd/goo"
 	"github.com/Ladicle/hack/pkg/cmd/ini"
 	"github.com/Ladicle/hack/pkg/cmd/open"
 	"github.com/Ladicle/hack/pkg/cmd/test"
@@ -34,6 +35,7 @@ func Run() error {
 	cmd.AddCommand(ini.NewCommand(f, out))
 	cmd.AddCommand(test.NewCommand(f, out))
 	cmd.AddCommand(open.NewCommand())
+	cmd.AddCommand(goo.NewCommand(f, out))
 
 	cobra.OnInitialize(func() {
 		config.MustUnmarshal(path, f)

pkg/cmd/goo/cmd.go

@@ -0,0 +1,57 @@
+package goo
+
+import (
+	"fmt"
+	"io"
+	"path/filepath"
+
+	"github.com/Ladicle/hack/pkg/config"
+	"github.com/Ladicle/hack/pkg/contest"
+	"github.com/spf13/cobra"
+)
+
+type Options struct {
+	contestID string
+	taskID    string
+}
+
+func NewCommand(f *config.File, out io.Writer) *cobra.Command {
+	return &cobra.Command{
+		Use:          "go [<CONTEST>] [<TASK>]",
+		Aliases:      []string{"g"},
+		Short:        "Print path to the directory",
+		SilenceUsage: true,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			var opts Options
+			if err := opts.Validate(args); err != nil {
+				return err
+			}
+			return opts.Run(f, out)
+		},
+	}
+}
+
+func (o *Options) Validate(args []string) error {
+	switch len(args) {
+	case 0:
+		// noop
+	case 1:
+		o.contestID = args[0]
+	case 2:
+		o.contestID = args[0]
+		o.taskID = args[1]
+	}
+	return nil
+}
+
+func (o *Options) Run(f *config.File, out io.Writer) error {
+	path := contest.GetAtCoderDir(f.BaseDir)
+	if o.contestID != "" {
+		path = filepath.Join(path, o.contestID)
+	}
+	if o.taskID != "" {
+		path = filepath.Join(path, o.taskID)
+	}
+	_, err := fmt.Fprint(out, path)
+	return err
+}

pkg/contest/atcoder.go

@@ -12,7 +12,10 @@ import (
 	"github.com/Ladicle/hack/pkg/session"
 )
 
-const atCoderHost = "atcoder.jp"
+const (
+	atCoderDir  = "atcoder"
+	atCoderHost = "atcoder.jp"
+)
 
 type AtCoder struct {
 	ContestID string
@@ -152,9 +155,14 @@ func GetTaskID(dir string) string {
 	return fmt.Sprintf("%s_%s", contestID, filepath.Base(dir))
 }
 
+// GetAtCoderDir return the AtCoder directory name.
+func GetAtCoderDir(baseDir string) string {
+	return filepath.Join(baseDir, atCoderDir)
+}
+
 // GetContestDir return the contest directory name.
 func GetContestDir(baseDir, contestID string) string {
-	return filepath.Join(baseDir, "atcoder", contestID)
+	return filepath.Join(baseDir, atCoderDir, contestID)
 }
 
 // GetTaskDir return the task directory name.