futriix/internal/repl/repl.go

/*
 * Copyright 2026 Safronov Grigorii
 *
 * Licensed under the CDDL, Version 1.0 (the "License");
 * you may not use this file except in compliance with the License.
 *
 * You may obtain a copy of the License at
 * https://opensource.org/licenses/CDDL-1.0
 */

// Файл: internal/repl/repl.go
// Назначение: REPL (Read-Eval-Print Loop) интерфейс для интерактивной работы с СУБД.
// Поддерживает автодополнение, историю команд, цветовой вывод и все операции с данными.

package repl

import (
    "bufio"
    "fmt"
    "os"
    "strings"
    "time"
    
    "futriis/internal/acl"
    "futriis/internal/cluster"
    "futriis/internal/compression"
    "futriis/internal/config"
    "futriis/internal/log"
    "futriis/internal/plugin"
    "futriis/internal/storage"
    "futriis/pkg/utils"
    
    "github.com/fatih/color"
)

// Repl представляет основную структуру REPL
type Repl struct {
    store         *storage.Storage
    coordinator   *cluster.RaftCoordinator
    logger        *log.Logger
    config        *config.Config
    aclManager    *acl.ACLManager
    pluginManager *plugin.PluginManager
    reader        *bufio.Reader
    currentDB     string
    currentUser   string
    currentRole   string
    authenticated bool
    sessionID     string
    commands      map[string]*Command
    history       []string
    historyPos    int
    historyFile   *History
}

// Command представляет команду REPL
type Command struct {
    Name        string
    Description string
    Handler     func(args []string) error
}

// NewRepl создаёт новый экземпляр REPL
func NewRepl(store *storage.Storage, coordinator *cluster.RaftCoordinator, logger *log.Logger, cfg *config.Config, aclManager *acl.ACLManager, pluginManager *plugin.PluginManager) *Repl {
    r := &Repl{
        store:         store,
        coordinator:   coordinator,
        logger:        logger,
        config:        cfg,
        aclManager:    aclManager,
        pluginManager: pluginManager,
        reader:        bufio.NewReader(os.Stdin),
        currentDB:     "",
        currentUser:   "",
        currentRole:   "anonymous",
        authenticated: false,
        sessionID:     "",
        commands:      make(map[string]*Command),
        history:       make([]string, 0, cfg.Repl.HistorySize),
        historyPos:    -1,
        historyFile:   NewHistory(cfg.Repl.HistorySize),
    }
    
    // Загружаем историю из файла
    if err := r.historyFile.Load(); err == nil {
        r.history = r.historyFile.GetEntries()
        r.historyPos = len(r.history)
    }
    
    r.registerCommands()
    return r
}

