initial functional file indexer, added Qt / C++ port of Cartographer

desktop/datalake/indexer.cpp

@@ -1,6 +1,90 @@
 #include "indexer.hpp"
+
+#include <KFileMetaData/ExtractorCollection>
+#include <QDebug>
 #include <QDirIterator>
+#include <QMimeDatabase>
+#include <iostream>
 
-FileIndexer::FileIndexer(KotoLibraryConfig * config) {
-
+FileIndexer::FileIndexer(KotoLibraryConfig* config) {
+  this->i_root = QString {config->getPath().c_str()};
+}
+
+FileIndexer::~FileIndexer() = default;
+
+void FileIndexer::index() {
+  QMimeDatabase                      db;
+  KFileMetaData::ExtractorCollection extractors;
+
+  QStringList root_dirs {this->i_root.split(QDir::separator())};
+
+  QDirIterator it {this->i_root, QDir::Dirs | QDir::Files | QDir::NoDotAndDotDot, QDirIterator::Subdirectories};
+
+  // std::cout << "Indexing " << this->i_root.toStdString();
+
+  while (it.hasNext()) {
+    QString   path = it.next();
+    QFileInfo info {path};
+
+    if (info.isDir()) {
+      auto diffPath     = info.dir().relativeFilePath(this->i_root);
+      auto diffDirs     = diffPath.split("..");
+      auto diffDirsSize = diffDirs.size() - 1;
+
+      // This is going to be an artist
+      if (diffDirsSize == 0) {
+        auto artist = new KotoArtist();
+        artist->setName(info.fileName());
+        artist->setPath(path);
+        this->i_artists.append(artist);
+        Cartographer::instance().addArtist(artist);
+        continue;
+      } else if (diffDirsSize == 1) {
+        auto album = new KotoAlbum();
+        album->setTitle(info.fileName());
+
+        auto artistDir      = QDir(info.dir());
+        auto artistName     = artistDir.dirName();
+        auto artistOptional = Cartographer::instance().getArtist(artistName);
+
+        if (artistOptional.has_value()) {
+          auto artist        = artistOptional.value();
+          album->artist_uuid = artist->uuid;
+          artist->addAlbum(album);
+        }
+
+        Cartographer::instance().addAlbum(album);
+        continue;
+      }
+    }
+
+    // This is a file
+    QMimeType mime = db.mimeTypeForFile(info);
+    if (mime.name().startsWith("audio/")) {
+      auto extractorList = extractors.fetchExtractors(mime.name());
+      if (extractorList.isEmpty()) { continue; }
+
+      auto result = KFileMetaData::SimpleExtractionResult(path, mime.name(), KFileMetaData::ExtractionResult::ExtractMetaData);
+      extractorList.first()->extract(&result);
+
+      if (!result.types().contains(KFileMetaData::Type::Audio)) { continue; }
+
+      auto track = KotoTrack::fromMetadata(result);
+      track->setPath(path);
+
+      this->i_tracks.append(track);
+      Cartographer::instance().addTrack(track);
+    } else if (mime.name().startsWith("image/")) {
+      // This is an image, TODO add cover art to album
+    }
+  }
+
+  std::cout << "===== Summary =====" << std::endl;
+  for (auto artist : this->i_artists) {
+    std::cout << "Artist: " << artist->getName().toStdString() << std::endl;
+    for (auto album : artist->getAlbums()) {
+      std::cout << "  Album: " << album->getTitle().toStdString() << std::endl;
+      for (auto track : album->getTracks()) { std::cout << "    Track: " << track->getTitle().toStdString() << std::endl; }
+    }
+  }
 }