source: trunk/lib/utility.cc @ 1321

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

updating includes

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