3 files changed, 120 insertions, 22 deletions

diff --git a/common/storage.h b/common/storage.h
index a4137ce..22bf06c 100644
--- a/common/storage.h
+++ b/common/storage.h
@@ -106,6 +106,8 @@ public:
             return (d != nullptr);
         }
 
+        qint64 getSize();
+
     private:
         friend Transaction;
         NamedDatabase(NamedDatabase& other);
@@ -123,6 +125,8 @@ public:
         bool commit(const std::function<void(const Storage::Error &error)> &errorHandler = std::function<void(const Storage::Error &error)>());
         void abort();
 
+        QList<QByteArray> getDatabaseNames() const;
+
         NamedDatabase openDatabase(const QByteArray &name = QByteArray("default"), const std::function<void(const Storage::Error &error)> &errorHandler = std::function<void(const Storage::Error &error)>(), bool allowDuplicates = false) const;
 
         Transaction(Transaction&& other) : d(other.d)
diff --git a/common/storage_lmdb.cpp b/common/storage_lmdb.cpp
index a247e38..e32618b 100644
--- a/common/storage_lmdb.cpp
+++ b/common/storage_lmdb.cpp
@@ -318,6 +318,27 @@ void Storage::NamedDatabase::findLatest(const QByteArray &k,
     return;
 }
 
+qint64 Storage::NamedDatabase::getSize()
+{
+    if (!d || !d->transaction) {
+        return -1;
+    }
+
+    int rc;
+    MDB_stat stat;
+    rc = mdb_stat(d->transaction, d->dbi, &stat);
+    if (rc) {
+        qWarning() << "Something went wrong " << rc;
+    }
+    // std::cout << "overflow_pages: " << stat.ms_overflow_pages << std::endl;
+    // std::cout << "page size: " << stat.ms_psize << std::endl;
+    // std::cout << "branch_pages: " << stat.ms_branch_pages << std::endl;
+    // std::cout << "leaf_pages: " << stat.ms_leaf_pages << std::endl;
+    // std::cout << "depth: " << stat.ms_depth << std::endl;
+    // std::cout << "entries: " << stat.ms_entries << std::endl;
+    return stat.ms_psize * (stat.ms_leaf_pages + stat.ms_branch_pages + stat.ms_overflow_pages);
+}
+
 
 
 
@@ -428,9 +449,34 @@ Storage::NamedDatabase Storage::Transaction::openDatabase(const QByteArray &db,
     return Storage::NamedDatabase(p);
 }
 
+QList<QByteArray> Storage::Transaction::getDatabaseNames() const
+{
+    if (!d) {
+        qWarning() << "Invalid transaction";
+        return QList<QByteArray>();
+    }
 
+    int rc;
+    QList<QByteArray> list;
+    if ((rc = mdb_dbi_open(d->transaction, nullptr, 0, &d->dbi) == 0)) {
+        MDB_val key;
+        MDB_val data;
+        MDB_cursor *cursor;
 
-
+        rc = mdb_cursor_open(d->transaction, d->dbi, &cursor);
+        if ((rc = mdb_cursor_get(cursor, &key, &data, MDB_FIRST)) == 0) {
+            list << QByteArray::fromRawData((char*)key.mv_data, key.mv_size);
+            while ((rc = mdb_cursor_get(cursor, &key, &data, MDB_NEXT)) == 0) {
+                list << QByteArray::fromRawData((char*)key.mv_data, key.mv_size);
+            }
+        } else {
+            qWarning() << "Failed to get a value" << rc;
+        }
+    } else {
+        qWarning() << "Failed to open db" << rc << QByteArray(mdb_strerror(rc));
+    }
+    return list;
+}
 
 
 
diff --git a/examples/client/main.cpp b/examples/client/main.cpp
index 7bbac56..f2fb237 100644
--- a/examples/client/main.cpp
+++ b/examples/client/main.cpp
@@ -144,14 +144,11 @@ int main(int argc, char *argv[])
                                      QObject::tr("A type to work with"));
     cliOptions.addPositionalArgument(QObject::tr("[resource]"),
                                      QObject::tr("A resource to connect to"));
-    cliOptions.addOption(QCommandLineOption("clear"));
-    cliOptions.addOption(QCommandLineOption("debuglevel"));
+    cliOptions.addOption(QCommandLineOption("debuglevel", "A debuglevel from 0-6", "debuglevel"));
     cliOptions.addHelpOption();
     cliOptions.process(app);
     QStringList args = cliOptions.positionalArguments();
     auto command = args.takeFirst();
