mainwindow.cpp

#include "mainwindow.h"
#include "noteeditor.h"
#include <QVBoxLayout>
#include <QPushButton>
#include <QLineEdit>
#include <QTreeWidget>
#include <QFileDialog>
#include <QDir>
#include <QFile>
#include <QTextStream>
#include <QDebug>
#include <QStatusBar>

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent), centralWidget(new QWidget(this))
{
    // UI Setup
    QVBoxLayout *layout = new QVBoxLayout(centralWidget);
    searchBar = new QLineEdit(this);
    noteList = new QTreeWidget(this);
    noteList->setHeaderLabels({"Title", "Modified"});

    QPushButton *btnNew = new QPushButton("New Note", this);
    QPushButton *btnEdit = new QPushButton("Edit Note", this);
    QPushButton *btnExport = new QPushButton("Export", this);

    // Layout organization
    layout->addWidget(searchBar);
    layout->addWidget(noteList);

    QHBoxLayout *buttonLayout = new QHBoxLayout;
    buttonLayout->addWidget(btnNew);
    buttonLayout->addWidget(btnEdit);
    buttonLayout->addWidget(btnExport);
    layout->addLayout(buttonLayout);

    centralWidget->setLayout(layout);
    setCentralWidget(centralWidget);

    // Connections
    connect(btnNew, &QPushButton::clicked, this, &MainWindow::onNewNote);
    connect(btnEdit, &QPushButton::clicked, this, &MainWindow::onEditNote);
    connect(btnExport, &QPushButton::clicked, this, &MainWindow::onExport);
    connect(searchBar, &QLineEdit::textChanged, this, &MainWindow::onSearch);

    loadNotes();
}

MainWindow::~MainWindow()
{
    // Cleanup if needed
}

void MainWindow::createNewNote(const QString &initialText)
{
    NoteEditor editor(this);
    editor.setContent(initialText);
    if (editor.exec() == QDialog::Accepted) {
        Note note;
        note.filename = QString("notes/note_%1.txt").arg(QDateTime::currentDateTime().toString("yyyyMMdd_hhmmss"));
        note.content = editor.getContent();
        note.title = note.content.left(30).replace('\n', ' ');
        note.modified = QDateTime::currentDateTime();
        note.created = note.modified;

        QDir().mkpath("notes");
        QFile file(note.filename);
        if (file.open(QIODevice::WriteOnly | QIODevice::Text)) {
            QTextStream out(&file);
            out << note.content;
            file.close();
            loadNotes();
            statusBar()->showMessage("Note saved", 2000);
        }
    }
}

void MainWindow::loadNotes()
{
    noteList->clear();
    notes.clear();

    QDir dir("notes");
    if (!dir.exists()) {
        dir.mkpath(".");
        return;
    }

    foreach (QString filename, dir.entryList(QStringList() << "*.txt")) {
        Note note;
        note.filename = dir.filePath(filename);
        QFile file(note.filename);
        if (file.open(QIODevice::ReadOnly | QIODevice::Text)) {
            QTextStream in(&file);
            note.content = in.readAll();
            note.title = filename.left(filename.lastIndexOf('.'));
            note.modified = QFileInfo(file).lastModified();
            note.created = note.modified;
            notes.append(note);
            file.close();
        }
    }

    foreach (const Note &note, notes) {
        QTreeWidgetItem *item = new QTreeWidgetItem(noteList);
        item->setText(0, note.title);
        item->setText(1, note.modified.toString("MMM d h:mm AP"));
    }
}

void MainWindow::onNewNote()
{
    createNewNote();
}

void MainWindow::onEditNote()
{
    if (noteList->currentItem()) {
        int index = noteList->indexOfTopLevelItem(noteList->currentItem());
        if (index >= 0 && index < notes.size()) {
            NoteEditor editor(this);
            editor.setContent(notes[index].content);
            if (editor.exec() == QDialog::Accepted) {
                notes[index].content = editor.getContent();
                notes[index].modified = QDateTime::currentDateTime();

                QFile file(notes[index].filename);
                if (file.open(QIODevice::WriteOnly | QIODevice::Text)) {
                    QTextStream out(&file);
                    out << notes[index].content;
                    file.close();
                    loadNotes();
                }
            }
        }
    }
}

void MainWindow::onSearch()
{
    QString term = searchBar->text().toLower();
    noteList->clear();

    foreach (const Note &note, notes) {
        if (note.content.toLower().contains(term) ||
            note.title.toLower().contains(term)) {
            QTreeWidgetItem *item = new QTreeWidgetItem(noteList);
            item->setText(0, note.title);
            item->setText(1, note.modified.toString("MMM d h:mm AP"));
        }
    }
}

void MainWindow::onExport()
{
    QString fileName = QFileDialog::getSaveFileName(this, "Export Notes", "", "Text Files (*.txt)");
    if (!fileName.isEmpty()) {
        QFile file(fileName);
        if (file.open(QIODevice::WriteOnly | QIODevice::Text)) {
            QTextStream out(&file);
            foreach (const Note &note, notes) {
                out << "=== " << note.title << " ===\n"
                    << "Created: " << note.created.toString("yyyy-MM-dd hh:mm") << "\n"
                    << note.content << "\n\n";
            }
        }
    }
}