#include "perechentablemodel.h"
#include "componentmodel.h"
#include "../view/pagebreakconfirmationdialog.h"
#include <QDebug>
#include <QJsonDocument>
#include <QJsonObject>
#include <QApplication>
#include <QJsonArray>
#include <QColor>
#include <QBrush>
#include <QFont>
#include <QFontMetrics>
#include <QSet>
#include <QPair>

// Регистрируем структуры для работы с QVariant
Q_DECLARE_METATYPE(PerechenCellData)
Q_DECLARE_METATYPE(PerechenRowData)

// Статические константы
const QStringList PerechenTableModel::DEFAULT_HEADERS = QStringList() << "Поз.обозначение" << "Наименование" << "Кол." << "Примечание";

PerechenTableModel::PerechenTableModel(QObject *parent)
    : QAbstractTableModel(parent)
{
    // Настройки колонок будут установлены позже из базы данных или через setColumnMappings
    // Не устанавливаем жестко заданные значения по умолчанию
    qDebug() << "PerechenTableModel::PerechenTableModel: Создана модель без настроек колонок по умолчанию";
}

int PerechenTableModel::rowCount(const QModelIndex &parent) const
{
    if (parent.isValid())
        return 0;
    return m_rows.size();
}

int PerechenTableModel::columnCount(const QModelIndex &parent) const
{
    if (parent.isValid())
        return 0;
    return COLUMN_COUNT;
}

QVariant PerechenTableModel::data(const QModelIndex &index, int role) const
{
    if (!index.isValid() || index.row() >= m_rows.size())
        return QVariant();

    const PerechenRowData &row = m_rows[index.row()];
    
    if (role == Qt::DisplayRole || role == Qt::EditRole) {
        switch (index.column()) {
        case 0: return row.position.value;
        case 1: return row.designation.value;
        case 2: return row.quantity.value;
        case 3: return row.note.value;
        default: return QVariant();
        }
    } else if (role == Qt::BackgroundRole) {
        if (row.isHeader) {
            return QBrush(QColor(200, 200, 200));
        } else {
            // Проверяем переполнение ячейки (используем приблизительную ширину колонки)
            // Приблизительные ширины колонок для Perechen (в пикселях)
            static const qreal columnWidths[] = {100, 200, 60, 150}; // Поз., Наименование, Кол., Примечание
            qreal columnWidth = (index.column() < 4) ? columnWidths[index.column()] : 150;
            
            // Проверяем переполнение ячейки (используем информацию из PDF или проверку)
            QPair<int, int> cellKey(index.row(), index.column());
            if (m_overflowCells.contains(cellKey)) {
                // Подсвечиваем красным, если ячейка переполнена (помечена из PDF)
                return QBrush(QColor(255, 200, 200)); // Светло-красный
            }
            
            // Чередуем оттенки фона для разных страниц
            // Нечетные страницы - светло-голубой, четные - светло-желтый
            if (row.pageNumber % 2 == 1) {
                return QBrush(QColor(240, 248, 255)); // AliceBlue - для нечетных страниц
            } else {
                return QBrush(QColor(255, 255, 240)); // Ivory - для четных страниц
            }
        }
    } else if (role == Qt::FontRole) {
        QFont font;
        bool isBold = false;
        bool isUnderline = false;
        
        // Проверяем свойства ячейки
        switch (index.column()) {
        case 0:
            isBold = row.position.isHeader;
            isUnderline = row.position.isUnderline;
            break;
        case 1:
            isBold = row.designation.isHeader;
            isUnderline = row.designation.isUnderline;
            break;
        case 2:
            isBold = row.quantity.isHeader;
            isUnderline = row.quantity.isUnderline;
            break;
        case 3:
            isBold = row.note.isHeader;
            isUnderline = row.note.isUnderline;
            break;
        default:
            break;
        }
        
        if (isBold) {
            font.setBold(true);
        }
        if (isUnderline) {
            font.setUnderline(true);
        }
        
        if (isBold || isUnderline) {
            return font;
        }
    } else if (role == Qt::UserRole) {
        // Возвращаем структуру ячейки для дополнительной информации
        switch (index.column()) {
        case 0: return QVariant::fromValue(row.position);
        case 1: return QVariant::fromValue(row.designation);
        case 2: return QVariant::fromValue(row.quantity);
        case 3: return QVariant::fromValue(row.note);
        default: return QVariant();
        }
    }
    
    return QVariant();
}

QVariant PerechenTableModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if (role != Qt::DisplayRole)
        return QVariant();

    if (orientation == Qt::Horizontal) {
        if (section < DEFAULT_HEADERS.size()) {
            return DEFAULT_HEADERS[section];
        }
    } else {
        // Вертикальные заголовки - номера строк с указанием страницы
        if (section < m_rows.size()) {
            const PerechenRowData &row = m_rows[section];
            return QString("%1 (стр. %2)").arg(section + 1).arg(row.pageNumber);
        } else {
        return section + 1;
        }
    }

    return QVariant();
}

