128 lines
2.8 KiB
Go
128 lines
2.8 KiB
Go
package forgepb
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/destel/rill"
|
|
"go.wit.com/lib/protobuf/gitpb"
|
|
"go.wit.com/log"
|
|
)
|
|
|
|
func (f *Forge) ScanGoSrc() (bool, error) {
|
|
dirs, err := gitDirectories(f.goSrc)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
var gopaths []string
|
|
for _, dir := range dirs {
|
|
if strings.HasPrefix(dir, f.goSrc) {
|
|
gopath := strings.TrimPrefix(dir, f.goSrc)
|
|
gopath = strings.Trim(gopath, "/")
|
|
gopaths = append(gopaths, gopath)
|
|
} else {
|
|
log.Log(FORGEPBWARN, "ScanGoSrc() bad:", dir)
|
|
return false, errors.New("forgepb.ScanGoSrc() bad dir: " + dir)
|
|
}
|
|
}
|
|
f.rillScanDirs(gopaths)
|
|
|
|
/*
|
|
for _, dir := range dirs {
|
|
if strings.HasPrefix(dir, f.goSrc) {
|
|
gopath := strings.TrimPrefix(dir, f.goSrc)
|
|
gopath = strings.Trim(gopath, "/")
|
|
repo, err := f.Repos.NewGoPath(f.goSrc, gopath)
|
|
} else {
|
|
log.Log(FORGEPBWARN, "ScanGoSrc() bad:", dir)
|
|
}
|
|
}
|
|
*/
|
|
return true, err
|
|
}
|
|
|
|
func gitDirectories(srcDir string) ([]string, error) {
|
|
var all []string
|
|
err := filepath.Walk(srcDir, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
log.Log(FORGEPBWARN, "Error accessing path:", path, err)
|
|
return nil
|
|
}
|
|
|
|
// Check if the path is a directory and has a .git subdirectory
|
|
if info.IsDir() && IsGitDir(path) {
|
|
all = append(all, path)
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
log.Log(FORGEPBWARN, "Error walking the path:", srcDir, err)
|
|
}
|
|
|
|
return all, err
|
|
}
|
|
|
|
// IsGitDir checks if a .git directory exists inside the given directory
|
|
func IsGitDir(dir string) bool {
|
|
gitDir := filepath.Join(dir, ".git")
|
|
info, err := os.Stat(gitDir)
|
|
if os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
return info.IsDir()
|
|
}
|
|
|
|
/*
|
|
// rill is awesome. long live rill
|
|
func rillAddDirs(gopaths []string) {
|
|
// Convert a slice of user IDs into a channel
|
|
ids := rill.FromSlice(gopaths, nil)
|
|
|
|
// Read users from the API.
|
|
// Concurrency = 20
|
|
dirs := rill.Map(ids, 20, func(id string) (*repolist.RepoRow, error) {
|
|
return me.repos.View.FindByName(id), nil
|
|
})
|
|
|
|
// Activate users.
|
|
// Concurrency = 10
|
|
err := rill.ForEach(dirs, 10, func(repo *repolist.RepoRow) error {
|
|
fmt.Printf("Repo found : %s\n", repo.GoPath())
|
|
repo.Run([]string{"git", "pull"})
|
|
return nil
|
|
})
|
|
|
|
// Handle errors
|
|
fmt.Println("Error:", err)
|
|
}
|
|
*/
|
|
|
|
// rill is awesome. long live rill
|
|
// attempt scan with rill
|
|
func (f *Forge) rillScanDirs(gopaths []string) error {
|
|
// Convert a slice of user IDs into a channel
|
|
ids := rill.FromSlice(gopaths, nil)
|
|
|
|
// Read users from the API.
|
|
// Concurrency = 20
|
|
dirs := rill.Map(ids, 20, func(id string) (*gitpb.Repo, error) {
|
|
return f.Repos.NewGoPath(f.goSrc, id)
|
|
})
|
|
|
|
// Activate users.
|
|
// Concurrency = 10
|
|
err := rill.ForEach(dirs, 10, func(repo *gitpb.Repo) error {
|
|
// could do something here
|
|
// fmt.Printf("Repo found : %s\n", repo.GoPath)
|
|
// repo.Run([]string{"git", "pull"})
|
|
return nil
|
|
})
|
|
|
|
return err
|
|
}
|