Initial commit.
This commit is contained in:
251
internal/instance/instance.go
Normal file
251
internal/instance/instance.go
Normal file
@@ -0,0 +1,251 @@
|
||||
package instance
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/wild-cloud/wild-central/daemon/internal/config"
|
||||
"github.com/wild-cloud/wild-central/daemon/internal/context"
|
||||
"github.com/wild-cloud/wild-central/daemon/internal/secrets"
|
||||
"github.com/wild-cloud/wild-central/daemon/internal/storage"
|
||||
)
|
||||
|
||||
// Manager handles instance lifecycle operations
|
||||
type Manager struct {
|
||||
dataDir string
|
||||
configMgr *config.Manager
|
||||
secretsMgr *secrets.Manager
|
||||
contextMgr *context.Manager
|
||||
}
|
||||
|
||||
// NewManager creates a new instance manager
|
||||
func NewManager(dataDir string) *Manager {
|
||||
return &Manager{
|
||||
dataDir: dataDir,
|
||||
configMgr: config.NewManager(),
|
||||
secretsMgr: secrets.NewManager(),
|
||||
contextMgr: context.NewManager(dataDir),
|
||||
}
|
||||
}
|
||||
|
||||
// Instance represents a Wild Cloud instance
|
||||
type Instance struct {
|
||||
Name string
|
||||
Path string
|
||||
ConfigPath string
|
||||
SecretsPath string
|
||||
}
|
||||
|
||||
// GetInstancePath returns the path to an instance directory
|
||||
func (m *Manager) GetInstancePath(name string) string {
|
||||
return filepath.Join(m.dataDir, "instances", name)
|
||||
}
|
||||
|
||||
// GetInstanceConfigPath returns the path to an instance's config file
|
||||
func (m *Manager) GetInstanceConfigPath(name string) string {
|
||||
return filepath.Join(m.GetInstancePath(name), "config.yaml")
|
||||
}
|
||||
|
||||
// GetInstanceSecretsPath returns the path to an instance's secrets file
|
||||
func (m *Manager) GetInstanceSecretsPath(name string) string {
|
||||
return filepath.Join(m.GetInstancePath(name), "secrets.yaml")
|
||||
}
|
||||
|
||||
// InstanceExists checks if an instance exists
|
||||
func (m *Manager) InstanceExists(name string) bool {
|
||||
return storage.FileExists(m.GetInstancePath(name))
|
||||
}
|
||||
|
||||
// CreateInstance creates a new Wild Cloud instance with initial structure
|
||||
func (m *Manager) CreateInstance(name string) error {
|
||||
if name == "" {
|
||||
return fmt.Errorf("instance name cannot be empty")
|
||||
}
|
||||
|
||||
instancePath := m.GetInstancePath(name)
|
||||
|
||||
// Check if instance already exists (idempotency - just return success)
|
||||
if m.InstanceExists(name) {
|
||||
return nil
|
||||
}
|
||||
|
||||
// Acquire lock for instance creation
|
||||
lockPath := filepath.Join(m.dataDir, "instances", ".lock")
|
||||
return storage.WithLock(lockPath, func() error {
|
||||
// Create instance directory
|
||||
if err := storage.EnsureDir(instancePath, 0755); err != nil {
|
||||
return fmt.Errorf("creating instance directory: %w", err)
|
||||
}
|
||||
|
||||
// Create config file
|
||||
if err := m.configMgr.EnsureInstanceConfig(instancePath); err != nil {
|
||||
return fmt.Errorf("creating config file: %w", err)
|
||||
}
|
||||
|
||||
// Create secrets file
|
||||
if err := m.secretsMgr.EnsureSecretsFile(instancePath); err != nil {
|
||||
return fmt.Errorf("creating secrets file: %w", err)
|
||||
}
|
||||
|
||||
// Create subdirectories
|
||||
subdirs := []string{"talos", "k8s", "logs", "backups"}
|
||||
for _, subdir := range subdirs {
|
||||
subdirPath := filepath.Join(instancePath, subdir)
|
||||
if err := storage.EnsureDir(subdirPath, 0755); err != nil {
|
||||
return fmt.Errorf("creating subdirectory %s: %w", subdir, err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
// DeleteInstance removes a Wild Cloud instance
|
||||
func (m *Manager) DeleteInstance(name string) error {
|
||||
if name == "" {
|
||||
return fmt.Errorf("instance name cannot be empty")
|
||||
}
|
||||
|
||||
instancePath := m.GetInstancePath(name)
|
||||
|
||||
// Check if instance exists
|
||||
if !m.InstanceExists(name) {
|
||||
return fmt.Errorf("instance %s does not exist", name)
|
||||
}
|
||||
|
||||
// Clear context if this is the current instance
|
||||
currentContext, err := m.contextMgr.GetCurrentContext()
|
||||
if err == nil && currentContext == name {
|
||||
if err := m.contextMgr.ClearCurrentContext(); err != nil {
|
||||
return fmt.Errorf("clearing current context: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Acquire lock for instance deletion
|
||||
lockPath := filepath.Join(m.dataDir, "instances", ".lock")
|
||||
return storage.WithLock(lockPath, func() error {
|
||||
// Remove instance directory
|
||||
if err := os.RemoveAll(instancePath); err != nil {
|
||||
return fmt.Errorf("removing instance directory: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
// ListInstances returns a list of all instance names
|
||||
func (m *Manager) ListInstances() ([]string, error) {
|
||||
instancesDir := filepath.Join(m.dataDir, "instances")
|
||||
|
||||
// Ensure instances directory exists
|
||||
if !storage.FileExists(instancesDir) {
|
||||
return []string{}, nil
|
||||
}
|
||||
|
||||
entries, err := os.ReadDir(instancesDir)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("reading instances directory: %w", err)
|
||||
}
|
||||
|
||||
var instances []string
|
||||
for _, entry := range entries {
|
||||
if entry.IsDir() && entry.Name() != ".lock" {
|
||||
instances = append(instances, entry.Name())
|
||||
}
|
||||
}
|
||||
|
||||
return instances, nil
|
||||
}
|
||||
|
||||
// GetInstance retrieves instance information
|
||||
func (m *Manager) GetInstance(name string) (*Instance, error) {
|
||||
if !m.InstanceExists(name) {
|
||||
return nil, fmt.Errorf("instance %s does not exist", name)
|
||||
}
|
||||
|
||||
return &Instance{
|
||||
Name: name,
|
||||
Path: m.GetInstancePath(name),
|
||||
ConfigPath: m.GetInstanceConfigPath(name),
|
||||
SecretsPath: m.GetInstanceSecretsPath(name),
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetCurrentInstance returns the current context instance
|
||||
func (m *Manager) GetCurrentInstance() (*Instance, error) {
|
||||
name, err := m.contextMgr.GetCurrentContext()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return m.GetInstance(name)
|
||||
}
|
||||
|
||||
// SetCurrentInstance sets the current instance context
|
||||
func (m *Manager) SetCurrentInstance(name string) error {
|
||||
if !m.InstanceExists(name) {
|
||||
return fmt.Errorf("instance %s does not exist", name)
|
||||
}
|
||||
|
||||
return m.contextMgr.SetCurrentContext(name)
|
||||
}
|
||||
|
||||
// ValidateInstance checks if an instance has valid structure
|
||||
func (m *Manager) ValidateInstance(name string) error {
|
||||
if !m.InstanceExists(name) {
|
||||
return fmt.Errorf("instance %s does not exist", name)
|
||||
}
|
||||
|
||||
instance, err := m.GetInstance(name)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Check config file exists and is valid
|
||||
if !storage.FileExists(instance.ConfigPath) {
|
||||
return fmt.Errorf("config file missing for instance %s", name)
|
||||
}
|
||||
|
||||
if err := m.configMgr.ValidateConfig(instance.ConfigPath); err != nil {
|
||||
return fmt.Errorf("invalid config for instance %s: %w", name, err)
|
||||
}
|
||||
|
||||
// Check secrets file exists with proper permissions
|
||||
if !storage.FileExists(instance.SecretsPath) {
|
||||
return fmt.Errorf("secrets file missing for instance %s", name)
|
||||
}
|
||||
|
||||
// Verify secrets file permissions
|
||||
info, err := os.Stat(instance.SecretsPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("checking secrets file permissions: %w", err)
|
||||
}
|
||||
|
||||
if info.Mode().Perm() != 0600 {
|
||||
return fmt.Errorf("secrets file has incorrect permissions (expected 0600, got %04o)", info.Mode().Perm())
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// InitializeInstance performs initial setup for a newly created instance
|
||||
func (m *Manager) InitializeInstance(name string, initialConfig map[string]string) error {
|
||||
if !m.InstanceExists(name) {
|
||||
return fmt.Errorf("instance %s does not exist", name)
|
||||
}
|
||||
|
||||
instance, err := m.GetInstance(name)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Set initial config values
|
||||
for key, value := range initialConfig {
|
||||
if err := m.configMgr.SetConfigValue(instance.ConfigPath, key, value); err != nil {
|
||||
return fmt.Errorf("setting config value %s: %w", key, err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
176
internal/instance/instance_test.go
Normal file
176
internal/instance/instance_test.go
Normal file
@@ -0,0 +1,176 @@
|
||||
package instance
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestManager_CreateInstance(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
m := NewManager(tmpDir)
|
||||
|
||||
instanceName := "test-cloud"
|
||||
|
||||
// Create instance
|
||||
err := m.CreateInstance(instanceName)
|
||||
if err != nil {
|
||||
t.Fatalf("CreateInstance failed: %v", err)
|
||||
}
|
||||
|
||||
// Verify instance directory structure
|
||||
instancePath := m.GetInstancePath(instanceName)
|
||||
expectedDirs := []string{
|
||||
instancePath,
|
||||
filepath.Join(instancePath, "talos"),
|
||||
filepath.Join(instancePath, "k8s"),
|
||||
filepath.Join(instancePath, "logs"),
|
||||
filepath.Join(instancePath, "backups"),
|
||||
}
|
||||
|
||||
for _, dir := range expectedDirs {
|
||||
info, err := os.Stat(dir)
|
||||
if err != nil {
|
||||
t.Errorf("Directory not created: %s: %v", dir, err)
|
||||
continue
|
||||
}
|
||||
if !info.IsDir() {
|
||||
t.Errorf("Path is not a directory: %s", dir)
|
||||
}
|
||||
}
|
||||
|
||||
// Verify config.yaml exists
|
||||
configPath := m.GetInstanceConfigPath(instanceName)
|
||||
if _, err := os.Stat(configPath); err != nil {
|
||||
t.Errorf("Config file not created: %v", err)
|
||||
}
|
||||
|
||||
// Verify secrets.yaml exists with correct permissions
|
||||
secretsPath := m.GetInstanceSecretsPath(instanceName)
|
||||
info, err := os.Stat(secretsPath)
|
||||
if err != nil {
|
||||
t.Errorf("Secrets file not created: %v", err)
|
||||
} else {
|
||||
// Check permissions (should be 0600)
|
||||
mode := info.Mode().Perm()
|
||||
if mode != 0600 {
|
||||
t.Errorf("Secrets file has wrong permissions: got %o, want 0600", mode)
|
||||
}
|
||||
}
|
||||
|
||||
// Test idempotency - creating again should not error
|
||||
err = m.CreateInstance(instanceName)
|
||||
if err != nil {
|
||||
t.Fatalf("CreateInstance not idempotent: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestManager_ListInstances(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
m := NewManager(tmpDir)
|
||||
|
||||
// Initially should be empty
|
||||
instances, err := m.ListInstances()
|
||||
if err != nil {
|
||||
t.Fatalf("ListInstances failed: %v", err)
|
||||
}
|
||||
if len(instances) != 0 {
|
||||
t.Fatalf("Expected 0 instances, got %d", len(instances))
|
||||
}
|
||||
|
||||
// Create instances
|
||||
instanceNames := []string{"cloud1", "cloud2", "cloud3"}
|
||||
for _, name := range instanceNames {
|
||||
err := m.CreateInstance(name)
|
||||
if err != nil {
|
||||
t.Fatalf("CreateInstance failed: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// List should return all instances
|
||||
instances, err = m.ListInstances()
|
||||
if err != nil {
|
||||
t.Fatalf("ListInstances failed: %v", err)
|
||||
}
|
||||
if len(instances) != len(instanceNames) {
|
||||
t.Fatalf("Expected %d instances, got %d", len(instanceNames), len(instances))
|
||||
}
|
||||
|
||||
// Verify all expected instances are present
|
||||
instanceMap := make(map[string]bool)
|
||||
for _, name := range instances {
|
||||
instanceMap[name] = true
|
||||
}
|
||||
for _, expected := range instanceNames {
|
||||
if !instanceMap[expected] {
|
||||
t.Errorf("Expected instance %q not found", expected)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestManager_DeleteInstance(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
m := NewManager(tmpDir)
|
||||
|
||||
instanceName := "test-cloud"
|
||||
|
||||
// Create instance
|
||||
err := m.CreateInstance(instanceName)
|
||||
if err != nil {
|
||||
t.Fatalf("CreateInstance failed: %v", err)
|
||||
}
|
||||
|
||||
// Verify it exists (by checking directory)
|
||||
instancePath := m.GetInstancePath(instanceName)
|
||||
if _, err := os.Stat(instancePath); err != nil {
|
||||
t.Fatalf("Instance should exist: %v", err)
|
||||
}
|
||||
|
||||
// Delete instance
|
||||
err = m.DeleteInstance(instanceName)
|
||||
if err != nil {
|
||||
t.Fatalf("DeleteInstance failed: %v", err)
|
||||
}
|
||||
|
||||
// Verify it's gone
|
||||
err = m.ValidateInstance(instanceName)
|
||||
if err == nil {
|
||||
t.Fatalf("Instance should not exist after deletion")
|
||||
}
|
||||
|
||||
// Deleting non-existent instance should error
|
||||
err = m.DeleteInstance(instanceName)
|
||||
if err == nil {
|
||||
t.Fatalf("Deleting non-existent instance should error")
|
||||
}
|
||||
}
|
||||
|
||||
func TestManager_ValidateInstance(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
m := NewManager(tmpDir)
|
||||
|
||||
instanceName := "test-cloud"
|
||||
|
||||
// Should fail for non-existent instance
|
||||
err := m.ValidateInstance(instanceName)
|
||||
if err == nil {
|
||||
t.Fatalf("ValidateInstance should fail for non-existent instance")
|
||||
}
|
||||
|
||||
// Create instance
|
||||
err = m.CreateInstance(instanceName)
|
||||
if err != nil {
|
||||
t.Fatalf("CreateInstance failed: %v", err)
|
||||
}
|
||||
|
||||
// Should succeed for existing instance (if yq is available)
|
||||
// Note: ValidateInstance requires yq for config validation
|
||||
err = m.ValidateInstance(instanceName)
|
||||
if err != nil {
|
||||
// It's OK if yq is not installed, just check instance exists
|
||||
if !m.InstanceExists(instanceName) {
|
||||
t.Fatalf("Instance should exist after creation")
|
||||
}
|
||||
t.Logf("ValidateInstance failed (likely yq not installed): %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user