repo_manipulator.h

#pragma once
#include <map>
#include <string>

#include "../core/model/bson/repo_bson_role_settings.h"
#include "../core/model/bson/repo_node_map.h"
#include "../core/model/bson/repo_node_reference.h"
#include "../core/model/bson/repo_node_transformation.h"
#include "../core/model/collection/repo_scene.h"
#include "diff/repo_diff_abstract.h"
#include "modelconvertor/export/repo_model_export_web.h"
#include "modelconvertor/import/repo_model_import_config.h"
#include "modelutility/spatialpartitioning/repo_spatial_partitioner_abstract.h"

namespace repo{
    namespace manipulator{
        class RepoManipulator
        {
        public:
            RepoManipulator();
            ~RepoManipulator();

            void cleanUp(
                const std::string                      &databaseAd,
                const repo::core::model::RepoBSON      *cred,
                const std::string                      &dbName,
                const std::string                      &projectName
                );

            bool connectAndAuthenticate(
                std::string       &errMsg,
                const std::string &address,
                const uint32_t    &port,
                const uint32_t    &maxConnections,
                const std::string &dbName,
                const std::string &username,
                const std::string &password,
                const bool        &pwDigested = false
                );

            bool connectAndAuthenticate(
                std::string       &errMsg,
                const std::string &address,
                const uint32_t    &port,
                const uint32_t    &maxConnections,
                const std::string &dbName,
                const repo::core::model::RepoBSON *credentials
                );

            bool connectAndAuthenticateWithAdmin(
                std::string       &errMsg,
                const std::string &address,
                const uint32_t    &port,
                const uint32_t    &maxConnections,
                const std::string &username,
                const std::string &password,
                const bool        &pwDigested = false
                );

            bool commitScene(
                const std::string                     &databaseAd,
                const repo::core::model::RepoBSON     *cred,
                repo::core::model::RepoScene          *scene,
                const std::string                     &owner = "",
                const std::string                     &tag = "",
                const std::string                     &desc = "");

            void compareScenes(
                repo::core::model::RepoScene                  *base,
                repo::core::model::RepoScene                  *compare,
                repo_diff_result_t                              &baseResults,
                repo_diff_result_t                              &compResults,
                const repo::DiffMode                            &diffMode,
                const repo::core::model::RepoScene::GraphType &gType
                = repo::core::model::RepoScene::GraphType::DEFAULT);

            repo::core::model::RepoBSON* createCredBSON(
                const std::string &databaseAd,
                const std::string &username,
                const std::string &password,
                const bool        &pwDigested);

            repo::core::model::RepoScene* createFederatedScene(
                const std::map<repo::core::model::TransformationNode, repo::core::model::ReferenceNode> &fedMap);

            repo::core::model::RepoScene* createMapScene(
                const repo::core::model::MapNode &mapNode);

            uint64_t countItemsInCollection(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                const std::string                             &database,
                const std::string                             &collection,
                std::string                                   &errMsg
                );

            void disconnectFromDatabase(const std::string &databaseAd);

            bool dropCollection(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                const std::string                             &databaseName,
                const std::string                             &collectionName,
                std::string                                   &errMsg
                );

            bool dropDatabase(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                const std::string                             &databaseName,
                std::string                                   &errMsg
                );

            std::list<std::string> fetchDatabases(
                const std::string                 &databaseAd,
                const repo::core::model::RepoBSON *cred
                );

            std::list<std::string> fetchCollections(
                const std::string                 &databaseAd,
                const repo::core::model::RepoBSON *cred,
                const std::string                 &database
                );

            repo::core::model::RepoScene* fetchScene(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                const std::string                             &database,
                const std::string                             &collection,
                const repoUUID                                &uuid,
                const bool                                    &headRevision = false,
                const bool                                    &lightFetch = false);

            void fetchScene(
                const std::string                         &databaseAd,
                const repo::core::model::RepoBSON         *cred,
                repo::core::model::RepoScene              *scene);

            bool generateAndCommitSelectionTree(
                const std::string                         &databaseAd,
                const repo::core::model::RepoBSON         *cred,
                repo::core::model::RepoScene              *scene
                );

            bool generateAndCommitStashGraph(
                const std::string                         &databaseAd,
                const repo::core::model::RepoBSON         *cred,
                repo::core::model::RepoScene* scene
                );

            bool generateAndCommitWebViewBuffer(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                repo::core::model::RepoScene                  *scene,
                repo_web_buffers_t                            &buffers,
                const modelconvertor::WebExportType           &exType);

            bool generateAndCommitGLTFBuffer(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                repo::core::model::RepoScene                  *scene);

            bool generateAndCommitSRCBuffer(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                repo::core::model::RepoScene                  *scene);

            repo_web_buffers_t generateGLTFBuffer(
                repo::core::model::RepoScene *scene);

            repo_web_buffers_t generateSRCBuffer(
                repo::core::model::RepoScene *scene);

            bool generateStashGraph(
                repo::core::model::RepoScene              *scene
                );
            std::vector<repo::core::model::RepoBSON>
                getAllFromCollectionTailable(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                const std::string                             &database,
                const std::string                             &collection,
                const uint64_t                                &skip = 0,
                const uint32_t                                &limit = 0);

