1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/*
Copyright © 2015 Gluzskiy Alexandr (sss)
This file is part of Unknown Download Manager (UDM).
UDM is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
UDM is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with UDM. If not, see <http://www.gnu.org/licenses/>.
*/
#include <boost/program_options.hpp>
#include <boost/filesystem.hpp>
#include <iostream>
#include "api_module_metadata_storage.h"
#include "modules_handler.h"
#include "utilities.h"
#include "main.h"
core_api *module_api = nullptr;
modules_handler *modules = nullptr;
namespace bpo = boost::program_options;
runtime_config_s runtime_config;
int main(int argc, char *argv[])
{
bpo::options_description desc("Available commands and options");
desc.add_options()
("help,h", "this message")
("daemon,d", "fork to background (must be combined with \"--run\", does nothing alone)")
("config,c", bpo::value<std::string>(), "use specified config file instead of \"~/.config/udm/udm.conf\"")
("list-modules", "list all installed modules (you can set verbosity level, accepted values from 0 to 1, default = 0)")
("verbosity,v", bpo::value<short>()->default_value(0), "set global verbosity level")
("run", "start UDM server")
;
bool daemon = false, run = false;
std::string config_path = "~/.config/udm/udm.conf", parsed_config_path;
try{
bpo::variables_map vm;
bpo::store(bpo::command_line_parser(argc, argv).options(desc).style(bpo::command_line_style::unix_style | bpo::command_line_style::allow_long_disguise).run(), vm);
bpo::notify(vm);
//load config first, as it may be needed for other commands
if(vm.count("config"))
config_path = vm["config"].as<std::string>();
{
parsed_config_path = replace_home_var(config_path);
if(boost::filesystem::exists(parsed_config_path) && boost::filesystem::is_regular(parsed_config_path))
bpt::read_info(parsed_config_path, runtime_config.config_file); //TODO: finish this
else
std::cerr<<"failed to load config: \"" + config_path + "\", file does not exists or is not regular file\n";
}
if(vm.count("verbosity"))
{
runtime_config.verbosity = vm["verbosity"].as<short>();
}
if(vm.count("help"))
{
std::cout<<desc<<std::endl;
return 0;
}
if(vm.count("daemon"))
{
daemon = true;
}
if(vm.count("list-modules"))
{
if(!module_api)
module_api = new core_api;
if(!modules)
modules = new modules_handler;
std::cout<<modules->list_modules();
bpt::write_info(parsed_config_path, runtime_config.config_file); //save config on exit
return 0;
}
if(vm.count("run"))
{
run = true;
}
}
catch(std::exception &e)
{
std::cerr<<"error: "<<e.what()<<std::endl;
std::cout<<desc<<std::endl;
return -1;
}
catch(...)
{
//noop
return -1;
}
if(run)
{
if(daemon)
{
//TODO: fork here
}
//TODO: run here
}
std::cerr<<"error: no command specified"<<std::endl;
std::cout<<desc<<std::endl;
bpt::write_info(parsed_config_path, runtime_config.config_file); //save config on exit
return 0;
}
|
