main.cpp

#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <limits>
#include "headers/headers.hpp"

class TextEditor {
private:
    std::string current_file;
    std::string content;
    Cryptograph crypto;
    bool is_modified;

    void display_menu() {
        std::cout << "\n=== ТЕКСТОВЫЙ РЕДАКТОР С ШИФРОВАНИЕМ ===\n";
        std::cout << "1. Создать новый документ\n";
        std::cout << "2. Открыть документ\n";
        std::cout << "3. Сохранить документ\n";
        std::cout << "4. Сохранить как...\n";
        std::cout << "5. Показать содержимое\n";
        std::cout << "6. Редактировать содержимое\n";
        std::cout << "7. Зашифровать и сохранить\n";
        std::cout << "8. Открыть зашифрованный документ\n";
        std::cout << "9. Настройки шифрования\n";
        std::cout << "0. Выход\n";
        std::cout << "Выберите действие: ";
    }

    void create_new_document() {
        if (is_modified && !content.empty()) {
            std::cout << "Есть несохраненные изменения. Сохранить? (y/n): ";
            char choice;
            std::cin >> choice;
            std::cin.ignore();
            if (choice == 'y' || choice == 'Y') {
                save_document();
            }
        }
        
        current_file.clear();
        content.clear();
        is_modified = false;
        std::cout << "Создан новый документ.\n";
    }

    void open_document() {
        std::string filename;
        std::cout << "Введите имя файла для открытия: ";
        std::cin >> filename;
        std::cin.ignore();

        std::ifstream file(filename);
        if (!file.is_open()) {
            std::cout << "Ошибка: Не удалось открыть файл " << filename << "\n";
            return;
        }

        content.clear();
        std::string line;
        while (std::getline(file, line)) {
            content += line + "\n";
        }
        file.close();

        current_file = filename;
        is_modified = false;
        std::cout << "Файл " << filename << " успешно открыт.\n";
    }

    void save_document() {
        if (current_file.empty()) {
            save_document_as();
            return;
        }

        std::ofstream file(current_file);
        if (!file.is_open()) {
            std::cout << "Ошибка: Не удалось сохранить файл " << current_file << "\n";
            return;
        }

        file << content;
        file.close();
        is_modified = false;
        std::cout << "Файл " << current_file << " успешно сохранен.\n";
    }

    void save_document_as() {
        std::string filename;
        std::cout << "Введите имя файла для сохранения: ";
        std::cin >> filename;
        std::cin.ignore();

        std::ofstream file(filename);
        if (!file.is_open()) {
            std::cout << "Ошибка: Не удалось создать файл " << filename << "\n";
            return;
        }

        file << content;
        file.close();
        current_file = filename;
        is_modified = false;
        std::cout << "Файл " << filename << " успешно сохранен.\n";
    }

    void show_content() {
        if (content.empty()) {
            std::cout << "Документ пуст.\n";
            return;
        }

        std::cout << "\n=== СОДЕРЖИМОЕ ДОКУМЕНТА ===\n";
        std::cout << "Файл: " << (current_file.empty() ? "Не сохранен" : current_file) << "\n";
        std::cout << "Размер: " << content.length() << " символов\n";
        std::cout << "Изменен: " << (is_modified ? "Да" : "Нет") << "\n";
        std::cout << "----------------------------------------\n";
        std::cout << content;
        std::cout << "----------------------------------------\n";
    }

    void edit_content() {
        std::cout << "Текущее содержимое будет заменено. Продолжить? (y/n): ";
        char choice;
        std::cin >> choice;
        std::cin.ignore();

        if (choice != 'y' && choice != 'Y') {
            return;
        }

        std::cout << "Введите новый текст (введите END на отдельной строке для завершения):\n";
        content.clear();
        std::string line;
        
        while (true) {
            std::getline(std::cin, line);
            if (line == "END") {
                break;
            }
            content += line + "\n";
        }

        is_modified = true;
        std::cout << "Текст успешно изменен.\n";
    }

    void encrypt_and_save() {
        if (content.empty()) {
            std::cout << "Документ пуст. Нечего шифровать.\n";
            return;
        }

        int mode = select_encryption_mode();
        if (mode == -1) return;

        std::string filename;
        std::cout << "Введите имя файла для сохранения: ";
        std::cin >> filename;
        std::cin.ignore();

        try {
            std::string encrypted_content = crypto.encrypt(content, mode);
            
            std::ofstream file(filename);
            if (!file.is_open()) {
                std::cout << "Ошибка: Не удалось создать файл " << filename << "\n";
                return;
            }

            file << encrypted_content;
            file.close();
            
            std::cout << "Документ успешно зашифрован и сохранен в " << filename << "\n";
            std::cout << "Использован режим шифрования: " << get_mode_name(mode) << "\n";
            
        } catch (const std::exception& e) {
            std::cout << "Ошибка при шифровании: " << e.what() << "\n";
        }
    }

