ref: 51eb2ed06f13e1477ebe3ebf7675688c8f8e1246
dir: /task.go/
package main
import (
"bufio"
"errors"
"fmt"
"log"
"os"
"path"
"path/filepath"
"strings"
"text/template"
)
const tmplt = `{{range .TaskList}}{{.Title}}
{{range .List}}{{if .Done}}[x] {{.Desc}}{{else}}[ ] {{.Desc}}{{end}}
{{end}}
{{end}}`
type layout struct {
TaskList []*entries
file string
}
// Entries - set of todo items for a give task
type entries struct {
Title string
List []*entry
}
// Entry - individual item of a set of tasks
type entry struct {
Desc string
Done bool
}
func parse(file string) (*layout, error) {
l := &layout{
TaskList: []*entries{},
file: file,
}
fl, err := os.Open(file)
if err != nil {
return nil, err
}
sc := bufio.NewScanner(fl)
for sc.Scan() {
if sc.Text() == "" {
continue
}
tl := &entries{
Title: sc.Text(),
List: parseEntries(sc),
}
l.TaskList = append(l.TaskList, tl)
}
if err := sc.Err(); err != nil {
return nil, err
}
return l, nil
}
func parseEntries(sc *bufio.Scanner) []*entry {
sc.Scan()
en := []*entry{}
for {
d := sc.Text()
if len(d) < 4 {
return en
}
if err := sc.Err(); err != nil {
log.Fatal(err)
}
switch d[:3] {
case "[ ]":
en = append(en, &entry{
Desc: d[4:],
Done: false,
})
case "[x]":
en = append(en, &entry{
Desc: d[4:],
Done: true,
})
default:
return en
}
sc.Scan()
}
}
// We want to return any matched string for entry, otherwise return $dirname.todo of current directory
func findEntry(entry string) string {
wd, _ := os.Getwd()
match := path.Base(wd) + ".todo"
filepath.Walk(".", func(p string, info os.FileInfo, err error) error {
if path.Ext(p) != ".todo" {
return nil
}
// Found a file! Search for entry
f, err := os.Open(p)
if err != nil {
return err
}
defer f.Close()
scanner := bufio.NewScanner(f)
for scanner.Scan() {
if strings.Contains(scanner.Text(), entry) {
match = p
return filepath.SkipDir
}
}
return nil
})
return match
}
func task(c *command) error {
if len(c.args) < 2 {
return errors.New("Too few arguments supplied")
}
fp := findEntry(c.args[1])
l, err := parse(fp)
if err != nil && c.args[0] != "create" {
return err
}
switch c.args[0] {
case "create":
l = &layout{
TaskList: []*entries{},
file: fp,
}
err = l.create(c.args[1])
if err != nil {
return err
}
l.write()
return nil
case "destroy":
err := l.destroy(c.args[1])
if err != nil {
return err
}
l.write()
return nil
case "add":
if l.exists(c.args[1], c.args[2]) {
return fmt.Errorf("Entry exists")
}
l.add(c.args[1], c.args[2])
l.write()
return nil
case "rm":
if !l.exists(c.args[1], c.args[2]) {
return fmt.Errorf("No entry found")
}
defer l.write()
return l.rm(c.args[1], c.args[2])
case "toggle":
if !l.exists(c.args[1], c.args[2]) {
return fmt.Errorf("No such task/entry")
}
defer l.write()
return l.toggle(c.args[1], c.args[2])
default:
return fmt.Errorf("Command not supported: %v", c.args[0])
}
}
func (l *layout) destroy(title string) error {
if _, err := os.Stat(l.file); err != nil {
return errors.New("Unable to locate .todo file")
}
for i, e := range l.TaskList {
if e.Title != title {
continue
}
if i < len(l.TaskList)-1 {
copy(l.TaskList[i:], l.TaskList[i+1:])
}
l.TaskList = l.TaskList[:len(l.TaskList)-1]
return nil
}
return errors.New("No such entry")
}
func (l *layout) create(title string) error {
if len(title) < 1 {
return errors.New("Unable to add nil entry")
}
l.TaskList = append(l.TaskList, &entries{
Title: title,
List: []*entry{},
})
return nil
}
func (l *layout) write() {
wr, err := os.Create(l.file)
defer wr.Close()
if err != nil {
log.Fatal(err)
}
t := template.Must(template.New("tmplt").Parse(tmplt))
err = t.Execute(wr, l)
if err != nil {
log.Fatal(err)
}
}
func (l *layout) exists(Title, item string) bool {
for _, e := range l.TaskList {
if e.Title != Title {
continue
}
for _, t := range e.List {
if t.Desc == item {
return true
}
}
}
return false
}
func (l *layout) add(Title, item string) error {
for _, e := range l.TaskList {
// Found existing task, append to List
if e.Title != Title {
continue
}
e.List = append(e.List, &entry{
Desc: item,
Done: false,
})
return nil
}
line := &entry{
Desc: item,
Done: false,
}
l.TaskList = append(l.TaskList, &entries{
Title: Title,
List: []*entry{line},
})
return nil
}
func (l *layout) rm(Title, item string) error {
for _, e := range l.TaskList {
if e.Title != Title {
continue
}
for i, t := range e.List {
if t.Desc != item {
continue
}
if i < len(e.List)-1 {
copy(e.List[i:], e.List[i+1:])
}
e.List = e.List[:len(e.List)-1]
return nil
}
}
return fmt.Errorf("No such task/entry")
}
func (l *layout) toggle(Title, item string) error {
for _, e := range l.TaskList {
if e.Title != Title {
continue
}
for _, t := range e.List {
if t.Desc != item {
continue
}
t.Done = !t.Done
return nil
}
}
return fmt.Errorf("No such task/entry")
}