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 OHOS_CLOUD_SYNC_SERVICE_ALBUM_DATA_CONVERTOR_H
17 #define OHOS_CLOUD_SYNC_SERVICE_ALBUM_DATA_CONVERTOR_H
18
19 #include <unordered_map>
20
21 #include <uuid/uuid.h>
22
23 #include "data_convertor.h"
24 #include "dfs_error.h"
25 #include "gallery_album_const.h"
26 #include "utils_log.h"
27
28 namespace OHOS {
29 namespace FileManagement {
30 namespace CloudSync {
31
32 #define UUID_STR_LENGTH 37
33
34 class AlbumDataConvertor : public DataConvertor {
35 public:
36 enum OperationType {
37 /* upload */
38 ALBUM_CREATE,
39 ALBUM_DELETE,
40 ALBUM_MODIFY,
41 };
42
43 AlbumDataConvertor(OperationType type);
44 ~AlbumDataConvertor() = default;
45
46 int32_t Convert(DriveKit::DKRecord &record, NativeRdb::ResultSet &resultSet) override;
47 int32_t Convert(DriveKit::DKRecord &record, NativeRdb::ValuesBucket &valueBucket) override;
48
49 private:
50 /* record id */
51 int32_t FillRecordId(DriveKit::DKRecord &record, NativeRdb::ResultSet &resultSet);
52
53 /* basic */
54 int32_t HandleAlbumId(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet);
55 int32_t HandleAlbumName(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet);
56 int32_t HandleType(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet);
57 int32_t HandleAlbumLogicType(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet);
58 int32_t HandlePath(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet);
59
60 /* properties */
61 int32_t HandleProperties(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet);
62 /* properties - general */
63 int32_t HandleGeneral(DriveKit::DKRecordFieldMap &map, NativeRdb::ResultSet &resultSet);
64
65 /* identifier */
66 static const std::string recordType_;
67 OperationType type_;
68 };
69
HandleAlbumId(DriveKit::DKRecordData & data,NativeRdb::ResultSet & resultSet)70 inline int32_t AlbumDataConvertor::HandleAlbumId(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet)
71 {
72 uuid_t uuid;
73 uuid_generate(uuid);
74 char str[UUID_STR_LENGTH] = {};
75 uuid_unparse(uuid, str);
76 data[ALBUM_ID] = DriveKit::DKRecordField(str);
77 return E_OK;
78 }
79
HandleAlbumName(DriveKit::DKRecordData & data,NativeRdb::ResultSet & resultSet)80 inline int32_t AlbumDataConvertor::HandleAlbumName(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet)
81 {
82 std::string val;
83 int32_t ret = GetString(Media::PhotoAlbumColumns::ALBUM_NAME, val, resultSet);
84 if (ret != E_OK) {
85 return ret;
86 }
87 data[ALBUM_NAME] = DriveKit::DKRecordField(val);
88 return E_OK;
89 }
90
HandleType(DriveKit::DKRecordData & data,NativeRdb::ResultSet & resultSet)91 inline int32_t AlbumDataConvertor::HandleType(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet)
92 {
93 data[ALBUM_TYPE] = DriveKit::DKRecordField(AlbumType::NORMAL);
94 return E_OK;
95 }
96
HandleAlbumLogicType(DriveKit::DKRecordData & data,NativeRdb::ResultSet & resultSet)97 inline int32_t AlbumDataConvertor::HandleAlbumLogicType(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet)
98 {
99 data[ALBUM_LOGIC_TYPE] = DriveKit::DKRecordField(LogicType::LOGICAL);
100 return E_OK;
101 }
102
HandlePath(DriveKit::DKRecordData & data,NativeRdb::ResultSet & resultSet)103 inline int32_t AlbumDataConvertor::HandlePath(DriveKit::DKRecordData &data, NativeRdb::ResultSet &resultSet)
104 {
105 std::string val;
106 int32_t ret = GetString(Media::PhotoAlbumColumns::ALBUM_NAME, val, resultSet);
107 if (ret != E_OK) {
108 return ret;
109 }
110 data[ALBUM_PATH] = DriveKit::DKRecordField(ALBUM_LOCAL_PATH_PREFIX + val);
111 return E_OK;
112 }
113 } // namespace CloudSync
114 } // namespace FileManagement
115 } // namespace OHOS
116 #endif // OHOS_CLOUD_SYNC_SERVICE_ALBUM_DATA_CONVERTOR_H
117