1 files changed, 272 insertions, 0 deletions
diff --git a/src/plugins/voiceagents/VoiceAgentsDataManagerImpl.cpp b/src/plugins/voiceagents/VoiceAgentsDataManagerImpl.cpp
new file mode 100644
index 0000000..626a7fc
--- /dev/null
+++ b/src/plugins/voiceagents/VoiceAgentsDataManagerImpl.cpp
@@ -0,0 +1,272 @@
+/*
+ * Copyright 2018-2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *     http://aws.amazon.com/apache2.0/
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+#include "voiceagents/VoiceAgentsDataManager.h"
+
+#include "voiceagents/include/VoiceAgentEventsHandler.h"
+
+static string TAG = "vshl::voiceagents::VoiceAgentsDataManager";
+
+/**
+ * Specifies the severity level of a log message
+ */
+using Level = vshl::common::interfaces::ILogger::Level;
+
+namespace vshl {
+namespace voiceagents {
+
+std::unique_ptr<VoiceAgentsDataManager> VoiceAgentsDataManager::create(
+    shared_ptr<vshl::common::interfaces::ILogger> logger,
+    shared_ptr<vshl::common::interfaces::IAFBApi> afbApi) {
+    return std::unique_ptr<VoiceAgentsDataManager>(new VoiceAgentsDataManager(logger, afbApi));
+}
+
+// Constructor
+VoiceAgentsDataManager::VoiceAgentsDataManager(
+    shared_ptr<vshl::common::interfaces::ILogger> logger,
+    shared_ptr<vshl::common::interfaces::IAFBApi> afbApi) :
+        mLogger(logger),
+        mAfbApi(afbApi) {
+    mVoiceAgentEventsHandler = VoiceAgentEventsHandler::create(mLogger, mAfbApi);
+}
+
+// Destructor
+VoiceAgentsDataManager::~VoiceAgentsDataManager() {
+    // Clear the observers
+    mVoiceAgentChangeObservers.clear();
+    // Clear the voiceagents
+    mVoiceAgents.clear();
+}
+
+uint32_t VoiceAgentsDataManager::activateVoiceAgents(const unordered_set<string>& activeVoiceAgentIds) {
+    if (activeVoiceAgentIds.empty() || mVoiceAgents.empty()) {
+        mLogger->log(Level::ERROR, TAG, "Failed to activate voiceagents");
+        return 0;
+    }
+
+    uint32_t agentsActivated = 0;
+    for (auto voiceAgentId : activeVoiceAgentIds) {
+        auto voiceAgentIt = mVoiceAgents.find(voiceAgentId);
+        if (voiceAgentIt != mVoiceAgents.end()) {
+            // activate the voiceagent
+            ++agentsActivated;
+            if (!voiceAgentIt->second->isActive()) {
+                voiceAgentIt->second->setIsActive(true);
+                // Notify observers
+                for (auto observer : mVoiceAgentChangeObservers) {
+                    observer->OnVoiceAgentActivated(voiceAgentIt->second);
+                }
+            }
+        }
+    }
+    return agentsActivated;
+}
+
+uint32_t VoiceAgentsDataManager::deactivateVoiceAgents(const unordered_set<string>& inactiveVoiceAgentIds) {
+    if (inactiveVoiceAgentIds.empty() || mVoiceAgents.empty()) {
+        mLogger->log(Level::ERROR, TAG, "Failed to deactivate voiceagents");
+        return 0;
+    }
+
+    uint32_t agentsDeactivated = 0;
+    for (auto voiceAgentId : inactiveVoiceAgentIds) {
+        auto voiceAgentIt = mVoiceAgents.find(voiceAgentId);
+        if (voiceAgentIt != mVoiceAgents.end()) {
+            ++agentsDeactivated;
+            if (voiceAgentIt->second->isActive()) {
+                // deactivate the voiceagent
+                voiceAgentIt->second->setIsActive(false);
+                // Notify observers
+                for (auto observer : mVoiceAgentChangeObservers) {
+                    observer->OnVoiceAgentDeactivated(voiceAgentIt->second);
+                }
+            }
+        }
+    }
+
+    return agentsDeactivated;
+}
+
+bool VoiceAgentsDataManager::setDefaultVoiceAgent(const string& voiceAgentId) {
+    if (mVoiceAgents.empty() || voiceAgentId.empty()) {
+        string message = string("Failed to set default voiceagent id: ") + voiceAgentId;
+        mLogger->log(Level::ERROR, TAG, message);
+        return false;
+    }
+
+    auto defaultVoiceAgentIt = mVoiceAgents.find(voiceAgentId);
+
+    if (defaultVoiceAgentIt != mVoiceAgents.end()) {
+        if (mDefaultVoiceAgentId != voiceAgentId) {
+            // Notify observers
+            for (auto observer : mVoiceAgentChangeObservers) {
+                observer->OnDefaultVoiceAgentChanged(defaultVoiceAgentIt->second);
+            }
+        }
+        mDefaultVoiceAgentId = voiceAgentId;
+    } else {
+        mLogger->log(Level::ERROR, TAG, "Can't set default agent. Invalid voice agent id:" + voiceAgentId);
+        return false;
+    }
+
+    return true;
+}
+
+std::string VoiceAgentsDataManager::getDefaultVoiceAgent() {
+    return mDefaultVoiceAgentId;
+}
+
+bool VoiceAgentsDataManager::setActiveWakeWord(const string& voiceAgentId, const string& wakeword) {
+    if (mVoiceAgents.empty() || wakeword.empty()) {
+        string message =
+            string("Failed to set active wakeword: ") + wakeword + string(" for voiceagent id: ") + voiceAgentId;
+        mLogger->log(Level::ERROR, TAG, message);
+        return false;
+    }
+
+    auto voiceAgentIt = mVoiceAgents.find(voiceAgentId);
+    if (voiceAgentIt == mVoiceAgents.end()) {
+        return false;
+    }
+
+    string oldWakeWord = voiceAgentIt->second->getActiveWakeword();
+    if (oldWakeWord != wakeword) {
+        voiceAgentIt->second->setActiveWakeWord(wakeword);
+        // Notify observers
+        for (auto observer : mVoiceAgentChangeObservers) {
+            observer->OnVoiceAgentActiveWakeWordChanged(voiceAgentIt->second);
+        }
+    }
+
+    return true;
+}
+
+bool VoiceAgentsDataManager::addNewVoiceAgent(
+    const string& id,
+    const string& name,
+    const string& description,
+    const string& api,
+    const string& vendor,
+    const string& activeWakeword,
+    const bool isActive,
+    const shared_ptr<unordered_set<string>> wakewords) {
+    shared_ptr<VoiceAgent> voiceAgent =
+        VoiceAgent::create(mLogger, id, name, description, api, vendor, activeWakeword, isActive, wakewords);
+
+    if (voiceAgent.get() == nullptr || voiceAgent->getId().empty()) {
+        string message = string("Invalid Arguments: Failed to add new voiceagent");
+        mLogger->log(Level::ERROR, TAG, message);
+        return false;
+    }
+
+    if (!mVoiceAgents.empty() && mVoiceAgents.find(voiceAgent->getId()) != mVoiceAgents.end()) {
+        string message =
+            string("Failed to add new voiceagent. Voiceagent: ") + voiceAgent->getId() + string(" already exists.");
+        mLogger->log(Level::ERROR, TAG, message);
+        return false;
+    }
+
+    mVoiceAgents.insert(make_pair(voiceAgent->getId(), voiceAgent));
+
+    // Notify the observers
+    for (auto observer : mVoiceAgentChangeObservers) {
+        observer->OnVoiceAgentAdded(voiceAgent);
+    }
+
+    // Create all vshl events for the voiceagent.
+    mVoiceAgentEventsHandler->createVshlEventsForVoiceAgent(voiceAgent->getId());
+
+    return true;
+}
+
+bool VoiceAgentsDataManager::removeVoiceAgent(const string& voiceAgentId) {
+    if (mVoiceAgents.empty()) {
+        string message = string("Failed to remove voiceagent: ") + voiceAgentId + string(". Voiceagents data empty.");
+        mLogger->log(Level::ERROR, TAG, message);
+        return false;
+    }
+
+    auto voiceAgentIt = mVoiceAgents.find(voiceAgentId);
+    if (voiceAgentIt == mVoiceAgents.end()) {
+        string message = string("Failed to remove voiceagent: ") + voiceAgentId + string(". Doesn't exist.");
+        mLogger->log(Level::ERROR, TAG, message);
+        return false;
+    }
+
+    auto voiceAgent = voiceAgentIt->second;
+    // Remove from the map
+    mVoiceAgents.erase(voiceAgentId);
+    // Notify the observers
+    for (auto observer : mVoiceAgentChangeObservers) {
+        observer->OnVoiceAgentRemoved(voiceAgent);
+    }
+
+    // Remove all vshl events for the voiceagent.
+    mVoiceAgentEventsHandler->removeVshlEventsForVoiceAgent(voiceAgent->getId());
+
+    return true;
+}
+
+std::set<std::shared_ptr<vshl::common::interfaces::IVoiceAgent>> VoiceAgentsDataManager::getAllVoiceAgents() {
+    std::set<std::shared_ptr<vshl::common::interfaces::IVoiceAgent>> voiceAgentsSet;
+    for (auto element : mVoiceAgents) {
+        voiceAgentsSet.insert(element.second);
+    }
+
+    return voiceAgentsSet;
+}
+
+// Returns the event filter that belongs to the core module.
+shared_ptr<vshl::common::interfaces::IEventFilter> VoiceAgentsDataManager::getEventFilter() const {
+    return mVoiceAgentEventsHandler;
+}
+
+bool VoiceAgentsDataManager::subscribeToVshlEventFromVoiceAgent(
+    vshl::common::interfaces::IAFBRequest& request,
+    const string eventName,
+    const string voiceAgentId) {
+    auto voiceAgentIt = mVoiceAgents.find(voiceAgentId);
+    if (voiceAgentIt == mVoiceAgents.end()) {
+        mLogger->log(
+            Level::ERROR,
+            TAG,
+            "\
+          Failed to subscribe to VSHL events from voiceagent. VoiceAgent: " +
+                voiceAgentId + " doesn't exist.");
+        return false;
+    }
+    return mVoiceAgentEventsHandler->subscribeToVshlEventFromVoiceAgent(request, eventName, voiceAgentIt->second);
+}
+
+bool VoiceAgentsDataManager::addVoiceAgentsChangeObserver(
+    shared_ptr<vshl::common::interfaces::IVoiceAgentsChangeObserver> observer) {
+    if (!observer) {
+        return false;
+    }
+
+    mVoiceAgentChangeObservers.insert(observer);
+    return true;
+}
+
+bool VoiceAgentsDataManager::removeVoiceAgentsChangeObserver(
+    shared_ptr<vshl::common::interfaces::IVoiceAgentsChangeObserver> observer) {
+    if (!observer) {
+        return false;
+    }
+
+    mVoiceAgentChangeObservers.erase(observer);
+    return true;
+}
+}  // namespace voiceagents
+}  // namespace vshl