bool PerechenTableModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
    if (!index.isValid() || index.row() >= m_rows.size() || role != Qt::EditRole)
        return false;

    // Проверяем, можно ли редактировать эту строку
    if (!canEditRow(index.row())) {
        qDebug() << "PerechenTableModel::setData: Строка" << index.row() << "не может быть отредактирована";
        return false;
    }

    PerechenRowData &row = m_rows[index.row()];
    QString newValue = value.toString();
    
    switch (index.column()) {
    case 0:
        row.position.value = newValue;
        row.position.isAutoGenerated = false;
        row.position.isEmpty = newValue.isEmpty();
        break;
    case 1:
        row.designation.value = newValue;
        row.designation.isAutoGenerated = false;
        row.designation.isEmpty = newValue.isEmpty();
        break;
    case 2:
        row.quantity.value = newValue;
        row.quantity.isAutoGenerated = false;
        row.quantity.isEmpty = newValue.isEmpty();
        break;
    case 3:
        row.note.value = newValue;
        row.note.isAutoGenerated = false;
        row.note.isEmpty = newValue.isEmpty();
        break;
    default:
        return false;
    }

    // Обновляем флаг isEmpty для строки, если все ячейки пустые
    updateRowEmptyStatus(index.row());

    emit dataChanged(index, index, QVector<int>() << role);
    return true;
}

bool PerechenTableModel::setCellStretch(int row, int column, int stretch)
{
    if (row < 0 || row >= m_rows.size() || column < 0 || column >= COLUMN_COUNT) {
        return false;
    }
    
    PerechenRowData &rowData = m_rows[row];
    bool changed = false;
    
    switch (column) {
    case 0:
        if (rowData.position.stretch != stretch) {
            rowData.position.stretch = stretch;
            changed = true;
        }
        break;
    case 1:
        if (rowData.designation.stretch != stretch) {
            rowData.designation.stretch = stretch;
            changed = true;
        }
        break;
    case 2:
        if (rowData.quantity.stretch != stretch) {
            rowData.quantity.stretch = stretch;
            changed = true;
        }
        break;
    case 3:
        if (rowData.note.stretch != stretch) {
            rowData.note.stretch = stretch;
            changed = true;
        }
        break;
    default:
        return false;
    }
    
    if (changed) {
        QModelIndex index = createIndex(row, column);
        emit dataChanged(index, index, QVector<int>() << Qt::UserRole);
        return true;
    }
    
    return false;
}

bool PerechenTableModel::setCellHeader(int row, int column, bool isHeader)
{
    if (row < 0 || row >= m_rows.size() || column < 0 || column >= COLUMN_COUNT) {
        return false;
    }
    
    PerechenRowData &rowData = m_rows[row];
    bool changed = false;
    
    switch (column) {
    case 0:
        if (rowData.position.isHeader != isHeader) {
            rowData.position.isHeader = isHeader;
            changed = true;
        }
        break;
    case 1:
        if (rowData.designation.isHeader != isHeader) {
            rowData.designation.isHeader = isHeader;
            changed = true;
        }
        break;
    case 2:
        if (rowData.quantity.isHeader != isHeader) {
            rowData.quantity.isHeader = isHeader;
            changed = true;
        }
        break;
    case 3:
        if (rowData.note.isHeader != isHeader) {
            rowData.note.isHeader = isHeader;
            changed = true;
        }
        break;
    default:
        return false;
    }
    
    if (changed) {
        QModelIndex index = createIndex(row, column);
        emit dataChanged(index, index, QVector<int>() << Qt::UserRole);
        return true;
    }
    
    return false;
}

bool PerechenTableModel::setCellUnderline(int row, int column, bool isUnderline)
{
    if (row < 0 || row >= m_rows.size() || column < 0 || column >= COLUMN_COUNT) {
        return false;
    }
    
    PerechenRowData &rowData = m_rows[row];
    bool changed = false;
    
    switch (column) {
    case 0:
        if (rowData.position.isUnderline != isUnderline) {
            rowData.position.isUnderline = isUnderline;
            changed = true;
        }
        break;
    case 1:
        if (rowData.designation.isUnderline != isUnderline) {
            rowData.designation.isUnderline = isUnderline;
            changed = true;
        }
        break;
    case 2:
        if (rowData.quantity.isUnderline != isUnderline) {
            rowData.quantity.isUnderline = isUnderline;
            changed = true;
        }
        break;
    case 3:
        if (rowData.note.isUnderline != isUnderline) {
            rowData.note.isUnderline = isUnderline;
            changed = true;
        }
        break;
    default:
        return false;
    }
    
    if (changed) {
        QModelIndex index = createIndex(row, column);
        emit dataChanged(index, index, QVector<int>() << Qt::UserRole);
        return true;
    }
    
    return false;
}

bool PerechenTableModel::isCellOverflow(int row, int column, qreal columnWidth, int fontSize) const
{
    if (row < 0 || row >= m_rows.size() || column < 0 || column >= COLUMN_COUNT) {
        return false;
    }
    
    const PerechenRowData &rowData = m_rows[row];
    QString cellText;
    int cellStretch = 100;
    
    // Получаем текст и поджим ячейки
    switch (column) {
    case 0: cellText = rowData.position.value; cellStretch = rowData.position.stretch; break;
    case 1: cellText = rowData.designation.value; cellStretch = rowData.designation.stretch; break;
    case 2: cellText = rowData.quantity.value; cellStretch = rowData.quantity.stretch; break;
    case 3: cellText = rowData.note.value; cellStretch = rowData.note.stretch; break;
    default: return false;
    }
    
    if (cellText.isEmpty()) {
        return false;
    }
    
    // Используем ту же логику, что и в PDF: масштабирование вместо setStretch
    // Создаем шрифт без поджима для измерения ширины
    QFont testFont("Arial", fontSize);
    QFontMetrics metrics(testFont);
    
    // Вычисляем коэффициент масштабирования (как в drawCellWithStretch)
    qreal scaleX = cellStretch / 100.0;
    
    // Вычисляем ширину текста после масштабирования
    qreal textWidth = metrics.horizontalAdvance(cellText) * scaleX;
    
    // Используем ту же логику, что и в PDF: если текст обрезается с многоточием, это переполнение
    // Вычисляем максимальную ширину текста до масштабирования (как в drawCellWithStretch)
    qreal maxTextWidth = columnWidth / scaleX;
    
    // Проверяем, будет ли текст обрезан с многоточием (как в PDF)
    // Если исходный текст шире, чем максимальная ширина, то он будет обрезан
    qreal originalTextWidth = metrics.horizontalAdvance(cellText);
    
    // Добавляем небольшой запас для учета возможных расхождений между Arial и GOST шрифтами
    qreal tolerance = 3.0;
    
    // Переполнение, если текст будет обрезан с многоточием
    return originalTextWidth > (maxTextWidth + tolerance);
}

