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 
16 #ifndef DATA_SHARE_HELPER_IMPL_H
17 #define DATA_SHARE_HELPER_IMPL_H
18 
19 #include "datashare_helper.h"
20 #include "ext_special_controller.h"
21 #include "general_controller.h"
22 #include "persistent_data_controller.h"
23 #include "published_data_controller.h"
24 
25 namespace OHOS::DataShare {
26 class DataShareHelperImpl : public DataShareHelper {
27 public:
28     DataShareHelperImpl(const Uri &uri, const sptr<IRemoteObject> &token,
29         std::shared_ptr<DataShareConnection> connection);
30     DataShareHelperImpl(std::string extUri = "");
31 
32     ~DataShareHelperImpl() override;
33 
34     bool Release() override;
35 
36     std::vector<std::string> GetFileTypes(Uri &uri, const string &mimeTypeFilter) override;
37 
38     int OpenFile(Uri &uri, const string &mode) override;
39 
40     int OpenRawFile(Uri &uri, const string &mode) override;
41 
42     int Insert(Uri &uri, const DataShareValuesBucket &value) override;
43 
44     int InsertExt(Uri &uri, const DataShareValuesBucket &value, std::string &result) override;
45 
46     int Update(Uri &uri, const DataSharePredicates &predicates, const DataShareValuesBucket &value) override;
47 
48     int BatchUpdate(const UpdateOperations &operations, std::vector<BatchUpdateResult> &results) override;
49 
50     int Delete(Uri &uri, const DataSharePredicates &predicates) override;
51 
52     std::shared_ptr<DataShareResultSet> Query(Uri &uri, const DataSharePredicates &predicates,
53         std::vector<std::string> &columns, DatashareBusinessError *businessError) override;
54 
55     string GetType(Uri &uri) override;
56 
57     int BatchInsert(Uri &uri, const std::vector<DataShareValuesBucket> &values) override;
58 
59     int ExecuteBatch(const std::vector<OperationStatement> &statements, ExecResultSet &result) override;
60 
61     int RegisterObserver(const Uri &uri, const sptr<AAFwk::IDataAbilityObserver> &dataObserver) override;
62 
63     int UnregisterObserver(const Uri &uri, const sptr<AAFwk::IDataAbilityObserver> &dataObserver) override;
64 
65     void NotifyChange(const Uri &uri) override;
66 
67     Uri NormalizeUri(Uri &uri) override;
68 
69     Uri DenormalizeUri(Uri &uri) override;
70 
71     int AddQueryTemplate(const string &uri, int64_t subscriberId, Template &tpl) override;
72 
73     int DelQueryTemplate(const string &uri, int64_t subscriberId) override;
74 
75     std::vector<OperationResult> Publish(const Data &data, const string &bundleName) override;
76 
77     Data GetPublishedData(const string &bundleName, int &resultCode) override;
78 
79     std::vector<OperationResult> SubscribeRdbData(const std::vector<std::string> &uris, const TemplateId &templateId,
80         const std::function<void(const RdbChangeNode &)> &callback) override;
81 
82     std::vector<OperationResult> UnsubscribeRdbData(const std::vector<std::string> &uris,
83         const TemplateId &templateId) override;
84 
85     std::vector<OperationResult> EnableRdbSubs(const std::vector<std::string> &uris,
86         const TemplateId &templateId) override;
87 
88     std::vector<OperationResult> DisableRdbSubs(const std::vector<std::string> &uris,
89         const TemplateId &templateId) override;
90 
91     std::vector<OperationResult> SubscribePublishedData(const std::vector<std::string> &uris, int64_t subscriberId,
92         const std::function<void(const PublishedDataChangeNode &)> &callback) override;
93 
94     std::vector<OperationResult> UnsubscribePublishedData(const std::vector<std::string> &uris,
95         int64_t subscriberId) override;
96 
97     std::vector<OperationResult> EnablePubSubs(const std::vector<std::string> &uris, int64_t subscriberId) override;
98 
99     std::vector<OperationResult> DisablePubSubs(const std::vector<std::string> &uris, int64_t subscriberId) override;
100 
101     std::pair<int32_t, int32_t> InsertEx(Uri &uri, const DataShareValuesBucket &value) override;
102 
103     std::pair<int32_t, int32_t> UpdateEx(
104         Uri &uri, const DataSharePredicates &predicates, const DataShareValuesBucket &value) override;
105 
106     std::pair<int32_t, int32_t> DeleteEx(Uri &uri, const DataSharePredicates &predicates) override;
107 
108 private:
109     std::shared_ptr<ExtSpecialController> extSpCtl_ = nullptr;
110     std::shared_ptr<GeneralController> generalCtl_ = nullptr;
111     std::shared_ptr<PersistentDataController> persistentDataCtl_ = nullptr;
112     std::shared_ptr<PublishedDataController> publishedDataCtl_ = nullptr;
113 };
114 } // namespace OHOS::DataShare
115 #endif // DATA_SHARE_HELPER_IMPL_H
116