Add support for boost command line options in prep for unit tests.

This commit is contained in:
Arthur Britto
2012-05-02 17:35:28 -07:00
parent f775bfd856
commit c6f6914dab
3 changed files with 67 additions and 43 deletions

View File

@@ -46,27 +46,20 @@ std::string EncodeBase64(const std::string& s)
return result;
}
int commandLineRPC(int argc, char *argv[])
int commandLineRPC(const std::vector<std::string>& vCmd)
{
std::string strPrint;
int nRet = 0;
try
{
// Skip switches
while ((argc > 1) && isSwitchChar(argv[1][0]))
{
argc--;
argv++;
}
if (vCmd.size() < 2) return 1;
if (argc < 2) return 0;
std::string strMethod = argv[1];
std::string strMethod = vCmd[0];
// Parameters default to strings
Json::Value params(Json::arrayValue);
for (int i = 2; i < argc; i++)
params.append(argv[i]);
for (int i = 1; i != vCmd.size(); i++)
params.append(vCmd[i]);
// Execute
Json::Value reply = callRPC(strMethod, params);

View File

@@ -3,5 +3,5 @@
#include "../json/value.h"
extern int commandLineRPC(int argc, char *argv[]);
extern int commandLineRPC(const std::vector<std::string>& vCmd);
extern Json::Value callRPC(const std::string& strMethod, const Json::Value& params);

View File

@@ -1,34 +1,31 @@
#include "Application.h"
#include "CallRPC.h"
#include "Config.h"
#include "utils.h"
#include <iostream>
#include "CallRPC.h"
#include "Config.h"
#include <boost/program_options.hpp>
namespace po = boost::program_options;
extern void runTests();
extern bool AddSystemEntropy();
using namespace std;
using namespace boost;
/*
Detect if another is running
If so message it with the users command
*/
void startApp()
void startServer()
{
theApp = new Application();
theApp->run(); // blocks till we get a stop RPC
theApp->run(); // Blocks till we get a stop RPC.
}
void printHelp()
void printHelp(const po::options_description& desc)
{
cout << "newcoin [options] <command> <params>" << endl;
cout << "options: " << endl;
cout << " -" << endl;
cout << "commands: " << endl;
cout << desc << endl;
cout << "Commands: " << endl;
cout << " accountinfo <family>:<key>" << endl;
cout << " connect <ip> [<port>]" << endl;
cout << " createfamily [<key>]" << endl;
@@ -48,35 +45,69 @@ void printHelp()
cout << " validation_create [<seed>|<pass_phrase>|<key>]" << endl;
}
int parseCommandline(int argc, char* argv[])
int main(int argc, char* argv[])
{
int ret=0;
int iResult = 0;
po::variables_map vm; // Map of options.
//
// Set up option parsing.
//
po::options_description desc("Options");
desc.add_options()
("help,h", "Display this message.")
("command", po::value< vector<string> >(), "Specify a comma seperated RPC command.")
;
po::positional_options_description p;
p.add("command", -1);
//
// Prepare to run
//
theConfig.load();
if (!AddSystemEntropy())
{
#ifdef DEBUG
std::cerr << "Unable to add system entropy" << std::endl;
#endif
iResult = 2;
}
if(argc>1)
// Parse options, if no error.
if (!iResult)
{
ret = commandLineRPC(argc, argv);
if(ret)
printHelp();
po::store(po::command_line_parser(argc, argv)
.options(desc) // Parse options.
.positional(p) // Remainder as --command.
.run(),
vm);
po::notify(vm); // Invoke option notify functions.
}
else startApp();
return ret;
}
if (iResult)
{
nothing();
}
else if (vm.count("help"))
{
iResult = 1;
}
else if (!vm.count("command"))
{
// No arguments. Run server.
startServer();
}
else
{
// Have a RPC command.
std::vector<std::string> vCmd = vm["command"].as<std::vector<std::string> >();
iResult = commandLineRPC(vCmd);
}
int main(int argc, char* argv[])
{
// runTests();
if (1 == iResult)
printHelp(desc);
return(parseCommandline(argc, argv));
return iResult;
}
// vim:ts=4