void PerechenTableModel::setCellOverflow(int row, int column, bool overflow)
{
    if (row < 0 || row >= m_rows.size() || column < 0 || column >= COLUMN_COUNT) {
        return;
    }
    
    QPair<int, int> cellKey(row, column);
    
    if (overflow) {
        if (!m_overflowCells.contains(cellKey)) {
            m_overflowCells.insert(cellKey);
            QModelIndex index = createIndex(row, column);
            emit dataChanged(index, index, QVector<int>() << Qt::BackgroundRole);
        }
    } else {
        if (m_overflowCells.contains(cellKey)) {
            m_overflowCells.remove(cellKey);
            QModelIndex index = createIndex(row, column);
            emit dataChanged(index, index, QVector<int>() << Qt::BackgroundRole);
        }
    }
}

void PerechenTableModel::clearOverflowMarks()
{
    if (m_overflowCells.isEmpty()) {
        return;
    }
    
    // Сохраняем индексы для обновления
    QList<QModelIndex> indicesToUpdate;
    QSet<QPair<int, int>>::const_iterator it;
    for (it = m_overflowCells.constBegin(); it != m_overflowCells.constEnd(); ++it) {
        const QPair<int, int> &cellKey = *it;
        indicesToUpdate.append(createIndex(cellKey.first, cellKey.second));
    }
    
    m_overflowCells.clear();
    
    // Обновляем все затронутые ячейки
    for (const QModelIndex &index : indicesToUpdate) {
        emit dataChanged(index, index, QVector<int>() << Qt::BackgroundRole);
    }
}

Qt::ItemFlags PerechenTableModel::flags(const QModelIndex &index) const
{
    if (!index.isValid())
        return Qt::NoItemFlags;

    Qt::ItemFlags flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable;
    
    // Разрешаем редактирование только для редактируемых строк
    if (canEditRow(index.row())) {
        flags |= Qt::ItemIsEditable;
    }
    
    return flags;
}

void PerechenTableModel::setColumnMappings(const QStringList &mappings)
{
    qDebug() << "PerechenTableModel::setColumnMappings: Устанавливаем настройки колонок:" << mappings;
    qDebug() << "PerechenTableModel::setColumnMappings: Текущие настройки:" << m_columnMappings;
    
    if (m_columnMappings != mappings) {
        m_columnMappings = mappings;
        qDebug() << "PerechenTableModel::setColumnMappings: Настройки обновлены";
        emit columnMappingsChanged();
    } else {
        qDebug() << "PerechenTableModel::setColumnMappings: Настройки не изменились";
    }
}

QStringList PerechenTableModel::columnMappings() const
{
    qDebug() << "PerechenTableModel::columnMappings: Возвращаем настройки:" << m_columnMappings;
    return m_columnMappings;
}

void PerechenTableModel::addRow(const PerechenRowData &rowData)
{
    beginInsertRows(QModelIndex(), m_rows.size(), m_rows.size());
    m_rows.append(rowData);
    endInsertRows();
    // Автоматический пересчет отключен - пересчет выполняется по кнопке "Пересчет таблицы"
    // updateRowNumbers();
    
    // Оптимизируем разрывы страниц после добавления строки
    // if (rowData.isHeader) {
    //     optimizePageBreaks();
    // }
    
    emit modelDataChanged();
}

void PerechenTableModel::insertRow(int position, const PerechenRowData &rowData)
{
    if (position < 0 || position > m_rows.size())
        return;

    beginInsertRows(QModelIndex(), position, position);
    m_rows.insert(position, rowData);
    endInsertRows();
    // Автоматический пересчет отключен - пересчет выполняется по кнопке "Пересчет таблицы"
    // updateRowNumbers();
    
    // Оптимизируем разрывы страниц после вставки строки
    // if (rowData.isHeader) {
    //     optimizePageBreaks();
    // }
    
    emit modelDataChanged();
}

void PerechenTableModel::removeRow(int row)
{
    if (row < 0 || row >= m_rows.size())
        return;

    beginRemoveRows(QModelIndex(), row, row);
    m_rows.removeAt(row);
    endRemoveRows();
    // Автоматический пересчет отключен - пересчет выполняется по кнопке "Пересчет таблицы"
    // updateRowNumbers();
    
    // Оптимизируем разрывы страниц после удаления строки
    // optimizePageBreaks();
    
    emit modelDataChanged();
}

void PerechenTableModel::clear()
{
    qDebug() << "PerechenTableModel::clear: Очищаем таблицу, строк:" << m_rows.size();
    
    if (m_rows.isEmpty()) {
        qDebug() << "PerechenTableModel::clear: Таблица уже пуста";
        return;
    }

    beginResetModel();
    m_rows.clear();
    endResetModel();
    
    qDebug() << "PerechenTableModel::clear: Таблица очищена";
    emit modelDataChanged();
}

