143 lines
3.0 KiB
Go
143 lines
3.0 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"isle/garage"
|
|
"isle/nebula"
|
|
"path/filepath"
|
|
|
|
"github.com/mediocregopher/mediocre-go-lib/v2/mctx"
|
|
"github.com/mediocregopher/mediocre-go-lib/v2/mlog"
|
|
"github.com/minio/minio-go/v7"
|
|
)
|
|
|
|
// Paths within garage's global bucket
|
|
const (
|
|
garageGlobalBucketBootstrapHostsDirPath = "bootstrap/hosts"
|
|
)
|
|
|
|
// PutGarageBoostrapHost places the <hostname>.json.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 {
|
|
var (
|
|
host = b.ThisHost()
|
|
client = b.GlobalBucketS3APIClient()
|
|
)
|
|
|
|
configured, err := nebula.Sign(
|
|
host.HostConfigured, b.PrivateCredentials.SigningPrivateKey,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("signing host configured data: %w", err)
|
|
}
|
|
|
|
hostB, err := json.Marshal(AuthenticatedHost{
|
|
Assigned: b.SignedHostAssigned,
|
|
Configured: configured,
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("encoding host data: %w", err)
|
|
}
|
|
|
|
filePath := filepath.Join(
|
|
garageGlobalBucketBootstrapHostsDirPath,
|
|
host.Name+".json.signed",
|
|
)
|
|
|
|
_, err = client.PutObject(
|
|
ctx,
|
|
garage.GlobalBucket,
|
|
filePath,
|
|
bytes.NewReader(hostB),
|
|
int64(len(hostB)),
|
|
minio.PutObjectOptions{},
|
|
)
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("writing to %q in global bucket: %w", filePath, err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// RemoveGarageBootstrapHost removes the <hostname>.json.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+".json.signed",
|
|
)
|
|
|
|
return client.RemoveObject(
|
|
ctx, garage.GlobalBucket, filePath,
|
|
minio.RemoveObjectOptions{},
|
|
)
|
|
}
|
|
|
|
// GetGarageBootstrapHosts loads the <hostname>.json.signed file for all hosts
|
|
// stored in garage.
|
|
func (b Bootstrap) GetGarageBootstrapHosts(
|
|
ctx context.Context,
|
|
logger *mlog.Logger,
|
|
) (
|
|
map[string]Host, error,
|
|
) {
|
|
|
|
client := b.GlobalBucketS3APIClient()
|
|
|
|
hosts := map[string]Host{}
|
|
|
|
objInfoCh := client.ListObjects(
|
|
ctx, garage.GlobalBucket,
|
|
minio.ListObjectsOptions{
|
|
Prefix: garageGlobalBucketBootstrapHostsDirPath,
|
|
Recursive: true,
|
|
},
|
|
)
|
|
|
|
for objInfo := range objInfoCh {
|
|
|
|
ctx := mctx.Annotate(ctx, "objectKey", objInfo.Key)
|
|
|
|
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)
|
|
}
|
|
|
|
var authedHost AuthenticatedHost
|
|
|
|
err = json.NewDecoder(obj).Decode(&authedHost)
|
|
obj.Close()
|
|
|
|
if err != nil {
|
|
logger.Warn(ctx, "object contains invalid json", err)
|
|
continue
|
|
}
|
|
|
|
host, err := authedHost.Unwrap(b.CAPublicCredentials)
|
|
if err != nil {
|
|
logger.Warn(ctx, "host could not be authenticated", err)
|
|
}
|
|
|
|
hosts[host.Name] = host
|
|
}
|
|
|
|
return hosts, nil
|
|
}
|