Add is adding to toml idempotently

Author: Eric Crosson

src/cmd/git-ledger/add.go

@@ -2,11 +2,11 @@ package main
 
 import (
 	"fmt"
+	"regexp"
 	"strings"
 
 	"github.com/urfave/cli"
-	// "github.com/BurntSushi/toml"
-	"github.com/ericcrosson/git-ledger/src/git-ledger"
+	"github.com/git-hook/git-ledger/src/git-ledger"
 	"github.com/codeskyblue/go-sh"
 )
 
@@ -23,12 +23,14 @@ func getRemote(dir string) string {
 func getSlug(dir string) string {
 	session := sh.NewSession()
 	session.SetDir(dir)
-	out, err := session.Command("git", "remote", "get-url", getRemote(dir)).Command("sed", "-r", "s#[^/:]*/##").Output()
+	out, err := session.Command("git", "remote", "get-url", getRemote(dir)).Output()
 	output := strings.TrimSpace(fmt.Sprintf("%s", out))
+	reg := regexp.MustCompile(`[^/:]*/[^/:]*$`)
+	res := reg.FindStringSubmatch(output)
 	if err != nil {
 		panic(err)
 	}
-	return output
+	return res[0]
 }
 
 func add(c *cli.Context) error {
@@ -44,5 +46,8 @@ func add(c *cli.Context) error {
 	fmt.Println("record is ", record)
 	fmt.Println("path is ", ledger.Path())
 
+	record.RemoveFromLedger()
+	record.WriteToLedger()
+
 	return nil
 }

src/git-ledger/ledger.go

@@ -1,16 +1,61 @@
 package ledger
 
 import (
+	"os"
+	"fmt"
 	"path"
+	"io/ioutil"
+
 	"github.com/jmalloc/grit/src/grit/pathutil"
+	"github.com/BurntSushi/toml"
 )
 
 type Record struct {
 	Path string
 	Slug string
 }
 
+type Records struct {
+	Record []Record
+}
+
 func Path() string {
 	home, _ := pathutil.HomeDir()
 	return path.Join(home, ".git-ledger")
 }
+
+func (r Record) String() string {
+	return fmt.Sprintf("[[Record]]\npath = \"%s\"\nslug = \"%s\"\n\n", r.Path, r.Slug)
+}
+
+func (r Record) RemoveFromLedger() {
+	b, err := ioutil.ReadFile(Path())
+	if err != nil {
+		fmt.Println(err)
+	}
+	str := string(b)
+
+	var ledger Records
+	if _, err := toml.Decode(str, &ledger); err != nil {
+		panic(err)
+	}
+
+	var content string
+	for _, s := range ledger.Record {
+		if s.Path != r.Path {
+			content = content + s.String() + "\n"
+		}
+	}
+
+	ioutil.WriteFile(Path(), []byte(content), 0644)
+
+}
+
+func (r Record) WriteToLedger() {
+	f, err := os.OpenFile(Path(), os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0644)
+	if err != nil {
+		panic(err)
+	}
+	defer f.Close()
+	fmt.Fprintf(f, r.String())
+}