void PerechenTableModel::setComponents(const QList<ComponentModel*> &components)
{
    qDebug() << "PerechenTableModel::setComponents: Получено" << components.size() << "компонентов";
    
    // Проверяем компоненты на null
    QList<ComponentModel*> validComponents;
    for (ComponentModel *component : components) {
        if (component) {
            validComponents.append(component);
        } else {
            qDebug() << "PerechenTableModel::setComponents: Пропускаем null компонент";
        }
    }
    
    // Сохраняем только валидные компоненты
    m_components = validComponents;
    
    qDebug() << "PerechenTableModel::setComponents: Сохранено" << m_components.size() << "валидных компонентов";
}



void PerechenTableModel::setPropertyNames(const QStringList &propertyNames)
{
    qDebug() << "PerechenTableModel::setPropertyNames: получено" << propertyNames;
    m_propertyNames = propertyNames;
    
    // Обновляем маппинг по умолчанию только если настройки колонок не были установлены явно
    // и если это первая установка свойств
    if (m_columnMappings.isEmpty() && m_propertyNames.contains("Designator")) {
        qDebug() << "PerechenTableModel::setPropertyNames: Устанавливаем настройки колонок по умолчанию";
        m_columnMappings << "<Авто>" << "Designator" << "<Пусто>" << "<Пусто>";
        emit columnMappingsChanged();
    } else {
        qDebug() << "PerechenTableModel::setPropertyNames: Настройки колонок уже установлены:" << m_columnMappings;
    }
}

QStringList PerechenTableModel::availableProperties() const
{
    QStringList properties = m_propertyNames;
    properties.prepend("<Пусто>");
    properties.prepend("<Авто>");
    qDebug() << "PerechenTableModel::availableProperties: m_propertyNames =" << m_propertyNames;
    qDebug() << "PerechenTableModel::availableProperties: возвращаем" << properties;
    return properties;
}

QByteArray PerechenTableModel::saveToDatabase() const
{
    qDebug() << "PerechenTableModel::saveToDatabase: Начинаем сохранение в БД";
    qDebug() << "PerechenTableModel::saveToDatabase: Количество строк:" << m_rows.size();
    qDebug() << "PerechenTableModel::saveToDatabase: Настройки колонок:" << m_columnMappings;
    
    QJsonObject root;
    QJsonArray rowsArray;
    
    for (const PerechenRowData &row : m_rows) {
        QJsonObject rowObj;
        
        // Сохраняем данные ячеек
        QJsonObject positionObj;
        positionObj["value"] = row.position.value;
        positionObj["rowNumber"] = row.position.rowNumber;
        positionObj["pageNumber"] = row.position.pageNumber;
        positionObj["isHeader"] = row.position.isHeader;
        positionObj["isAutoGenerated"] = row.position.isAutoGenerated;
        positionObj["isEmpty"] = row.position.isEmpty;
        positionObj["stretch"] = row.position.stretch;
        positionObj["isUnderline"] = row.position.isUnderline;
        rowObj["position"] = positionObj;
        
        QJsonObject designationObj;
        designationObj["value"] = row.designation.value;
        designationObj["rowNumber"] = row.designation.rowNumber;
        designationObj["pageNumber"] = row.designation.pageNumber;
        designationObj["isHeader"] = row.designation.isHeader;
        designationObj["isAutoGenerated"] = row.designation.isAutoGenerated;
        designationObj["isEmpty"] = row.designation.isEmpty;
        designationObj["stretch"] = row.designation.stretch;
        designationObj["isUnderline"] = row.designation.isUnderline;
        rowObj["designation"] = designationObj;
        
        QJsonObject quantityObj;
        quantityObj["value"] = row.quantity.value;
        quantityObj["rowNumber"] = row.quantity.rowNumber;
        quantityObj["pageNumber"] = row.quantity.pageNumber;
        quantityObj["isHeader"] = row.quantity.isHeader;
        quantityObj["isAutoGenerated"] = row.quantity.isAutoGenerated;
        quantityObj["isEmpty"] = row.quantity.isEmpty;
        quantityObj["stretch"] = row.quantity.stretch;
        quantityObj["isUnderline"] = row.quantity.isUnderline;
        rowObj["quantity"] = quantityObj;
        
        QJsonObject noteObj;
        noteObj["value"] = row.note.value;
        noteObj["rowNumber"] = row.note.rowNumber;
        noteObj["pageNumber"] = row.note.pageNumber;
        noteObj["isHeader"] = row.note.isHeader;
        noteObj["isAutoGenerated"] = row.note.isAutoGenerated;
        noteObj["isEmpty"] = row.note.isEmpty;
        noteObj["stretch"] = row.note.stretch;
        noteObj["isUnderline"] = row.note.isUnderline;
        rowObj["note"] = noteObj;
        
        // Сохраняем массив поджимов для быстрого доступа
        QJsonArray stretchesArray;
        stretchesArray.append(row.position.stretch);
        stretchesArray.append(row.designation.stretch);
        stretchesArray.append(row.quantity.stretch);
        stretchesArray.append(row.note.stretch);
        rowObj["cell_stretches"] = stretchesArray;
        
        rowObj["isHeader"] = row.isHeader;
        rowObj["isEmpty"] = row.isEmpty;
        rowObj["rowNumber"] = row.rowNumber;
        rowObj["pageNumber"] = row.pageNumber;
        
        rowsArray.append(rowObj);
    }
    
    root["rows"] = rowsArray;
    root["columnMappings"] = QJsonArray::fromStringList(m_columnMappings);
    root["propertyNames"] = QJsonArray::fromStringList(m_propertyNames);
    
    QJsonDocument doc(root);
    QByteArray result = doc.toJson();
    
    qDebug() << "PerechenTableModel::saveToDatabase: Сохранение завершено, размер данных:" << result.size() << "байт";
    return result;
}

