// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include "document_db_explorer.h"
#include "document_meta_store_read_guards.h"
#include "document_subdb_collection_explorer.h"
#include "executor_threading_service_explorer.h"
#include "maintenance_controller_explorer.h"
#include "documentdb.h"
#include <vespa/searchcore/proton/bucketdb/bucket_db_explorer.h>
#include <vespa/searchcore/proton/common/state_reporter_utils.h>
#include <vespa/vespalib/data/slime/slime.h>

using vespalib::StateExplorer;
using namespace vespalib::slime;

namespace proton {

DocumentDBExplorer::DocumentDBExplorer(std::shared_ptr<DocumentDB> docDb)
    : _docDb(std::move(docDb))
{
}

DocumentDBExplorer::~DocumentDBExplorer() = default;

void
DocumentDBExplorer::get_state(const Inserter &inserter, bool full) const
{
    (void) full;
    Cursor &object = inserter.insertObject();
    object.setString("documentType", _docDb->getDocTypeName().toString());
    {
        StateReporterUtils::convertToSlime(*_docDb->reportStatus(), ObjectInserter(object, "status"));
    }
    {
        DocumentMetaStoreReadGuards dmss(_docDb->getDocumentSubDBs());
        Cursor &documents = object.setObject("documents");
        documents.setLong("active", dmss.numActiveDocs());
        documents.setLong("ready", dmss.numReadyDocs());
        documents.setLong("total", dmss.numTotalDocs());
        documents.setLong("removed", dmss.numRemovedDocs());
    }
}

const vespalib::string SUB_DB = "subdb";
const vespalib::string THREADING_SERVICE = "threadingservice";
const vespalib::string BUCKET_DB = "bucketdb";
const vespalib::string MAINTENANCE_CONTROLLER = "maintenancecontroller";

std::vector<vespalib::string>
DocumentDBExplorer::get_children_names() const
{
    return {SUB_DB, THREADING_SERVICE, BUCKET_DB, MAINTENANCE_CONTROLLER};
}

std::unique_ptr<StateExplorer>
DocumentDBExplorer::get_child(vespalib::stringref name) const
{
    if (name == SUB_DB) {
        return std::make_unique<DocumentSubDBCollectionExplorer>(_docDb->getDocumentSubDBs());
    } else if (name == THREADING_SERVICE) {
        return std::make_unique<ExecutorThreadingServiceExplorer>(_docDb->getWriteService());
    } else if (name == BUCKET_DB) {
        // TODO(geirst): const_cast can be avoided if we add const guard to BucketDBOwner.
        return std::make_unique<BucketDBExplorer>(
            (const_cast<DocumentSubDBCollection &>(_docDb->getDocumentSubDBs())).getBucketDB().takeGuard());
    } else if (name == MAINTENANCE_CONTROLLER) {
        return std::make_unique<MaintenanceControllerExplorer>(_docDb->getMaintenanceController().getJobList());
    }
    return std::unique_ptr<StateExplorer>();
}

} // namespace proton