1 /*
2 * Copyright (c) 2021 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 "xml_key_node.h"
17 #include <algorithm>
18 #include <fstream>
19 #include <iostream>
20 #include <vector>
21 #include "resource_util.h"
22
23 namespace OHOS {
24 namespace Global {
25 namespace Restool {
26 using namespace std;
27 const map<XmlKeyNode::KeyType, string> XmlKeyNode::KEY_TO_FILE_NAME = {
28 {KeyType::NODE, "nodes.key" },
29 {KeyType::ATTRIBUTE, "attributes.key" },
30 {KeyType::CONSTANT, "constants.key" },
31 {KeyType::CONTENT, "contents.key" }
32 };
XmlKeyNode()33 XmlKeyNode::XmlKeyNode() : keyId_(0)
34 {
35 PushKey("");
36 }
37
~XmlKeyNode()38 XmlKeyNode::~XmlKeyNode()
39 {
40 }
41
PushKey(const string & name)42 int32_t XmlKeyNode::PushKey(const string &name)
43 {
44 auto result = keyMap_.emplace(name, keyId_);
45 if (result.second) {
46 return keyId_++;
47 }
48 return result.first->second;
49 }
50
SaveToFile(const string & filePath) const51 bool XmlKeyNode::SaveToFile(const string &filePath) const
52 {
53 ofstream out(filePath, ofstream::out | ofstream::binary);
54 if (!out.is_open()) {
55 cerr << "Error: open failed '" << filePath << "', reason: " << strerror(errno) << endl;
56 return false;
57 }
58
59 vector<pair<string, int32_t>> sets(keyMap_.begin(), keyMap_.end());
60 sort(sets.begin(), sets.end(), [](const auto &a, const auto &b) {
61 return a.second < b.second;
62 });
63
64 char null = 0;
65 for (const auto &iter : sets) {
66 out.write(reinterpret_cast<const char *>(iter.first.c_str()), iter.first.length());
67 out.write(&null, sizeof(char));
68 }
69 return true;
70 }
71
LoadFromFile(const string & filePath)72 bool XmlKeyNode::LoadFromFile(const string &filePath)
73 {
74 return LoadFromFile(filePath, nullptr);
75 }
76
LoadFromFile(const std::string & filePath,RefParser parser)77 bool XmlKeyNode::LoadFromFile(const std::string &filePath, RefParser parser)
78 {
79 ifstream in(filePath, ifstream::in | ifstream::binary);
80 if (!in.is_open()) {
81 cerr << "Error: open failed '" << filePath << "', reason: " << strerror(errno) << endl;
82 return false;
83 }
84
85 string inputLine;
86 getline(in, inputLine);
87 if (inputLine.empty()) {
88 return true;
89 }
90
91 string::size_type offset = 0;
92 while (offset < inputLine.length()) {
93 string item(inputLine.c_str() + offset);
94 offset = offset + item.length() + sizeof(char);
95 if (parser && !parser(item)) {
96 return false;
97 }
98 PushKey(item);
99 }
100 return true;
101 }
102
GetKeyValue(int32_t keyId,std::string & value) const103 bool XmlKeyNode::GetKeyValue(int32_t keyId, std::string &value) const
104 {
105 auto result = find_if(keyMap_.begin(), keyMap_.end(), [&keyId](const auto &iter) {
106 return keyId == iter.second;
107 });
108 if (result == keyMap_.end()) {
109 return false;
110 }
111 value = result->first;
112 return true;
113 }
114 }
115 }
116 }