    void open_encrypted_document() {
        std::string filename;
        std::cout << "Введите имя зашифрованного файла: ";
        std::cin >> filename;
        std::cin.ignore();

        int mode = select_encryption_mode();
        if (mode == -1) return;

        std::ifstream file(filename);
        if (!file.is_open()) {
            std::cout << "Ошибка: Не удалось открыть файл " << filename << "\n";
            return;
        }

        std::string encrypted_content;
        std::string line;
        while (std::getline(file, line)) {
            encrypted_content += line;
        }
        file.close();

        try {
            content = crypto.decrypt(encrypted_content, mode);
            current_file = filename + ".decrypted";
            is_modified = false;
            
            std::cout << "Зашифрованный документ успешно открыт и расшифрован.\n";
            std::cout << "Использован режим шифрования: " << get_mode_name(mode) << "\n";
            
        } catch (const std::exception& e) {
            std::cout << "Ошибка при дешифровании: " << e.what() << "\n";
            std::cout << "Проверьте правильность выбранного режима шифрования и ключа.\n";
        }
    }

    int select_encryption_mode() {
        std::cout << "\n=== ВЫБОР РЕЖИМА ШИФРОВАНИЯ ===\n";
        std::cout << "0 - Случайный ключ (автогенерация)\n";
        std::cout << "1 - Ключ на основе времени\n";
        std::cout << "2 - Пользовательский ключ\n";
        std::cout << "Выберите режим (0-2): ";
        
        int mode;
        std::cin >> mode;
        std::cin.ignore();

        if (mode < 0 || mode > 2) {
            std::cout << "Неверный режим шифрования.\n";
            return -1;
        }

        if (mode == 2) {
            std::string user_key;
            std::cout << "Введите пользовательский ключ: ";
            std::getline(std::cin, user_key);
            crypto.reset(user_key);
        }

        return mode;
    }

    std::string get_mode_name(int mode) {
        switch (mode) {
            case 0: return "Случайный ключ";
            case 1: return "Временной ключ";
            case 2: return "Пользовательский ключ";
            default: return "Неизвестный";
        }
    }

    void encryption_settings() {
        std::cout << "\n=== НАСТРОЙКИ ШИФРОВАНИЯ ===\n";
        std::cout << "1. Сбросить случайные ключи\n";
        std::cout << "2. Установить новый пользовательский ключ\n";
        std::cout << "3. Показать информацию о ключах\n";
        std::cout << "0. Назад\n";
        std::cout << "Выберите действие: ";
        
        int choice;
        std::cin >> choice;
        std::cin.ignore();

        switch (choice) {
            case 1:
                crypto.reset();
                std::cout << "Случайные ключи сброшены.\n";
                break;
            case 2: {
                std::string new_key;
                std::cout << "Введите новый пользовательский ключ: ";
                std::getline(std::cin, new_key);
                crypto.reset(new_key);
                std::cout << "Пользовательский ключ установлен.\n";
                break;
            }
            case 3:
                std::cout << "Информация о ключах (hex):\n";
                std::cout << "Случайный ключ: " << crypto.get_random_key_hex() << "\n";
                std::cout << "Временной ключ: " << crypto.get_time_key_hex() << "\n";
                std::cout << "Пользовательский ключ: " << crypto.get_user_key_hex() << "\n";
                break;
            case 0:
                break;
            default:
                std::cout << "Неверный выбор.\n";
        }
    }

public:
    TextEditor() : is_modified(false) {}

    void run() {
        std::cout << "Добро пожаловать в текстовый редактор с шифрованием!\n";
        
        while (true) {
            display_menu();
            
            int choice;
            std::cin >> choice;
            std::cin.ignore();

            try {
                switch (choice) {
                    case 1:
                        create_new_document();
                        break;
                    case 2:
                        open_document();
                        break;
                    case 3:
                        save_document();
                        break;
                    case 4:
                        save_document_as();
                        break;
                    case 5:
                        show_content();
                        break;
                    case 6:
                        edit_content();
                        break;
                    case 7:
                        encrypt_and_save();
                        break;
                    case 8:
                        open_encrypted_document();
                        break;
                    case 9:
                        encryption_settings();
                        break;
                    case 0:
                        if (is_modified) {
                            std::cout << "Есть несохраненные изменения. Выйти без сохранения? (y/n): ";
                            char exit_choice;
                            std::cin >> exit_choice;
                            std::cin.ignore();
                            if (exit_choice != 'y' && exit_choice != 'Y') {
                                continue;
                            }
                        }
                        std::cout << "До свидания!\n";
                        return;
                    default:
                        std::cout << "Неверный выбор. Попробуйте снова.\n";
                }
            } catch (const std::exception& e) {
                std::cout << "Произошла ошибка: " << e.what() << "\n";
            }
        }
    }
};

int main() {
    try {
        TextEditor editor;
        editor.run();
    } catch (const std::exception& e) {
        std::cerr << "Критическая ошибка: " << e.what() << std::endl;
        return 1;
    }
    
    return 0;
}