// registerCommands регистрирует все команды REPL
func (r *Repl) registerCommands() {
    // Команды управления базами данных
    r.commands["create database"] = &Command{
        Name:        "create database",
        Description: "Create a new database",
        Handler:     r.handleCreateDatabase,
    }
    
    r.commands["drop database"] = &Command{
        Name:        "drop database",
        Description: "Drop a database",
        Handler:     r.handleDropDatabase,
    }
    
    r.commands["use"] = &Command{
        Name:        "use",
        Description: "Switch to a database",
        Handler:     r.handleUseDatabase,
    }
    
    r.commands["show databases"] = &Command{
        Name:        "show databases",
        Description: "List all databases",
        Handler:     r.handleShowDatabases,
    }
    
    // Команды управления коллекциями
    r.commands["create collection"] = &Command{
        Name:        "create collection",
        Description: "Create a new collection in current database",
        Handler:     r.handleCreateCollection,
    }
    
    r.commands["drop collection"] = &Command{
        Name:        "drop collection",
        Description: "Drop a collection from current database",
        Handler:     r.handleDropCollection,
    }
    
    r.commands["show collections"] = &Command{
        Name:        "show collections",
        Description: "List all collections in current database",
        Handler:     r.handleShowCollections,
    }
    
    // Команды работы с документами
    r.commands["insert"] = &Command{
        Name:        "insert",
        Description: "Insert a document into a collection (JSON format)",
        Handler:     r.handleInsert,
    }
    
    r.commands["find"] = &Command{
        Name:        "find",
        Description: "Find a document by ID",
        Handler:     r.handleFind,
    }
    
    r.commands["findbyindex"] = &Command{
        Name:        "findbyindex",
        Description: "Find documents by index",
        Handler:     r.handleFindByIndex,
    }
    
    r.commands["update"] = &Command{
        Name:        "update",
        Description: "Update a document",
        Handler:     r.handleUpdate,
    }
    
    r.commands["delete"] = &Command{
        Name:        "delete",
        Description: "Delete a document",
        Handler:     r.handleDelete,
    }
    
    r.commands["count"] = &Command{
        Name:        "count",
        Description: "Count documents in a collection",
        Handler:     r.handleCount,
    }
    
    // Команды управления индексами
    r.commands["create index"] = &Command{
        Name:        "create index",
        Description: "Create an index on a collection",
        Handler:     r.handleCreateIndex,
    }
    
    r.commands["drop index"] = &Command{
        Name:        "drop index",
        Description: "Drop an index from a collection",
        Handler:     r.handleDropIndex,
    }
    
    r.commands["show indexes"] = &Command{
        Name:        "show indexes",
        Description: "Show all indexes in a collection",
        Handler:     r.handleShowIndexes,
    }
    
    // Команды ограничений
    r.commands["add required"] = &Command{
        Name:        "add required",
        Description: "Add a required field constraint",
        Handler:     r.handleAddRequired,
    }
    
    r.commands["add unique"] = &Command{
        Name:        "add unique",
        Description: "Add a unique constraint",
        Handler:     r.handleAddUnique,
    }
    
    r.commands["add min"] = &Command{
        Name:        "add min",
        Description: "Add a minimum value constraint",
        Handler:     r.handleAddMin,
    }
    
    r.commands["add max"] = &Command{
        Name:        "add max",
        Description: "Add a maximum value constraint",
        Handler:     r.handleAddMax,
    }
    
    r.commands["add enum"] = &Command{
        Name:        "add enum",
        Description: "Add an enum constraint (allowed values)",
        Handler:     r.handleAddEnum,
    }
    
    // Команды триггеров (MongoDB-like syntax)
    r.commands["create trigger"] = &Command{
        Name:        "create trigger",
        Description: "Create a trigger on a collection (MongoDB-like syntax)",
        Handler:     r.handleCreateTrigger,
    }
    
    r.commands["drop trigger"] = &Command{
        Name:        "drop trigger",
        Description: "Drop a trigger from a collection",
        Handler:     r.handleDropTrigger,
    }
    
    r.commands["show triggers"] = &Command{
        Name:        "show triggers",
        Description: "Show all triggers on a collection",
        Handler:     r.handleShowTriggers,
    }
    
    r.commands["enable trigger"] = &Command{
        Name:        "enable trigger",
        Description: "Enable a trigger",
        Handler:     r.handleEnableTrigger,
    }
    
    r.commands["disable trigger"] = &Command{
        Name:        "disable trigger",
        Description: "Disable a trigger",
        Handler:     r.handleDisableTrigger,
    }
    
    r.commands["trigger log"] = &Command{
        Name:        "trigger log",
        Description: "Show trigger execution log",
        Handler:     r.handleTriggerLog,
    }
    
    // Команды транзакций
    r.commands["begin transaction"] = &Command{
        Name:        "begin transaction",
        Description: "Start a new transaction",
        Handler:     r.handleBeginTransaction,
    }
    
    r.commands["commit"] = &Command{
        Name:        "commit",
        Description: "Commit current transaction",
        Handler:     r.handleCommitTransaction,
    }
    
    r.commands["rollback"] = &Command{
        Name:        "rollback",
        Description: "Rollback current transaction",
        Handler:     r.handleRollbackTransaction,
    }
    
    r.commands["show transactions"] = &Command{
        Name:        "show transactions",
        Description: "Show active transactions",
        Handler:     r.handleShowTransactions,
    }
    
    // Команды плагинов
    r.commands["plugin list"] = &Command{
        Name:        "plugin list",
        Description: "List all loaded plugins",
        Handler:     r.handlePluginList,
    }
    
    r.commands["plugin load"] = &Command{
        Name:        "plugin load",
        Description: "Load a plugin from file",
        Handler:     r.handlePluginLoad,
    }
    
    r.commands["plugin unload"] = &Command{
        Name:        "plugin unload",
        Description: "Unload a plugin",
        Handler:     r.handlePluginUnload,
    }
    
    r.commands["plugin start"] = &Command{
        Name:        "plugin start",
        Description: "Start a plugin",
        Handler:     r.handlePluginStart,
    }
    
    r.commands["plugin stop"] = &Command{
        Name:        "plugin stop",
        Description: "Stop a plugin",
        Handler:     r.handlePluginStop,
    }
    
    r.commands["plugin exec"] = &Command{
        Name:        "plugin exec",
        Description: "Execute a plugin function",
        Handler:     r.handlePluginExec,
    }
    
    // Команды импорта/экспорта
    r.commands["export"] = &Command{
        Name:        "export",
        Description: "Export database to MessagePack file",
        Handler:     r.handleExport,
    }
    
    r.commands["import"] = &Command{
        Name:        "import",
        Description: "Import database from MessagePack file",
        Handler:     r.handleImport,
    }
    
    // Команды ACL
    r.commands["acl login"] = &Command{
        Name:        "acl login",
        Description: "Authenticate with username and password",
        Handler:     r.handleACLLogin,
    }
    
    r.commands["acl logout"] = &Command{
        Name:        "acl logout",
        Description: "Logout current user session",
        Handler:     r.handleACLLogout,
    }
    
    r.commands["acl grant"] = &Command{
        Name:        "acl grant",
        Description: "Grant permissions (r=read,w=write,d=delete,a=admin)",
        Handler:     r.handleACLGrant,
    }
    
    r.commands["acl users"] = &Command{
        Name:        "acl users",
        Description: "List all users",
        Handler:     r.handleACLUsers,
    }
    
    r.commands["acl roles"] = &Command{
        Name:        "acl roles",
        Description: "List all roles",
        Handler:     r.handleACLRoles,
    }
    
    // Команды сжатия
    r.commands["compression stats"] = &Command{
        Name:        "compression stats",
        Description: "Show compression statistics for the database",
        Handler:     r.handleCompressionStats,
    }
    
    r.commands["compress collection"] = &Command{
        Name:        "compress collection",
        Description: "Manually compress all documents in a collection",
        Handler:     r.handleCompressCollection,
    }
    
    r.commands["doc compression"] = &Command{
        Name:        "doc compression",
        Description: "Show compression ratio for a document",
        Handler:     r.handleDocCompression,
    }
    
    r.commands["compression config"] = &Command{
        Name:        "compression config",
        Description: "Show current compression configuration",
        Handler:     r.handleCompressionConfig,
    }
    
    // Команды кластера
    r.commands["status"] = &Command{
        Name:        "status",
        Description: "Show cluster status",
        Handler:     r.handleStatus,
    }
    
    r.commands["nodes"] = &Command{
        Name:        "nodes",
        Description: "List cluster nodes",
        Handler:     r.handleNodes,
    }
    
    // Команды системы
    r.commands["help"] = &Command{
        Name:        "help",
        Description: "Show this help message",
        Handler:     r.handleHelp,
    }
    
    r.commands["clear"] = &Command{
        Name:        "clear",
        Description: "Clear the screen",
        Handler:     r.handleClear,
    }
    
    r.commands["quit"] = &Command{
        Name:        "quit",
        Description: "Exit the REPL",
        Handler:     r.handleQuit,
    }
    
    r.commands["exit"] = &Command{
        Name:        "exit",
        Description: "Exit the REPL",
        Handler:     r.handleQuit,
    }
}

// Run запускает основной цикл REPL
func (r *Repl) Run() error {
    utils.Println("")
    utils.PrintInfo("Type 'help' for available commands")
    utils.Println("")
    
    for {
        // Формируем приглашение к вводу
        prompt := r.buildPrompt()
        
        // Читаем ввод пользователя
        fmt.Print(prompt)
        input, err := r.reader.ReadString('\n')
        if err != nil {
            if err.Error() == "EOF" {
                return nil
            }
            return err
        }
        
        input = strings.TrimSpace(input)
        if input == "" {
            continue
        }
        
        // Сохраняем в историю
        r.addToHistory(input)
        
        // Обрабатываем команду
        if err := r.executeCommand(input); err != nil {
            utils.PrintError(err.Error())
            if r.logger != nil {
                r.logger.Error("REPL command error: " + err.Error())
            }
        }
    }
}

// buildPrompt формирует строку приглашения
func (r *Repl) buildPrompt() string {
    prompt := color.New(color.FgHiCyan).Sprint("futriiS")
    
    if r.currentDB != "" {
        prompt += color.New(color.FgHiYellow).Sprint(":" + r.currentDB)
    }
    
    if r.authenticated && r.currentUser != "" {
        prompt += color.New(color.FgHiGreen).Sprint(" (" + r.currentUser + ")")
    }
    
    prompt += color.New(color.FgHiCyan).Sprint("> ")
    return prompt
}

