package main

import (
    "fmt"
    "github.com/hallucinaut/keyvault/pkg/lifecycle"
    "github.com/hallucinaut/keyvault/pkg/storage"
    "github.com/hallucinaut/keyvault/pkg/rotation"
)

func main() {
    // Create lifecycle manager
    manager := lifecycle.NewKeyLifecycleManager()
    
    // Generate key
    key, err := manager.GenerateKey(lifecycle.AlgorithmRSA, 2048, []lifecycle.KeyUsage{
        lifecycle.UsageEncryption, lifecycle.UsageDecryption,
    })
    if err != nil {
        panic(err)
    }
    
    fmt.Printf("Generated key: %s\n", key.ID)
    
    // Activate key
    err = manager.ActivateKey(key.ID)
    if err != nil {
        panic(err)
    }
    
    // Store key
    keyStorage := storage.NewKeyStorage(&storage.StorageConfig{
        Backend: storage.BackendMemory,
    })
    
    metadata := storage.KeyMetadata{
        ID:        key.ID,
        Algorithm: string(key.Algorithm),
        KeySize:   key.KeySize,
    }
    
    keyData, _ := manager.ExportKeyPEM(key.ID)
    keyStorage.StoreKey(key.ID, keyData, metadata)
    
    // Set up rotation
    rotationManager := rotation.NewRotationManager()
    rotationManager.AddPolicy(rotation.CreateDefaultPolicy())
    
    schedule, _ := rotationManager.CreateSchedule(key.ID, "default")
    fmt.Printf("Next rotation: %s\n", schedule.NextRotation.Format("2006-01-02"))
    
    // Check rotation status
    overdue := rotationManager.GetOverdueKeys()
    fmt.Printf("Overdue keys: %d\n", len(overdue))
}