2 * This program is free software; you can redistribute it and/or modify
3 * it under the terms of the GNU General Public License as published by
4 * the Free Software Foundation; either version 2 of the License, or
5 * (at your option) any later version.
7 * This program is distributed in the hope that it will be useful,
8 * but WITHOUT ANY WARRANTY; without even the implied warranty of
9 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10 * GNU General Public License for more details.
12 * You should have received a copy of the GNU General Public License
13 * along with this program; if not, write to the Free Software
14 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 * Author : Maxim Mamontov <faust@stargazer.dp.ua>
24 #include "parser_state.h"
26 #include "stg/common.h"
40 void ReadConfigFile(const std::string & filePath, void (C::* callback)(const std::string&, const std::string&), C * obj)
42 std::ifstream stream(filePath.c_str());
45 while (std::getline(stream, line))
49 std::string::size_type pos = line.find_first_of('#');
50 if (pos != std::string::npos)
51 line = line.substr(0, pos);
54 pos = line.find_first_of('=');
55 if (pos == std::string::npos)
57 std::ostringstream error;
58 error << "Bad file format, missing '=' in '" << filePath << ":" << num << "'.";
59 throw std::runtime_error(error.str().c_str());
61 (obj->*callback)(Trim(line.substr(0, pos)), Trim(line.substr(pos + 1, line.length() - pos - 1)));
65 } // namespace anonymous
68 using SGCONF::OPTION_BLOCK;
69 using SGCONF::OPTION_BLOCKS;
71 using SGCONF::PARSER_STATE;
73 OPTION::OPTION(const std::string & shortName,
74 const std::string & longName,
75 std::unique_ptr<ACTION> action,
76 const std::string & description)
77 : m_shortName(shortName),
79 m_action(std::move(action)),
80 m_description(description)
84 OPTION::OPTION(const std::string & longName,
85 std::unique_ptr<ACTION> action,
86 const std::string & description)
87 : m_longName(longName),
88 m_action(std::move(action)),
89 m_description(description)
93 void OPTION::Help(size_t level) const
96 throw ERROR("Option is not defined.");
97 std::string indent(level, '\t');
99 if (!m_shortName.empty())
100 std::cout << "-" << m_shortName << ", ";
101 std::cout << "--" << m_longName << " " << m_action->ParamDescription()
102 << "\t" << m_description << m_action->DefaultDescription() << "\n";
103 m_action->Suboptions().Help(level);
106 bool OPTION::Check(const char * arg) const
115 return m_longName == arg + 1;
117 return m_shortName == arg;
120 PARSER_STATE OPTION::Parse(int argc, char ** argv, void * data)
123 throw ERROR("Option is not defined.");
126 return m_action->Parse(argc, argv, data);
128 catch (const ACTION::ERROR & ex)
130 if (m_longName.empty())
131 throw ERROR("-" + m_shortName + ": " + ex.what());
133 throw m_shortName.empty() ? ERROR("--" + m_longName + ": " + ex.what())
134 : ERROR("--" + m_longName + ", -" + m_shortName + ": " + ex.what());
138 void OPTION::ParseValue(const std::string & value)
141 throw ERROR("Option is not defined.");
144 return m_action->ParseValue(value);
146 catch (const ACTION::ERROR & ex)
148 throw ERROR(m_longName + ": " + ex.what());
152 OPTION_BLOCK & OPTION_BLOCK::Add(const std::string & shortName,
153 const std::string & longName,
154 std::unique_ptr<ACTION> action,
155 const std::string & description)
157 m_options.emplace_back(shortName, longName, std::move(action), description);
161 OPTION_BLOCK & OPTION_BLOCK::Add(const std::string & longName,
162 std::unique_ptr<ACTION> action,
163 const std::string & description)
165 m_options.emplace_back(longName, std::move(action), description);
169 void OPTION_BLOCK::Help(size_t level) const
171 if (m_options.empty())
173 if (!m_description.empty())
174 std::cout << m_description << ":\n";
175 for (const auto& option : m_options)
176 option.Help(level + 1);
179 PARSER_STATE OPTION_BLOCK::Parse(int argc, char ** argv, void * data)
181 PARSER_STATE state(false, argc, argv);
184 while (state.argc > 0 && !state.stop)
186 const auto it = std::find_if(m_options.begin(), m_options.end(), [&state](const auto& opt){ return opt.Check(*state.argv); });
187 if (it != m_options.end())
188 state = it->Parse(--state.argc, ++state.argv, data);
195 void OPTION_BLOCK::ParseFile(const std::string & filePath)
197 if (access(filePath.c_str(), R_OK))
198 throw ERROR("File '" + filePath + "' does not exists.");
199 ReadConfigFile(filePath, &OPTION_BLOCK::OptionCallback, this);
202 void OPTION_BLOCK::OptionCallback(const std::string & key, const std::string & value)
204 for (auto& option : m_options)
205 if (option.Name() == key)
206 option.ParseValue(value);
209 void OPTION_BLOCKS::Help(size_t level) const
211 for (const auto& block : m_blocks)
218 PARSER_STATE OPTION_BLOCKS::Parse(int argc, char ** argv)
220 PARSER_STATE state(false, argc, argv);
221 auto it = m_blocks.begin();
222 while (state.argc > 0 && !state.stop && it != m_blocks.end())
224 state = it->Parse(state.argc, state.argv);