Refactor

Signed-off-by: Oliver Davies <oliver@oliverdavies.uk>

internal/zet/parse.go

@@ -1,21 +0,0 @@
-package zet
-
-import (
-	"fmt"
-	"strconv"
-)
-
-func ParseZetList(ids []int) []string {
-	var lines []string
-
-	green := "\033[32m"
-	reset := "\033[0m"
-
-	for _, num := range ids {
-		line := fmt.Sprintf("%s%s%s %s", green, strconv.Itoa(num), reset, getTitle(num))
-
-		lines = append(lines, line)
-	}
-
-	return lines
-}

internal/zet/zet.go

@@ -12,6 +12,31 @@ import (
 	"code.oliverdavies.uk/opdavies/cmd-zet/internal/git"
 )
 
+type ZetList struct {
+	Ids []int
+}
+
+func CreateZetList(ids []int) *ZetList {
+	return &ZetList{
+		Ids: ids,
+	}
+}
+
+func (z *ZetList) Parse() []string {
+	var lines []string
+
+	green := "\033[32m"
+	reset := "\033[0m"
+
+	for _, num := range z.Ids {
+		line := fmt.Sprintf("%s%s%s %s", green, strconv.Itoa(num), reset, getTitle(num))
+
+		lines = append(lines, line)
+	}
+
+	return lines
+}
+
 func getTitle(id int) string {
 	return getTitleFromFile(path.Join(strconv.Itoa(id), "index.adoc"))
 }