void PerechenTableModel::loadFromDatabase(const QByteArray &data)
{
    qDebug() << "PerechenTableModel::loadFromDatabase: Начинаем загрузку из БД, размер данных:" << data.size();
    qDebug() << "PerechenTableModel::loadFromDatabase: Текущие настройки колонок:" << m_columnMappings;
    
    // Проверяем, есть ли уже установленные настройки колонок
    if (!m_columnMappings.isEmpty()) {
        qDebug() << "PerechenTableModel::loadFromDatabase: ВНИМАНИЕ! Настройки колонок уже установлены, будем их защищать от перезаписи";
    }
    
    QJsonDocument doc = QJsonDocument::fromJson(data);
    if (!doc.isObject()) {
        qDebug() << "PerechenTableModel::loadFromDatabase: Ошибка парсинга JSON";
        return;
    }
    
    qDebug() << "PerechenTableModel::loadFromDatabase: JSON успешно распарсен";

    QJsonObject root = doc.object();
    
    // Загружаем маппинг колонок
    if (root.contains("columnMappings")) {
        QJsonArray mappingsArray = root["columnMappings"].toArray();
        QStringList mappings;
        for (const QJsonValue &value : mappingsArray) {
            mappings.append(value.toString());
        }
        qDebug() << "PerechenTableModel::loadFromDatabase: Загружены настройки колонок из БД:" << mappings;
        
        // Проверяем, не перезаписываем ли мы уже установленные правильные настройки
        if (!m_columnMappings.isEmpty() && m_columnMappings != mappings) {
            qDebug() << "PerechenTableModel::loadFromDatabase: ВНИМАНИЕ! Настройки колонок уже установлены:" << m_columnMappings;
            qDebug() << "PerechenTableModel::loadFromDatabase: НЕ перезаписываем настройки колонок из БД, оставляем текущие";
        } else {
            qDebug() << "PerechenTableModel::loadFromDatabase: Устанавливаем настройки колонок из БД";
            m_columnMappings = mappings;
            emit columnMappingsChanged();
        }
    } else {
        qDebug() << "PerechenTableModel::loadFromDatabase: Настройки колонок не найдены в БД";
    }
    
    // Загружаем свойства
    if (root.contains("propertyNames")) {
        QJsonArray propertiesArray = root["propertyNames"].toArray();
        QStringList properties;
        for (const QJsonValue &value : propertiesArray) {
            properties.append(value.toString());
        }
        m_propertyNames = properties;
    }
    
    // Загружаем строки
    clear();
    if (root.contains("rows")) {
        QJsonArray rowsArray = root["rows"].toArray();
        for (const QJsonValue &value : rowsArray) {
            QJsonObject rowObj = value.toObject();
            PerechenRowData row;
            
            // Загружаем данные ячеек
            // Сначала загружаем массив поджимов, если он есть
            QJsonArray stretchesArray;
            if (rowObj.contains("cell_stretches")) {
                stretchesArray = rowObj["cell_stretches"].toArray();
            }
            
            if (rowObj.contains("position")) {
                QJsonObject posObj = rowObj["position"].toObject();
                int stretch = 100;
                if (posObj.contains("stretch")) {
                    stretch = posObj["stretch"].toInt(100);
                } else if (stretchesArray.size() > 0) {
                    stretch = stretchesArray[0].toInt(100);
                }
                bool isUnderline = posObj["isUnderline"].toBool(false);
                row.position = PerechenCellData(
                    posObj["value"].toString(),
                    posObj["rowNumber"].toInt(),
                    posObj["pageNumber"].toInt(),
                    posObj["isHeader"].toBool(),
                    posObj["isAutoGenerated"].toBool(),
                    posObj["isEmpty"].toBool(),
                    stretch,
                    isUnderline
                );
            }
            
            if (rowObj.contains("designation")) {
                QJsonObject desObj = rowObj["designation"].toObject();
                int stretch = 100;
                if (desObj.contains("stretch")) {
                    stretch = desObj["stretch"].toInt(100);
                } else if (stretchesArray.size() > 1) {
                    stretch = stretchesArray[1].toInt(100);
                }
                bool isUnderline = desObj["isUnderline"].toBool(false);
                row.designation = PerechenCellData(
                    desObj["value"].toString(),
                    desObj["rowNumber"].toInt(),
                    desObj["pageNumber"].toInt(),
                    desObj["isHeader"].toBool(),
                    desObj["isAutoGenerated"].toBool(),
                    desObj["isEmpty"].toBool(),
                    stretch,
                    isUnderline
                );
            }
            
            if (rowObj.contains("quantity")) {
                QJsonObject qtyObj = rowObj["quantity"].toObject();
                int stretch = 100;
                if (qtyObj.contains("stretch")) {
                    stretch = qtyObj["stretch"].toInt(100);
                } else if (stretchesArray.size() > 2) {
                    stretch = stretchesArray[2].toInt(100);
                }
                bool isUnderline = qtyObj["isUnderline"].toBool(false);
                row.quantity = PerechenCellData(
                    qtyObj["value"].toString(),
                    qtyObj["rowNumber"].toInt(),
                    qtyObj["pageNumber"].toInt(),
                    qtyObj["isHeader"].toBool(),
                    qtyObj["isAutoGenerated"].toBool(),
                    qtyObj["isEmpty"].toBool(),
                    stretch,
                    isUnderline
                );
            }
            
            if (rowObj.contains("note")) {
                QJsonObject noteObj = rowObj["note"].toObject();
                int stretch = 100;
                if (noteObj.contains("stretch")) {
                    stretch = noteObj["stretch"].toInt(100);
                } else if (stretchesArray.size() > 3) {
                    stretch = stretchesArray[3].toInt(100);
                }
                bool isUnderline = noteObj["isUnderline"].toBool(false);
                row.note = PerechenCellData(
                    noteObj["value"].toString(),
                    noteObj["rowNumber"].toInt(),
                    noteObj["pageNumber"].toInt(),
                    noteObj["isHeader"].toBool(),
                    noteObj["isAutoGenerated"].toBool(),
                    noteObj["isEmpty"].toBool(),
                    stretch,
                    isUnderline
                );
            }
            
            row.isHeader = rowObj["isHeader"].toBool();
            row.isEmpty = rowObj["isEmpty"].toBool();
            row.rowNumber = rowObj["rowNumber"].toInt();
            row.pageNumber = rowObj["pageNumber"].toInt();
            
            addRow(row);
        }
    }
    
    qDebug() << "PerechenTableModel::loadFromDatabase: Загружено строк:" << m_rows.size();
    qDebug() << "PerechenTableModel::loadFromDatabase: Маппинг колонок:" << m_columnMappings;
    qDebug() << "PerechenTableModel::loadFromDatabase: Свойства:" << m_propertyNames;
    
    // Проверяем, что настройки колонок загрузились корректно
    if (m_columnMappings.isEmpty()) {
        qDebug() << "PerechenTableModel::loadFromDatabase: ВНИМАНИЕ! Настройки колонок пусты после загрузки!";
    } else {
        qDebug() << "PerechenTableModel::loadFromDatabase: Настройки колонок после загрузки:" << m_columnMappings;
        
        // Проверяем, не были ли настройки перезаписаны неправильными данными
        if (m_columnMappings.contains("<Авто>") || m_columnMappings.contains("Designator")) {
            qDebug() << "PerechenTableModel::loadFromDatabase: ВНИМАНИЕ! Настройки колонок содержат значения по умолчанию!";
        }
    }
    
    emit columnMappingsChanged();
}

