db word search implemented in Ui for word manipulation.

Author: khumnath

src/lekhika-trainer.cpp

@@ -435,7 +435,7 @@ class DbEditorTab : public QWidget
     explicit DbEditorTab(QWidget *parent = nullptr) : QWidget(parent)
     {
         auto *v = new QVBoxLayout(this);
-        auto *h = new QHBoxLayout;
+        auto *topControls = new QHBoxLayout;
 
         reloadBtn = new QPushButton("Reload");
         newBtn    = new QPushButton("New word");
@@ -444,12 +444,20 @@ class DbEditorTab : public QWidget
         editBtn   = new QPushButton("Edit word");
         editBtn->setEnabled(false);
 
-        h->addWidget(reloadBtn);
-        h->addWidget(newBtn);
-        h->addWidget(delBtn);
-        h->addWidget(editBtn);
-        h->addWidget(resetBtn);
-        h->addStretch();
+        topControls->addWidget(reloadBtn);
+        topControls->addWidget(newBtn);
+        topControls->addWidget(delBtn);
+        topControls->addWidget(editBtn);
+        topControls->addWidget(resetBtn);
+        topControls->addStretch();
+
+        auto *searchLay = new QHBoxLayout;
+        searchEdit_ = new QLineEdit;
+        searchEdit_->setPlaceholderText("Search for a word (Latin or Devanagari)...");
+        clearSearchBtn_ = new QPushButton("Clear");
+        searchLay->addWidget(new QLabel("Search:"));
+        searchLay->addWidget(searchEdit_);
+        searchLay->addWidget(clearSearchBtn_);
 
         table = new QTableWidget(0, 2);
         table->setSelectionBehavior(QAbstractItemView::SelectRows);
@@ -473,7 +481,8 @@ class DbEditorTab : public QWidget
         log->setMaximumHeight(120);
         logLayout->addWidget(log);
 
-        v->addLayout(h);
+        v->addLayout(topControls);
+        v->addLayout(searchLay);
         v->addWidget(table, 1);
         v->addWidget(logFrame, 0);
 
@@ -486,13 +495,48 @@ class DbEditorTab : public QWidget
         connect(table->selectionModel(), &QItemSelectionModel::selectionChanged,
                 this, &DbEditorTab::onSelectionChanged);
         connect(table->verticalScrollBar(), &QScrollBar::valueChanged, this, &DbEditorTab::onScroll);
+        connect(searchEdit_, &QLineEdit::textChanged, this, &DbEditorTab::onSearchTextChanged);
+        connect(clearSearchBtn_, &QPushButton::clicked, this, [this](){ searchEdit_->clear(); });
     }
 
     void setOnDatabaseUpdateCallback(std::function<void()> callback) {
         onDatabaseUpdate_ = callback;
     }
 
 private:
