lure-updater/internal/builtins/updater.go

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
	})
}