package bootstrap

import (
	"bytes"
	"context"
	"cryptic-net/garage"
	"cryptic-net/nebula"
	"fmt"
	"os"
	"path/filepath"

	"github.com/minio/minio-go/v7"
	"gopkg.in/yaml.v3"
)

// Paths within garage's global bucket
const (
	garageGlobalBucketBootstrapHostsDirPath = "bootstrap/hosts"
)

// PutGarageBoostrapHost places the <hostname>.yml.signed file for this host
// into garage so that other hosts are able to see relevant configuration for
// it.
func (b Bootstrap) PutGarageBoostrapHost(ctx context.Context) error {

	host := b.ThisHost()
	client := b.GlobalBucketS3APIClient()

	hostB, err := yaml.Marshal(host)
	if err != nil {
		return fmt.Errorf("yaml encoding host data: %w", err)
	}

	buf := new(bytes.Buffer)

	err = nebula.SignAndWrap(buf, b.Nebula.HostCredentials.HostKeyPEM, hostB)
	if err != nil {
		return fmt.Errorf("signing encoded host data: %w", err)
	}

	filePath := filepath.Join(
		garageGlobalBucketBootstrapHostsDirPath,
		host.Name+".yml.signed",
	)

	_, err = client.PutObject(
		ctx, garage.GlobalBucket, filePath, buf, int64(buf.Len()),
		minio.PutObjectOptions{},
	)

	if err != nil {
		return fmt.Errorf("writing to %q in global bucket: %w", filePath, err)
	}

	return nil
}

// RemoveGarageBootstrapHost removes the <hostname>.yml.signed for the given
// host from garage.
//
// The given client should be for the global bucket.
func RemoveGarageBootstrapHost(
	ctx context.Context, client garage.S3APIClient, hostName string,
) error {

	filePath := filepath.Join(
		garageGlobalBucketBootstrapHostsDirPath,
		hostName+".yml.signed",
	)

	return client.RemoveObject(
		ctx, garage.GlobalBucket, filePath,
		minio.RemoveObjectOptions{},
	)
}

// GetGarageBootstrapHosts loads the <hostname>.yml.signed file for all hosts
// stored in garage.
func (b Bootstrap) GetGarageBootstrapHosts(
	ctx context.Context,
) (
	map[string]Host, error,
) {

	caCertPEM := b.Nebula.HostCredentials.CACertPEM
	client := b.GlobalBucketS3APIClient()

	hosts := map[string]Host{}

	objInfoCh := client.ListObjects(
		ctx, garage.GlobalBucket,
		minio.ListObjectsOptions{
			Prefix:    garageGlobalBucketBootstrapHostsDirPath,
			Recursive: true,
		},
	)

	for objInfo := range objInfoCh {

		if objInfo.Err != nil {
			return nil, fmt.Errorf("listing objects: %w", objInfo.Err)
		}

		obj, err := client.GetObject(
			ctx, garage.GlobalBucket, objInfo.Key, minio.GetObjectOptions{},
		)

		if err != nil {
			return nil, fmt.Errorf("retrieving object %q: %w", objInfo.Key, err)
		}

		hostB, sig, err := nebula.Unwrap(obj)
		obj.Close()

		if err != nil {
			return nil, fmt.Errorf("unwrapping signature from %q: %w", objInfo.Key, err)
		}

		var host Host
		if err = yaml.Unmarshal(hostB, &host); err != nil {
			return nil, fmt.Errorf("yaml decoding object %q: %w", objInfo.Key, err)
		}

		hostCertPEM := host.Nebula.CertPEM

		if err := nebula.ValidateSignature(hostCertPEM, hostB, sig); err != nil {
			fmt.Fprintf(os.Stderr, "invalid host data for %q: %v\n", objInfo.Key, err)
			continue
		}

		if err := nebula.ValidateHostCertPEM(caCertPEM, hostCertPEM); err != nil {
			fmt.Fprintf(os.Stderr, "invalid nebula cert for %q: %v\n", objInfo.Key, err)
			continue
		}

		hosts[host.Name] = host
	}

	return hosts, nil
}