Mercurial > hg > orthanc

comparison Core/Compression/HierarchicalZipWriter.cpp @ 247:c9b3ba0fd140

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
path management in zip files
author Sebastien Jodogne <s.jodogne@gmail.com>
date Tue, 04 Dec 2012 17:27:23 +0100
parents
children 2442033911d6
comparison
equal deleted inserted replaced
246:fe6ba20d00a8 247:c9b3ba0fd140
1 /**
2 * Orthanc - A Lightweight, RESTful DICOM Store
3 * Copyright (C) 2012 Medical Physics Department, CHU of Liege,
4 * Belgium
5 *
6 * This program is free software: you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License as
8 * published by the Free Software Foundation, either version 3 of the
9 * License, or (at your option) any later version.
10 *
11 * In addition, as a special exception, the copyright holders of this
12 * program give permission to link the code of its release with the
13 * OpenSSL project's "OpenSSL" library (or with modified versions of it
14 * that use the same license as the "OpenSSL" library), and distribute
15 * the linked executables. You must obey the GNU General Public License
16 * in all respects for all of the code used other than "OpenSSL". If you
17 * modify file(s) with this exception, you may extend this exception to
18 * your version of the file(s), but you are not obligated to do so. If
19 * you do not wish to do so, delete this exception statement from your
20 * version. If you delete this exception statement from all source files
21 * in the program, then also delete it here.
22 *
23 * This program is distributed in the hope that it will be useful, but
24 * WITHOUT ANY WARRANTY; without even the implied warranty of
25 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
26 * General Public License for more details.
27 *
28 * You should have received a copy of the GNU General Public License
29 * along with this program. If not, see <http://www.gnu.org/licenses/>.
30 **/
31
32
33 #include "HierarchicalZipWriter.h"
34
35 #include "../Toolbox.h"
36 #include "../OrthancException.h"
37
38 #include <boost/lexical_cast.hpp>
39
40 namespace Orthanc
41 {
42 std::string HierarchicalZipWriter::Index::KeepAlphanumeric(const std::string& source)
43 {
44 std::string result;
45
46 bool lastSpace = false;
47
48 result.reserve(source.size());
49 for (size_t i = 0; i < source.size(); i++)
50 {
51 if (source[i] < 128 &&
52 source[i] >= 0)
53 {
54 if (isspace(source[i]))
55 {
56 if (!lastSpace)
57 {
58 lastSpace = true;
59 result.push_back(' ');
60 }
61 }
62 else if (isalnum(source[i]))
63 {
64 result.push_back(source[i]);
65 lastSpace = false;
66 }
67 }
68 }
69
70 return Toolbox::StripSpaces(result);
71 }
72
73 std::string HierarchicalZipWriter::Index::GetCurrentDirectoryPath() const
74 {
75 std::string result;
76
77 Stack::const_iterator it = stack_.begin();
78 it++; // Skip the root node (to avoid absolute paths)
79
80 while (it != stack_.end())
81 {
82 result += (*it)->name_ + "/";
83 it++;
84 }
85
86 return result;
87 }
88
89 std::string HierarchicalZipWriter::Index::EnsureUniqueFilename(const char* filename)
90 {
91 std::string standardized = KeepAlphanumeric(filename);
92
93 Directory& d = *stack_.back();
94 Directory::Content::iterator it = d.content_.find(standardized);
95
96 if (it == d.content_.end())
97 {
98 d.content_[standardized] = 1;
99 return standardized;
100 }
101 else
102 {
103 it->second++;
104 return standardized + "-" + boost::lexical_cast<std::string>(it->second);
105 }
106 }
107
108 HierarchicalZipWriter::Index::Index()
109 {
110 stack_.push_back(new Directory);
111 }
112
113 HierarchicalZipWriter::Index::~Index()
114 {
115 for (Stack::iterator it = stack_.begin(); it != stack_.end(); it++)
116 {
117 delete *it;
118 }
119 }
120
121 std::string HierarchicalZipWriter::Index::CreateFile(const char* name)
122 {
123 return GetCurrentDirectoryPath() + EnsureUniqueFilename(name);
124 }
125
126 void HierarchicalZipWriter::Index::CreateDirectory(const char* name)
127 {
128 std::string d = EnsureUniqueFilename(name);
129
130 // Push the new directory onto the stack
131 stack_.push_back(new Directory);
132 stack_.back()->name_ = d;
133 }
134
135 void HierarchicalZipWriter::Index::CloseDirectory()
136 {
137 if (IsRoot())
138 {
139 // Cannot close the root node
140 throw OrthancException(ErrorCode_BadSequenceOfCalls);
141 }
142
143 delete stack_.back();
144 stack_.pop_back();
145 }
146
147
148 HierarchicalZipWriter::HierarchicalZipWriter(const char* path)
149 {
150 writer_.SetOutputPath(path);
151 writer_.Open();
152 }
153
154 HierarchicalZipWriter::~HierarchicalZipWriter()
155 {
156 writer_.Close();
157 }
158
159 void HierarchicalZipWriter::CreateFile(const char* name)
160 {
161 std::string p = indexer_.CreateFile(name);
162 writer_.CreateFileInZip(p.c_str());
163 }
164
165 void HierarchicalZipWriter::CreateDirectory(const char* name)
166 {
167 indexer_.CreateDirectory(name);
168 }
169
170 void HierarchicalZipWriter::CloseDirectory()
171 {
172 indexer_.CloseDirectory();
173 }
174 }