Monorepo for Tangled forked from tangled.org/core
Something went wrong. Try again.
3.9 kB · 173 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174package git
import ( "bufio" "context" "errors" "fmt" "io" "strings" "time"
"tangled.org/core/api/tangled"
"github.com/go-git/go-git/v5/plumbing")
type PostReceiveLine struct { OldSha plumbing.Hash // old sha of reference being updated NewSha plumbing.Hash // new sha of reference being updated Ref string // the reference being updated}
func ParsePostReceive(buf io.Reader) ([]PostReceiveLine, error) { scanner := bufio.NewScanner(buf) var lines []PostReceiveLine for scanner.Scan() { line := scanner.Text() parts := strings.SplitN(line, " ", 3) if len(parts) != 3 { continue }
oldSha := parts[0] newSha := parts[1] ref := parts[2]
lines = append(lines, PostReceiveLine{ OldSha: plumbing.NewHash(oldSha), NewSha: plumbing.NewHash(newSha), Ref: ref, }) }
if err := scanner.Err(); err != nil { return nil, err }
return lines, nil}
type RefUpdateMeta struct { CommitCount CommitCount IsDefaultRef bool LangBreakdown LangBreakdown}
type CommitCount struct { ByEmail map[string]int}
func (g *GitRepo) RefUpdateMeta(line PostReceiveLine) (RefUpdateMeta, error) { var errs error
commitCount, err := g.newCommitCount(line) errors.Join(errs, err)
isDefaultRef, err := g.isDefaultBranch(line) errors.Join(errs, err)
ctx, cancel := context.WithTimeout(context.Background(), time.Second*2) defer cancel() breakdown, err := g.AnalyzeLanguages(ctx) errors.Join(errs, err)
return RefUpdateMeta{ CommitCount: commitCount, IsDefaultRef: isDefaultRef, LangBreakdown: breakdown, }, errs}
func (g *GitRepo) newCommitCount(line PostReceiveLine) (CommitCount, error) { byEmail := make(map[string]int) commitCount := CommitCount{ ByEmail: byEmail, }
if line.NewSha.IsZero() { return commitCount, nil }
args := []string{fmt.Sprintf("--max-count=%d", 100)}
if line.OldSha.IsZero() { // git rev-list <newsha> ^other-branches --not ^this-branch args = append(args, line.NewSha.String())
branches, _ := g.Branches() for _, b := range branches { if !strings.Contains(line.Ref, b.Name) { args = append(args, fmt.Sprintf("^%s", b.Name)) } }
args = append(args, "--not") args = append(args, fmt.Sprintf("^%s", line.Ref)) } else { // git rev-list <oldsha>..<newsha> args = append(args, fmt.Sprintf("%s..%s", line.OldSha.String(), line.NewSha.String())) }
output, err := g.revList(args...) if err != nil { return commitCount, fmt.Errorf("failed to run rev-list: %w", err) }
lines := strings.Split(strings.TrimSpace(string(output)), "\n") if len(lines) == 1 && lines[0] == "" { return commitCount, nil }
for _, item := range lines { obj, err := g.r.CommitObject(plumbing.NewHash(item)) if err != nil { continue } commitCount.ByEmail[obj.Author.Email] += 1 }
return commitCount, nil}
func (g *GitRepo) isDefaultBranch(line PostReceiveLine) (bool, error) { defaultBranch, err := g.FindMainBranch() if err != nil { return false, err }
refName := plumbing.ReferenceName(line.Ref) if refName.IsBranch() { return defaultBranch == refName.Short(), nil }
return false, err}
func (m RefUpdateMeta) AsRecord() tangled.GitRefUpdate_Meta { var byEmail []*tangled.GitRefUpdate_IndividualEmailCommitCount for e, v := range m.CommitCount.ByEmail { byEmail = append(byEmail, &tangled.GitRefUpdate_IndividualEmailCommitCount{ Email: e, Count: int64(v), }) }
var langs []*tangled.GitRefUpdate_IndividualLanguageSize for lang, size := range m.LangBreakdown { langs = append(langs, &tangled.GitRefUpdate_IndividualLanguageSize{ Lang: lang, Size: size, }) }
return tangled.GitRefUpdate_Meta{ CommitCount: &tangled.GitRefUpdate_CommitCountBreakdown{ ByEmail: byEmail, }, IsDefaultRef: m.IsDefaultRef, LangBreakdown: &tangled.GitRefUpdate_LangBreakdown{ Inputs: langs, }, }}