source: trunk/lib/utility.cc @ 1320

Last change on this file since 1320 was 1320, checked in by Peter Johansson, 11 years ago

prefer PATH_MAX over MAXPATHLEN

  • Property svn:eol-style set to native
  • Property svn:keywords set to Id
File size: 8.6 KB
Line 
1// $Id: utility.cc 1320 2010-11-28 16:43:54Z peter $
2
3/*
4  Copyright (C) 2006, 2007, 2008, 2009 Jari Häkkinen, Peter Johansson
5  Copyright (C) 2010 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 "subversion_info.h"
28#include "yat/Exception.h"
29
30#include <cassert>
31#include <cerrno> 
32#include <cstdio>
33#include <cstdlib>
34#include <cstring>
35#include <fnmatch.h>
36#include <fstream>
37#include <sstream>
38#include <stdexcept>
39#include <string>
40#include <sys/stat.h>
41#include <unistd.h>
42
43#include <iostream>
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  std::string concatenate_path(std::string dir, std::string base)
79  {
80    if (dir.empty() || dir==".")
81      return base;
82    if (dir[dir.size()-1]!='/')
83      dir.append("/");
84    return dir+base;
85  }
86
87
88  void copy_file(const std::string& source, const std::string& target)
89  {
90    std::ofstream o(target.c_str());
91    std::ifstream i(source.c_str());
92    while (i.good()) {
93      char ch=i.get();
94      if (i.good())
95        o.put(ch);
96      if (!o.good())
97        throw std::runtime_error(std::string("copy_file: ") +
98                                 "writing target file failed '" + target + "'");
99    }
100    if (!i.eof() && (i.bad() || i.fail()))  // fail on everything except eof
101      throw std::runtime_error(std::string("copy_file: ") +
102                               "error reading source file '" + source + "'");
103    i.close(); o.close();
104    // copy permissions
105    struct stat nodestat;
106    stat(source.c_str(),&nodestat);   
107    chmod(target.c_str(), nodestat.st_mode);
108  }
109
110
111  std::string directory_name(std::string path)
112  {
113    size_t pos = path.find_last_of("/");
114    if (pos==std::string::npos)
115      return ".";
116    if (pos==path.size()-1)
117      return directory_name(path.substr(0,path.size()-2));
118    return path.substr(0,pos+1);
119  }
120
121
122  std::string file_name(const std::string& full_path)
123  {
124    std::stringstream ss(full_path);
125    std::string name;
126    while (getline(ss,name,'/')) {}
127    return name;
128  }
129
130
131  std::string getenv(const std::string& var)
132  {
133    char* buffer=std::getenv(var.c_str());
134    if (!buffer)
135      throw std::runtime_error("Environment variable "+var+" is not set");
136    return std::string(buffer);
137  }
138
139
140  std::string hex(int x, unsigned int width)
141  {
142    std::stringstream ss;
143    ss << std::hex << x;
144    if (!width)
145      return ss.str();
146    if (ss.str().size()<width) 
147      return std::string(width-ss.str().size(), '0') + ss.str();
148    return ss.str().substr(0, width);
149  }
150
151
152  std::string htrim(std::string str)
153  {
154    size_t length=str.size();
155    while(length && isspace(str[length-1]))
156      --length;
157    return str.substr(0,length);
158  }
159
160
161  bool fnmatch(const std::string& pattern, const std::string& str)
162  {
163    int res = ::fnmatch(pattern.c_str(), str.c_str(), 0);
164    if (res==0)
165      return true;
166    if (res!=FNM_NOMATCH) {
167      std::stringstream ss;
168      ss << "fnmatch with args: " << pattern << ", " << str;
169      throw std::runtime_error(ss.str());
170    }                     
171    return false;
172  }
173
174
175  void lstat(const std::string path, struct stat* nodestat)
176  {
177    int res = ::lstat(path.c_str(), nodestat);
178    if (res) {
179      throw yat::utility::errno_error("lstat: ");
180    }
181  }
182
183
184  std::string ltrim(std::string str)
185  {
186    size_t i = 0;
187    while(i<str.size() && isspace(str[i]))
188      ++i;
189    return str.substr(i);
190  }
191
192  void mkdir(const std::string& dir)
193  { 
194    int code = ::mkdir(dir.c_str(),0777);
195    if (code){
196      std::stringstream ss;
197      ss << "mkdir: `" << dir << "': ";
198      throw yat::utility::errno_error(ss.str());
199    }
200  }
201
202
203  void mkdir_p(const std::string& dir)
204  { 
205    if (node_exist(dir))
206      return;
207    std::string mother = directory_name(dir);
208    mkdir_p(mother);
209    mkdir(dir);
210  }
211
212
213  bool node_exist(const std::string& path)
214  {
215    struct stat buf;
216    return !stat(path.c_str(),&buf);
217  }
218
219
220  int percent(int a, int b)
221  {
222    if (b)
223      return (100*a)/b;
224    return 0;
225  }
226
227
228  std::string pwd(void)
229  {
230    char buffer[PATH_MAX];
231    if (!getcwd(buffer, PATH_MAX))
232      throw yat::utility::errno_error("pwd: ");
233    return std::string(buffer);
234  }
235
236
237  bool regexp(const std::string& pattern, const std::string& str,
238              std::vector<std::string>& vec)
239  {
240    bool regexp__(std::string::const_iterator first1, 
241                  std::string::const_iterator last1,
242                  std::string::const_iterator first2,
243                  std::string::const_iterator last2,
244                  std::vector<std::string>::iterator item);
245   
246    // find number of special chars
247    size_t count=0;
248    for (std::string::const_iterator i=pattern.begin(); i!=pattern.end(); ++i)
249      if (*i=='*' || *i=='?' || *i=='[')
250        ++count;
251    vec.resize(count);
252    return regexp__(pattern.begin(), pattern.end(), str.begin(), str.end(),
253                    vec.begin());
254  }
255
256  bool regexp__(std::string::const_iterator first1, 
257                std::string::const_iterator last1,
258                std::string::const_iterator first2,
259                std::string::const_iterator last2,
260                std::vector<std::string>::iterator item)
261  {
262    if (first1==last1) {
263      return first2==last2;
264    }
265    if (*first1 == '*') {
266      if (first2<last2) {
267        item->push_back(*first2);
268        if (regexp__(first1, last1, first2+1, last2, item))
269          return true;
270        item->resize(item->size()-1);
271      }
272      return regexp__(first1+1, last1, first2, last2, item+1);
273    }
274    if (*first1 == '?') {
275      if (first2==last2)
276        return false;
277      *item = *first2;
278      return regexp__(first1+1, last1, first2+1, last2, item+1);
279    }
280    if (*first1 == '[') {
281      if (first2==last2)
282        return false;
283      bool found = false;
284      while (*first1 != ']') {
285        if (*first1 == *first2) {
286          found = true;
287          *item = *first2;
288        }
289        ++first1;
290        assert(first1!=last1);
291      }
292      return regexp__(first1+1, last1, first2+1, last2, item+1);
293    }
294
295    if (first2==last2)
296      return false;
297    if (*first1 != *first2)
298      return false;
299    return regexp__(first1+1, last1, first2+1, last2, item);
300  }
301
302
303  void rename(const std::string& from, const std::string to)
304  {
305    int code = ::rename(from.c_str(), to.c_str());
306    if (code){
307      std::stringstream ss;
308      ss << "rename" << from << " to " << to << ": ";
309      throw yat::utility::errno_error(ss.str());
310    }
311  }
312
313
314  void replace(std::string& str, std::string old_str, std::string new_str)
315  {
316    std::string::iterator iter(str.begin());
317    while ((iter=search(iter, str.end(), old_str)) != str.end()) {
318      size_t i = iter-str.begin();
319      str = std::string(str.begin(), iter) + new_str + 
320        std::string(iter+old_str.size(), str.end());
321      // pointing to char after substr we just inserted
322      iter = str.begin() + (i+new_str.size()); 
323    }
324  }
325
326
327  void touch(std::string str)
328  {
329    if (!node_exist(str)) {
330      std::ofstream os(str.c_str());
331      os.close();
332    }
333  }
334
335  time_t str2time(const std::string& str)
336  {
337    //  str in format 2006-09-09T10:55:52.132733Z
338    std::stringstream sstream(str);
339    time_t rawtime;
340    struct tm * timeinfo;
341    time ( &rawtime );
342    timeinfo =  gmtime ( &rawtime );
343
344    unsigned int year, month, day, hour, minute, second;
345    std::string tmp;
346    getline(sstream,tmp,'-');
347    year=atoi(tmp.c_str());
348    timeinfo->tm_year = year - 1900;
349
350    getline(sstream,tmp,'-');
351    month=atoi(tmp.c_str());
352    timeinfo->tm_mon = month - 1;
353
354    getline(sstream,tmp,'T');
355    day=atoi(tmp.c_str());
356    timeinfo->tm_mday = day;
357
358    getline(sstream,tmp,':');
359    hour=atoi(tmp.c_str());
360    timeinfo->tm_hour = hour;
361
362    getline(sstream,tmp,':');
363    minute=atoi(tmp.c_str());
364    timeinfo->tm_min = minute;
365
366    getline(sstream,tmp,'.');
367    second=atoi(tmp.c_str());
368    timeinfo->tm_sec = second;
369
370    return mktime(timeinfo);
371  }
372
373
374  std::string match(std::string::const_iterator& first,
375                    const std::string::const_iterator& last,
376                    std::string str)
377  {
378    if (match_begin(first, last, str)){
379      first+=str.size();
380      return str;
381    }
382    return std::string();
383  }
384
385}} // end of namespace svndigest and namespace theplu
Note: See TracBrowser for help on using the repository browser.