Implement commands

Author: Eric Crosson

src/cmd/git-ledger/add.go

@@ -1,13 +1,17 @@
 package main
 
+// TODO: test
+// TODO: doxygenize
+
 import (
 	"fmt"
 	"regexp"
 	"strings"
+	"path/filepath"
 
 	"github.com/urfave/cli"
-	"github.com/git-hook/git-ledger/src/git-ledger"
 	"github.com/codeskyblue/go-sh"
+	"github.com/git-hook/git-ledger/src/git-ledger"
 )
 
 func getRemote(dir string) string {
@@ -34,20 +38,23 @@ func getSlug(dir string) string {
 }
 
 func add(c *cli.Context) error {
+	// TODO: print message to user
 	fmt.Println("I am in add!")
 	// Default: add cwd to toml
 	project := c.Args().First()
 
+	// TODO: ensure project contains a '.git' directory
+
 	// TODO: clean up this declaration
 	var record ledger.Record
-	record.Path = project
+	record.Path, _ = filepath.Abs(project)
 	record.Slug = getSlug(project)
 
 	fmt.Println("record is ", record)
 	fmt.Println("path is ", ledger.Path())
 
 	record.RemoveFromLedger()
-	record.WriteToLedger()
+	record.AddToLedger()
 
 	return nil
 }

src/cmd/git-ledger/find.go

@@ -4,9 +4,19 @@ import (
 	"fmt"
 
 	"github.com/urfave/cli"
+	"github.com/git-hook/git-ledger/src/git-ledger"
 )
 
 func find(c *cli.Context) error {
-	fmt.Println("I am in find!")
+	slug := c.Args().First()
+
+	record, err := ledger.GetBySlug(slug)
+	if err != nil {
+		// TODO: print a helpful message to stderr
+		fmt.Println(err)
+		// panic(err)
+	}
+	fmt.Println(record.Path)
+
 	return nil
 }

src/cmd/git-ledger/ls.go

@@ -4,9 +4,20 @@ import (
 	"fmt"
 
 	"github.com/urfave/cli"
+	"github.com/git-hook/git-ledger/src/git-ledger"
 )
 
 func ls(c *cli.Context) error {
-	fmt.Println("I am in ls!")
+	records, err := ledger.GetRecords()
+	if err != nil {
+		// TODO: stderr time
+		// TODO exit code
+		panic(err)
+	}
+
+	for _, rec := range records {
+		fmt.Println(fmt.Sprintf("%s: %s", rec.Slug, rec.Path))
+	}
+	// TODO: handle no size, print "none" to stderr? if git does something similar
 	return nil
 }

src/cmd/git-ledger/main.go

@@ -2,10 +2,8 @@ package main
 
 import (
 	"os"
-	"fmt"
 
 	"github.com/Masterminds/semver"
-	"github.com/jmalloc/grit/src/grit/pathutil"
 	"github.com/urfave/cli"
 )
 
@@ -14,10 +12,9 @@ var VERSION = semver.MustParse("0.0.0")
 func main () {
 
 	app := cli.NewApp()
-	homeDir, _ := pathutil.HomeDir()
 
-	app.Name = "grip"
-	app.Usage = "Index your Git clones."
+	app.Name = "git-ledger"
+	app.Usage = "Index your git clones."
 	app.Version = VERSION.String()
 
 	app.Commands = []cli.Command {
@@ -36,7 +33,7 @@ func main () {
 		{
 			Name:   "ls",
 			Usage:  "Print all tracked repositories",
-			Action: find,
+			Action: ls,
 		},
 		{
 			Name:   "rm",
@@ -46,8 +43,6 @@ func main () {
 		},
 	}
 
-	fmt.Println("Made it here with homeDir ", homeDir)
-
 	if err := app.Run(os.Args); err != nil {
 		os.Exit(1)
 	}

src/cmd/git-ledger/rm.go

@@ -4,9 +4,30 @@ import (
 	"fmt"
 
 	"github.com/urfave/cli"
+	"github.com/git-hook/git-ledger/src/git-ledger"
 )
 
+// TODO document that this command:
+// takes PATH
+
+// TODO: set exit code (project-wide)
 func rm(c *cli.Context) error {
 	fmt.Println("I am in rm!")
+	slug := c.Args().First()
+
+	record, err := ledger.GetBySlug(slug)
+	if err != nil {
+		// TODO: print a helpful message to stderr
+		fmt.Println(err)
+		// panic(err)
+	}
+
+	err = record.RemoveFromLedger()
+	if err != nil {
+		// TODO: print a helpful message to stderr
+		fmt.Println(err)
+		// panic(err)
+	}
+
 	return nil
 }

src/git-ledger/ledger.go

@@ -4,6 +4,8 @@ import (
 	"os"
 	"fmt"
 	"path"
+	"errors"
+	"strings"
 	"io/ioutil"
 
 	"github.com/jmalloc/grit/src/grit/pathutil"
@@ -24,20 +26,60 @@ func Path() string {
 	return path.Join(home, ".git-ledger")
 }
 
+func GetRecords() ([]Record, error) {
+	var ledger Records
+
+	b, err := ioutil.ReadFile(Path())
+	if err != nil {
+		return ledger.Record, err
+	}
+	str := string(b)
+
+	if _, err := toml.Decode(str, &ledger); err != nil {
+		return ledger.Record, err
+	}
+	return ledger.Record, err
+}
+
+func GetBySlug(slug string) (Record, error) {
+	var match Record
+	var ledger Records
+
+	// TODO: use getrecords
+	b, err := ioutil.ReadFile(Path())
+	if err != nil {
+		return match, err
+	}
+	str := string(b)
+
+	if _, err := toml.Decode(str, &ledger); err != nil {
+		return match, err
+	}
+
+	for _, r := range ledger.Record {
+		if strings.Contains(r.Slug, slug) {
+			return r, nil
+		}
+	}
+	return match, errors.New(fmt.Sprintf("Unknown project: %s", slug))
+}
+
+// fixme: remove duplicated code
 func (r Record) String() string {
 	return fmt.Sprintf("[[Record]]\npath = \"%s\"\nslug = \"%s\"\n\n", r.Path, r.Slug)
 }
 
-func (r Record) RemoveFromLedger() {
+// comparison by Path
+func (r Record) RemoveFromLedger() error {
 	b, err := ioutil.ReadFile(Path())
 	if err != nil {
-		fmt.Println(err)
+		return err
 	}
 	str := string(b)
 
 	var ledger Records
 	if _, err := toml.Decode(str, &ledger); err != nil {
-		panic(err)
+		return err
 	}
 
 	var content string
@@ -48,10 +90,10 @@ func (r Record) RemoveFromLedger() {
 	}
 
 	ioutil.WriteFile(Path(), []byte(content), 0644)
-
+	return nil
 }
 
-func (r Record) WriteToLedger() {
+func (r Record) AddToLedger() {
 	f, err := os.OpenFile(Path(), os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0644)
 	if err != nil {
 		panic(err)