package builtins import ( "io" "os" "path/filepath" "strings" "sync" "time" "github.com/go-git/go-git/v5" "github.com/go-git/go-git/v5/plumbing/object" "github.com/go-git/go-git/v5/plumbing/transport/http" "go.elara.ws/logger/log" "go.elara.ws/lure-updater/internal/config" "go.starlark.net/starlark" "go.starlark.net/starlarkstruct" ) func updaterModule(cfg *config.Config) *starlarkstruct.Module { return &starlarkstruct.Module{ Name: "updater", Members: starlark.StringDict{ "repo_dir": starlark.String(cfg.Git.RepoDir), "pull": updaterPull(cfg), "push_changes": updaterPushChanges(cfg), "get_package_file": getPackageFile(cfg), "write_package_file": writePackageFile(cfg), }, } } // repoMtx makes sure two starlark threads can // never access the repo at the same time var repoMtx = &sync.Mutex{} func updaterPull(cfg *config.Config) *starlark.Builtin { return starlark.NewBuiltin("updater.pull", func(thread *starlark.Thread, fn *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) { repoMtx.Lock() defer repoMtx.Unlock() repo, err := git.PlainOpen(cfg.Git.RepoDir) if err != nil { return nil, err } w, err := repo.Worktree() if err != nil { return nil, err } err = w.Pull(&git.PullOptions{Progress: os.Stderr}) if err != git.NoErrAlreadyUpToDate && err != nil { return nil, err } return starlark.None, nil }) } func updaterPushChanges(cfg *config.Config) *starlark.Builtin { return starlark.NewBuiltin("updater.push_changes", func(thread *starlark.Thread, b *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) { var msg string err := starlark.UnpackArgs("updater.push_changes", args, kwargs, "msg", &msg) if err != nil { return nil, err } repoMtx.Lock() defer repoMtx.Unlock() repo, err := git.PlainOpen(cfg.Git.RepoDir) if err != nil { return nil, err } w, err := repo.Worktree() if err != nil { return nil, err } status, err := w.Status() if err != nil { return nil, err } if status.IsClean() { return starlark.None, nil } err = w.Pull(&git.PullOptions{Progress: os.Stderr}) if err != git.NoErrAlreadyUpToDate && err != nil { return nil, err } _, err = w.Add(".") if err != nil { return nil, err } sig := &object.Signature{ Name: cfg.Git.Commit.Name, Email: cfg.Git.Commit.Email, When: time.Now(), } h, err := w.Commit(msg, &git.CommitOptions{ Author: sig, Committer: sig, }) if err != nil { return nil, err } log.Debug("Created new commit").Stringer("hash", h).Stringer("pos", thread.CallFrame(1).Pos).Send() err = repo.Push(&git.PushOptions{ Progress: os.Stderr, Auth: &http.BasicAuth{ Username: cfg.Git.Credentials.Username, Password: cfg.Git.Credentials.Password, }, }) if err != nil { return nil, err } log.Debug("Successfully pushed to repo").Stringer("pos", thread.CallFrame(1).Pos).Send() return starlark.None, nil }) } func getPackageFile(cfg *config.Config) *starlark.Builtin { return starlark.NewBuiltin("updater.get_package_file", func(thread *starlark.Thread, b *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) { var pkg, filename string err := starlark.UnpackArgs("updater.get_package_file", args, kwargs, "pkg", &pkg, "filename", &filename) if err != nil { return nil, err } repoMtx.Lock() defer repoMtx.Unlock() path := filepath.Join(cfg.Git.RepoDir, pkg, filename) data, err := os.ReadFile(path) if err != nil { return nil, err } log.Debug("Got package file").Str("package", pkg).Str("filename", filename).Stringer("pos", thread.CallFrame(1).Pos).Send() return starlark.String(data), nil }) } func writePackageFile(cfg *config.Config) *starlark.Builtin { return starlark.NewBuiltin("updater.write_package_file", func(thread *starlark.Thread, b *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) { var pkg, filename, content string err := starlark.UnpackArgs("updater.write_package_file", args, kwargs, "pkg", &pkg, "filename", &filename, "content", &content) if err != nil { return nil, err } repoMtx.Lock() defer repoMtx.Unlock() path := filepath.Join(cfg.Git.RepoDir, pkg, filename) fl, err := os.Create(path) if err != nil { return nil, err } _, err = io.Copy(fl, strings.NewReader(content)) if err != nil { return nil, err } log.Debug("Wrote package file").Str("package", pkg).Str("filename", filename).Stringer("pos", thread.CallFrame(1).Pos).Send() return starlark.None, nil }) }