// executeCommand выполняет введённую команду
func (r *Repl) executeCommand(input string) error {
    parts := strings.Fields(input)
    if len(parts) == 0 {
        return nil
    }
    
    // Ищем команду по префиксу
    for cmdName, cmd := range r.commands {
        if strings.HasPrefix(input, cmdName) {
            // Извлекаем аргументы
            args := strings.TrimPrefix(input, cmdName)
            args = strings.TrimSpace(args)
            argList := strings.Fields(args)
            return cmd.Handler(argList)
        }
    }
    
    return fmt.Errorf("unknown command: %s", parts[0])
}

// addToHistory добавляет команду в историю
func (r *Repl) addToHistory(cmd string) {
    if len(r.history) > 0 && r.history[len(r.history)-1] == cmd {
        return
    }
    
    if len(r.history) >= r.config.Repl.HistorySize {
        r.history = r.history[1:]
    }
    r.history = append(r.history, cmd)
    r.historyPos = len(r.history)
    
    // Сохраняем в файл
    r.historyFile.Add(cmd)
    r.historyFile.Save()
}

// ========== Обработчики команд ==========

func (r *Repl) handleCreateDatabase(args []string) error {
    if len(args) < 1 {
        return fmt.Errorf("usage: create database <name>")
    }
    
    name := args[0]
    if err := r.store.CreateDatabase(name); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Database '%s' created", name))
    return nil
}

func (r *Repl) handleDropDatabase(args []string) error {
    if len(args) < 1 {
        return fmt.Errorf("usage: drop database <name>")
    }
    
    name := args[0]
    if err := r.store.DropDatabase(name); err != nil {
        return err
    }
    
    if r.currentDB == name {
        r.currentDB = ""
    }
    
    utils.PrintSuccess(fmt.Sprintf("Database '%s' dropped", name))
    return nil
}

func (r *Repl) handleUseDatabase(args []string) error {
    if len(args) < 1 {
        return fmt.Errorf("usage: use <database>")
    }
    
    name := args[0]
    if !r.store.ExistsDatabase(name) {
        return fmt.Errorf("database '%s' does not exist", name)
    }
    
    r.currentDB = name
    utils.PrintSuccess(fmt.Sprintf("Switched to database '%s'", name))
    return nil
}

func (r *Repl) handleShowDatabases(args []string) error {
    databases := r.store.ListDatabases()
    if len(databases) == 0 {
        utils.PrintInfo("No databases found")
        return nil
    }
    
    utils.PrintInfo("Databases:")
    for _, db := range databases {
        prefix := "  "
        if db == r.currentDB {
            prefix = " *"
        }
        utils.Println(fmt.Sprintf("%s %s", prefix, db))
    }
    return nil
}

func (r *Repl) handleCreateCollection(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 1 {
        return fmt.Errorf("usage: create collection <name>")
    }
    
    name := args[0]
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    if err := db.CreateCollection(name); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Collection '%s' created in database '%s'", name, r.currentDB))
    return nil
}

func (r *Repl) handleDropCollection(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 1 {
        return fmt.Errorf("usage: drop collection <name>")
    }
    
    name := args[0]
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    if err := db.DropCollection(name); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Collection '%s' dropped from database '%s'", name, r.currentDB))
    return nil
}

func (r *Repl) handleShowCollections(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    collections := db.ListCollections()
    if len(collections) == 0 {
        utils.PrintInfo("No collections found")
        return nil
    }
    
    utils.PrintInfo(fmt.Sprintf("Collections in database '%s':", r.currentDB))
    for _, coll := range collections {
        utils.Println(fmt.Sprintf("  - %s", coll))
    }
    return nil
}

func (r *Repl) handleInsert(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 2 {
        return fmt.Errorf("usage: insert <collection> <json>")
    }
    
    collName := args[0]
    jsonStr := strings.Join(args[1:], " ")
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    doc := storage.NewDocument()
    
    // Проверяем, является ли ввод JSON
    if strings.Contains(jsonStr, "{") {
        // Упрощённый парсинг JSON (для реального использования нужен json.Unmarshal)
        // Здесь оставляем простой парсинг key=value
        pairs := strings.Split(jsonStr, ",")
        for _, pair := range pairs {
            pair = strings.TrimSpace(pair)
            kv := strings.SplitN(pair, "=", 2)
            if len(kv) == 2 {
                doc.SetField(kv[0], kv[1])
            }
        }
    } else {
        // Формат key=value
        pairs := strings.Split(jsonStr, ",")
        for _, pair := range pairs {
            pair = strings.TrimSpace(pair)
            kv := strings.SplitN(pair, "=", 2)
            if len(kv) == 2 {
                doc.SetField(kv[0], kv[1])
            }
        }
    }
    
    if err := coll.Insert(doc); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Document inserted with ID: %s", doc.ID))
    return nil
}

func (r *Repl) handleFind(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 2 {
        return fmt.Errorf("usage: find <collection> <id>")
    }
    
    collName := args[0]
    docID := args[1]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    doc, err := coll.Find(docID)
    if err != nil {
        return err
    }
    
    utils.PrintInfo(fmt.Sprintf("Document found:"))
    utils.PrintJSON(doc.GetFields())
    return nil
}

func (r *Repl) handleFindByIndex(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: findbyindex <collection> <index> <value>")
    }
    
    collName := args[0]
    indexName := args[1]
    value := args[2]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    docs, err := coll.FindByIndex(indexName, value)
    if err != nil {
        return err
    }
    
    utils.PrintInfo(fmt.Sprintf("Found %d document(s):", len(docs)))
    for i, doc := range docs {
        utils.PrintInfo(fmt.Sprintf("  [%d] ID: %s", i+1, doc.ID))
        utils.PrintJSON(doc.GetFields())
    }
    return nil
}

func (r *Repl) handleUpdate(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: update <collection> <id> <field=value>...")
    }
    
    collName := args[0]
    docID := args[1]
    
    updates := make(map[string]interface{})
    for i := 2; i < len(args); i++ {
        kv := strings.SplitN(args[i], "=", 2)
        if len(kv) == 2 {
            updates[kv[0]] = kv[1]
        }
    }
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    if err := coll.Update(docID, updates); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Document '%s' updated", docID))
    return nil
}

func (r *Repl) handleDelete(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 2 {
        return fmt.Errorf("usage: delete <collection> <id>")
    }
    
    collName := args[0]
    docID := args[1]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    if err := coll.Delete(docID); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Document '%s' deleted", docID))
    return nil
}

func (r *Repl) handleCount(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 1 {
        return fmt.Errorf("usage: count <collection>")
    }
    
    collName := args[0]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    count := coll.Count()
    utils.PrintInfo(fmt.Sprintf("Collection '%s' has %d document(s)", collName, count))
    return nil
}