-    auto type = !args.isEmpty() ? args.takeFirst() : QByteArray();
-    auto resources = args;
 
     if (cliOptions.isSet("debuglevel")) {
         Akonadi2::Log::setDebugOutputLevel(static_cast<Akonadi2::Log::DebugLevel>(cliOptions.value("debuglevel").toInt()));
@@ -163,31 +160,31 @@ int main(int argc, char *argv[])
     //     ResourceConfig::addResource(resource.toLatin1(), Akonadi2::Store::resourceName(resource.toLatin1()));
     // }
 
-    Akonadi2::Query query;
-    for (const auto &res : resources) {
-        query.resources << res.toLatin1();
-    }
-    query.syncOnDemand = false;
-    query.processAll = false;
-    query.liveQuery = true;
-
-    qDebug() << "Type: " << type;
-
     if (command == "list") {
+        auto type = !args.isEmpty() ? args.takeFirst() : QByteArray();
+        auto resources = args;
+
+        Akonadi2::Query query;
+        for (const auto &res : resources) {
+            query.resources << res.toLatin1();
+        }
+        query.syncOnDemand = false;
+        query.processAll = false;
         query.liveQuery = false;
+
         auto model = loadModel(type, query);
         qDebug() << "Listing";
-        std::cout << "\tColumn\t ";
+        std::cout << "\tColumn\t\t Identifier\t\t\t\t";
         for (int i = 0; i < model->columnCount(QModelIndex()); i++) {
-            std::cout << "\t" << model->headerData(i, Qt::Horizontal).toString().toStdString();
+            std::cout << "\t|" << model->headerData(i, Qt::Horizontal).toString().toStdString();
         }
         std::cout << std::endl;
         QObject::connect(model.data(), &QAbstractItemModel::rowsInserted, [model](const QModelIndex &index, int start, int end) {
             for (int i = start; i <= end; i++) {
                 std::cout << "\tRow " << model->rowCount() << ":\t ";
-                std::cout << "\t" << model->data(model->index(i, 0, index), Akonadi2::Store::DomainObjectBaseRole).value<Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr>()->identifier().toStdString();
+                std::cout << "\t" << model->data(model->index(i, 0, index), Akonadi2::Store::DomainObjectBaseRole).value<Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr>()->identifier().toStdString() << "\t";
                 for (int col = 0; col < model->columnCount(QModelIndex()); col++) {
-                    std::cout << "\t" << model->data(model->index(i, col, index)).toString().toStdString();
+                    std::cout << "\t|" << model->data(model->index(i, col, index)).toString().toStdString();
                 }
                 std::cout << std::endl;
             }
@@ -201,6 +198,15 @@ int main(int argc, char *argv[])
             return app.exec();
         }
     } else if (command == "count") {
+        auto type = !args.isEmpty() ? args.takeFirst() : QByteArray();
+        auto resources = args;
+
+        Akonadi2::Query query;
+        for (const auto &res : resources) {
+            query.resources << res.toLatin1();
+        }
+        query.syncOnDemand = false;
+        query.processAll = false;
         query.liveQuery = false;
         auto model = loadModel(type, query);
         QObject::connect(model.data(), &QAbstractItemModel::dataChanged, [model, &app](const QModelIndex &, const QModelIndex &, const QVector<int> &roles) {
@@ -211,6 +217,11 @@ int main(int argc, char *argv[])
         });
         return app.exec();
     } else if (command == "synchronize") {
+        auto resources = args;
+        Akonadi2::Query query;
+        for (const auto &res : resources) {
+            query.resources << res.toLatin1();
+        }
         query.syncOnDemand = true;
         query.processAll = true;
         Akonadi2::Store::synchronize(query).then<void>([&app]() {
@@ -218,6 +229,15 @@ int main(int argc, char *argv[])
         }).exec();
         app.exec();
     } else if (command == "show") {
+        auto type = !args.isEmpty() ? args.takeFirst() : QByteArray();
+        auto resources = args;
+
+        Akonadi2::Query query;
+        for (const auto &res : resources) {
+            query.resources << res.toLatin1();
+        }
+        query.syncOnDemand = false;
+        query.processAll = false;
         query.liveQuery = true;
         if (type == "folder") {
             query.parentProperty = "parent";
@@ -239,17 +259,45 @@ int main(int argc, char *argv[])
             app.exec();
         }
     } else if (command == "clear") {
+        auto resources = args;
+
         qDebug() << "Clearing";
         for (const auto &resource : resources) {
             Akonadi2::Store::removeFromDisk(resource.toLatin1());
         }
     } else if (command == "create") {
+        auto type = !args.isEmpty() ? args.takeFirst() : QByteArray();
         if (type == "resource") {
             Akonadi2::ApplicationDomain::AkonadiResource resource;
-            resource.setProperty("identifier", resources.at(0));
-            resource.setProperty("type", resources.at(1));
+            resource.setProperty("identifier", args.at(0));
+            resource.setProperty("type", args.at(1));
             Akonadi2::Store::create<Akonadi2::ApplicationDomain::AkonadiResource>(resource).exec().waitForFinished();
-            qDebug() << "Created resource " << resources;
+            qDebug() << "Created resource " << args;
+        }
+    } else if (command == "remove") {
+        auto type = !args.isEmpty() ? args.takeFirst() : QByteArray();
+        if (type == "resource") {
+            Akonadi2::ApplicationDomain::AkonadiResource resource;
+            resource.setProperty("identifier", args.at(0));
+            Akonadi2::Store::remove<Akonadi2::ApplicationDomain::AkonadiResource>(resource).exec().waitForFinished();
+            qDebug() << "Created resource " << args;
+        }
+    } else if (command == "stat") {
+        auto resources = args;
+        for (const auto &resource : resources) {
+            Akonadi2::Storage::Storage storage(Akonadi2::storageLocation(), resource, Akonadi2::Storage::ReadOnly);
+            auto transaction = storage.createTransaction(Akonadi2::Storage::ReadOnly);
+
+            QList<QByteArray> databases = transaction.getDatabaseNames();
+            qint64 total = 0;
+            for (const auto &databaseName : databases) {
+                std::cout << "Database: " << databaseName.toStdString() << std::endl;
+                auto db = transaction.openDatabase(databaseName);
+                auto size = db.getSize();
+                std::cout << "\tSize [kb]: " << size / 1024 << std::endl;
+                total += size;
+            }
+            std::cout << "Total [kb]: " << total / 1024 << std::endl;
         }
     } else {
         qWarning() << "Unknown command " << command;