summaryrefslogtreecommitdiff
path: root/stat.makeRemoteRefs.go
blob: b1a5736f0c448c3546eb0c20b9493f97d9c44fa5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
package gitpb

import (
	"errors"
	"fmt"
	"path/filepath"
	"slices"
	"strings"

	"go.wit.com/lib/config"
	"go.wit.com/lib/env"
	"go.wit.com/log"
)

// returns err if anything changes or anything is wrong (todo: should these be different?)
func (r *Repo) LoadRemoteRefs(remoteName string) (*Stats, error) {
	fullname := filepath.Join(r.FullPath, ".git", remoteName+".refs.pb")
	stats := NewStats()
	stats.Filename = fullname
	err := config.ForceCreatePB(stats)
	return stats, err
}

// returns err if anything changes or anything is wrong (todo: should these be different?)
func (r *Repo) MakeRemoteRefs(remoteName string) (*Stats, error) {
	stats, err := r.LoadRemoteRefs(remoteName)
	if err != nil {
		return stats, err
	}
	if stats.Len() != 0 {
		return stats, errors.New("refs file was already created")
	}
	cmd := []string{"git", "ls-remote", remoteName} // must use 'master' as queried from the git server // GO has 250k remote refs
	if env.True("stats") {
		log.Info("STATS VERBOSE Run:", cmd)
	}
	cmdout := r.Run(cmd)
	if len(cmdout.Stdout) == 0 {
		return stats, errors.New("got nothing from git ls-remote")
	}
	var counter int
	for _, line := range cmdout.Stdout {
		line = strings.TrimSpace(line)
		parts := strings.Fields(line)
		if len(parts) != 2 {
			log.Printf("Repo: %s\n", r.FullPath)
			log.Printf("CMD: %v\n", cmd)
			log.Printf("LINE:%s\n", line)
			return stats, errors.New(line)
		}
		if env.True("stats") {
			log.Printf("LINE:%v %d %s\n", parts, counter, r.FullPath)
		}
		counter += 1
		newstat := new(Stat)
		newstat.Hash = parts[0]
		stats.Append(newstat)
	}
	if counter > 0 {
		stats.SaveByHash()
	}
	return stats, nil
}

func (x *Stats) SaveByHash() error {
	newx := NewStats()

	all := x.SortByHash()
	for all.Scan() {
		r := all.Next()
		newx.Clone(r)
	}

	newx.Filename = x.Filename
	err := newx.Save()

	return err
}

// returns err if anything changes or anything is wrong (todo: should these be different?)
func (r *Repo) UpdateRemoteRefs(remoteName string) error {
	stats, err := r.LoadRemoteRefs(remoteName)
	if err != nil {
		return err
	}
	if stats.Len() == 0 {
		return errors.New("need to make file instead")
	}
	cmd := []string{"git", "ls-remote", remoteName} // must use 'master' as queried from the git server // GO has 250k remote refs
	if env.True("stats") {
		log.Info("STATS VERBOSE Run:", cmd)
	}
	cmdout := r.Run(cmd)
	if len(cmdout.Stdout) == 0 {
		return errors.New("got nothing back")
	}
	var ticker int
	var counter int
	var allnew []*Stat
	for _, line := range cmdout.Stdout {
		line = strings.TrimSpace(line)
		parts := strings.Fields(line)
		if len(parts) != 2 {
			log.Printf("Repo: %s\n", r.FullPath)
			log.Printf("CMD: %v\n", cmd)
			log.Printf("LINE:%s\n", line)
			return errors.New(line)
		}
		if env.True("stats") {
			log.Printf("LINE:%v %d %s\n", parts, ticker, r.FullPath)
		}
		// refpath := filepath.Join("refs/remote", remoteName)
		// if strings.HasPrefix(parts[1], refpath) {
		// this ref is not from the remote
		// 	continue
		// }
		counter += 1
		newstat := new(Stat)
		newstat.Hash = parts[0]
		n, found := slices.BinarySearchFunc(stats.Stats, newstat, func(a, b *Stat) int {
			return strings.Compare(a.Hash, b.Hash)
		})
		_ = n

		if found {
			// log.Info("found", n)
			continue
		}
		ticker += 1
		log.Info("not found", newstat)
		allnew = append(allnew, newstat)
	}
	if ticker > 0 {
		stats.Stats = append(stats.Stats, allnew...)
		stats.SaveByHash()
		return errors.New(fmt.Sprintf("len(%d), ticker(%d) counter(%d) refs changed", stats.Len(), ticker, counter))
	}
	return nil
}