func (r *Repl) handleCreateIndex(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: create index <collection> <name> <fields> [unique]")
    }
    
    collName := args[0]
    indexName := args[1]
    fields := strings.Split(args[2], ",")
    unique := len(args) > 3 && args[3] == "unique"
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    if err := coll.CreateIndex(indexName, fields, unique); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Index '%s' created on collection '%s'", indexName, collName))
    return nil
}

func (r *Repl) handleDropIndex(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 2 {
        return fmt.Errorf("usage: drop index <collection> <name>")
    }
    
    collName := args[0]
    indexName := args[1]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    if err := coll.DropIndex(indexName); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Index '%s' dropped from collection '%s'", indexName, collName))
    return nil
}

func (r *Repl) handleShowIndexes(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 1 {
        return fmt.Errorf("usage: show indexes <collection>")
    }
    
    collName := args[0]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    indexes := coll.GetIndexesInfo()
    if len(indexes) == 0 {
        utils.PrintInfo(fmt.Sprintf("No indexes found on collection '%s'", collName))
        return nil
    }
    
    utils.PrintInfo(fmt.Sprintf("Indexes on collection '%s':", collName))
    for _, idx := range indexes {
        uniqueStr := ""
        if idx["unique"].(bool) {
            uniqueStr = " (unique)"
        }
        utils.Println(fmt.Sprintf("  - %s: %v%s", idx["name"], idx["fields"], uniqueStr))
    }
    return nil
}

func (r *Repl) handleAddRequired(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 2 {
        return fmt.Errorf("usage: add required <collection> <field>")
    }
    
    collName := args[0]
    field := args[1]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    coll.AddRequiredField(field)
    utils.PrintSuccess(fmt.Sprintf("Required field '%s' added to collection '%s'", field, collName))
    return nil
}

func (r *Repl) handleAddUnique(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 2 {
        return fmt.Errorf("usage: add unique <collection> <field>")
    }
    
    collName := args[0]
    field := args[1]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    coll.AddUniqueConstraint(field)
    utils.PrintSuccess(fmt.Sprintf("Unique constraint added for field '%s' on collection '%s'", field, collName))
    return nil
}

func (r *Repl) handleAddMin(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: add min <collection> <field> <value>")
    }
    
    collName := args[0]
    field := args[1]
    var minVal float64
    if _, err := fmt.Sscanf(args[2], "%f", &minVal); err != nil {
        return fmt.Errorf("invalid minimum value: %s", args[2])
    }
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    coll.AddMinConstraint(field, minVal)
    utils.PrintSuccess(fmt.Sprintf("Min constraint added for field '%s' on collection '%s' (min: %.2f)", field, collName, minVal))
    return nil
}

func (r *Repl) handleAddMax(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: add max <collection> <field> <value>")
    }
    
    collName := args[0]
    field := args[1]
    var maxVal float64
    if _, err := fmt.Sscanf(args[2], "%f", &maxVal); err != nil {
        return fmt.Errorf("invalid maximum value: %s", args[2])
    }
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    coll.AddMaxConstraint(field, maxVal)
    utils.PrintSuccess(fmt.Sprintf("Max constraint added for field '%s' on collection '%s' (max: %.2f)", field, collName, maxVal))
    return nil
}

func (r *Repl) handleAddEnum(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: add enum <collection> <field> <values...>")
    }
    
    collName := args[0]
    field := args[1]
    values := make([]interface{}, len(args[2:]))
    for i, v := range args[2:] {
        values[i] = v
    }
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    coll.AddEnumConstraint(field, values)
    utils.PrintSuccess(fmt.Sprintf("Enum constraint added for field '%s' on collection '%s' (allowed: %v)", field, collName, values))
    return nil
}

// ========== Обработчики команд транзакций ==========

func (r *Repl) handleBeginTransaction(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    tx := storage.BeginTransaction()
    if tx == nil {
        return fmt.Errorf("failed to begin transaction")
    }
    
    utils.PrintSuccess(fmt.Sprintf("Transaction %d started", tx.ID))
    return nil
}

func (r *Repl) handleCommitTransaction(args []string) error {
    if !storage.HasActiveTransaction() {
        return fmt.Errorf("no active transaction to commit")
    }
    
    if err := storage.CommitCurrentTransaction(); err != nil {
        return err
    }
    
    utils.PrintSuccess("Transaction committed successfully")
    return nil
}

func (r *Repl) handleRollbackTransaction(args []string) error {
    if !storage.HasActiveTransaction() {
        return fmt.Errorf("no active transaction to rollback")
    }
    
    if err := storage.AbortCurrentTransaction(); err != nil {
        return err
    }
    
    utils.PrintSuccess("Transaction rolled back")
    return nil
}

func (r *Repl) handleShowTransactions(args []string) error {
    transactions := storage.GetActiveTransactions()
    
    if len(transactions) == 0 {
        utils.PrintInfo("No active transactions")
        return nil
    }
    
    utils.PrintHeader("Active Transactions")
    for _, tx := range transactions {
        utils.PrintInfo(fmt.Sprintf("  ID: %s, Status: %s, Operations: %d, Started: %s",
            tx.ID, tx.Status, tx.OperationCount, time.UnixMilli(tx.StartTime).Format("15:04:05")))
        
        for _, op := range tx.Operations {
            utils.Println(fmt.Sprintf("      - %s: %s.%s [%s]", op.Type, op.Database, op.Collection, op.DocumentID))
        }
    }
    return nil
}

// ========== Обработчики команд плагинов ==========

func (r *Repl) handlePluginList(args []string) error {
    if r.pluginManager == nil || !r.pluginManager.IsEnabled() {
        return fmt.Errorf("plugin system is disabled")
    }
    
    plugins := r.pluginManager.ListPlugins()
    
    if len(plugins) == 0 {
        utils.PrintInfo("No plugins loaded")
        utils.PrintInfo(fmt.Sprintf("Plugins directory: %s", r.pluginManager.GetPluginsDir()))
        return nil
    }
    
    utils.PrintHeader("Loaded Plugins")
    for _, p := range plugins {
        status := "loaded"
        switch p.Status.Load() {
        case 1:
            status = "running"
        case 2:
            status = "stopped"
        case 3:
            status = "error"
        }
        utils.PrintInfo(fmt.Sprintf("  %s v%s - %s [%s]", p.Name, p.Version(), status, p.Description()))
        utils.Println(fmt.Sprintf("      Author: %s, Loaded: %s", p.Author(), p.LoadedAt().Format("2006-01-02 15:04:05")))
    }
    return nil
}

