deps.go

package config

import (
	"context"
	"fmt"

	"github.com/magefile/mage/sh"
	"github.com/wavesoftware/go-magetasks/pkg/tasks"
)

type Dependencies interface {
	Configurator
	Install(ctx context.Context, t *tasks.Task, dest string) error
	Count() int

	merge(other dependencies)
	installs() []string
}

func NewDependencies(deps ...string) Dependencies {
	s := make(map[string]bool, len(deps))
	for _, dep := range deps {
		s[dep] = exists
	}
	return dependencies{
		set: s,
	}
}

const exists = true

type dependencies struct {
	set map[string]bool
}

func (d dependencies) Count() int {
	return len(d.set)
}

func (d dependencies) Install(_ context.Context, t *tasks.Task, dest string) error {
	for _, dep := range d.installs() {
		pp := t.Part(fmt.Sprintf("Go install %q", dep)).Starting()
		env := map[string]string{
			"GOBIN": dest,
		}
		if err := sh.RunWith(env, "go", "install", dep); err != nil {
			pp.Done(err)
			return err
		}
		pp.Notify("installed")
	}
	return nil
}

func (d dependencies) installs() []string {
	keys := make([]string, len(d.set))

	i := 0
	for k := range d.set {
		keys[i] = k
		i++
	}

	return keys
}

func (d dependencies) Configure(cfg Configurable) {
	cfg.Config().Dependencies.Golang.merge(d)
}

func (d dependencies) merge(other dependencies) {
	for _, dep := range other.installs() {
		d.set[dep] = exists
	}
}