QString PerechenTableModel::getDocumentTitle() const
{
    return "Перечень элементов";
}

QString PerechenTableModel::getDocumentType() const
{
    return "Perechen";
}

void PerechenTableModel::updateRowNumbers()
{
    int rowNumber = 1;
    for (PerechenRowData &row : m_rows) {
        if (!row.isHeader) {
            row.rowNumber = rowNumber++;
        }
    }
}

void PerechenTableModel::updatePageNumbers()
{
    qDebug() << "PerechenTableModel::updatePageNumbers: Обновляем номера страниц";
    
    int currentRow = 0;
    int currentPage = 1;
    int rowsOnCurrentPage = 0;
    
    for (int i = 0; i < m_rows.size(); ++i) {
        PerechenRowData &row = m_rows[i];
        
        // Определяем, сколько строк помещается на текущей странице
        int maxRowsOnPage = (currentPage == 1) ? ROWS_PER_FIRST_PAGE : ROWS_PER_OTHER_PAGE;
        
        // Если текущая страница заполнена, переходим на следующую
        if (rowsOnCurrentPage >= maxRowsOnPage) {
            currentPage++;
            rowsOnCurrentPage = 0;
            maxRowsOnPage = ROWS_PER_OTHER_PAGE; // Начиная со второй страницы
        }
        
        // Устанавливаем номер страницы для строки
        row.pageNumber = currentPage;
        row.position.pageNumber = currentPage;
        row.designation.pageNumber = currentPage;
        row.quantity.pageNumber = currentPage;
        row.note.pageNumber = currentPage;
        
        // Увеличиваем счетчик строк на текущей странице
        rowsOnCurrentPage++;
        currentRow++;
        
        qDebug() << "PerechenTableModel::updatePageNumbers: Строка" << i << "на странице" << currentPage 
                 << "(строк на странице:" << rowsOnCurrentPage << "/" << maxRowsOnPage << ")";
    }
    
    qDebug() << "PerechenTableModel::updatePageNumbers: Обновление завершено, всего страниц:" << currentPage;
    
    // Уведомляем UI об изменении данных для обновления отображения страниц
    if (!m_rows.isEmpty()) {
        emit dataChanged(index(0, 0), index(m_rows.size() - 1, COLUMN_COUNT - 1), 
                        QVector<int>() << Qt::BackgroundRole);
        emit headerDataChanged(Qt::Vertical, 0, m_rows.size() - 1);
    }
}

int PerechenTableModel::getPageForRow(int rowNumber) const
{
    if (rowNumber <= 0) return 1;
    
    int currentRow = 0;
    int currentPage = 1;
    
    for (int i = 0; i < m_rows.size(); ++i) {
        int maxRowsOnPage = (currentPage == 1) ? ROWS_PER_FIRST_PAGE : ROWS_PER_OTHER_PAGE;
        
        if (currentRow + 1 > maxRowsOnPage) {
            currentPage++;
            currentRow = 0;
            maxRowsOnPage = ROWS_PER_OTHER_PAGE;
        }
        
        currentRow++;
        
        if (i + 1 == rowNumber) {
            return currentPage;
        }
    }
    
    return currentPage;
}

