return the config when initializing

Author: jahvon

vault.go

@@ -23,24 +23,26 @@ type Provider interface {
 type Option func(*Config)
 
 // New creates a new vault instance with the provided ID and options
-func New(id string, opts ...Option) (Provider, error) {
+func New(id string, opts ...Option) (Provider, *Config, error) {
 	config := &Config{ID: id}
 	for _, opt := range opts {
 		opt(config)
 	}
 	if err := config.Validate(); err != nil {
-		return nil, err
+		return nil, config, err
 	}
 
 	switch config.Type {
 	case ProviderTypeAge:
-		return NewAgeVault(config)
+		provider, err := NewAgeVault(config)
+		return provider, config, err
 	case ProviderTypeAES256:
-		return NewAES256Vault(config)
+		provider, err := NewAES256Vault(config)
+		return provider, config, err
 	case ProviderTypeExternal:
-		return nil, fmt.Errorf("external vault provider not implemented yet")
+		return nil, nil, fmt.Errorf("external vault provider not implemented yet")
 	}
-	return nil, fmt.Errorf("unsupported vault type: %s", config.Type)
+	return nil, nil, fmt.Errorf("unsupported vault type: %s", config.Type)
 }
 
 // WithProvider sets the vault provider type

vault_test.go

@@ -53,7 +53,7 @@ func setupAESVault(t *testing.T, dir string) vault.Provider {
 		t.Setenv(vault.DefaultVaultKeyEnv, key)
 	}
 
-	v, err := vault.New("test-aes",
+	v, _, err := vault.New("test-aes",
 		vault.WithProvider(vault.ProviderTypeAES256),
 		vault.WithAESPath(dir),
 		vault.WithAESKeyFromEnv(vault.DefaultVaultKeyEnv),
@@ -75,7 +75,7 @@ func setupAgeVault(t *testing.T, dir string) vault.Provider {
 		t.Fatalf("Failed to write test key file: %v", err)
 	}
 
-	v, err := vault.New("test-age",
+	v, _, err := vault.New("test-age",
 		vault.WithProvider(vault.ProviderTypeAge),
 		vault.WithAgePath(dir),
 		vault.WithAgeIdentityFromFile(keyFile),