66490c1aeaeb001a83a4ac723cd94211307cf9b3
[libsub.git] / src / util.cc
1 /*
2     Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include "util.h"
21 #include <string>
22 #include <sstream>
23 #include <iostream>
24 #include <cstdio>
25
26 using std::string;
27 using std::stringstream;
28 using std::getline;
29 using boost::optional;
30
31 /** @param s A string.
32  *  @return true if the string contains only space, newline or tab characters, or is empty.
33  */
34 bool
35 sub::empty_or_white_space (string s)
36 {
37         for (size_t i = 0; i < s.length(); ++i) {
38                 if (s[i] != ' ' && s[i] != '\n' && s[i] != '\t') {
39                         return false;
40                 }
41         }
42
43         return true;
44 }
45
46 optional<string>
47 sub::get_line_stringstream (stringstream* str)
48 {
49         if (!str->good ()) {
50                 return optional<string> ();
51         }
52
53         string s;
54         getline (*str, s);
55         return s;
56 }
57
58 optional<string>
59 sub::get_line_file (FILE* f)
60 {
61         char buffer[256];
62         char* r = fgets (buffer, sizeof (buffer), f);
63         if (r == 0) {
64                 return optional<string> ();
65         }
66
67         return string (buffer);
68 }
69
70 void
71 sub::remove_unicode_bom (optional<string>& line)
72 {
73         if (
74                 line->length() >= 3 &&
75                 static_cast<unsigned char> (line.get()[0]) == 0xef &&
76                 static_cast<unsigned char> (line.get()[1]) == 0xbb &&
77                 static_cast<unsigned char> (line.get()[2]) == 0xbf
78                 ) {
79
80                 /* Skip Unicode byte order mark */
81                 line = line->substr (3);
82         }
83 }