source: trunk/lib/utility.cc @ 1459

Last change on this file since 1459 was 1459, checked in by Peter Johansson, 9 years ago

quote consistently. update release year

  • Property svn:eol-style set to native
  • Property svn:keywords set to Id
File size: 8.3 KB
Line 
1// $Id: utility.cc 1459 2012-01-09 02:49:37Z peter $
2
3/*
4  Copyright (C) 2006, 2007, 2008, 2009 Jari Häkkinen, Peter Johansson
5  Copyright (C) 2010, 2011 Peter Johansson
6
7  This file is part of svndigest, http://dev.thep.lu.se/svndigest
8
9  svndigest is free software; you can redistribute it and/or modify it
10  under the terms of the GNU General Public License as published by
11  the Free Software Foundation; either version 3 of the License, or
12  (at your option) any later version.
13
14  svndigest is distributed in the hope that it will be useful, but
15  WITHOUT ANY WARRANTY; without even the implied warranty of
16  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17  General Public License for more details.
18
19  You should have received a copy of the GNU General Public License
20  along with svndigest. If not, see <http://www.gnu.org/licenses/>.
21*/
22
23#include <config.h>
24
25#include "utility.h"
26
27#include "yat/Exception.h"
28
29#include <cassert>
30#include <cstdio>
31#include <cstdlib>
32#include <cstring>
33#include <fnmatch.h>
34#include <fstream>
35#include <sstream>
36#include <stdexcept>
37#include <string>
38#include <sys/stat.h>
39#include <unistd.h>
40
41#include <iostream>
42
43#include <libgen.h>
44
45namespace theplu{
46namespace svndigest{
47
48  int access_rights(const std::string& path, const std::string& bits)
49  {
50    if (access(path.c_str(),F_OK)) {
51      throw std::runtime_error(std::string("access_rights: ") + path +
52                               "' does not exist.");
53    }
54    int mode=0;
55    for (unsigned int i=0; i<bits.length(); ++i)
56      switch (bits[i]) {
57          case 'r':
58            mode|=R_OK;
59            break;
60          case 'w':
61            mode|=W_OK;
62            break;
63          case 'x':
64            mode|=X_OK;
65            break;
66      }
67    return access(path.c_str(),mode);
68  }
69
70
71  void chdir(const std::string& dir)
72  {
73    if (::chdir(dir.c_str()) )
74      throw yat::utility::errno_error("chdir: " + dir + ":");
75  }
76
77
78  void chmod(const std::string& filename, mode_t mode)
79  {
80    if (::chmod(filename.c_str(), mode))
81      throw yat::utility::errno_error("chmod: ");
82  }
83
84
85  std::string concatenate_path(std::string dir, std::string base)
86  {
87    if (dir.empty() || dir==".")
88      return base;
89    if (dir[dir.size()-1]!='/')
90      dir.append("/");
91    return dir+base;
92  }
93
94
95  void copy_file(const std::string& source, const std::string& target)
96  {
97    std::ofstream o(target.c_str());
98    std::ifstream i(source.c_str());
99    while (i.good()) {
100      char ch=i.get();
101      if (i.good())
102        o.put(ch);
103      if (!o.good())
104        throw std::runtime_error(std::string("copy_file: ") +
105                                 "writing target file failed '" + target + "'");
106    }
107    if (!i.eof() && (i.bad() || i.fail()))  // fail on everything except eof
108      throw std::runtime_error(std::string("copy_file: ") +
109                               "error reading source file '" + source + "'");
110    i.close(); o.close();
111    // copy permissions
112    struct stat nodestat;
113    stat(source.c_str(),&nodestat);
114    chmod(target.c_str(), nodestat.st_mode);
115  }
116
117
118  std::string directory_name(const std::string& path)
119  {
120    if (path.size()==1) {
121      if (path=="/")
122        return path;
123      else
124        return ".";
125    }
126    assert(path.size()>=2);
127    size_t pos = path.find_last_of('/', path.size()-2);
128    if (pos==std::string::npos)
129      return ".";
130    if (pos==0)
131      return "/";
132    return path.substr(0,pos);
133    return "";
134  }
135
136
137  std::string file_name(const std::string& path)
138  {
139    if (path.size()==1)
140      return path;
141    size_t pos = path.find_last_of('/');
142    if (pos==std::string::npos)
143      return path;
144    if (pos==path.size()-1)
145      return file_name(path.substr(0, path.size()-1));
146    return path.substr(pos+1);
147  }
148
149
150  std::string getenv(const std::string& var)
151  {
152    char* buffer=std::getenv(var.c_str());
153    if (!buffer)
154      throw std::runtime_error("Environment variable "+var+" is not set");
155    return std::string(buffer);
156  }
157
158
159  std::string hex(int x, unsigned int width)
160  {
161    std::stringstream ss;
162    ss << std::hex << x;
163    if (!width)
164      return ss.str();
165    if (ss.str().size()<width)
166      return std::string(width-ss.str().size(), '0') + ss.str();
167    return ss.str().substr(0, width);
168  }
169
170
171  std::string htrim(std::string str)
172  {
173    size_t length=str.size();
174    while(length && isspace(str[length-1]))
175      --length;
176    return str.substr(0,length);
177  }
178
179
180  bool fnmatch(const std::string& pattern, const std::string& str)
181  {
182    int res = ::fnmatch(pattern.c_str(), str.c_str(), 0);
183    if (res==0)
184      return true;
185    if (res!=FNM_NOMATCH) {
186      std::stringstream ss;
187      ss << "fnmatch with args: " << pattern << ", " << str;
188      throw std::runtime_error(ss.str());
189    }
190    return false;
191  }
192
193
194  void lstat(const std::string path, struct stat* nodestat)
195  {
196    int res = ::lstat(path.c_str(), nodestat);
197    if (res) {
198      throw yat::utility::errno_error("lstat: ");
199    }
200  }
201
202
203  std::string ltrim(std::string str)
204  {
205    size_t i = 0;
206    while(i<str.size() && isspace(str[i]))
207      ++i;
208    return str.substr(i);
209  }
210
211  void mkdir(const std::string& dir)
212  {
213    int code = ::mkdir(dir.c_str(),0777);
214    if (code){
215      std::stringstream ss;
216      ss << "mkdir: '" << dir << "': ";
217      throw yat::utility::errno_error(ss.str());
218    }
219  }
220
221
222  void mkdir_p(const std::string& dir)
223  {
224    if (node_exist(dir))
225      return;
226    std::string mother = directory_name(dir);
227    mkdir_p(mother);
228    mkdir(dir);
229  }
230
231
232  bool node_exist(const std::string& path)
233  {
234    struct stat buf;
235    return !stat(path.c_str(),&buf);
236  }
237
238
239  int percent(int a, int b)
240  {
241    if (b)
242      return (100*a)/b;
243    return 0;
244  }
245
246
247  std::string pwd(void)
248  {
249    char buffer[FILENAME_MAX];
250    if (!getcwd(buffer, FILENAME_MAX))
251      throw yat::utility::errno_error("pwd: ");
252    return std::string(buffer);
253  }
254
255
256  bool regexp(const std::string& pattern, const std::string& str,
257              std::vector<std::string>& vec)
258  {
259    bool regexp__(std::string::const_iterator first1,
260                  std::string::const_iterator last1,
261                  std::string::const_iterator first2,
262                  std::string::const_iterator last2,
263                  std::vector<std::string>::iterator item);
264
265    // find number of special chars
266    size_t count=0;
267    for (std::string::const_iterator i=pattern.begin(); i!=pattern.end(); ++i)
268      if (*i=='*' || *i=='?' || *i=='[')
269        ++count;
270    vec.resize(count);
271    return regexp__(pattern.begin(), pattern.end(), str.begin(), str.end(),
272                    vec.begin());
273  }
274
275  bool regexp__(std::string::const_iterator first1,
276                std::string::const_iterator last1,
277                std::string::const_iterator first2,
278                std::string::const_iterator last2,
279                std::vector<std::string>::iterator item)
280  {
281    if (first1==last1) {
282      return first2==last2;
283    }
284    if (*first1 == '*') {
285      if (first2<last2) {
286        item->push_back(*first2);
287        if (regexp__(first1, last1, first2+1, last2, item))
288          return true;
289        item->resize(item->size()-1);
290      }
291      return regexp__(first1+1, last1, first2, last2, item+1);
292    }
293    if (*first1 == '?') {
294      if (first2==last2)
295        return false;
296      *item = *first2;
297      return regexp__(first1+1, last1, first2+1, last2, item+1);
298    }
299    if (*first1 == '[') {
300      if (first2==last2)
301        return false;
302      bool found = false;
303      while (*first1 != ']') {
304        if (*first1 == *first2) {
305          found = true;
306          *item = *first2;
307        }
308        ++first1;
309        assert(first1!=last1);
310      }
311      return regexp__(first1+1, last1, first2+1, last2, item+1);
312    }
313
314    if (first2==last2)
315      return false;
316    if (*first1 != *first2)
317      return false;
318    return regexp__(first1+1, last1, first2+1, last2, item);
319  }
320
321
322  void remove(const std::string& fn)
323  {
324    if (::remove(fn.c_str())) {
325      std::string msg("remove: ");
326      msg += fn;
327      throw yat::utility::errno_error(msg);
328    }
329  }
330
331
332  void rename(const std::string& from, const std::string to)
333  {
334    int code = ::rename(from.c_str(), to.c_str());
335    if (code){
336      std::stringstream ss;
337      ss << "rename" << from << " to " << to << ": ";
338      throw yat::utility::errno_error(ss.str());
339    }
340  }
341
342
343  void replace(std::string& str, std::string old_str, std::string new_str)
344  {
345    std::string::iterator iter(str.begin());
346    while ((iter=search(iter, str.end(), old_str)) != str.end()) {
347      size_t i = iter-str.begin();
348      str = std::string(str.begin(), iter) + new_str +
349        std::string(iter+old_str.size(), str.end());
350      // pointing to char after substr we just inserted
351      iter = str.begin() + (i+new_str.size());
352    }
353  }
354
355
356  void touch(std::string str)
357  {
358    if (!node_exist(str)) {
359      std::ofstream os(str.c_str());
360      os.close();
361    }
362  }
363
364
365  std::string match(std::string::const_iterator& first,
366                    const std::string::const_iterator& last,
367                    std::string str)
368  {
369    if (match_begin(first, last, str)){
370      first+=str.size();
371      return str;
372    }
373    return std::string();
374  }
375
376}} // end of namespace svndigest and namespace theplu
Note: See TracBrowser for help on using the repository browser.