source: trunk/lib/utility.cc @ 1291

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

fixes #263. copy_file now copies permissions too.

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