bool PerechenTableModel::needsPageBreakProtection(int rowIndex) const
{
    if (rowIndex < 0 || rowIndex >= m_rows.size()) {
        return false;
    }
    
    const PerechenRowData &row = m_rows[rowIndex];
    
    // Проверяем только заголовки групп
    if (!row.isHeader) {
        return false;
    }
    
    // Получаем номер страницы для текущей строки
    int currentPage = row.pageNumber;
    
    // Определяем максимальное количество строк на текущей странице
    int maxRowsOnPage = (currentPage == 1) ? ROWS_PER_FIRST_PAGE : ROWS_PER_OTHER_PAGE;
    
    // Подсчитываем количество строк на текущей странице до этой строки
    int rowsOnCurrentPage = 0;
    for (int i = 0; i < rowIndex; ++i) {
        if (m_rows[i].pageNumber == currentPage) {
            rowsOnCurrentPage++;
        }
    }
    
    // Проверяем, находится ли заголовок на последней или предпоследней строке страницы
    // Заголовки обособлены пустыми строками с двух сторон, поэтому:
    // - Если заголовок на последней строке страницы - нужна защита
    // - Если заголовок на предпоследней строке страницы - нужна защита (чтобы пустая строка после заголовка не оказалась на следующей странице)
    if (rowsOnCurrentPage == maxRowsOnPage - 1 || rowsOnCurrentPage == maxRowsOnPage - 2) {
        // Проверяем, есть ли элементы группы на следующей странице
        for (int i = rowIndex + 1; i < m_rows.size(); ++i) {
            // Если встретили другой заголовок, значит это уже другая группа
            if (m_rows[i].isHeader) {
                break;
            }
            
            // Если элемент группы находится на следующей странице
            if (m_rows[i].pageNumber == currentPage + 1 && !m_rows[i].isEmpty) {
                qDebug() << "PerechenTableModel::needsPageBreakProtection: Заголовок на строке" << rowIndex 
                         << "нуждается в защите от разрыва страницы (элемент группы на странице" << (currentPage + 1) 
                         << ", заголовок на строке" << (rowsOnCurrentPage + 1) << "из" << maxRowsOnPage << ")";
                return true;
            }
        }
    }
    
    return false;
}

void PerechenTableModel::addPageBreakProtection()
{
    qDebug() << "PerechenTableModel::addPageBreakProtection: Добавляем защиту от разрывов страниц";
    
    // Сначала обновляем номера страниц
    updatePageNumbers();
    
    // Собираем информацию о заголовках, которые нуждаются в защите
    QList<PageBreakInfo> pageBreakInfos;
    
    for (int i = 0; i < m_rows.size(); ++i) {
        if (needsPageBreakProtection(i)) {
            const PerechenRowData &row = m_rows[i];
            int currentPage = row.pageNumber;
            int maxRowsOnPage = (currentPage == 1) ? ROWS_PER_FIRST_PAGE : ROWS_PER_OTHER_PAGE;
            
            // Подсчитываем количество строк на текущей странице до этой строки
            int rowsOnCurrentPage = 0;
            for (int j = 0; j < i; ++j) {
                if (m_rows[j].pageNumber == currentPage) {
                    rowsOnCurrentPage++;
                }
            }
            
            // Определяем, сколько пустых строк нужно добавить
            int emptyRowsToAdd = 1; // По умолчанию одна строка
            
            if (rowsOnCurrentPage == maxRowsOnPage - 2) {
                // Заголовок на предпоследней строке - добавляем две пустые строки
                emptyRowsToAdd = 2;
                qDebug() << "PerechenTableModel::addPageBreakProtection: Заголовок на предпоследней строке, добавляем" << emptyRowsToAdd << "пустые строки перед строкой" << i;
            } else {
                qDebug() << "PerechenTableModel::addPageBreakProtection: Заголовок на последней строке, добавляем" << emptyRowsToAdd << "пустую строку перед строкой" << i;
            }
            
            // Создаем информацию о переносе
            PageBreakInfo info;
            info.rowIndex = i;
            info.headerText = row.designation.value;
            info.currentPage = currentPage;
            info.nextPage = currentPage + 1;
            info.emptyRowsToAdd = emptyRowsToAdd;
            
            pageBreakInfos.append(info);
        }
    }
    
    // Если есть заголовки для защиты, запрашиваем подтверждение
    if (!pageBreakInfos.isEmpty()) {
        qDebug() << "PerechenTableModel::addPageBreakProtection: Найдено" << pageBreakInfos.size() << "заголовков для защиты";
        
        // Находим главное окно приложения
        QWidget *mainWindow = nullptr;
        for (QWidget *widget : QApplication::topLevelWidgets()) {
            if (widget->isVisible() && widget->windowTitle().contains("GostGenerator", Qt::CaseInsensitive)) {
                mainWindow = widget;
                break;
            }
        }
        
        // Создаем и показываем диалог подтверждения
        PageBreakConfirmationDialog dialog(pageBreakInfos, mainWindow);
        if (dialog.exec() == QDialog::Accepted) {
            // Получаем подтвержденные строки
            QList<int> confirmedRows = dialog.getConfirmedRows();
            
            // Добавляем пустые строки только для подтвержденных заголовков
            for (int i = 0; i < m_rows.size(); ++i) {
                if (needsPageBreakProtection(i) && confirmedRows.contains(i)) {
                    const PerechenRowData &row = m_rows[i];
                    int currentPage = row.pageNumber;
                    int maxRowsOnPage = (currentPage == 1) ? ROWS_PER_FIRST_PAGE : ROWS_PER_OTHER_PAGE;
                    
                    // Подсчитываем количество строк на текущей странице до этой строки
                    int rowsOnCurrentPage = 0;
                    for (int j = 0; j < i; ++j) {
                        if (m_rows[j].pageNumber == currentPage) {
                            rowsOnCurrentPage++;
                        }
                    }
                    
                    // Определяем, сколько пустых строк нужно добавить
                    int emptyRowsToAdd = 1; // По умолчанию одна строка
                    
                    if (rowsOnCurrentPage == maxRowsOnPage - 2) {
                        // Заголовок на предпоследней строке - добавляем две пустые строки
                        emptyRowsToAdd = 2;
                        qDebug() << "PerechenTableModel::addPageBreakProtection: Добавляем" << emptyRowsToAdd << "пустые строки перед строкой" << i;
                    } else {
                        qDebug() << "PerechenTableModel::addPageBreakProtection: Добавляем" << emptyRowsToAdd << "пустую строку перед строкой" << i;
                    }
                    
                    // Добавляем нужное количество пустых строк перед заголовком
                    for (int k = 0; k < emptyRowsToAdd; ++k) {
                        PerechenRowData emptyRow = createEmptyRow();
                        insertRow(i, emptyRow);
                    }
                    
                    // Обновляем номера страниц после вставки
                    updatePageNumbers();
                    
                    // Пропускаем вставленные строки
                    i += emptyRowsToAdd;
                }
            }
        } else {
            qDebug() << "PerechenTableModel::addPageBreakProtection: Пользователь отменил добавление защиты от разрывов страниц";
        }
    }
    
    // Обновляем отображение страниц в UI
    updatePageDisplay();
    
    qDebug() << "PerechenTableModel::addPageBreakProtection: Защита от разрывов страниц добавлена";
}