            std::vector<repo::core::model::RepoBSON>
                getAllFromCollectionTailable(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                const std::string                             &database,
                const std::string                             &collection,
                const std::list<std::string>                  &fields,
                const std::string                             &sortField = std::string(),
                const int                                     &sortOrder = -1,
                const uint64_t                                &skip = 0,
                const uint32_t                                &limit = 0);

            repo::core::model::CollectionStats getCollectionStats(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                const std::string                             &database,
                const std::string                             &collection,
                std::string                                   &errMsg
                );

            std::map<std::string, std::list<std::string>>
                getDatabasesWithProjects(
                const std::string                 &databaseAd,
                const repo::core::model::RepoBSON *cred,
                const std::list<std::string>      &databases);

            std::list<std::string> getAdminDatabaseRoles(
                const std::string                     &databaseAd);

            repo::core::model::RepoRoleSettings getRoleSettingByName(
                const std::string                   &databaseAd,
                const repo::core::model::RepoBSON   *cred,
                const std::string                   &database,
                const std::string                   &uniqueRoleName
                );

            std::shared_ptr<repo_partitioning_tree_t> getScenePartitioning(
                const repo::core::model::RepoScene *scene,
                const uint32_t                     &maxDepth = 8
                );

            std::list<std::string> getStandardDatabaseRoles(
                const std::string                             &databaseAd);

            std::string getNameOfAdminDatabase(
                const std::string                             &databaseAd) const;

            bool hasCollection(
                const std::string                      &databaseAd,
                const repo::core::model::RepoBSON      *cred,
                const std::string                      &dbName,
                const std::string                      &project);

            bool hasDatabase(
                const std::string                      &databaseAd,
                const repo::core::model::RepoBSON      *cred,
                const std::string                      &dbName);

            void insertBinaryFileToDatabase(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                const std::string                             &database,
                const std::string                             &collection,
                const std::string                             &name,
                const std::vector<uint8_t>                    &rawData,
                const std::string                             &mimeType = "");

            void insertRole(
                const std::string                             &databaseAd,
                const repo::core::model::RepoBSON             *cred,
                const repo::core::model::RepoRole             &role);

            void insertUser(
                const std::string                       &databaseAd,
                const repo::core::model::RepoBSON       *cred,
                const repo::core::model::RepoUser       &user);

            repo::core::model::RepoNodeSet
                loadMetadataFromFile(
                const std::string &filePath,
                const char        &delimiter = ',');

            repo::core::model::RepoScene*
                loadSceneFromFile(
                const std::string                                          &filePath,
                std::string                                                &msg,
                const bool                                                 &applyReduction = true,
                const bool                                                 &rotateModel = false,
                const repo::manipulator::modelconvertor::ModelImportConfig *config = nullptr);

            void removeDocument(
                const std::string                       &databaseAd,
                const repo::core::model::RepoBSON       *cred,
                const std::string                       &databaseName,
                const std::string                       &collectionName,
                const repo::core::model::RepoBSON       &bson);

            bool removeProject(
                const std::string                       &databaseAd,
                const repo::core::model::RepoBSON       *cred,
                const std::string                        &databaseName,
                const std::string                        &projectName,
                std::string                              &errMsg
                );

            void reduceTransformations(
                repo::core::model::RepoScene                  *scene,
                const repo::core::model::RepoScene::GraphType &gType
                = repo::core::model::RepoScene::GraphType::DEFAULT);

            bool removeStashGraphFromDatabase(
                const std::string                         &databaseAd,
                const repo::core::model::RepoBSON         *cred,
                repo::core::model::RepoScene              *scene
                );

            void removeRole(
                const std::string                       &databaseAd,
                const repo::core::model::RepoBSON       *cred,
                const repo::core::model::RepoRole       &role);

            void removeUser(
                const std::string                       &databaseAd,
                const repo::core::model::RepoBSON       *cred,
                const repo::core::model::RepoUser       &user);

            void saveOriginalFiles(
                const std::string                    &databaseAd,
                const repo::core::model::RepoBSON    *cred,
                const repo::core::model::RepoScene   *scene,
                const std::string                    &directory);

            void saveOriginalFiles(
                const std::string                    &databaseAd,
                const repo::core::model::RepoBSON    *cred,
                const std::string                    &database,
                const std::string                    &project,
                const std::string                    &directory);

            bool saveSceneToFile(
                const std::string &filePath,
                const repo::core::model::RepoScene* scene);

            void updateRole(
                const std::string                       &databaseAd,
                const repo::core::model::RepoBSON       *cred,
                const repo::core::model::RepoRole       &role);

            void updateUser(
                const std::string                       &databaseAd,
                const repo::core::model::RepoBSON       *cred,
                const repo::core::model::RepoUser       &user);
            void upsertDocument(
                const std::string                       &databaseAd,
                const repo::core::model::RepoBSON       *cred,
                const std::string                       &databaseName,
                const std::string                       &collectionName,
                const repo::core::model::RepoBSON       &bson);

        private:

            repo::core::model::RepoRole findRole(
                const std::string                      &databaseAd,
                const repo::core::model::RepoBSON      *cred,
                const std::string                      &dbName,
                const std::string                      &roleName
                );

            repo::core::model::RepoUser findUser(
                const std::string                      &databaseAd,
                const repo::core::model::RepoBSON      *cred,
                const std::string                      &username
                );
        };
    }
}