func (r *Repl) handlePluginLoad(args []string) error {
    if r.pluginManager == nil || !r.pluginManager.IsEnabled() {
        return fmt.Errorf("plugin system is disabled")
    }
    
    if len(args) < 2 {
        return fmt.Errorf("usage: plugin load <name> <filepath>")
    }
    
    name := args[0]
    filepath := args[1]
    
    if err := r.pluginManager.LoadPlugin(name, filepath); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Plugin '%s' loaded from %s", name, filepath))
    return nil
}

func (r *Repl) handlePluginUnload(args []string) error {
    if r.pluginManager == nil || !r.pluginManager.IsEnabled() {
        return fmt.Errorf("plugin system is disabled")
    }
    
    if len(args) < 1 {
        return fmt.Errorf("usage: plugin unload <name>")
    }
    
    name := args[0]
    
    if err := r.pluginManager.UnloadPlugin(name); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Plugin '%s' unloaded", name))
    return nil
}

func (r *Repl) handlePluginStart(args []string) error {
    if r.pluginManager == nil || !r.pluginManager.IsEnabled() {
        return fmt.Errorf("plugin system is disabled")
    }
    
    if len(args) < 1 {
        return fmt.Errorf("usage: plugin start <name>")
    }
    
    name := args[0]
    
    if err := r.pluginManager.StartPlugin(name); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Plugin '%s' started", name))
    return nil
}

func (r *Repl) handlePluginStop(args []string) error {
    if r.pluginManager == nil || !r.pluginManager.IsEnabled() {
        return fmt.Errorf("plugin system is disabled")
    }
    
    if len(args) < 1 {
        return fmt.Errorf("usage: plugin stop <name>")
    }
    
    name := args[0]
    
    if err := r.pluginManager.StopPlugin(name); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Plugin '%s' stopped", name))
    return nil
}

func (r *Repl) handlePluginExec(args []string) error {
    if r.pluginManager == nil || !r.pluginManager.IsEnabled() {
        return fmt.Errorf("plugin system is disabled")
    }
    
    if len(args) < 2 {
        return fmt.Errorf("usage: plugin exec <plugin> <function> [args...]")
    }
    
    pluginName := args[0]
    funcName := args[1]
    
    var execArgs []interface{}
    for _, arg := range args[2:] {
        execArgs = append(execArgs, arg)
    }
    
    result, err := r.pluginManager.ExecutePlugin(pluginName, funcName, execArgs...)
    if err != nil {
        return err
    }
    
    utils.PrintSuccess("Result:")
    utils.PrintJSON(result)
    return nil
}

// ========== Обработчики команд триггеров ==========

func (r *Repl) handleCreateTrigger(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 5 {
        return fmt.Errorf("usage: create trigger <collection> <name> <event> <action> [options]\n"+
            "  Events: BEFORE_INSERT, AFTER_INSERT, BEFORE_UPDATE, AFTER_UPDATE, BEFORE_DELETE, AFTER_DELETE\n"+
            "  Actions: abort, skip, modify, log, notify\n"+
            "  Options: --description <text>, --set <field> <value>, --inc <field> <value>, --currentDate <field>, --condition <field> <op> <value>")
    }
    
    collName := args[0]
    triggerName := args[1]
    event := args[2]
    action := args[3]
    
    tm := storage.GetTriggerManager()
    
    config := storage.MongoDBLikeTriggerConfig().
        On(event).
        Action(action)
    
    // Парсим опции
    for i := 4; i < len(args); i++ {
        switch args[i] {
        case "--description":
            if i+1 < len(args) {
                config.Description(args[i+1])
                i++
            }
        case "--set":
            if i+2 < len(args) {
                config.Set(args[i+1], args[i+2])
                i += 2
            }
        case "--inc":
            if i+2 < len(args) {
                var val float64
                fmt.Sscanf(args[i+2], "%f", &val)
                config.Inc(args[i+1], val)
                i += 2
            }
        case "--currentDate":
            if i+1 < len(args) {
                config.CurrentDate(args[i+1])
                i++
            }
        case "--condition":
            if i+3 < len(args) {
                config.Condition(args[i+1], args[i+2], args[i+3])
                i += 3
            }
        }
    }
    
    var eventType storage.TriggerEvent
    switch event {
    case "BEFORE_INSERT":
        eventType = storage.TriggerBeforeInsert
    case "AFTER_INSERT":
        eventType = storage.TriggerAfterInsert
    case "BEFORE_UPDATE":
        eventType = storage.TriggerBeforeUpdate
    case "AFTER_UPDATE":
        eventType = storage.TriggerAfterUpdate
    case "BEFORE_DELETE":
        eventType = storage.TriggerBeforeDelete
    case "AFTER_DELETE":
        eventType = storage.TriggerAfterDelete
    default:
        return fmt.Errorf("unknown event type: %s", event)
    }
    
    if err := tm.CreateTrigger(r.currentDB, collName, triggerName, eventType, config.Build()); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Trigger '%s' created on collection '%s' for event %s", triggerName, collName, event))
    return nil
}

func (r *Repl) handleDropTrigger(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: drop trigger <collection> <event> <name>")
    }
    
    collName := args[0]
    event := args[1]
    triggerName := args[2]
    
    tm := storage.GetTriggerManager()
    if err := tm.DropTrigger(collName, event, triggerName); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Trigger '%s' dropped from collection '%s'", triggerName, collName))
    return nil
}

func (r *Repl) handleShowTriggers(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 1 {
        return fmt.Errorf("usage: show triggers <collection>")
    }
    
    collName := args[0]
    
    tm := storage.GetTriggerManager()
    triggers := tm.ListTriggers(collName)
    
    if len(triggers) == 0 {
        utils.PrintInfo(fmt.Sprintf("No triggers found on collection '%s'", collName))
        return nil
    }
    
    utils.PrintHeader(fmt.Sprintf("Triggers on collection '%s':", collName))
    for _, t := range triggers {
        status := "enabled"
        if !t.Enabled {
            status = "disabled"
        }
        utils.PrintInfo(fmt.Sprintf("  %s (%s) - %s [%s]", t.Name, t.Event, status, t.Action))
        if t.Description != "" {
            utils.Println(fmt.Sprintf("      Description: %s", t.Description))
        }
        if t.Condition != nil {
            utils.Println(fmt.Sprintf("      Condition: %s %s %v", t.Condition.Field, t.Condition.Operator, t.Condition.Value))
        }
        if len(t.Operations) > 0 {
            utils.Println("      Operations:")
            for _, op := range t.Operations {
                utils.Println(fmt.Sprintf("        - %s: %s = %v", op.Type, op.Field, op.Value))
            }
        }
    }
    return nil
}

