Merge pull request #61 from augmentable-dev/helper-func

loads a helper function for splitting strings

pkg/gitqlite/gitqlite.go

@@ -48,6 +48,11 @@ func init() {
 				return err
 			}
 
+			err = loadHelperFuncs(conn)
+			if err != nil {
+				return err
+			}
+
 			return nil
 		},
 	})
@@ -109,3 +114,20 @@ func (g *GitQLite) ensureTables(options *Options) error {
 
 	return nil
 }
+
+func loadHelperFuncs(conn *sqlite3.SQLiteConn) error {
+	// str_split(inputString, splitCharacter, index) string
+	split := func(s, c string, i int) string {
+		split := strings.Split(s, c)
+		if i < len(split) {
+			return split[i]
+		}
+		return ""
+	}
+
+	if err := conn.RegisterFunc("str_split", split, true); err != nil {
+		return err
+	}
+
+	return nil
+}

pkg/gitqlite/helpers_test.go

@@ -0,0 +1,39 @@
+package gitqlite
+
+import (
+	"testing"
+)
+
+func TestStrSplit(t *testing.T) {
+	instance, err := New(fixtureRepoDir, &Options{})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	rows, err := instance.DB.Query("SELECT str_split('hello world', ' ', 0)")
+	if err != nil {
+		t.Fatal(err)
+	}
+	rowNum, contents, err := GetContents(rows)
+	if err != nil {
+		t.Fatalf("err %d at row Number %d", err, rowNum)
+	}
+
+	if contents[0][0] != "hello" {
+		t.Fatalf("expected string: %s, got %s", "hello", contents[0][0])
+	}
+
+
+	rows, err = instance.DB.Query("SELECT str_split('hello world', ' ', 10)")
+	if err != nil {
+		t.Fatal(err)
+	}
+	rowNum, contents, err = GetContents(rows)
+	if err != nil {
+		t.Fatalf("err %d at row Number %d", err, rowNum)
+	}
+
+	if contents[0][0] != "" {
+		t.Fatalf("expected string: %s, got %s", "", contents[0][0])
+	}
+}