gitforgefs/git/client.go

112 lines
3.0 KiB
Go
Raw Normal View History

2020-12-29 02:37:18 +00:00
package git
import (
"context"
"fmt"
2024-06-06 05:51:34 +00:00
"log/slog"
"os"
"path/filepath"
"regexp"
"strconv"
"time"
"github.com/badjware/gitlabfs/config"
"github.com/badjware/gitlabfs/fstree"
"github.com/badjware/gitlabfs/utils"
"github.com/vmihailenco/taskq/v3"
"github.com/vmihailenco/taskq/v3/memqueue"
2020-12-29 02:37:18 +00:00
)
type gitClient struct {
config.GitClientConfig
2024-06-06 05:51:34 +00:00
logger *slog.Logger
hostnameProg *regexp.Regexp
majorVersion int
minorVersion int
patchVersion string
queue taskq.Queue
cloneTask *taskq.Task
pullTask *taskq.Task
2020-12-29 02:37:18 +00:00
}
func NewClient(logger *slog.Logger, p config.GitClientConfig) (*gitClient, error) {
queueFactory := memqueue.NewFactory()
2020-12-29 02:37:18 +00:00
// Create the client
c := &gitClient{
GitClientConfig: p,
2020-12-29 02:37:18 +00:00
2024-06-06 05:51:34 +00:00
logger: logger,
2024-05-08 23:34:29 +00:00
hostnameProg: regexp.MustCompile(`([a-z0-1:\-]+\.)+[a-z0-1:\-]+`),
queue: queueFactory.RegisterQueue(&taskq.QueueOptions{
Name: "git-queue",
MaxNumWorker: int32(p.QueueWorkerCount),
BufferSize: p.QueueSize,
Storage: taskq.NewLocalStorage(),
}),
2020-12-29 02:37:18 +00:00
}
// Parse git version
2024-06-06 05:51:34 +00:00
gitVersionOutput, err := utils.ExecProcess(logger, "git", "--version")
if err != nil {
return nil, fmt.Errorf("failed to run \"git --version\": %v", err)
}
prog := regexp.MustCompile(`([0-9]+)\.([0-9]+)\.(.+)`)
gitVersionMatches := prog.FindStringSubmatch(gitVersionOutput)
c.majorVersion, err = strconv.Atoi(gitVersionMatches[1])
if err != nil {
return nil, fmt.Errorf("failed to parse git major version \"%v\": %v", gitVersionOutput, err)
}
c.minorVersion, err = strconv.Atoi(gitVersionMatches[2])
if err != nil {
return nil, fmt.Errorf("failed to parse git minor version \"%v\": %v", gitVersionOutput, err)
}
c.patchVersion = gitVersionMatches[3]
2024-06-06 05:51:34 +00:00
logger.Info("Detected git version", "major", c.majorVersion, "minor", c.minorVersion, "patch", c.patchVersion)
// Register tasks
c.cloneTask = taskq.RegisterTask(&taskq.TaskOptions{
Name: "git-clone",
Handler: c.clone,
RetryLimit: 1,
})
c.pullTask = taskq.RegisterTask(&taskq.TaskOptions{
Name: "git-pull",
Handler: c.pull,
RetryLimit: 1,
})
2020-12-29 02:37:18 +00:00
return c, nil
}
func (c *gitClient) FetchLocalRepositoryPath(source fstree.RepositorySource) (localRepoLoc string, err error) {
rid := source.GetRepositoryID()
cloneUrl := source.GetCloneURL()
defaultBranch := source.GetDefaultBranch()
// Parse the url
hostname := c.hostnameProg.FindString(cloneUrl)
if hostname == "" {
return "", fmt.Errorf("failed to match a valid hostname from \"%v\"", cloneUrl)
}
localRepoLoc = filepath.Join(c.CloneLocation, hostname, strconv.Itoa(int(rid)))
if _, err := os.Stat(localRepoLoc); os.IsNotExist(err) {
// Dispatch clone msg
msg := c.cloneTask.WithArgs(context.Background(), cloneUrl, defaultBranch, localRepoLoc)
msg.OnceInPeriod(time.Second, rid)
c.queue.Add(msg)
} else if c.AutoPull {
// Dispatch pull msg
msg := c.pullTask.WithArgs(context.Background(), localRepoLoc, defaultBranch)
msg.OnceInPeriod(time.Second, rid)
c.queue.Add(msg)
}
return localRepoLoc, nil
}