add add-task command (#52)

add add-task command

Author: Yusaku Hatanaka

.golangci.yml

@@ -289,6 +289,11 @@ issues:
       linters:
         - gosec
 
+    - path: editor/editor.go
+      text: "G204: Subprocess launched with variable"
+      linters:
+        - gosec
+
     - path: notify
       text: "Subprocess launched with variable"
       linters:

cmd/addTask.go

@@ -0,0 +1,70 @@
+package cmd
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/spf13/cobra"
+
+	"github.com/hatappi/gomodoro/config"
+	"github.com/hatappi/gomodoro/editor"
+	"github.com/hatappi/gomodoro/task"
+)
+
+const initialText = `# Please write one task per line`
+
+// addTaskCmd represents the addTask command
+var addTaskCmd = &cobra.Command{
+	Use:   "add-task TASK_NAME",
+	Short: "add task",
+	Long: `This command adds a task.
+Please specify the task name in the argument.
+if you doesn't specify task name, editor starts up.
+And add a task using the editor.
+`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		config, err := config.GetConfig()
+		if err != nil {
+			return err
+		}
+		taskPath, err := config.ExpandTaskFile()
+		if err != nil {
+			return err
+		}
+
+		name := strings.Join(args, " ")
+		if name != "" {
+			err = task.AddTask(taskPath, name)
+			if err != nil {
+				return err
+			}
+			fmt.Printf("add %s\n", name)
+			return nil
+		}
+
+		ts, err := editor.GetSliceText(initialText)
+		if err != nil {
+			return err
+		}
+
+		for _, t := range ts {
+			if t == "" {
+				continue
+			}
+			if strings.HasPrefix(t, "#") {
+				continue
+			}
+			err = task.AddTask(taskPath, t)
+			if err != nil {
+				return err
+			}
+			fmt.Printf("add %s\n", t)
+		}
+
+		return nil
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(addTaskCmd)
+}

editor/editor.go

@@ -0,0 +1,65 @@
+// Package editor manage EDITOR
+package editor
+
+import (
+	"bufio"
+	"io/ioutil"
+	"os"
+	"os/exec"
+)
+
+// GetSliceText get slice text edited with editor
+func GetSliceText(initialText string) ([]string, error) {
+	tmpfile, err := ioutil.TempFile("", "gomodoro")
+	if err != nil {
+		return nil, err
+	}
+	defer func() {
+		_ = os.Remove(tmpfile.Name())
+	}()
+	defer func() {
+		_ = tmpfile.Close()
+	}()
+
+	_, err = tmpfile.Write([]byte(initialText))
+	if err != nil {
+		return nil, err
+	}
+
+	if err = openEditor(tmpfile.Name()); err != nil {
+		return nil, err
+	}
+
+	f, err := os.Open(tmpfile.Name())
+	if err != nil {
+		return nil, err
+	}
+	defer func() {
+		_ = f.Close()
+	}()
+
+	ts := make([]string, 0)
+	scanner := bufio.NewScanner(f)
+	for scanner.Scan() {
+		ts = append(ts, scanner.Text())
+	}
+
+	if err := scanner.Err(); err != nil {
+		return nil, err
+	}
+
+	return ts, nil
+}
+
+func openEditor(filepath string) error {
+	cmdName := "vi"
+	if e := os.Getenv("EDITOR"); e != "" {
+		cmdName = e
+	}
+
+	c := exec.Command(cmdName, filepath)
+	c.Stdin = os.Stdin
+	c.Stdout = os.Stdout
+	c.Stderr = os.Stderr
+	return c.Run()
+}

task/task.go

@@ -234,3 +234,28 @@ func createTaskName(c screen.Client) (string, error) {
 		}
 	}
 }
+
+// AddTask save task to file
+func AddTask(taskFile, name string) error {
+	if name == "" {
+		return xerrors.New("task name is empty")
+	}
+
+	c := &clientImpl{
+		taskFile: taskFile,
+	}
+
+	tasks, err := c.loadTasks()
+	if err != nil {
+		return err
+	}
+
+	tasks = append(tasks, &Task{Name: name})
+
+	err = c.saveTasks(tasks)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}