+    void onSearchTextChanged(const QString &text) {
+        if (text.isEmpty()) {
+            m_isSearchActive = false;
+            reload();
+            return;
+        }
+
+        m_isSearchActive = true;
+
+        Transliteration transliterator;
+        std::string devanagari_search_term = transliterator.transliterate(text.toStdString());
+
+        log->appendPlainText(QString("Searching for '%1' (%2)...").arg(text, QString::fromStdString(devanagari_search_term)));
+
+        DictionaryManager dm;
+        auto words = dm.searchWords(devanagari_search_term);
+
+        table->setSortingEnabled(false);
+        table->setRowCount(0); // Clear table before populating search results
+        table->setRowCount(words.size());
+
+        for(size_t i = 0; i < words.size(); ++i) {
+            auto *wItem = new QTableWidgetItem(QString::fromStdString(words[i].first));
+            auto *fItem = new QTableWidgetItem(QString::number(words[i].second));
+            wItem->setFlags(wItem->flags() & ~Qt::ItemIsEditable);
+            fItem->setFlags(fItem->flags() & ~Qt::ItemIsEditable);
+            table->setItem(i, 0, wItem);
+            table->setItem(i, 1, fItem);
+        }
+        table->setSortingEnabled(true);
+        log->appendPlainText(QString("Found %1 match(es).").arg(words.size()));
+    }
+
     void onSortColumnChanged(int logicalIndex) {
         m_currentSortColumn = logicalIndex;
         m_currentSortOrder = table->horizontalHeader()->sortIndicatorOrder();
@@ -505,6 +549,10 @@ class DbEditorTab : public QWidget
     }
 
     void reload() {
+        if (m_isSearchActive) {
+            onSearchTextChanged(searchEdit_->text());
+            return;
+        }
         log->clear();
         currentPage_ = 0;
         table->setRowCount(0);
@@ -514,7 +562,7 @@ class DbEditorTab : public QWidget
     }
 
     void loadMore() {
-        if (isLoading_) return;
+        if (isLoading_ || m_isSearchActive) return;
         isLoading_ = true;
 
         if (currentPage_ == 0) {
@@ -556,7 +604,7 @@ class DbEditorTab : public QWidget
     }
 
     void onScroll(int value) {
-        if (!isLoading_ && value == table->verticalScrollBar()->maximum()) {
+        if (!isLoading_ && !m_isSearchActive && value == table->verticalScrollBar()->maximum()) {
             loadMore();
         }
     }
@@ -578,7 +626,6 @@ class DbEditorTab : public QWidget
         auto sel = table->selectionModel()->selectedRows();
         if (sel.empty()) { log->appendPlainText("Nothing selected."); return; }
 
-        // Sort indexes in descending order to safely remove rows from bottom to top
         std::sort(sel.begin(), sel.end(), [](const QModelIndex &a, const QModelIndex &b) {
             return a.row() > b.row();
         });
@@ -649,12 +696,14 @@ class DbEditorTab : public QWidget
 
 private:
     QTableWidget *table;
-    QPushButton *reloadBtn, *newBtn, *delBtn, *resetBtn, *editBtn;
+    QPushButton *reloadBtn, *newBtn, *delBtn, *resetBtn, *editBtn, *clearSearchBtn_;
+    QLineEdit *searchEdit_;
     QPlainTextEdit *log;
-    std::function<void()> onDatabaseUpdate_;
+    std::function<void()> onDatabaseUpdate_ = nullptr;
     int currentPage_ = 0;
     const int pageSize_ = 50;
     bool isLoading_ = false;
+    bool m_isSearchActive = false;
     int m_currentSortColumn = 0;
     Qt::SortOrder m_currentSortOrder = Qt::AscendingOrder;
 };

src/lekhika_core.cpp

@@ -298,6 +298,31 @@ void DictionaryManager::removeWord(const std::string &word) {
     sqlite3_finalize(stmt);
 }
 
+std::vector<std::pair<std::string, int>>
+DictionaryManager::searchWords(const std::string& searchTerm) {
+    std::vector<std::pair<std::string, int>> results;
+    if (!db_ || searchTerm.empty()) return results;
+    sqlite3_stmt *stmt;
+
+    // Use LIKE with wildcards to search for the term anywhere in the word
+    std::string sql_str = "SELECT word, frequency FROM words WHERE word LIKE ? ORDER BY frequency DESC;";
+
+    if (sqlite3_prepare_v2(db_, sql_str.c_str(), -1, &stmt, NULL) != SQLITE_OK) {
+        return results;
+    }
+
+    std::string pattern = "%" + searchTerm + "%";
+    sqlite3_bind_text(stmt, 1, pattern.c_str(), -1, SQLITE_TRANSIENT);
+
+    while (sqlite3_step(stmt) == SQLITE_ROW) {
+        results.emplace_back((const char *)sqlite3_column_text(stmt, 0),
+                             sqlite3_column_int(stmt, 1));
+    }
+
+    sqlite3_finalize(stmt);
+    return results;
+}
+
 bool DictionaryManager::updateWordFrequency(const std::string &word,
                                             int frequency) {
     if (!db_) return false;

src/lekhika_core.h

@@ -39,6 +39,7 @@ class DictionaryManager {
 
     enum SortColumn { ByWord = 0, ByFrequency = 1 };
     std::vector<std::pair<std::string, int>> getAllWords(int limit = -1, int offset = 0, SortColumn sortBy = ByWord, bool ascending = true);
+    std::vector<std::pair<std::string, int>> searchWords(const std::string& searchTerm);
 
     // Transaction management
     void beginTransaction();