src/configuration.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2008 by Andrzej Rybczak   *
   3  *   electricityispower@gmail.com   *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include <iostream>
  22 #include <fstream>
  23
  24 #include "configuration.h"
  25 #include "misc.h"
  26
  27 using std::string;
  28
  29 namespace
  30 {
  31         std::string GetLineValue(const string &line, char a = '"', char b = '"')
  32         {
  33                 int i = 0;
  34                 int begin = -1, end = -1;
  35                 for (string::const_iterator it = line.begin(); it != line.end(); i++, it++)
  36                 {
  37                         if (*it == a || *it == b)
  38                         {
  39                                 if (begin < 0)
  40                                         begin = i+1;
  41                                 else
  42                                         end = i;
  43                         }
  44                 }
  45                 if (begin >= 0 && end >= 0)
  46                         return line.substr(begin, end-begin);
  47                 else
  48                         return "";
  49         }
  50
  51         LogLevel IntoLogLevel(const string &value)
  52         {
  53                 LogLevel result = llInfo;
  54                 if (value == "none")
  55                 {
  56                         result = llNone;
  57                 }
  58                 else if (value == "info")
  59                 {
  60                         result = llInfo;
  61                 }
  62                 else if (value == "verbose")
  63                 {
  64                         result = llVerbose;
  65                 }
  66                 return result;
  67         }
  68
  69 }
  70
  71 bool CheckFiles(ScrobbyConfig &conf)
  72 {
  73         std::ofstream f;
  74         std::ifstream g;
  75
  76         g.open(conf.file_pid.c_str());
  77         if (g.is_open())
  78         {
  79                 string strpid;
  80                 getline(g, strpid);
  81                 g.close();
  82                 pid_t pid = StrToInt(strpid);
  83                 if (pid < 1)
  84                 {
  85                         std::cerr << "pid file: " << conf.file_pid << " is invalid, trying to remove...\n";
  86                         if (unlink(conf.file_pid.c_str()) == 0)
  87                         {
  88                                 std::cout << "pid file succesfully removed.\n";
  89                         }
  90                         else
  91                         {
  92                                 std::cerr << "couldn't remove pid file.\n";
  93                                 return false;
  94                         }
  95                 }
  96                 else
  97                 {
  98                         std::cerr << "scrobby is already running with PID " << pid << "!\n";
  99                         return false;
 100                 }
 101         }
 102
 103         f.open(conf.file_pid.c_str(), std::ios_base::app);
 104         if (!f.is_open())
 105         {
 106                 std::cerr << "Cannot create/open pid file: " << conf.file_pid << std::endl;
 107                 return false;
 108         }
 109         f.close();
 110
 111         f.open(conf.file_log.c_str(), std::ios_base::app);
 112         if (!f.is_open())
 113         {
 114                 std::cerr << "Cannot create/open log file: " << conf.file_log << std::endl;
 115                 return false;
 116         }
 117         f.close();
 118
 119         f.open(conf.file_cache.c_str(), std::ios_base::app);
 120         if (!f.is_open())
 121         {
 122                 std::cerr << "Cannot create/open cache file: " << conf.file_cache << std::endl;
 123                 return false;
 124         }
 125         f.close();
 126
 127         return true;
 128 }
 129
 130 void DefaultConfiguration(ScrobbyConfig &conf)
 131 {
 132         conf.mpd_host = "localhost";
 133         conf.mpd_port = 6600;
 134         conf.mpd_timeout = 15;
 135
 136         conf.file_log = "/var/log/scrobby.log";
 137         conf.file_pid = "/var/run/scrobby.pid";
 138         conf.file_cache = "/var/cache/scrobby/scrobby.cache";
 139
 140         conf.log_level = llInfo;
 141 }
 142
 143 bool ReadConfiguration(ScrobbyConfig &conf, const string &file)
 144 {
 145         string line, v;
 146         std::ifstream f(file.c_str());
 147
 148         if (!f.is_open())
 149                 return false;
 150
 151         while (!f.eof())
 152         {
 153                 getline(f, line);
 154                 if (!line.empty() && line[0] != '#')
 155                 {
 156                         v = GetLineValue(line);
 157
 158                         if (line.find("mpd_host") != string::npos)
 159                         {
 160                                 if (!line.empty())
 161                                         conf.mpd_host = v;
 162                         }
 163                         else if (line.find("mpd_password") != string::npos)
 164                         {
 165                                 if (!v.empty())
 166                                         conf.mpd_password = v;
 167                         }
 168                         else if (line.find("mpd_port") != string::npos)
 169                         {
 170                                 if (!v.empty())
 171                                         conf.mpd_port = StrToInt(v);
 172                         }
 173                         else if (line.find("mpd_timeout") != string::npos)
 174                         {
 175                                 if (!v.empty())
 176                                         conf.mpd_timeout = StrToInt(v);
 177                         }
 178                         else if (line.find("log_file") != string::npos)
 179                         {
 180                                 if (!v.empty())
 181                                         conf.file_log = v;
 182                         }
 183                         else if (line.find("pid_file") != string::npos)
 184                         {
 185                                 if (!v.empty())
 186                                         conf.file_pid = v;
 187                         }
 188                         else if (line.find("cache_file") != string::npos)
 189                         {
 190                                 if (!v.empty())
 191                                         conf.file_cache = v;
 192                         }
 193                         else if (line.find("lastfm_user") != string::npos)
 194                         {
 195                                 if (!v.empty())
 196                                         conf.lastfm_user = v;
 197                         }
 198                         else if (line.find("lastfm_password") != string::npos)
 199                         {
 200                                 if (!v.empty())
 201                                         conf.lastfm_password = v;
 202                         }
 203                         else if (line.find("lastfm_md5_password") != string::npos)
 204                         {
 205                                 if (!v.empty())
 206                                         conf.lastfm_md5_password = v;
 207                         }
 208                         else if (line.find("log_level") != string::npos)
 209                         {
 210                                 if (!v.empty())
 211                                         conf.log_level = IntoLogLevel(v);
 212                         }
 213                 }
 214         }
 215         f.close();
 216         return true;
 217 }
 218