func (r *Repl) handleEnableTrigger(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: enable trigger <collection> <event> <name>")
    }
    
    collName := args[0]
    event := args[1]
    triggerName := args[2]
    
    tm := storage.GetTriggerManager()
    if err := tm.EnableTrigger(collName, event, triggerName); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Trigger '%s' enabled", triggerName))
    return nil
}

func (r *Repl) handleDisableTrigger(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: disable trigger <collection> <event> <name>")
    }
    
    collName := args[0]
    event := args[1]
    triggerName := args[2]
    
    tm := storage.GetTriggerManager()
    if err := tm.DisableTrigger(collName, event, triggerName); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Trigger '%s' disabled", triggerName))
    return nil
}

func (r *Repl) handleTriggerLog(args []string) error {
    tm := storage.GetTriggerManager()
    logs := tm.GetTriggerExecutionLog()
    
    if len(logs) == 0 {
        utils.PrintInfo("No trigger executions logged")
        return nil
    }
    
    utils.PrintHeader("Trigger Execution Log")
    for i, logEntry := range logs {
        utils.PrintInfo(fmt.Sprintf("[%d] %s - Trigger: %s, Event: %s, Collection: %s, Document: %s",
            i+1, logEntry.Timestamp.Format("2006-01-02 15:04:05"), logEntry.TriggerName, logEntry.Event, logEntry.Collection, logEntry.DocumentID))
    }
    return nil
}

// ========== Обработчики команд импорта/экспорта ==========

func (r *Repl) handleExport(args []string) error {
    if len(args) < 2 {
        return fmt.Errorf("usage: export <database> <filename>")
    }
    
    dbName := args[0]
    fileName := args[1]
    
    if !r.store.ExistsDatabase(dbName) {
        return fmt.Errorf("database '%s' does not exist", dbName)
    }
    
    utils.PrintInfo(fmt.Sprintf("Exporting database '%s' to %s...", dbName, fileName))
    
    // Экспорт данных
    db, err := r.store.GetDatabase(dbName)
    if err != nil {
        return err
    }
    
    data, err := db.SerializeDatabase()
    if err != nil {
        return err
    }
    
    if err := os.WriteFile(fileName, data, 0644); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Database '%s' exported to %s", dbName, fileName))
    return nil
}

func (r *Repl) handleImport(args []string) error {
    if len(args) < 2 {
        return fmt.Errorf("usage: import <database> <filename>")
    }
    
    dbName := args[0]
    fileName := args[1]
    
    utils.PrintInfo(fmt.Sprintf("Importing data from %s to database '%s'...", fileName, dbName))
    
    data, err := os.ReadFile(fileName)
    if err != nil {
        return err
    }
    
    if !r.store.ExistsDatabase(dbName) {
        if err := r.store.CreateDatabase(dbName); err != nil {
            return err
        }
    }
    
    db, err := r.store.GetDatabase(dbName)
    if err != nil {
        return err
    }
    
    if err := db.DeserializeDatabase(data); err != nil {
        return err
    }
    
    utils.PrintSuccess(fmt.Sprintf("Data imported to database '%s' from %s", dbName, fileName))
    return nil
}

// ========== Обработчики команд ACL ==========

func (r *Repl) handleACLLogin(args []string) error {
    if len(args) < 2 {
        return fmt.Errorf("usage: acl login <username> <password>")
    }
    
    username := args[0]
    password := args[1]
    
    if r.aclManager == nil {
        return fmt.Errorf("ACL manager not initialized")
    }
    
    sessionID, err := r.aclManager.Authenticate(username, password)
    if err != nil {
        return err
    }
    
    r.authenticated = true
    r.currentUser = username
    r.sessionID = sessionID
    
    // Получаем роли пользователя
    roles := r.aclManager.GetUserRoles(sessionID)
    if len(roles) > 0 {
        r.currentRole = roles[0]
    }
    
    utils.PrintSuccess(fmt.Sprintf("Logged in as '%s' with role '%s'", username, r.currentRole))
    return nil
}

func (r *Repl) handleACLLogout(args []string) error {
    if r.sessionID != "" && r.aclManager != nil {
        r.aclManager.Logout(r.sessionID)
    }
    
    r.authenticated = false
    r.currentUser = ""
    r.currentRole = "anonymous"
    r.sessionID = ""
    
    utils.PrintSuccess("Logged out")
    return nil
}

func (r *Repl) handleACLGrant(args []string) error {
    if !r.authenticated || r.currentRole != "admin" {
        return fmt.Errorf("permission denied: admin access required")
    }
    
    if len(args) < 3 {
        return fmt.Errorf("usage: acl grant <collection> <role> <permissions>\n"+
            "  Permissions: r=read, w=write, d=delete, a=admin\n"+
            "  Example: acl grant users admin rwa")
    }
    
    collName := args[0]
    role := args[1]
    perms := args[2]
    
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    canRead := strings.Contains(perms, "r")
    canWrite := strings.Contains(perms, "w")
    canDelete := strings.Contains(perms, "d")
    isAdmin := strings.Contains(perms, "a")
    
    coll.SetACL(role, canRead, canWrite, canDelete, isAdmin)
    utils.PrintSuccess(fmt.Sprintf("Permissions '%s' granted to role '%s' on collection '%s'", perms, role, collName))
    return nil
}

func (r *Repl) handleACLUsers(args []string) error {
    if r.aclManager == nil {
        return fmt.Errorf("ACL manager not initialized")
    }
    
    users := r.aclManager.ListUsers()
    
    if len(users) == 0 {
        utils.PrintInfo("No users found")
        return nil
    }
    
    utils.PrintHeader("Users")
    for _, username := range users {
        userInfo, err := r.aclManager.GetUserInfo(username)
        if err != nil {
            continue
        }
        status := "active"
        if !userInfo.Active {
            status = "disabled"
        }
        utils.PrintInfo(fmt.Sprintf("  %s - Roles: %v [%s]", username, userInfo.Roles, status))
    }
    return nil
}

func (r *Repl) handleACLRoles(args []string) error {
    if r.aclManager == nil {
        return fmt.Errorf("ACL manager not initialized")
    }
    
    roles := r.aclManager.ListRoles()
    
    if len(roles) == 0 {
        utils.PrintInfo("No roles found")
        return nil
    }
    
    utils.PrintHeader("Roles")
    for _, roleName := range roles {
        perms, err := r.aclManager.GetRolePermissions(roleName)
        if err != nil {
            continue
        }
        utils.PrintInfo(fmt.Sprintf("  %s - Permissions: %v", roleName, perms))
    }
    return nil
}

// ========== Обработчики команд сжатия ==========

