package main

import (
    "context"
    "log"
    
    "github.com/khoomi/idxr"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

func main() {
    client, _ := mongo.Connect(context.Background(), 
        options.Client().ApplyURI("mongodb://localhost:27017"))
    
    db := client.Database("khoomi")
    manager := indexer.NewManager(db)
    
    // Add a unique index
    manager.AddIndex("users", mongo.IndexModel{
        Keys:    bson.D{{Key: "email", Value: 1}},
        Options: options.Index().SetUnique(true).SetName("email_unique"),
    })
    
    // Add a compound index
    manager.AddCompoundIndex("orders", []string{"user_id", "created_at"})
    
    // Add a text search index
    manager.AddTextIndex("products", "name", "description", "tags")
    
    // Create all indexes
    result, err := manager.Create(context.Background())
    if err != nil {
        log.Printf("Some indexes failed: %v", err)
    }
    
    log.Printf("Created %d indexes, %d failed", 
        result.SuccessCount, result.FailedCount)
}

migrationManager := indexer.NewMigrationManager(db)

migrationManager.AddMigration(indexer.Migration{
    Version:     "001_initial",
    Description: "Initial indexes",
    Up: func(db *mongo.Database) error {
        manager := indexer.NewManager(db)
        manager.AddIndex("users", mongo.IndexModel{
            Keys:    bson.D{{Key: "email", Value: 1}},
            Options: options.Index().SetUnique(true),
        })
        result, _ := manager.Create(context.Background())
        return nil
    },
    Down: func(db *mongo.Database) error {
        return indexer.NewManager(db).Drop(context.Background(), "users")
    },
})

// Run migrations
err := migrationManager.Run(context.Background())

// Check status
statuses, _ := migrationManager.Status(context.Background())

// Rollback to specific version
err = migrationManager.Rollback(context.Background(), "001_initial")

opts := &indexer.Options{
    Timeout:         60 * time.Second,
    ContinueOnError: true,
    SkipIfExists:    true,
    Environment:     "production",
}

manager := indexer.NewManager(db, opts)

manager.AddCompoundIndex("collection", []string{"field1", "field2"})

manager.AddTextIndex("collection", "title", "content", "tags")

manager.AddIndex("users", mongo.IndexModel{
    Keys: bson.D{{Key: "email", Value: 1}},
    Options: options.Index().
        SetPartialFilterExpression(bson.M{"active": true}),
})

manager.AddIndex("sessions", mongo.IndexModel{
    Keys: bson.D{{Key: "expires_at", Value: 1}},
    Options: options.Index().
        SetExpireAfterSeconds(3600),
})

// Get stats for single collection
stats, err := manager.Stats(context.Background(), "users")
for _, stat := range stats {
    fmt.Printf("%s: %d accesses since %v\n", 
        stat.Name, stat.Accesses, stat.Since)
}

// Get stats for all managed collections
allStats, err := manager.StatsAll(context.Background())

result, err := manager.Create(context.Background())

if result.FailedCount > 0 {
    for _, failure := range result.Failures {
        log.Printf("Failed: %s.%s - %v", 
            failure.Collection, 
            failure.IndexName, 
            failure.Error)
    }
}