package vcs

import (
	"os"

	"go.alanpearce.eu/website/internal/config"
	"go.alanpearce.eu/website/internal/log"

	"github.com/go-git/go-git/v5"
	"github.com/pkg/errors"
)

type Config struct {
	LocalPath string
	RemoteURL config.URL
	Branch    string `conf:"default:main"`
}

type Repository struct {
	repo *git.Repository
}

func CloneOrUpdate(cfg *Config) (*Repository, error) {
	gr, err := git.PlainClone(cfg.LocalPath, false, &git.CloneOptions{
		URL:      cfg.RemoteURL.String(),
		Progress: os.Stdout,
	})
	if err != nil {
		if !errors.Is(err, git.ErrRepositoryAlreadyExists) {
			return nil, err
		}
		gr, err = git.PlainOpen(cfg.LocalPath)
		if err != nil {
			return nil, err
		}
		repo := &Repository{
			repo: gr,
		}
		_, err := repo.Update(cfg)
		if err != nil {
			return nil, err
		}

		return repo, nil
	}

	return &Repository{
		repo: gr,
	}, nil
}

func (r *Repository) Update(cfg *Config) (bool, error) {
	log.Info("updating repository")

	head, err := r.repo.Head()
	if err != nil {
		return false, err
	}

	log.Info("updating from", "rev", head.Hash().String())
	wt, err := r.repo.Worktree()
	if err != nil {
		return false, err
	}
	err = wt.Pull(&git.PullOptions{
		SingleBranch: true,
		Force:        true,
	})
	if err != nil {
		if errors.Is(err, git.NoErrAlreadyUpToDate) {
			log.Info("already up-to-date")

			return true, nil
		}

		return false, err
	}

	head, err = r.repo.Head()
	if err != nil {
		return false, err
	}
	log.Info("updated to", "rev", head.Hash().String())

	return true, r.Clean(wt)
}

func (r *Repository) Clean(wt *git.Worktree) error {
	st, err := wt.Status()
	if err != nil {
		return err
	}

	if !st.IsClean() {
		err = wt.Clean(&git.CleanOptions{
			Dir: true,
		})
		if err != nil {
			return err
		}
	}

	return nil
}