func (r *Repl) handleCompressionStats(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    collections := db.ListCollections()
    totalDocs := int64(0)
    compressedDocs := int64(0)
    totalOriginalSize := int64(0)
    totalCompressedSize := int64(0)
    
    for _, collName := range collections {
        coll, err := db.GetCollection(collName)
        if err != nil {
            continue
        }
        
        docs := coll.GetAllDocuments()
        for _, doc := range docs {
            totalDocs++
            if doc.Compressed {
                compressedDocs++
                totalOriginalSize += doc.OriginalSize
                if data, err := doc.Serialize(); err == nil {
                    totalCompressedSize += int64(len(data))
                }
            }
        }
    }
    
    utils.PrintHeader("Compression Statistics")
    utils.PrintInfo(fmt.Sprintf("  Total Documents:      %d", totalDocs))
    utils.PrintInfo(fmt.Sprintf("  Compressed Documents: %d", compressedDocs))
    if totalDocs > 0 {
        utils.PrintInfo(fmt.Sprintf("  Compression Rate:     %.2f%%", float64(compressedDocs)/float64(totalDocs)*100))
    }
    if totalOriginalSize > 0 {
        ratio := float64(totalCompressedSize) / float64(totalOriginalSize)
        utils.PrintInfo(fmt.Sprintf("  Size Reduction:       %.2f%%", (1-ratio)*100))
        utils.PrintInfo(fmt.Sprintf("  Original Size:        %s", utils.FormatBytes(totalOriginalSize)))
        utils.PrintInfo(fmt.Sprintf("  Compressed Size:      %s", utils.FormatBytes(totalCompressedSize)))
    }
    utils.PrintInfo(fmt.Sprintf("  Algorithm:            %s", r.config.Compression.Algorithm))
    utils.PrintInfo(fmt.Sprintf("  Compression Level:    %d", r.config.Compression.Level))
    utils.PrintInfo(fmt.Sprintf("  Min Size Threshold:   %s", utils.FormatBytes(int64(r.config.Compression.MinSize))))
    
    return nil
}

func (r *Repl) handleCompressCollection(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 1 {
        return fmt.Errorf("usage: compress collection <name>")
    }
    
    collName := args[0]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    docs := coll.GetAllDocuments()
    compressed := 0
    
    utils.PrintInfo(fmt.Sprintf("Compressing collection '%s'...", collName))
    
    for _, doc := range docs {
        if !doc.Compressed {
            if err := doc.Compress(&compression.Config{
                Enabled:   r.config.Compression.Enabled,
                Algorithm: r.config.Compression.Algorithm,
                Level:     r.config.Compression.Level,
                MinSize:   r.config.Compression.MinSize,
            }); err == nil {
                compressed++
            }
        }
    }
    
    utils.PrintSuccess(fmt.Sprintf("Compressed %d documents in collection '%s'", compressed, collName))
    return nil
}

func (r *Repl) handleDocCompression(args []string) error {
    if r.currentDB == "" {
        return fmt.Errorf("no database selected")
    }
    
    if len(args) < 2 {
        return fmt.Errorf("usage: doc compression <collection> <id>")
    }
    
    collName := args[0]
    docID := args[1]
    
    db, err := r.store.GetDatabase(r.currentDB)
    if err != nil {
        return err
    }
    
    coll, err := db.GetCollection(collName)
    if err != nil {
        return err
    }
    
    doc, err := coll.Find(docID)
    if err != nil {
        return err
    }
    
    utils.PrintHeader(fmt.Sprintf("Compression Info for Document: %s", docID))
    utils.PrintInfo(fmt.Sprintf("  Compressed:     %v", doc.Compressed))
    if doc.Compressed {
        ratio := doc.GetCompressionRatio()
        utils.PrintInfo(fmt.Sprintf("  Ratio:          %.2f%%", (1-ratio)*100))
        utils.PrintInfo(fmt.Sprintf("  Original Size:  %s", utils.FormatBytes(doc.OriginalSize)))
        
        if data, err := doc.Serialize(); err == nil {
            utils.PrintInfo(fmt.Sprintf("  Current Size:   %s", utils.FormatBytes(int64(len(data)))))
        }
    }
    
    return nil
}

func (r *Repl) handleCompressionConfig(args []string) error {
    utils.PrintHeader("Compression Configuration")
    utils.PrintInfo(fmt.Sprintf("  Enabled:    %v", r.config.Compression.Enabled))
    utils.PrintInfo(fmt.Sprintf("  Algorithm:  %s", r.config.Compression.Algorithm))
    utils.PrintInfo(fmt.Sprintf("  Level:      %d", r.config.Compression.Level))
    utils.PrintInfo(fmt.Sprintf("  Min Size:   %s", utils.FormatBytes(int64(r.config.Compression.MinSize))))
    
    utils.PrintInfo("")
    utils.PrintInfo("Available Algorithms:")
    utils.PrintInfo("  snappy  - Fast compression/decompression, good balance (default)")
    utils.PrintInfo("  lz4     - Extremely fast, lower compression ratio")
    utils.PrintInfo("  zstd    - High compression ratio, slower")
    
    return nil
}

// ========== Обработчики команд кластера ==========

func (r *Repl) handleStatus(args []string) error {
    utils.PrintHeader("Cluster Status")
    
    if r.coordinator == nil {
        utils.PrintWarning("  Cluster coordinator not available")
        return nil
    }
    
    isLeader := r.coordinator.IsLeader()
    if isLeader {
        utils.PrintSuccess("  Role: LEADER")
    } else {
        utils.PrintWarning("  Role: FOLLOWER")
    }
    
    leader := r.coordinator.GetLeader()
    if leader != nil {
        utils.PrintInfo(fmt.Sprintf("  Leader: %s:%d", leader.IP, leader.Port))
    }
    
    clusterStatus := r.coordinator.GetClusterStatus()
    utils.PrintInfo(fmt.Sprintf("  Cluster Name: %s", clusterStatus.Name))
    utils.PrintInfo(fmt.Sprintf("  Total Nodes: %d", clusterStatus.TotalNodes))
    utils.PrintInfo(fmt.Sprintf("  Active Nodes: %d", clusterStatus.ActiveNodes))
    utils.PrintInfo(fmt.Sprintf("  Health: %s", clusterStatus.Health))
    utils.PrintInfo(fmt.Sprintf("  Replication Factor: %d", clusterStatus.ReplicationFactor))
    
    return nil
}

