source: trunk/lib/utility.cc @ 482

Last change on this file since 482 was 482, checked in by Peter Johansson, 14 years ago

fixes #79 and #5. Only updating changed files.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Id
File size: 5.2 KB
Line 
1// $Id: utility.cc 482 2007-10-13 01:40:38Z peter $
2
3/*
4  Copyright (C) 2006, 2007 Jari Häkkinen, Peter Johansson
5
6  This file is part of svndigest, http://trac.thep.lu.se/trac/svndigest
7
8  svndigest is free software; you can redistribute it and/or modify it
9  under the terms of the GNU General Public License as published by
10  the Free Software Foundation; either version 2 of the License, or
11  (at your option) any later version.
12
13  svndigest is distributed in the hope that it will be useful, but
14  WITHOUT ANY WARRANTY; without even the implied warranty of
15  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  General Public License for more details.
17
18  You should have received a copy of the GNU General Public License
19  along with this program; if not, write to the Free Software
20  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
21  02111-1307, USA.
22*/
23
24#include "utility.h"
25
26#include <cerrno> 
27#include <cstdlib>
28#include <fstream>
29#include <sstream>
30#include <stdexcept>
31#include <string>
32#include <sys/param.h>
33#include <unistd.h>
34
35#include <iostream>
36
37namespace theplu{
38namespace svndigest{
39
40  int access_rights(const std::string& path, const std::string& bits)
41  {
42    if (access(path.c_str(),F_OK)) {
43      throw std::runtime_error(std::string("access_rights: ") + path +
44                               "' does not exist.");
45    }
46    int mode=0;
47    for (u_int i=0; i<bits.length(); ++i)
48      switch (bits[i]) {
49          case 'r':
50            mode|=R_OK;
51            break;
52          case 'w':
53            mode|=W_OK;
54            break;
55          case 'x':
56            mode|=X_OK;
57            break;
58      }
59    return access(path.c_str(),mode);
60  }
61
62
63  void copy_file(const std::string& source, const std::string& target)
64  {
65    std::ofstream o(target.c_str());
66    std::ifstream i(source.c_str());
67    while (i.good()) {
68      char ch=i.get();
69      if (i.good())
70        o.put(ch);
71      if (!o.good())
72        throw std::runtime_error(std::string("copy_file: ") +
73                                 "writing target file failed '" + target + "'");
74    }
75    if (!i.eof() && (i.bad() || i.fail()))  // fail on everything except eof
76      throw std::runtime_error(std::string("copy_file: ") +
77                               "error reading source file '" + source + "'");
78    i.close(); o.close();
79  }
80
81
82  std::string directory_name(std::string path)
83  {
84    size_t pos = path.find_last_of("/");
85    if (pos==path.size()-1)
86      return directory_name(path.substr(0,path.size()-2));
87    return path.substr(0,pos+1);
88  }
89
90
91  std::string file_name(const std::string& full_path)
92  {
93    std::stringstream ss(full_path);
94    std::string name;
95    while (getline(ss,name,'/')) {}
96    return name;
97  }
98
99
100  std::string getenv(const std::string& var)
101  {
102    char* buffer=std::getenv(var.c_str());
103    if (!buffer)
104      throw std::runtime_error("Environment variable "+var+" is not set");
105    return std::string(buffer);
106  }
107
108
109  std::string hex(int x, u_int width)
110  {
111    std::stringstream ss;
112    ss << std::hex << x;
113    if (!width)
114      return ss.str();
115    if (ss.str().size()<width) 
116      return std::string(width-ss.str().size(), '0') + ss.str();
117    return ss.str().substr(0, width);
118  }
119
120
121  std::string htrim(std::string str)
122  {
123    size_t length=str.size();
124    while(length && isspace(str[length-1]))
125      --length;
126    return str.substr(0,length);
127  }
128
129
130  bool is_int(std::string s)
131  {
132    std::stringstream ss(s);
133    int a;
134    ss >> a;
135    if(ss.fail()) 
136      return false;
137    // Check that nothing is left on stream
138    std::string b;
139    ss >> b;
140    return b.empty();
141  }
142
143
144  std::string ltrim(std::string str)
145  {
146    size_t i = 0;
147    while(i<str.size() && isspace(str[i]))
148      ++i;
149    return str.substr(i);
150  }
151
152  void mkdir(const std::string& dir)
153  { 
154    int code = ::mkdir(dir.c_str(),0777);
155    if (code){
156      std::stringstream ss;
157      ss << "mkdir(" << dir << "): failed with error code: errno=" << errno;
158      throw std::runtime_error(ss.str());
159    }
160  }
161
162
163  bool node_exist(const std::string& path)
164  {
165    struct stat buf;
166    return !stat(path.c_str(),&buf);
167  }
168
169
170  int percent(int a, int b)
171  {
172    if (b)
173      return (100*a)/b;
174    return 0;
175  }
176
177
178  std::string pwd(void)
179  {
180    char buffer[MAXPATHLEN];
181    if (!getcwd(buffer, MAXPATHLEN))
182      throw std::runtime_error("Failed to get current working directory");
183    return std::string(buffer);
184  }
185
186
187  void touch(std::string str)
188  {
189    if (!node_exist(str)) {
190      std::ofstream os(str.c_str());
191      os.close();
192    }
193  }
194
195  time_t str2time(const std::string& str)
196  {
197    //  str in format 2006-09-09T10:55:52.132733Z
198    std::stringstream sstream(str);
199    time_t rawtime;
200    struct tm * timeinfo;
201    time ( &rawtime );
202    timeinfo =  gmtime ( &rawtime );
203
204    u_int year, month, day, hour, minute, second;
205    std::string tmp;
206    getline(sstream,tmp,'-');
207    year=atoi(tmp.c_str());
208    timeinfo->tm_year = year - 1900;
209
210    getline(sstream,tmp,'-');
211    month=atoi(tmp.c_str());
212    timeinfo->tm_mon = month - 1;
213
214    getline(sstream,tmp,'T');
215    day=atoi(tmp.c_str());
216    timeinfo->tm_mday = day;
217
218    getline(sstream,tmp,':');
219    hour=atoi(tmp.c_str());
220    timeinfo->tm_hour = hour;
221
222    getline(sstream,tmp,':');
223    minute=atoi(tmp.c_str());
224    timeinfo->tm_min = minute;
225
226    getline(sstream,tmp,'.');
227    second=atoi(tmp.c_str());
228    timeinfo->tm_sec = second;
229
230    return mktime(timeinfo);
231  }
232
233
234  std::string match(std::string::const_iterator& first,
235                    const std::string::const_iterator& last,
236                    std::string str)
237  {
238    if (match_begin(first, last, str)){
239      first+=str.size();
240      return str;
241    }
242    return std::string();
243  }
244
245}} // end of namespace svndigest and namespace theplu
Note: See TracBrowser for help on using the repository browser.