utils: Add helper to get all repos

This does a filepath.WalkDir of the config.ScanPath and tries to find
all valid git repos. It returns a list of repoInfo structs holding basic
information of each repository.
This commit is contained in:
Daniele Sluijters 2023-01-03 14:42:34 +01:00 committed by Anirudh Oppiliappan
parent ee459fce7b
commit ee5ab32a40
1 changed files with 57 additions and 0 deletions

View File

@ -1,8 +1,11 @@
package routes package routes
import ( import (
"io/fs"
"log"
"os" "os"
"path/filepath" "path/filepath"
"strings"
"git.icyphox.sh/legit/git" "git.icyphox.sh/legit/git"
) )
@ -31,3 +34,57 @@ func (d *deps) isIgnored(name string) bool {
return false return false
} }
type repoInfo struct {
Git *git.GitRepo
Path string
Category string
}
func (d *deps) getAllRepos() ([]repoInfo, error) {
repos := []repoInfo{}
max := strings.Count(d.c.Repo.ScanPath, string(os.PathSeparator)) + 2
err := filepath.WalkDir(d.c.Repo.ScanPath, func(path string, de fs.DirEntry, err error) error {
if err != nil {
return err
}
if de.IsDir() {
// Check if we've exceeded our recursion depth
if strings.Count(path, string(os.PathSeparator)) > max {
return fs.SkipDir
}
if d.isIgnored(path) {
return fs.SkipDir
}
// A bare repo should always have at least a HEAD file, if it
// doesn't we can continue recursing
if _, err := os.Lstat(filepath.Join(path, "HEAD")); err == nil {
repo, err := git.Open(path, "")
if err != nil {
log.Println(err)
} else {
relpath, _ := filepath.Rel(d.c.Repo.ScanPath, path)
repos = append(repos, repoInfo{
Git: repo,
Path: relpath,
Category: d.category(path),
})
// Since we found a Git repo, we don't want to recurse
// further
return fs.SkipDir
}
}
}
return nil
})
return repos, err
}
func (d *deps) category(path string) string {
return strings.TrimPrefix(filepath.Dir(strings.TrimPrefix(path, d.c.Repo.ScanPath)), string(os.PathSeparator))
}