func (r *Repl) handleNodes(args []string) error {
    if r.coordinator == nil {
        return fmt.Errorf("cluster coordinator not available")
    }
    
    nodes := r.coordinator.GetAllNodes()
    
    if len(nodes) == 0 {
        utils.PrintInfo("No nodes in cluster")
        return nil
    }
    
    utils.PrintHeader("Cluster Nodes")
    leader := r.coordinator.GetLeader()
    
    for _, node := range nodes {
        prefix := "  "
        if leader != nil && node.ID == leader.ID {
            prefix = " *"
        }
        statusColor := ""
        switch node.Status {
        case "active":
            statusColor = "\033[32m" // green
        case "syncing":
            statusColor = "\033[33m" // yellow
        default:
            statusColor = "\033[31m" // red
        }
        utils.Println(fmt.Sprintf("%s %s:%d [%s%s\033[0m] (last seen: %s)", 
            prefix, node.IP, node.Port, statusColor, node.Status, time.Unix(node.LastSeen, 0).Format("15:04:05")))
    }
    
    return nil
}

// ========== Системные обработчики ==========

func (r *Repl) handleHelp(args []string) error {
    utils.Println("")
    //utils.PrintHeader("Available Commands") Old prompt
    fmt.Println(color.New(color.FgHiCyan).Sprint("\n=== Available Commands ==="))
    
    categories := map[string][]struct {
        cmd         string
        description string
    }{
        "Database Management": {
            {"create database <name>", "Create a new database"},
            {"drop database <name>", "Delete an existing database"},
            {"use <database>", "Switch to a specific database"},
            {"show databases", "List all available databases"},
        },
        "Collection Management": {
            {"create collection <name>", "Create a new collection in current database"},
            {"drop collection <name>", "Delete a collection from current database"},
            {"show collections", "List all collections in current database"},
        },
        "Document Operations": {
            {"insert <collection> <json>", "Insert a new document (JSON format: key=value,key2=value2)"},
            {"find <collection> <id>", "Find a document by its ID"},
            {"findbyindex <collection> <index> <value>", "Find documents using an index"},
            {"update <collection> <id> <field=value>...", "Update fields of an existing document"},
            {"delete <collection> <id>", "Delete a document by its ID"},
            {"count <collection>", "Count total documents in a collection"},
        },
        "Index Management": {
            {"create index <collection> <name> <fields> [unique]", "Create a new index on specified fields"},
            {"drop index <collection> <name>", "Remove an existing index"},
            {"show indexes <collection>", "List all indexes on a collection"},
        },
        "Constraints": {
            {"add required <collection> <field>", "Add a required field constraint"},
            {"add unique <collection> <field>", "Add a unique constraint on a field"},
            {"add min <collection> <field> <value>", "Add a minimum value constraint for numeric fields"},
            {"add max <collection> <field> <value>", "Add a maximum value constraint for numeric fields"},
            {"add enum <collection> <field> <values...>", "Add allowed values constraint (enum)"},
        },
        "Transactions": {
            {"begin transaction", "Start a new transaction"},
            {"commit", "Commit the current transaction"},
            {"rollback", "Rollback the current transaction"},
            {"show transactions", "List all active transactions"},
        },
        "Triggers (MongoDB-like)": {
            {"create trigger <collection> <name> <event> <action> [options]", "Create a trigger on collection events"},
            {"  Events: BEFORE_INSERT, AFTER_INSERT, BEFORE_UPDATE, AFTER_UPDATE, BEFORE_DELETE, AFTER_DELETE", ""},
            {"  Actions: abort, skip, modify, log, notify", ""},
            {"  Special values: $$NOW (current timestamp), $$USER (current user), $$ROLE (current role)", ""},
            {"drop trigger <collection> <event> <name>", "Remove a trigger from collection"},
            {"show triggers <collection>", "List all triggers on a collection"},
            {"enable trigger <collection> <event> <name>", "Enable a disabled trigger"},
            {"disable trigger <collection> <event> <name>", "Disable a trigger without removing it"},
            {"trigger log", "Show trigger execution history"},
        },
        "Plugins": {
            {"plugin list", "List all loaded plugins"},
            {"plugin load <name> <filepath>", "Load a plugin from Lua file"},
            {"plugin unload <name>", "Unload a plugin"},
            {"plugin start <name>", "Start a loaded plugin"},
            {"plugin stop <name>", "Stop a running plugin"},
            {"plugin exec <plugin> <function> [args...]", "Execute a plugin function"},
        },
        "Import/Export": {
            {"export <database> <filename>", "Export entire database to file"},
            {"import <database> <filename>", "Import database from file"},
        },
        "Compression": {
            {"compression stats", "Show compression statistics for current database"},
            {"compression config", "Display current compression settings"},
            {"compress collection <name>", "Manually compress all documents in a collection"},
            {"doc compression <collection> <id>", "Show compression info for a specific document"},
        },
        "Access Control": {
            {"acl login <username> <password>", "Authenticate with username and password"},
            {"acl logout", "Logout current user session"},
            {"acl grant <collection> <role> <permissions>", "Grant permissions (r=read,w=write,d=delete,a=admin)"},
            {"acl users", "List all users"},
            {"acl roles", "List all roles"},
        },
        "Cluster": {
            {"status", "Show current cluster status and role (leader/follower)"},
            {"nodes", "List all nodes in the cluster"},
        },
        "System": {
            {"help", "Display this help message with all available commands"},
            {"clear", "Clear the terminal screen"},
            {"quit", "Exit the futriix database REPL"},
            {"exit", "Exit the futriix database REPL (alias for quit)"},
        },
    }
    
    for category, commands := range categories {
        utils.PrintInfo(fmt.Sprintf("\n%s:", category))
        for _, cmd := range commands {
            if cmd.description != "" {
                utils.Println(fmt.Sprintf("  %-50s %s", cmd.cmd, cmd.description))
            } else {
                utils.Println(fmt.Sprintf("  %s", cmd.cmd))
            }
        }
    }
    
    utils.Println("")
    utils.PrintInfo("Examples:")
    utils.Println("  create database test")
    utils.Println("  use test")
    utils.Println("  create collection users")
    utils.Println("  insert users name=John,age=30")
    utils.Println("  find users john123")
    utils.Println("  create index users idx_name name")
    utils.Println("  begin transaction")
    utils.Println("  update users john123 age=31")
    utils.Println("  commit")
    utils.Println("  create trigger users audit_log AFTER_INSERT log")
    utils.Println("  plugin list")
    utils.Println("  acl login admin admin")
    utils.Println("  status")
    utils.Println("")
    
    return nil
}

func (r *Repl) handleClear(args []string) error {
    fmt.Print("\033[2J\033[H")
    return nil
}

func (r *Repl) handleQuit(args []string) error {
    utils.PrintInfo("Goodbye!")
    os.Exit(0)
    return nil
}

// Close закрывает REPL и сохраняет историю
func (r *Repl) Close() error {
    if r.historyFile != nil {
        r.historyFile.Save()
    }
    return nil
}