From fbe047a35fff100cb5f42d10984bccde137f586e Mon Sep 17 00:00:00 2001
From: jacqueline <me@jacqueline.id.au>
Date: Wed, 26 Apr 2023 15:21:32 +1000
Subject: Add some basic data and retrieval

---
 src/database/database.cpp | 59 +++++++++++++++++++++++++----------------------
 1 file changed, 31 insertions(+), 28 deletions(-)

(limited to 'src/database/database.cpp')

diff --git a/src/database/database.cpp b/src/database/database.cpp
index cfac77af..b677f4ba 100644
--- a/src/database/database.cpp
+++ b/src/database/database.cpp
@@ -4,6 +4,10 @@
 #include "ff.h"
 #include "leveldb/cache.h"
 
+#include "file_gatherer.hpp"
+#include "leveldb/iterator.h"
+#include "leveldb/slice.h"
+#include "tag_processor.hpp"
 #include "env_esp.hpp"
 #include "leveldb/options.h"
 
@@ -38,38 +42,37 @@ Database::Database(leveldb::DB* db, leveldb::Cache* cache)
 
 Database::~Database() {}
 
-FRESULT scan_files(const std::string &path) {
-    FRESULT res;
-    FF_DIR dir;
-    static FILINFO fno;
-
-    res = f_opendir(&dir, path.c_str());
-    if (res == FR_OK) {
-        for (;;) {
-            res = f_readdir(&dir, &fno);
-            if (res != FR_OK || fno.fname[0] == 0) break;
-            if (fno.fname[0] == '.') continue;
-            if (fno.fattrib & AM_DIR) {
-                std::string new_path = path + "/" + fno.fname;
-                res = scan_files(new_path);
-                if (res != FR_OK) break;
-            } else {
-              ESP_LOGI(kTag, "found %s", fno.fname);
-            }
-        }
-        f_closedir(&dir);
-    }
-
-    return res;
+auto Database::Initialise() -> void {
+  leveldb::WriteOptions opt;
+  opt.sync = true;
+  FindFiles("", [&](const std::string &path) {
+      ESP_LOGI(kTag, "considering %s", path.c_str());
+      FileInfo info;
+      if (GetInfo(path, &info)) {
+        ESP_LOGI(kTag, "added as '%s'", info.title.c_str());
+        db_->Put(opt, "title:" + info.title, path);
+      }
+  });
+  db_->Put(opt, "title:coolkeywithoutval", leveldb::Slice());
 }
 
-auto Database::Initialise() -> void {
-  // TODO(jacqueline): Abstractions lol
-  scan_files("/");
+auto Database::ByTitle() -> Iterator {
+  leveldb::Iterator *it = db_->NewIterator(leveldb::ReadOptions());
+  it->Seek("title:");
+  while (it->Valid()) {
+    ESP_LOGI(kTag, "%s : %s", it->key().ToString().c_str(), it->value().ToString().c_str());
+    it->Next();
+  }
+  return Iterator(it);
 }
 
-auto Database::Update() -> void {
-  // TODO(jacqueline): Incremental updates!
+auto Iterator::Next() -> std::optional<std::string> {
+  if (!it_->Valid()) {
+    return {};
+  }
+  std::string ret = it_->key().ToString();
+  it_->Next();
+  return ret;
 }
 
 }  // namespace database
-- 
cgit v1.2.3