package provisioner

import (
	"context"
	"errors"
	"io"
	"strings"

	"github.com/FAU-CDI/wisski-distillery/internal/models"
	"github.com/FAU-CDI/wisski-distillery/internal/wisski/ingredient"
	"github.com/FAU-CDI/wisski-distillery/internal/wisski/ingredient/barrel"
	"github.com/FAU-CDI/wisski-distillery/internal/wisski/ingredient/bookkeeping"
	"github.com/alessio/shellescape"
	"github.com/tkw1536/pkglib/stream"
)

// Provisioner provides provisioning for a barrel
// NOTE(twiesing): This should be refactored to not use the provision script.
// Instead, this should code directly defined in go.
type Provisioner struct {
	ingredient.Base
	Dependencies struct {
		Barrel      *barrel.Barrel
		Bookkeeping *bookkeeping.Bookkeeping
	}
}

// ApplyFlags applies flags to an already provisioned instance.
func (provision *Provisioner) ApplyFlags(ctx context.Context, progress io.Writer, phpversion string) (err error) {
	// setup the new docker image
	provision.Instance.DockerBaseImage, err = models.GetBaseImage(phpversion)
	if err != nil {
		return err
	}

	// save in bookkeeping
	if err := provision.Dependencies.Bookkeeping.Save(ctx); err != nil {
		return err
	}

	return provision.Dependencies.Barrel.Build(ctx, progress, true)
}

// Provision provisions an instance, assuming that the required databases already exist.
func (provision *Provisioner) Provision(ctx context.Context, progress io.Writer) error {

	// build the container
	if err := provision.Dependencies.Barrel.Build(ctx, progress, false); err != nil {
		return err
	}

	provisionParams := []string{
		provision.Domain(),

		provision.SqlDatabase,
		provision.SqlUsername,
		provision.SqlPassword,

		provision.GraphDBRepository,
		provision.GraphDBUsername,
		provision.GraphDBPassword,

		provision.DrupalUsername,
		provision.DrupalPassword,

		"", // TODO: DrupalVersion
		"", // TODO: WissKIVersion
	}

	// escape the parameter
	for i, param := range provisionParams {
		provisionParams[i] = shellescape.Quote(param)
	}

	// figure out the provision script
	// TODO: Move the provision script into the control plane!
	provisionScript := "sudo PATH=$PATH -u www-data /bin/bash /provision_container.sh " + strings.Join(provisionParams, " ")

	code, err := provision.Dependencies.Barrel.Stack().Run(ctx, stream.NonInteractive(progress), true, "barrel", "/bin/bash", "-c", provisionScript)
	if err != nil {
		return err
	}
	if code != 0 {
		return errors.New("unable to run provision script")
	}

	return nil
}