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 #include "native_calendar_manager.h"
17 #include "data_share_helper_manager.h"
18 #include "calendar_env.h"
19 #include "calendar_log.h"
20 #include "native_util.h"
21
22 namespace OHOS::CalendarApi::Native {
23 DataShare::DataSharePredicates BuildCalendarFilter(const CalendarAccount& account);
24 const CalendarAccount defaultAccount {"phone", "local", ""};
25 const string calendarUrl = "datashare:///calendardata/Calendars";
26
CalendarManager()27 CalendarManager::CalendarManager()
28 {
29 uint64_t tokenId = CalendarEnv::GetInstance().GetTokenId();
30 auto bumdleName = CalendarEnv::GetInstance().GetBundleName();
31 auto bundleName_tokeId = "?bundleName=" + bumdleName + "&tokenId=" + std::to_string(tokenId);
32 LOG_INFO("bundleName_tokeId: %{public}s", bundleName_tokeId.c_str());
33 m_calendarUri = std::make_unique<Uri>(calendarUrl + bundleName_tokeId);
34 }
35
GetInstance()36 CalendarManager &CalendarManager::GetInstance()
37 {
38 static CalendarManager instance;
39 return instance;
40 }
41
IsDefaultAccount(const CalendarAccount & account)42 bool CalendarManager::IsDefaultAccount(const CalendarAccount &account)
43 {
44 return account.name == defaultAccount.name && account.type == defaultAccount.type;
45 }
46
BuildValueCalendarAccount(const CalendarAccount & account)47 auto BuildValueCalendarAccount(const CalendarAccount &account)
48 {
49 DataShare::DataShareValuesBucket valuesBucket;
50 valuesBucket.Put("account_name", account.name);
51 valuesBucket.Put("account_type", account.type);
52 if (account.displayName) {
53 valuesBucket.Put("calendar_displayName", account.displayName.value());
54 }
55
56 return valuesBucket;
57 }
58
CreateCalendar(const CalendarAccount & account)59 std::shared_ptr<Calendar> CalendarManager::CreateCalendar(const CalendarAccount& account)
60 {
61 auto valueEvent = BuildValueCalendarAccount(account);
62 auto index = DataShareHelperManager::GetInstance().Insert(*(m_calendarUri.get()), valueEvent);
63 LOG_DEBUG("Insert index %{public}d", index);
64 if (index <= 0) {
65 LOG_ERROR("Insert failed");
66 return nullptr;
67 }
68 return std::make_shared<Calendar>(account, index);
69 }
70
71
BuildCalendarFilter(const CalendarAccount & account)72 DataShare::DataSharePredicates BuildCalendarFilter(const CalendarAccount& account)
73 {
74 DumpCalendarAccount(account);
75 DataShare::DataSharePredicates predicates;
76 predicates.EqualTo("account_name", account.name);
77 predicates.And();
78 predicates.EqualTo("account_type", account.type);
79 return predicates;
80 }
81
GetCalendar(const std::optional<CalendarAccount> & account)82 std::shared_ptr<Calendar> CalendarManager::GetCalendar(const std::optional<CalendarAccount>& account)
83 {
84 DataShare::DataSharePredicates predicates;
85 if (account) {
86 predicates = BuildCalendarFilter(account.value());
87 } else {
88 LOG_WARN("get defaultAccount");
89 predicates = BuildCalendarFilter(defaultAccount);
90 }
91 std::vector<std::string> columns = {"_id", "account_name", "account_type", "calendar_displayName"};
92 DataShare::DatashareBusinessError error;
93 auto resultSet = DataShareHelperManager::GetInstance().Query(*(m_calendarUri.get()), predicates, columns, &error);
94 if (!resultSet) {
95 LOG_ERROR("query failed %{public}d, %{public}s", error.GetCode(), error.GetMessage().c_str());
96 return nullptr;
97 }
98 auto calendarSet = ResultSetToCalendars(resultSet);
99 if (calendarSet.empty()) {
100 LOG_WARN("calendarSet empty");
101 return std::make_shared<Calendar>(-1);
102 }
103 LOG_INFO("GetCalendar successed");
104 return std::move(calendarSet.at(0));
105 }
106
GetAllCalendars()107 std::vector<std::shared_ptr<Calendar>> CalendarManager::GetAllCalendars()
108 {
109 std::vector<std::shared_ptr<Calendar>> results;
110 DataShare::DataSharePredicates predicates;
111 std::vector<std::string> columns = {"_id", "account_name", "account_type", "calendar_displayName"};
112 DataShare::DatashareBusinessError error;
113 auto queryResult = DataShareHelperManager::GetInstance().Query(*(m_calendarUri.get()), predicates, columns, &error);
114 if (!queryResult) {
115 LOG_ERROR("query failed %{public}d, %{public}s", error.GetCode(), error.GetMessage().c_str());
116 return results;
117 }
118 return ResultSetToCalendars(queryResult);
119 }
120
121
DeleteCalendar(const Calendar & calendar)122 bool CalendarManager::DeleteCalendar(const Calendar& calendar)
123 {
124 DataShare::DataSharePredicates predicates;
125 predicates.EqualTo("_id", calendar.GetId());
126 auto result = DataShareHelperManager::GetInstance().Delete(*(m_calendarUri.get()), predicates);
127 LOG_INFO("DeleteCalendar %{public}d", result);
128 return result == 1;
129 }
130
DeleteAllCalendars()131 int CalendarManager::DeleteAllCalendars()
132 {
133 auto calendars = GetAllCalendars();
134 int count = 0;
135 for (const auto &calendar: calendars) {
136 if (DeleteCalendar(*calendar.get())) {
137 count +=1;
138 }
139 }
140 return count;
141 }
142
143 }
144