6b59e728 |
1 | // BotConfig.C -*- C++ -*- |
2 | // Copyright (C) 2004 Clinton Ebadi |
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 | // 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA. |
17 | |
18 | #include "BotConfig.H" |
19 | #include "Bot.H" |
20 | #include "StringTokenizer.H" |
21 | #include <iostream> |
22 | #include <string> |
23 | #include <list> |
24 | #include <algorithm> |
25 | |
26 | BotConfig::BotConfig (std::string cf) |
27 | : config_filename (cf) |
28 | { } |
29 | |
30 | bool BotConfig::read_config () |
31 | { |
32 | std::ifstream config_file (config_filename.c_str ()); |
33 | std::string current_line; |
34 | unsigned int line_number = 1; |
35 | |
36 | options_db.clear (); |
37 | |
38 | if (!config_file) |
39 | { |
40 | // FIXME: log lines (needs local bot object, probably a |
41 | // redesign. UGH). |
42 | |
43 | // Bot::logLine ("Error: Config File Not Found `" + |
44 | // config_filename + "'"); |
45 | return false; |
46 | } |
47 | |
48 | while (!config_file.eof ()) |
49 | { |
50 | std::getline (config_file, current_line); |
51 | if (current_line.length () == 0 || current_line[0] == '#') |
52 | { |
53 | line_number++; |
54 | continue; |
55 | } |
56 | |
57 | StringTokenizer st (current_line); |
58 | std::string command = st.nextToken('=').trim().toUpper(); |
59 | StringTokenizer params (st.nextToken('=').trim()); |
60 | |
61 | options_db[command] = t_option_values (t_value_list (), |
62 | t_watcher_list ()); |
63 | while (params.hasMoreTokens (',')) |
64 | { |
65 | options_db[command].first.push_back (params.nextToken (',').trim()); |
66 | } |
67 | } |
68 | return true; |
69 | } |
70 | |
71 | |
72 | BotConfig::t_option_values BotConfig::get_option_values (std::string key) |
73 | { |
74 | t_options_db::const_iterator cf_iter = |
75 | options_db.find (key); |
76 | |
77 | if (cf_iter != options_db.end ()) |
78 | return cf_iter->second; |
79 | else |
80 | return t_option_values (); // Empty vector |
81 | } |
82 | |
83 | std::string BotConfig::set_config_file (std::string fname) |
84 | { |
85 | std::string old_config_filename = config_filename; |
86 | config_filename = fname; |
87 | options_db.clear (); |
88 | read_config (); |
89 | return old_config_filename; |
90 | } |
91 | |
92 | namespace |
93 | { |
94 | struct push_back_ |
95 | { |
96 | BotConfig::t_value_list* lst; |
97 | |
98 | push_back_ (BotConfig::t_value_list* l) |
99 | : lst (l) |
100 | {} |
101 | |
102 | void operator()(BotConfig::t_value val) |
103 | { |
104 | lst->push_back (val); |
105 | } |
106 | }; |
107 | } |
108 | |
109 | void BotConfig::set_option_value (std::string key, t_value_list values, |
110 | bool append) |
111 | { |
112 | t_options_db::iterator cf_iter = options_db.find (key); |
113 | if (cf_iter != options_db.end ()) |
114 | { |
115 | if (append) |
116 | { |
117 | // FIXME: Potentially slow (make this push_front in 3.0) |
118 | std::for_each (values.begin (), values.end (), |
119 | push_back_ (&cf_iter->second.first)); |
120 | } |
121 | else |
122 | cf_iter->second.first = values; |
123 | } |
124 | } |
125 | |
126 | |