bubble root.yaml to first

cmd/add.go

@@ -18,23 +18,18 @@ func add(config *config) error {
 		Schedule: pttodo.Schedule(config.addSchedule),
 		Tags:     config.addTags,
 	}
-	newTarget, err := _add(config.targets, v)
-	if err != nil {
-		return err
-	}
-	config.targets = append(config.targets, newTarget)
-	return nil
+	return _add(config.Targets()[0], v)
 }
 
-func _add(filepaths []string, todo pttodo.Todo) (string, error) {
-	target := filepaths[0] + ".todo." + uuid.New().String()
+func _add(filepath string, todo pttodo.Todo) error {
+	target := filepath + ".todo." + uuid.New().String()
 
 	c, err := yaml.Marshal([]pttodo.Todo{todo})
 	if err != nil {
-		return "", err
+		return err
 	} else if err := ioutil.WriteFile(target, c, os.ModePerm); err != nil {
-		return "", err
+		return err
 	}
 
-	return target, nil
+	return nil
 }

cmd/config.go

@@ -4,11 +4,14 @@ import (
 	"flag"
 	"fmt"
 	"os"
+	"path"
+	"path/filepath"
+	"sort"
 	"strings"
 )
 
 type config struct {
-	targets     []string
+	target      string
 	root        string
 	tags        []string
 	search      string
@@ -25,8 +28,7 @@ func getConfig() config {
 	}
 
 	var config config
-	var target string
-	flag.StringVar(&target, "f", defaultFilepath, "($PTTODO_FILE) path to yaml file or dir (starting with root then alphabetical for dir)")
+	flag.StringVar(&config.target, "f", defaultFilepath, "($PTTODO_FILE) path to yaml file or dir (starting with root then alphabetical for dir)")
 	flag.StringVar(&config.root, "root", DUMP_TODO, "path to pretty print ("+fmt.Sprint([]string{DUMP_ALL, DUMP_TODO, DUMP_SCHEDULED, DUMP_DONE})+")")
 	var tagss string
 	flag.StringVar(&tagss, "tags", "", "csv of all tags to find, -x to invert")
@@ -38,10 +40,53 @@ func getConfig() config {
 	flag.Parse()
 
 	config.tags = strings.Split(tagss, ",")
-	config.targets = []string{target}
-	if stat, err := os.Stat(target); err == nil && stat.IsDir() {
-		config.targets, _ = listDir(target)
-	}
+
+	config.Targets()
 
 	return config
 }
+
+func (config config) Targets() []string {
+	result, err := config.targets()
+	if err != nil {
+		panic(err)
+	}
+	sort.Strings(result)
+	for i := range result {
+		if path.Base(result[i]) == "root.yaml" {
+			newresult := append([]string{result[i]}, result[:i]...)
+			newresult = append(newresult, result[i+1:]...)
+			result = newresult
+			break
+		}
+	}
+	return result
+}
+
+func (config config) targets() ([]string, error) {
+	patterns := []string{config.target, fmt.Sprintf("%s.*", config.target)}
+	isDir := false
+	if stat, err := os.Stat(config.target); err == nil {
+		isDir = stat.IsDir()
+	}
+	if isDir {
+		patterns = []string{fmt.Sprintf("%s/*", config.target)}
+	}
+
+	result := make([]string, 0, 1)
+	for _, pattern := range patterns {
+		results, err := filepath.Glob(pattern)
+		if err != nil {
+			return nil, err
+		}
+		result = append(result, results...)
+	}
+
+	if len(result) == 0 {
+		if isDir {
+			return []string{path.Join(config.target, "root.yaml")}, nil
+		}
+		return []string{config.target}, nil
+	}
+	return result, nil
+}

cmd/config_test.go

@@ -0,0 +1,87 @@
+package main
+
+import (
+	"os"
+	"path"
+	"testing"
+)
+
+func TestConfigTargets(t *testing.T) {
+	touch := func(t *testing.T, p string) {
+		if err := os.WriteFile(p, nil, os.ModePerm); err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	cases := map[string]struct {
+		setup func(*testing.T, string)
+		given string
+		want  []string
+	}{
+		"does not exist": {
+			given: "x",
+			want:  []string{"x"},
+		},
+		"one file": {
+			setup: func(t *testing.T, d string) {
+				touch(t, "x")
+			},
+			given: "x",
+			want:  []string{"x"},
+		},
+		"two files": {
+			setup: func(t *testing.T, d string) {
+				touch(t, path.Join(d, "a"))
+				touch(t, path.Join(d, "root.yaml"))
+				touch(t, path.Join(d, "z"))
+			},
+			given: "",
+			want:  []string{"root.yaml", "a", "z"},
+		},
+		"empty dir": {
+			setup: func(t *testing.T, d string) {
+				os.Mkdir(path.Join(d, "x"), os.ModePerm)
+			},
+			given: "x",
+			want:  []string{"root.yaml"},
+		},
+		"dir": {
+			setup: func(t *testing.T, d string) {
+				os.Mkdir(path.Join(d, "x"), os.ModePerm)
+				touch(t, path.Join(d, "x", "a"))
+				touch(t, path.Join(d, "x", "a.todo.xyz"))
+				touch(t, path.Join(d, "x", "b"))
+			},
+			given: "x",
+			want:  []string{"a", "a.todo.xyz", "b"},
+		},
+	}
+
+	for name, d := range cases {
+		c := d
+		t.Run(name, func(t *testing.T) {
+			d := t.TempDir()
+			if c.setup != nil {
+				c.setup(t, d)
+			}
+			config := config{target: path.Join(d, c.given)}
+			got := config.Targets()
+			for i := range got {
+				got[i] = path.Base(got[i])
+			}
+			for i := range c.want {
+				c.want[i] = path.Base(c.want[i])
+			}
+
+			t.Logf("want\n\t%+v, got \n\t%+v", c.want, got)
+			if len(got) != len(c.want) {
+				t.Error(c.want, got)
+			}
+			for i := range got {
+				if got[i] != c.want[i] {
+					t.Errorf("[%d] wanted %s, got %s", i, c.want[i], got[i])
+				}
+			}
+		})
+	}
+}

cmd/dump.go

@@ -6,6 +6,7 @@ import (
 	"io/ioutil"
 	"log"
 	"os"
+	"path/filepath"
 	"strings"
 
 	"gogs.inhome.blapointe.com/bel/pttodo/pttodo"
@@ -13,12 +14,25 @@ import (
 )
 
 func dump(config config) error {
-	return _dump(os.Stdout, config.targets, config.tags, config.search, config.root)
+	return _dump(os.Stdout, config.Targets(), config.tags, config.search, config.root)
 }
 
 func _dump(writer io.Writer, filepaths []string, tags []string, search, rootDisplay string) error {
 	var root pttodo.Root
 
+	for _, p := range filepaths {
+		results, err := filepath.Glob(p + ".*")
+		if err != nil {
+			return err
+		}
+		for _, result := range results {
+			if result == p {
+				continue
+			}
+			filepaths = append(filepaths, result)
+		}
+	}
+
 	for _, filepath := range filepaths {
 		reader, err := filePathReader(filepath)
 		if err != nil {

cmd/edit.go

@@ -15,11 +15,11 @@ import (
 	"gopkg.in/yaml.v2"
 )
 
-func edit(config config) error {
+func edit(config *config) error {
 	if !config.edit {
 		return nil
 	}
-	return _edit(config.targets)
+	return _edit(config.Targets())
 }
 
 func _edit(filepaths []string) error {

cmd/main.go

@@ -32,7 +32,7 @@ func _main() error {
 	if err := add(&config); err != nil {
 		return err
 	}
-	if err := edit(config); err != nil {
+	if err := edit(&config); err != nil {
 		return err
 	}
 	return dump(config)