From c4a74123ef4ebf39730bf8f98f04dfaadbeaed0c Mon Sep 17 00:00:00 2001
From: Tai Groot <tai@taigrr.com>
Date: Sat, 28 Jan 2023 00:24:00 -0800
Subject: [PATCH] add mgfetch

---
 cmd/mgfetch/mgfetch.go    | 123 ++++++++++++++++++++++++++++++++++++++
 gitgraph/common/common.go |   2 +-
 go.mod                    |   3 +
 3 files changed, 127 insertions(+), 1 deletion(-)
 create mode 100644 cmd/mgfetch/mgfetch.go

diff --git a/cmd/mgfetch/mgfetch.go b/cmd/mgfetch/mgfetch.go
new file mode 100644
index 0000000..ea428ca
--- /dev/null
+++ b/cmd/mgfetch/mgfetch.go
@@ -0,0 +1,123 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"time"
+
+	git "github.com/go-git/go-git/v5"
+	"github.com/go-git/go-git/v5/plumbing/object"
+	gterm "github.com/taigrr/gico/gitgraph/term"
+
+	"github.com/taigrr/gico/types"
+	"github.com/taigrr/mg/parse"
+)
+
+func main() {
+	year := time.Now().Year() - 1
+	yearLength := 365
+	if year%4 == 0 {
+		yearLength++
+	}
+
+	gfreq := make(YearFreq, yearLength)
+
+	mrconf, err := parse.LoadMRConfig()
+	if err != nil {
+		panic(err)
+	}
+	paths := mrconf.GetRepoPaths()
+	for _, p := range paths {
+		repo, err := OpenRepo(p)
+		if err != nil {
+			panic(err)
+		}
+		freq, err := repo.GetYear(year)
+		gfreq = gfreq.Merge(freq)
+		if err != nil {
+			panic(err)
+		}
+	}
+	fmt.Print(gfreq.String())
+}
+
+type Repo git.Repository
+
+func (a YearFreq) Merge(b YearFreq) YearFreq {
+	x := len(a)
+	y := len(b)
+	if x < y {
+		x = y
+	}
+	c := make(YearFreq, x)
+	copy(c, a)
+	for i := 0; i < y; i++ {
+		c[i] += b[i]
+	}
+	return c
+}
+
+func OpenRepo(directory string) (Repo, error) {
+	if s, err := os.Stat(directory); err != nil {
+		return Repo{}, err
+	} else {
+		if !s.IsDir() {
+			return Repo{}, errors.New("received path to non-directory for git repo")
+		}
+	}
+
+	r, err := git.PlainOpenWithOptions(directory, &(git.PlainOpenOptions{DetectDotGit: true}))
+	return Repo(*r), err
+}
+
+type YearFreq []int
+
+func (yf YearFreq) String() string {
+	return gterm.GetYearUnicode(yf)
+}
+
+func (repo Repo) GetYear(year int) (YearFreq, error) {
+	yearLength := 365
+	if year%4 == 0 {
+		yearLength++
+	}
+	freq := make([]int, yearLength)
+	data := types.NewDataSet()
+	r := git.Repository(repo)
+	ref, err := r.Head()
+	if err != nil {
+		return freq, err
+	}
+	cIter, err := r.Log(&git.LogOptions{From: ref.Hash()})
+	if err != nil {
+		return freq, err
+	}
+	err = cIter.ForEach(func(c *object.Commit) error {
+		ts := c.Author.When
+		commit := types.Commit{Author: c.Author.Name, Message: c.Message, TimeStamp: ts}
+		roundedTS := ts.Round(time.Hour * 24)
+		wd, ok := data[roundedTS]
+		if !ok {
+			wd = types.WorkDay{}
+			wd.Commits = []types.Commit{}
+		}
+		wd.Commits = append(wd.Commits, commit)
+		wd.Count++
+		wd.Day = roundedTS
+		data[roundedTS] = wd
+		return nil
+	})
+
+	for k, v := range data {
+		if k.Year() != year {
+			continue
+		}
+		// this is equivalent to adding len(commits) to the freq total, but
+		// it's a stub for later when we do more here
+		for range v.Commits {
+			freq[k.YearDay()-1]++
+		}
+	}
+	return freq, nil
+}
diff --git a/gitgraph/common/common.go b/gitgraph/common/common.go
index 621be6a..35e10fa 100644
--- a/gitgraph/common/common.go
+++ b/gitgraph/common/common.go
@@ -21,7 +21,7 @@ func CreateGraph() bytes.Buffer {
 
 func init() {
 	colors := []string{"#767960", "#a7297f", "#e8ca89", "#f5efd6", "#158266"}
-	// colors = []string{"#000000", "#0e4429", "#006d32", "#26a641", "#39d353"}
+	colors = []string{"#000000", "#0e4429", "#006d32", "#26a641", "#39d353"}
 	for _, c := range colors {
 		color := sc.FromHexString(c)
 		colorScheme = append(colorScheme, color)
diff --git a/go.mod b/go.mod
index dd1cba8..b67057e 100644
--- a/go.mod
+++ b/go.mod
@@ -4,6 +4,8 @@ go 1.19
 
 replace github.com/charmbracelet/bubbles => github.com/taigrr/bubbles v0.10.5
 
+replace github.com/taigrr/mg => ../mg
+
 replace github.com/muesli/termenv => github.com/taigrr/termenv v0.11.2
 
 replace github.com/charmbracelet/lipgloss => github.com/taigrr/lipgloss v0.5.1
@@ -39,6 +41,7 @@ require (
 	github.com/rivo/uniseg v0.4.3 // indirect
 	github.com/sergi/go-diff v1.3.1 // indirect
 	github.com/skeema/knownhosts v1.1.0 // indirect
+	github.com/taigrr/mg v0.0.0-00010101000000-000000000000 // indirect
 	github.com/taigrr/simplecolorpalettes v0.9.6 // indirect
 	github.com/xanzy/ssh-agent v0.3.3 // indirect
 	golang.org/x/crypto v0.5.0 // indirect