void PerechenTableModel::optimizePageBreaks()
{
    qDebug() << "PerechenTableModel::optimizePageBreaks: Оптимизируем разрывы страниц";
    
    // Добавляем защиту от разрывов страниц
    addPageBreakProtection();
    
    qDebug() << "PerechenTableModel::optimizePageBreaks: Оптимизация завершена";
}

void PerechenTableModel::updatePageDisplay()
{
    qDebug() << "PerechenTableModel::updatePageDisplay: Обновляем отображение страниц в UI";
    
    // Обновляем номера страниц
    updatePageNumbers();
    
    // Уведомляем UI об изменении всех данных для обновления отображения
    if (!m_rows.isEmpty()) {
        emit dataChanged(index(0, 0), index(m_rows.size() - 1, COLUMN_COUNT - 1), 
                        QVector<int>() << Qt::BackgroundRole);
        
        // Обновляем вертикальные заголовки
        emit headerDataChanged(Qt::Vertical, 0, m_rows.size() - 1);
    }
    
    qDebug() << "PerechenTableModel::updatePageDisplay: Отображение страниц обновлено";
}

PerechenCellData PerechenTableModel::createCellData(const QString &value, bool isHeader, 
                                                   bool isAuto, bool isEmpty)
{
    return PerechenCellData(value, 0, 1, isHeader, isAuto, isEmpty);
} 

// Новые методы для управления строками
void PerechenTableModel::addEmptyRow()
{
    PerechenRowData emptyRow = createEmptyRow();
    addRow(emptyRow);
}

void PerechenTableModel::addEmptyRowAt(int position)
{
    PerechenRowData emptyRow = createEmptyRow();
    insertRow(position, emptyRow);
}

PerechenRowData PerechenTableModel::createEmptyRow() const
{
    PerechenRowData row;
    row.isHeader = false;
    row.isEmpty = false; // Строка не пустая, так как может быть отредактирована
    row.rowNumber = 0; // Будет обновлено в updateRowNumbers()
    row.pageNumber = 1;
    
    // Создаем пустые ячейки, но они могут быть отредактированы
    row.position = PerechenCellData("", 0, 1, false, false, true);
    row.designation = PerechenCellData("", 0, 1, false, false, true);
    row.quantity = PerechenCellData("", 0, 1, false, false, true);
    row.note = PerechenCellData("", 0, 1, false, false, true);
    
    return row;
}

bool PerechenTableModel::canEditRow(int row) const
{
    if (row < 0 || row >= m_rows.size())
        return false;
    
    // Разрешаем редактирование и удаление всех строк, включая заголовки и пустые
    return true;
} 

void PerechenTableModel::updateRowEmptyStatus(int row)
{
    if (row < 0 || row >= m_rows.size())
        return;
    
    PerechenRowData &rowData = m_rows[row];
    
    // Проверяем, пустые ли все ячейки
    bool allEmpty = rowData.position.value.isEmpty() &&
                   rowData.designation.value.isEmpty() &&
                   rowData.quantity.value.isEmpty() &&
                   rowData.note.value.isEmpty();
    
    // Обновляем флаг isEmpty для строки
    if (rowData.isEmpty != allEmpty) {
        rowData.isEmpty = allEmpty;
        qDebug() << "PerechenTableModel::updateRowEmptyStatus: Строка" << row << "обновлена, isEmpty =" << allEmpty;
    }
} 

bool PerechenTableModel::isRowHeader(int row) const
{
    if (row < 0 || row >= m_rows.size())
        return false;
    
    return m_rows[row].isHeader;
} 
