From e73ca6a0e7b62add68cee0e258bb46d15c4741aa Mon Sep 17 00:00:00 2001
From: Kostyantyn Ovechko <fastinetserver@gmail.com>
Date: Fri, 23 Jul 2010 16:38:06 +0300
Subject: Add command line arguments: --no-daemon and
 --conf-dir=specify_conf_dir_here

Arguments are optional. If no arguments provided, segget will run in a daemon
mode and use /etc/seggetd dir to read configuration files.
---
 segget/config.cpp |  4 ++--
 segget/segget.cpp | 33 ++++++++++++++++++++++++++-------
 segget/segget.h   |  3 ++-
 segget/settings.h |  4 ++++
 4 files changed, 34 insertions(+), 10 deletions(-)

diff --git a/segget/config.cpp b/segget/config.cpp
index 6c926df..fbbbab0 100644
--- a/segget/config.cpp
+++ b/segget/config.cpp
@@ -31,10 +31,10 @@ void Tconfig::load_settings_from_config_file(){
 		ifstream file;
 		file.exceptions (ifstream::failbit | ifstream::badbit);
 		try{
-			file.open(config_file_name.c_str());
+			file.open((settings.conf_dir+"/"+config_file_name).c_str());
 		}
 		catch(...){
-			error_log("Error opening settings file: "+config_file_name+". Default settings will be used. Check if config file: "+config_file_name+" exists and segget has rights to access it.");
+			error_log("Error opening settings file: "+settings.conf_dir+"/"+config_file_name+". Default settings will be used. Check if config file: "+config_file_name+" exists and segget has rights to access it.");
 			return;
 		}
 		try{
diff --git a/segget/segget.cpp b/segget/segget.cpp
index 59c96ed..214c452 100644
--- a/segget/segget.cpp
+++ b/segget/segget.cpp
@@ -451,18 +451,37 @@ int init_curses(){
 	}
 	return 1;
 }
+int parse_cli_arguments(int argc, char* argv[]){
+	try{
+		string option,name,value;
+		cout << "argc = " << argc << endl;
+		int posEqual;
+		for(int i = 0; i < argc; i++){
+			cout << "argv[" << i << "] = " << argv[i] << endl; 
+			option=argv[i];
+			posEqual=option.find('=');
+			name  = trim(option.substr(0,posEqual));
+			value = trim(option.substr(posEqual+1));
+			if (name=="--conf-dir") {settings.conf_dir=value; continue;};
+			if (name=="--no-daemon"){settings.no_daemon_flag=true; continue;}
+		}
+		return 0;
+	}catch(...){
+		perror("Error in segget.cpp: init_curses()");
+	}
+	return 1;
+}
 
-int main()
+int main(int argc, char* argv[])
 {
 	try{
-		bool daemon_mode_flag=true;
-		if (daemon_mode_flag){
-			start_daemon_mode();
-		}else{
+		parse_cli_arguments(argc, argv);
+		if (settings.no_daemon_flag){
 			init_curses();
 			routine();
-			// exit curses
-			endwin();
+			endwin(); // exit curses
+		}else{
+			start_daemon_mode();
 		}
 		exit (0);
 	}catch(...){
diff --git a/segget/segget.h b/segget/segget.h
index c03e6ea..45b4abd 100644
--- a/segget/segget.h
+++ b/segget/segget.h
@@ -62,13 +62,14 @@ CURLM *cm;
 
 int routine();
 void start_daemon_mode();
+int parse_cli_arguments(int argc, char* argv[]);
 int init_curses();
 int load_pkgs();
 void show_pkgs();
 int pkg_choose_segment(Tpkg * cur_pkg, uint connection_num);
 int choose_segment(uint connection_num);
 int download_pkgs();
-int main();
+int main(int argc, char* argv[]);
 void *print_message_function( void *);
 
 #endif
\ No newline at end of file
diff --git a/segget/settings.h b/segget/settings.h
index bfe9c02..28d910e 100644
--- a/segget/settings.h
+++ b/segget/settings.h
@@ -39,6 +39,8 @@ class Tsettings{
 	private:
 		void load_provide_mirror_files_restricted_patterns_vector();
 	public:
+		string conf_dir;
+		bool no_daemon_flag;
 	//folders
 		string distfiles_dir;
 		string segments_dir;
@@ -81,6 +83,8 @@ class Tsettings{
 		string debug_log_file;
 
 	Tsettings():
+			conf_dir("/etc/seggetd"),
+			no_daemon_flag(false),
 		//folders
 			distfiles_dir("./distfiles"),
 			segments_dir("./tmp"),
-- 
cgit v1.2.3-65-gdbad