1 /*
2  * Copyright (C) 2024 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 "ringtone_rdb_callbacks.h"
17 
18 #include <sys/stat.h>
19 
20 #include "rdb_sql_utils.h"
21 #include "ringtone_errno.h"
22 #include "ringtone_log.h"
23 #include "ringtone_db_const.h"
24 #include "ringtone_file_utils.h"
25 #include "ringtone_mimetype_utils.h"
26 #include "ringtone_utils.h"
27 #include "result_set_utils.h"
28 #include "preferences_helper.h"
29 #include "dfx_const.h"
30 
31 namespace OHOS {
32 namespace Media {
33 using namespace std;
34 
35 const string DEFAULT_MIME_TYPE = "application/octet-stream";
36 static const char RINGTONE_PARAMETER_SCANNER_COMPLETED_KEY[] = "ringtone.scanner.completed";
37 static const int RINGTONE_PARAMETER_SCANNER_COMPLETED_FALSE = 0;
38 
39 const std::string CREATE_RINGTONE_TABLE = "CREATE TABLE IF NOT EXISTS " + RINGTONE_TABLE + "(" +
40     RINGTONE_COLUMN_TONE_ID                       + " INTEGER  PRIMARY KEY AUTOINCREMENT, " +
41     RINGTONE_COLUMN_DATA                          + " TEXT              , " +
42     RINGTONE_COLUMN_SIZE                          + " BIGINT   DEFAULT 0, " +
43     RINGTONE_COLUMN_DISPLAY_NAME                  + " TEXT              , " +
44     RINGTONE_COLUMN_TITLE                         + " TEXT              , " +
45     RINGTONE_COLUMN_MEDIA_TYPE                    + " INT      DEFAULT 0, " +
46     RINGTONE_COLUMN_TONE_TYPE                     + " INT      DEFAULT 0, " +
47     RINGTONE_COLUMN_MIME_TYPE                     + " TEXT              , " +
48     RINGTONE_COLUMN_SOURCE_TYPE                   + " INT      DEFAULT 0, " +
49     RINGTONE_COLUMN_DATE_ADDED                    + " BIGINT   DEFAULT 0, " +
50     RINGTONE_COLUMN_DATE_MODIFIED                 + " BIGINT   DEFAULT 0, " +
51     RINGTONE_COLUMN_DATE_TAKEN                    + " BIGINT   DEFAULT 0, " +
52     RINGTONE_COLUMN_DURATION                      + " INT      DEFAULT 0, " +
53     RINGTONE_COLUMN_SHOT_TONE_TYPE                + " INT      DEFAULT 0, " +
54     RINGTONE_COLUMN_SHOT_TONE_SOURCE_TYPE         + " INT      DEFAULT 0, " +
55     RINGTONE_COLUMN_NOTIFICATION_TONE_TYPE        + " INT      DEFAULT 0, " +
56     RINGTONE_COLUMN_NOTIFICATION_TONE_SOURCE_TYPE + " INT      DEFAULT 0, " +
57     RINGTONE_COLUMN_RING_TONE_TYPE                + " INT      DEFAULT 0, " +
58     RINGTONE_COLUMN_RING_TONE_SOURCE_TYPE         + " INT      DEFAULT 0, " +
59     RINGTONE_COLUMN_ALARM_TONE_TYPE               + " INT      DEFAULT 0, " +
60     RINGTONE_COLUMN_ALARM_TONE_SOURCE_TYPE        + " INT      DEFAULT 0, " +
61     RINGTONE_COLUMN_DISPLAY_LANGUAGE_TYPE         + " TEXT                " + ")";
62 
63 const std::string CREATE_SIMCARD_SETTING_TABLE = "CREATE TABLE IF NOT EXISTS " + SIMCARD_SETTING_TABLE + "(" +
64     SIMCARD_SETTING_COLUMN_MODE                   + " INTEGER            ," +
65     SIMCARD_SETTING_COLUMN_RINGTONE_TYPE          + " INTEGER            ," +
66     SIMCARD_SETTING_COLUMN_TONE_FILE              + " TEXT               ," +
67     SIMCARD_SETTING_COLUMN_VIBRATE_FILE           + " TEXT               ," +
68     SIMCARD_SETTING_COLUMN_VIBRATE_MODE           + " INTEGER            ," +
69     SIMCARD_SETTING_COLUMN_RING_MODE              + " INTEGER            ," +
70     " PRIMARY KEY (" + SIMCARD_SETTING_COLUMN_MODE + ", " + SIMCARD_SETTING_COLUMN_RINGTONE_TYPE + "))";
71 
72 const std::string INIT_SIMCARD_SETTING_TABLE = "INSERT OR IGNORE INTO " + SIMCARD_SETTING_TABLE + " (" +
73     SIMCARD_SETTING_COLUMN_MODE                   + ", " +
74     SIMCARD_SETTING_COLUMN_RINGTONE_TYPE          + ") VALUES (1, 0), (1, 1), (1, 2), (1, 3), \
75                                                         (2, 0), (2, 1), (2, 2), (2, 3),        \
76                                                         (3, 0), (3, 1), (3, 2), (3, 3);";
77 
78 const std::string CREATE_VIBRATE_TABLE = "CREATE TABLE IF NOT EXISTS " + VIBRATE_TABLE + "(" +
79     VIBRATE_COLUMN_VIBRATE_ID                     + " INTEGER  PRIMARY KEY AUTOINCREMENT, " +
80     VIBRATE_COLUMN_DATA                           + " TEXT              , " +
81     VIBRATE_COLUMN_SIZE                           + " BIGINT   DEFAULT 0, " +
82     VIBRATE_COLUMN_DISPLAY_NAME                   + " TEXT              , " +
83     VIBRATE_COLUMN_TITLE                          + " TEXT              , " +
84     VIBRATE_COLUMN_DISPLAY_LANGUAGE               + " TEXT              , " +
85     VIBRATE_COLUMN_VIBRATE_TYPE                   + " INT      DEFAULT 0, " +
86     VIBRATE_COLUMN_SOURCE_TYPE                    + " INT      DEFAULT 0, " +
87     VIBRATE_COLUMN_DATE_ADDED                     + " BIGINT   DEFAULT 0, " +
88     VIBRATE_COLUMN_DATE_MODIFIED                  + " BIGINT   DEFAULT 0, " +
89     VIBRATE_COLUMN_DATE_TAKEN                     + " BIGINT   DEFAULT 0, " +
90     VIBRATE_COLUMN_PLAY_MODE                      + " INT      DEFAULT 0  " + ")";
91 
92 const std::string CREATE_PRELOAD_CONF_TABLE = "CREATE TABLE IF NOT EXISTS " + PRELOAD_CONFIG_TABLE + "(" +
93     PRELOAD_CONFIG_COLUMN_RING_TONE_TYPE          + " INTEGER  PRIMARY KEY," +
94     PRELOAD_CONFIG_COLUMN_TONE_ID                 + " INTEGER             ," +
95     PRELOAD_CONFIG_COLUMN_DISPLAY_NAME            + " TEXT                 " + ")";
96 
97 const std::string INIT_PRELOAD_CONF_TABLE = "INSERT OR IGNORE INTO " + PRELOAD_CONFIG_TABLE + " (" +
98     PRELOAD_CONFIG_COLUMN_RING_TONE_TYPE + ") VALUES (1), (2), (3), (4), (5), (6);";
99 
100 static const vector<string> g_initSqls = {
101     CREATE_RINGTONE_TABLE,
102     CREATE_VIBRATE_TABLE,
103     CREATE_SIMCARD_SETTING_TABLE,
104     INIT_SIMCARD_SETTING_TABLE,
105     CREATE_PRELOAD_CONF_TABLE,
106     INIT_PRELOAD_CONF_TABLE,
107 };
108 
RingtoneDataCallBack(void)109 RingtoneDataCallBack::RingtoneDataCallBack(void)
110 {
111 }
112 
~RingtoneDataCallBack(void)113 RingtoneDataCallBack::~RingtoneDataCallBack(void)
114 {
115 }
116 
InitSql(NativeRdb::RdbStore &store)117 int32_t RingtoneDataCallBack::InitSql(NativeRdb::RdbStore &store)
118 {
119     for (const string &sqlStr : g_initSqls) {
120         if (store.ExecuteSql(sqlStr) != NativeRdb::E_OK) {
121             RINGTONE_ERR_LOG("Failed to execute sql");
122             return NativeRdb::E_ERROR;
123         }
124     }
125     return NativeRdb::E_OK;
126 }
127 
OnCreate(NativeRdb::RdbStore &store)128 int32_t RingtoneDataCallBack::OnCreate(NativeRdb::RdbStore &store)
129 {
130     if (InitSql(store) != NativeRdb::E_OK) {
131         RINGTONE_DEBUG_LOG("Failed to init sql");
132         return NativeRdb::E_ERROR;
133     }
134 
135     RingtoneFileUtils::CreateRingtoneDir();
136     return NativeRdb::E_OK;
137 }
138 
ExecSqls(const vector<string> &sqls, NativeRdb::RdbStore &store)139 static void ExecSqls(const vector<string> &sqls, NativeRdb::RdbStore &store)
140 {
141     int32_t err = NativeRdb::E_OK;
142     for (const auto &sql : sqls) {
143         err = store.ExecuteSql(sql);
144         if (err != NativeRdb::E_OK) {
145             RINGTONE_ERR_LOG("Failed to exec: %{private}s", sql.c_str());
146             continue;
147         }
148     }
149 }
150 
AddDisplayLanguageColumn(NativeRdb::RdbStore &store)151 static void AddDisplayLanguageColumn(NativeRdb::RdbStore &store)
152 {
153     const vector<string> sqls = {
154         "ALTER TABLE " + RINGTONE_TABLE + " ADD COLUMN " + RINGTONE_COLUMN_DISPLAY_LANGUAGE_TYPE + " TEXT",
155     };
156     RINGTONE_INFO_LOG("Add display language column");
157     ExecSqls(sqls, store);
158 }
159 
AddVibrateTable(NativeRdb::RdbStore &store)160 static void AddVibrateTable(NativeRdb::RdbStore &store)
161 {
162     const vector<string> sqls = {
163         CREATE_VIBRATE_TABLE,
164         CREATE_SIMCARD_SETTING_TABLE,
165         INIT_SIMCARD_SETTING_TABLE,
166     };
167     int32_t errCode;
168     shared_ptr<NativePreferences::Preferences> prefs =
169         NativePreferences::PreferencesHelper::GetPreferences(COMMON_XML_EL1, errCode);
170     if (!prefs) {
171         RINGTONE_ERR_LOG("AddVibrateTable: update faild errCode=%{public}d", errCode);
172     } else {
173         prefs->PutInt(RINGTONE_PARAMETER_SCANNER_COMPLETED_KEY, RINGTONE_PARAMETER_SCANNER_COMPLETED_FALSE);
174         prefs->FlushSync();
175     }
176 
177     RINGTONE_INFO_LOG("Add vibrate table");
178     ExecSqls(sqls, store);
179 }
180 
UpdateMimeType(NativeRdb::RdbStore &store)181 static void UpdateMimeType(NativeRdb::RdbStore &store)
182 {
183     RINGTONE_INFO_LOG("Update MimeType Begin");
184     RingtoneMimeTypeUtils::InitMimeTypeMap();
185     const string sql = "SELECT * FROM " + RINGTONE_TABLE;
186     auto resultSet = store.QuerySql(sql);
187     if (resultSet == nullptr) {
188         RINGTONE_ERR_LOG("error query sql %{public}s", sql.c_str());
189         return;
190     }
191     while (resultSet->GoToNextRow() == NativeRdb::E_OK) {
192         std::string mimeType = GetStringVal(RINGTONE_COLUMN_MIME_TYPE, resultSet);
193         if (mimeType != DEFAULT_MIME_TYPE) {
194             continue;
195         }
196         string displayName = GetStringVal(RINGTONE_COLUMN_DISPLAY_NAME, resultSet);
197         int32_t toneid = GetInt32Val(RINGTONE_COLUMN_TONE_ID, resultSet);
198         std::string extension = RingtoneFileUtils::GetFileExtension(displayName);
199         mimeType = RingtoneMimeTypeUtils::GetMimeTypeFromExtension(extension);
200         int32_t mime = RingtoneMimeTypeUtils::GetMediaTypeFromMimeType(mimeType);
201         RINGTONE_INFO_LOG("extension: %{public}s, mimeType: %{public}s, toneid: %{public}d mime: %{public}d",
202             extension.c_str(), mimeType.c_str(), toneid, mime);
203 
204         NativeRdb::ValuesBucket values;
205         values.PutString(RINGTONE_COLUMN_MIME_TYPE, mimeType);
206         values.PutInt(RINGTONE_COLUMN_MEDIA_TYPE, mime);
207         NativeRdb::AbsRdbPredicates absRdbPredicates(RINGTONE_TABLE);
208         absRdbPredicates.EqualTo(RINGTONE_COLUMN_TONE_ID, toneid);
209         int32_t changedRows;
210         int32_t result = store.Update(changedRows, values, absRdbPredicates);
211         if (result != E_OK || changedRows <= 0) {
212             RINGTONE_ERR_LOG("Update operation failed. Result %{public}d. Updated %{public}d", result, changedRows);
213         }
214     }
215     resultSet->Close();
216 }
217 
AddPreloadConfTable(NativeRdb::RdbStore &store)218 static void AddPreloadConfTable(NativeRdb::RdbStore &store)
219 {
220     const vector<string> sqls = {
221         CREATE_PRELOAD_CONF_TABLE,
222         INIT_PRELOAD_CONF_TABLE
223     };
224     RINGTONE_INFO_LOG("Add preload config table");
225     ExecSqls(sqls, store);
226 }
227 
UpdateDefaultSystemTone(NativeRdb::RdbStore &store)228 static void UpdateDefaultSystemTone(NativeRdb::RdbStore &store)
229 {
230     RINGTONE_INFO_LOG("setting system tone begin");
231     auto infos = RingtoneUtils::GetDefaultSystemtoneInfo();
232     for (auto info : infos) {
233         const string querySql = "SELECT tone_id FROM ToneFiles WHERE display_name = "s + "\"" + info.second + "\"";
234         auto resultSet = store.QuerySql(querySql);
235         if (resultSet == nullptr || resultSet->GoToFirstRow() != NativeRdb::E_OK) {
236             RINGTONE_ERR_LOG("Update operation failed. no resultSet");
237             continue;
238         }
239 
240         int32_t tone_id = GetInt32Val("tone_id", resultSet);
241         NativeRdb::ValuesBucket values;
242         values.PutString(PRELOAD_CONFIG_COLUMN_DISPLAY_NAME, info.second);
243         values.PutInt(PRELOAD_CONFIG_COLUMN_TONE_ID, tone_id);
244         NativeRdb::AbsRdbPredicates absRdbPredicates(PRELOAD_CONFIG_TABLE);
245         absRdbPredicates.EqualTo(PRELOAD_CONFIG_COLUMN_RING_TONE_TYPE, std::to_string(info.first));
246         int32_t changedRows = 0;
247         int32_t result = store.Update(changedRows, values, absRdbPredicates);
248         if (result != E_OK || changedRows <= 0) {
249             RINGTONE_ERR_LOG("Update operation failed. Result %{public}d. Updated %{public}d", result, changedRows);
250         }
251     }
252 }
253 
UpgradeExtension(NativeRdb::RdbStore &store, int32_t oldVersion)254 static void UpgradeExtension(NativeRdb::RdbStore &store, int32_t oldVersion)
255 {
256     if (oldVersion < VERSION_ADD_DISPLAY_LANGUAGE_COLUMN) {
257         AddDisplayLanguageColumn(store);
258     }
259     if (oldVersion < VERSION_ADD_VIBRATE_TABLE) {
260         AddVibrateTable(store);
261     }
262     if (oldVersion < VERSION_UPDATE_MIME_TYPE) {
263         UpdateMimeType(store);
264     }
265     if (oldVersion < VERSION_ADD_PRELOAD_CONF_TABLE) {
266         AddPreloadConfTable(store);
267         UpdateDefaultSystemTone(store);
268     }
269 }
270 
OnUpgrade(NativeRdb::RdbStore &store, int32_t oldVersion, int32_t newVersion)271 int32_t RingtoneDataCallBack::OnUpgrade(NativeRdb::RdbStore &store, int32_t oldVersion, int32_t newVersion)
272 {
273     RINGTONE_INFO_LOG("OnUpgrade old:%d, new:%d", oldVersion, newVersion);
274     UpgradeExtension(store, oldVersion);
275     return NativeRdb::E_OK;
276 }
277 } // namespace Media
278 } // namespace OHOS
279