source: trunk/lib/File.cc @ 446

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

fixes #243

  • Property svn:eol-style set to native
  • Property svn:keywords set to Id
File size: 9.9 KB
Line 
1// $Id: File.cc 446 2007-08-16 15:57:18Z peter $
2
3/*
4  Copyright (C) 2005, 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 "File.h"
25
26#include "Alias.h"
27#include "Configuration.h"
28#include "Date.h"
29#include "GnuplotFE.h"
30#include "html_utility.h"
31#include "HtmlStream.h"
32#include "Stats.h"
33#include "SVNblame.h"
34#include "SVNlog.h"
35
36#include <cassert>
37#include <ctime>
38#include <fstream>
39#include <iostream>
40#include <map>
41#include <string>
42#include <sys/stat.h>
43
44namespace theplu{
45namespace svndigest{
46
47
48  File::File(const u_int level, const std::string& path, 
49             const std::string& output) 
50    : Node(level,path,output) 
51  {
52    output_dir_=output;
53    if (!output_dir_.empty())
54      output_dir_+='/';
55  }
56
57
58  std::map<int, std::set<Alias> >
59  File::copyright_map(std::map<std::string, Alias>& alias) const
60  {
61    using namespace std;
62    map<int, set<Alias> > year_authors;
63    SVNlog log(path());
64
65    assert(log.author().size()==log.date().size());
66    vector<string>::const_iterator author=log.author().begin();
67    for (vector<string>::const_iterator date=log.date().begin();
68         date!=log.date().end(); ++date, ++author) {
69      time_t sec = str2time(*date);
70      tm* timeinfo = gmtime(&sec);
71
72      // find username in map of aliases
73      std::map<string,Alias>::iterator name(alias.lower_bound(*author));
74
75      // if alias exist insert alias
76      if (name != alias.end() && name->first==*author)
77        year_authors[timeinfo->tm_year].insert(name->second);
78      else {
79        // else insert user name
80        Alias a(*author,alias.size());
81        year_authors[timeinfo->tm_year].insert(a);
82        std::cerr << "svndigest: warning: no copyright alias found for `" 
83                  << *author << "'\n";
84        // insert alias to avoid multiple warnings.
85        alias.insert(name, std::make_pair(*author, a));
86      }
87    }
88    return year_authors;
89  }
90
91
92  std::string
93  File::copyright_block(const std::map<int, std::set<Alias> >& year_authors,
94                        const std::string& prefix) const
95  {
96    using namespace std;
97    stringstream ss;
98    for (map<int, set<Alias> >::const_iterator i(year_authors.begin());
99         i!=year_authors.end();) {
100      ss << prefix << "Copyright (C) "
101          << 1900+i->first;
102      map<int, set<Alias> >::const_iterator j = i;
103      assert(i!=year_authors.end());
104      while (++j!=year_authors.end() && 
105             i->second == j->second){
106        ss << ", " << 1900+(j->first);
107      }
108      // printing authors
109      std::vector<Alias> vec_alias;
110      back_insert_iterator<std::vector<Alias> > ii(vec_alias);
111      std::copy(i->second.begin(), i->second.end(), ii);
112      // sort with respect to id
113      std::sort(vec_alias.begin(), vec_alias.end(), IdCompare());
114      for (std::vector<Alias>::iterator a=vec_alias.begin();
115           a!=vec_alias.end(); ++a){
116        if (a!=vec_alias.begin())
117          ss << ",";
118        ss << " " << a->name();
119      }
120      ss << "\n";
121      i = j;
122    }
123    return ss.str();
124  }
125
126  bool File::detect_copyright(std::string& block, size_t& start_at_line,
127                              size_t& end_at_line, std::string& prefix) const
128  {
129    using namespace std;
130    ifstream is(path().c_str());
131    assert(is.good());
132    string line;
133    bool found_copyright = false;
134    bool after_copyright = false;
135    size_t line_no=0;
136    while(getline(is, line) && !after_copyright){
137      ++line_no;
138      if (found_copyright){
139        // check if line is end of copyright statement, i.e. contains
140        // no alphanumerical character (except in prefix).
141        after_copyright = true;
142        for (size_t i=0; i<line.size()&&after_copyright; ++i)
143          if (isalnum(line[i]) && !(i<prefix.size() && prefix[i]==line[i])) {
144            after_copyright = false;
145            block += line + "\n";
146          }
147        if (after_copyright)
148          end_at_line=line_no;
149      }
150      else {
151        // check whether copyright starts on this line
152        string::iterator i = search(line.begin(), line.end(), "Copyright (C)");
153        if (i!=line.end()) {
154          start_at_line=line_no;
155          prefix = line.substr(0, distance(line.begin(), i));
156          found_copyright = true;
157          block = line + "\n";
158        }
159      }
160    }
161    is.close();
162    return found_copyright;
163  }
164
165
166  std::string File::href(void) const
167  { 
168    return name()+".html"; 
169  }
170
171
172  std::string File::node_type(void) const
173  {
174    return std::string("file");
175  }
176
177
178  std::string File::output_path(void) const
179  {
180    return output_dir()+name()+".html";
181  }
182
183
184  const Stats& File::parse(const bool verbose)
185  {
186    if (verbose)
187      std::cout << "Parsing " << path_ << std::endl; 
188    stats_.reset();
189    stats_.parse(path_);
190    return stats_;
191  }
192
193
194  void File::print_blame(std::ofstream& os) const
195  {
196    os << "<br /><h3>Blame Information</h3>";
197    os << "<table class=\"blame\">\n";
198    os << "<thead>\n";
199    os << "<tr>\n";
200    os << "<th class=\"rev\">Rev</th>\n";
201    os << "<th class=\"date\">Date</th>\n";
202    os << "<th class=\"author\">Author</th>\n";
203    os << "<th class=\"line\">Line</th>\n";
204    os << "<th></th>\n";
205    os << "</tr>\n</thead>\n";
206    os << "<tbody>\n";
207    HtmlStream hs(os);
208    SVNblame blame(path_);
209    Parser parser(path_);
210    std::vector<Parser::line_type>::const_iterator
211      line_type(parser.type().begin());
212    int last=0;
213    int first=0;
214    bool using_dates=true;
215    if (GnuplotFE::instance()->dates().empty()){
216      using_dates=false;
217      last = stats_.revision();
218    }
219    else {
220      last = Date(GnuplotFE::instance()->dates().back()).seconds();
221      first = Date(GnuplotFE::instance()->dates()[0]).seconds();
222    }
223    // color is calculated linearly on time, c = kt + m
224    // brightest color (for oldest rev in log) is set to 192.
225    double k = 192.0/(first-last);
226    double m = -last*k; 
227    while (blame.valid()) {
228      std::string color;
229      if (using_dates)
230        color = hex(static_cast<int>(k*Date(blame.date()).seconds()+m),2);
231      else
232        color = hex(static_cast<int>(k*blame.revision()+m),2);
233      os << "<tr>\n<td class=\"rev\"><font color=\"#" << color
234         << color << color << "\">" << blame.revision()
235         << "</font></td>\n<td class=\"date\"><font color=\"#" << color
236         << color << color << "\">" ;
237      hs << Date(blame.date())("%d %b %y");
238      os << "</font></td>\n<td class=\"author\">";
239      hs << blame.author();
240      os << "</td>\n<td class=\"";
241      assert(line_type!=parser.type().end());
242      if (*line_type==Parser::empty)
243        os << "line-other";
244      else if (*line_type==Parser::comment)
245        os << "line-comment";
246      else
247        os << "line-code";
248      os << "\">" << blame.line_no()+1
249         << "</td>\n<td>";
250      hs << blame.line();
251      os << "</td>\n</tr>\n";
252      blame.next_line();
253      ++line_type;
254    }
255    os << "</tbody>\n";
256    os << "</table>\n";
257  }
258
259
260  void File::print_copyright(std::map<std::string, Alias>& alias, 
261                             bool verbose) const 
262  {
263    if (ignore())
264      return;
265
266    std::string old_block;
267    size_t start_line=0;
268    size_t end_line=0;
269    std::string prefix;
270    if (!detect_copyright(old_block, start_line, end_line, prefix)){
271      if (Configuration::instance().missing_copyright_warning())
272        std::cerr << "svndigest: warning: no copyright statement found in `" 
273                  << path_ << "'\n";
274      return;
275    }
276    std::map<int, std::set<Alias> > map=copyright_map(alias);
277    std::string new_block = copyright_block(map, prefix);
278    if (old_block==new_block)
279      return;
280    if (verbose)
281      std::cout << "Updating copyright in " << path_ << std::endl; 
282    update_copyright(new_block, start_line, end_line);
283  }
284
285
286  void File::print_core(const bool verbose) const 
287  {
288  }
289
290
291  void File::print_core(const std::string& user, const std::string& line_type,
292                        const SVNlog& log) const 
293  {
294    std::string outpath = user+"/"+line_type+"/"+local_path();
295    std::string imagefile = "images/"+line_type+"/"+local_path_+".png";
296    std::string html_name(outpath + ".html");
297    std::ofstream os(html_name.c_str());
298    print_header(os, name(), level_+2, user, line_type, local_path()+".html");
299    path_anchor(os);
300
301    os << "<p class=\"plot\">\n<img src='"; 
302    for (size_t i=0; i<level_; ++i)
303      os << "../";
304    os << "../../";
305    if (user=="all")
306      os << stats_.plot(imagefile,line_type);
307    else
308      os << imagefile;
309    os << "' alt='[plot]' />\n</p>";
310
311    print_author_summary(os, line_type, log);
312    os << "\n";
313
314    print_blame(os);
315
316    print_footer(os);
317    os.close(); 
318  }
319
320  void File::update_copyright(const std::string& new_block,
321                              size_t start_at_line, size_t end_at_line) const
322  {
323    // Code copied from Gnuplot -r70
324    char tmpname[]="/tmp/svndigestXXXXXX";
325    int fd=mkstemp(tmpname);  // mkstemp return a file descriptor
326    if (fd == -1)
327      throw std::runtime_error(std::string("Failed to get unique filename: ") +
328                               tmpname);
329    // Jari would like to do something like 'std::ofstream tmp(fd);'
330    // but has to settle for (which is stupid since the file is
331    // already open for writing.
332    std::ofstream tmp(tmpname);
333
334    using namespace std;
335    ifstream is(path().c_str());
336    assert(is.good());
337    string line;
338    // Copy lines before block
339    for (size_t i=1; i<start_at_line; ++i){
340      assert(is.good());
341      getline(is, line);
342      tmp << line << "\n";
343    }
344    // Printing copyright statement
345    tmp << new_block;
346    // Ignore old block lines
347    for (size_t i=start_at_line; i<end_at_line; ++i){
348      assert(is.good());
349      getline(is, line);
350    }
351    // Copy lines after block
352    while(getline(is, line))
353      tmp << line << "\n";
354
355    is.close();
356    tmp.close();
357    close(fd);
358    // get file permission
359    struct stat nodestat;
360    stat(path().c_str(),&nodestat);
361   
362    // finally move printed temporary file to original file
363    rename(tmpname, path().c_str());
364    chmod(path().c_str(), nodestat.st_mode);
365  }
366
367
368}} // end of namespace svndigest and namespace theplu
Note: See TracBrowser for help on using the repository browser.