1 /*
2  * Copyright (c) 2023 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 #ifndef INTELL_VOICE_TRIGGER_CONNECTOR_INTERNAL_VALIDATION_H
16 #define INTELL_VOICE_TRIGGER_CONNECTOR_INTERNAL_VALIDATION_H
17 
18 #include <mutex>
19 #include <string>
20 #include <set>
21 #include "v1_0/intell_voice_trigger_types.h"
22 #include "i_intell_voice_trigger_connector_internal.h"
23 
24 namespace OHOS {
25 namespace IntellVoiceTrigger {
26 using OHOS::HDI::IntelligentVoice::Trigger::V1_0::IntellVoiceRecognitionEvent;
27 
28 class TriggerConnectorInternalValidation : public IIntellVoiceTriggerConnectorInternal {
29 public:
30     explicit TriggerConnectorInternalValidation(std::unique_ptr<IIntellVoiceTriggerConnectorInternal> delegate);
31     std::vector<TriggerConnectorModuleDesc> ListModuleDescriptors() override;
32     std::shared_ptr<IIntellVoiceTriggerConnectorModule> GetModule(const std::string &adapterName,
33         std::shared_ptr<IIntellVoiceTriggerConnectorCallback> callback) override;
34 
35 private:
36     std::mutex mutex_;
37     std::set<std::string> moduleDescs_;
38     std::unique_ptr<IIntellVoiceTriggerConnectorInternal> delegate_ = nullptr;
39 
40 private:
41     class TriggerConnectorModuleValidation : public IIntellVoiceTriggerConnectorModule {
42     public:
43         explicit TriggerConnectorModuleValidation(std::shared_ptr<IIntellVoiceTriggerConnectorCallback> callback);
GetCallbackWrapper()44         std::shared_ptr<IIntellVoiceTriggerConnectorCallback> GetCallbackWrapper()
45         {
46             return callbackWrapper_;
47         }
SetDelegate(std::shared_ptr<IIntellVoiceTriggerConnectorModule> delegate)48         void SetDelegate(std::shared_ptr<IIntellVoiceTriggerConnectorModule> delegate)
49         {
50             delegate_ = delegate;
51         }
52         int32_t LoadModel(std::shared_ptr<GenericTriggerModel> model, int32_t &modelHandle) override;
53         int32_t UnloadModel(int32_t modelHandle) override;
54         int32_t Start(int32_t modelHandle) override;
55         int32_t Stop(int32_t modelHandle) override;
56         int32_t SetParams(const std::string &key, const std::string &value) override;
57         int32_t GetParams(const std::string& key, std::string &value) override;
58 
59     private:
60         class TriggerConnectorCallbackValidation : public IIntellVoiceTriggerConnectorCallback {
61         public:
TriggerConnectorCallbackValidation(std::shared_ptr<IIntellVoiceTriggerConnectorCallback> delegate)62             explicit TriggerConnectorCallbackValidation(std::shared_ptr<IIntellVoiceTriggerConnectorCallback> delegate)
63                 : delegate_(delegate) {}
OnRecognition(int32_t modelHandle,const IntellVoiceRecognitionEvent & event)64             void OnRecognition(int32_t modelHandle, const IntellVoiceRecognitionEvent &event) override
65             {
66                 delegate_->OnRecognition(modelHandle, event);
67             }
68 
69         private:
70             std::shared_ptr<IIntellVoiceTriggerConnectorCallback> delegate_;
71         };
72 
73         class ValidationUtils {
74         public:
75             static bool ValidateGenericModel(std::shared_ptr<GenericTriggerModel> model);
76         };
77     private:
78         std::shared_ptr<IIntellVoiceTriggerConnectorModule> delegate_ = nullptr;
79         std::shared_ptr<TriggerConnectorCallbackValidation> callbackWrapper_ = nullptr;
80     };
81 };
82 }
83 }
84 #endif
85