int main(int argc, char* argv[]) {
setupSignals();
mongo::shellUtils::RecordMyLocation( argv[ 0 ] );
mongo::ScriptEngine::setup();
auto_ptr< mongo::Scope > scope( mongo::globalScriptEngine->createScope() );
string url = "test";
string dbhost;
string port;
string username;
string password;
bool runShell = false;
bool nodb = false;
string script;
int argNumber = 1;
for ( ; argNumber < argc; argNumber++) {
const char* str = argv[argNumber];
if (strcmp(str, "--shell") == 0) {
runShell = true;
continue;
}
if (strcmp(str, "--nodb") == 0) {
nodb = true;
continue;
}
if ( strcmp( str , "--port" ) == 0 ){
port = argv[argNumber+1];
argNumber++;
continue;
}
if ( strcmp( str , "--host" ) == 0 ){
dbhost = argv[argNumber+1];
argNumber++;
continue;
}
if ( strcmp( str , "--eval" ) == 0 ){
script = argv[argNumber+1];
argNumber++;
continue;
}
if ( strcmp( str , "-u" ) == 0 ){
username = argv[argNumber+1];
argNumber++;
continue;
}
if ( strcmp( str , "-p" ) == 0 ){
password = argv[argNumber+1];
argNumber++;
continue;
}
if ( strstr( str , "-p" ) == str ){
password = str;
password = password.substr( 2 );
continue;
}
if ( strcmp(str, "--help") == 0 ||
strcmp(str, "-h" ) == 0 ) {
cout
<< "usage: " << argv[0] << " [options] [db address] [file names]\n"
<< "db address can be:\n"
<< " foo = foo database on local machine\n"
<< " 192.169.0.5/foo = foo database on 192.168.0.5 machine\n"
<< " 192.169.0.5:9999/foo = foo database on 192.168.0.5 machine on port 9999\n"
<< "options\n"
<< " --shell run the shell after executing files\n"
<< " -u <username>\n"
<< " -p<password> - notice no space\n"
<< " --host <host> - server to connect to\n"
<< " --port <port> - port to connect to\n"
<< " --nodb don't connect to mongo program on startup. No 'db address' arg expected.\n"
<< " --eval <script> evaluate javascript.\n"
<< "file names: a list of files to run. will exit after unless --shell is specified\n"
;
return 0;
}
if (strcmp(str, "-f") == 0) {
continue;
}
if (strncmp(str, "--", 2) == 0) {
printf("Warning: unknown flag %s.\nTry --help for options\n", str);
continue;
}
if ( nodb )
break;
const char * last = strstr( str , "/" );
if ( last )
last++;
else
last = str;
if ( ! strstr( last , "." ) ){
url = str;
continue;
}
if ( strstr( last , ".js" ) )
break;
path p( str );
if ( ! boost::filesystem::exists( p ) ){
url = str;
continue;
}
break;
}
scope->externalSetup();
mongo::shellUtils::installShellUtils( *scope );
if ( !nodb ) { // connect to db
cout << "url: " << url << endl;
string setup = (string)"db = connect( \"" + fixHost( url , dbhost , port ) + "\")";
if ( ! scope->exec( setup , "(connect)" , false , true , false ) )
return -1;
if ( username.size() && password.size() ){
stringstream ss;
ss << "if ( ! db.auth( \"" << username << "\" , \"" << password << "\" ) ){ throw 'login failed'; }";
if ( ! scope->exec( ss.str() , "(auth)" , true , true , false ) ){
cout << "login failed" << endl;
return -1;
}
}
}
if ( !script.empty() ) {
script = "function() { " + script + " }";
mongo::shellUtils::MongoProgramScope s;
if ( scope->invoke( script.c_str(), mongo::BSONObj() ) )
return -4;
}
int numFiles = 0;
for ( ; argNumber < argc; argNumber++) {
const char* str = argv[argNumber];
mongo::shellUtils::MongoProgramScope s;
if ( ! scope->execFile( str , false , true , false ) ){
return -3;
}
numFiles++;
}
if ( numFiles == 0 && script.empty() )
runShell = true;
if ( runShell ){
mongo::shellUtils::MongoProgramScope s;
shellHistoryInit();
cout << "type \"help\" for help" << endl;
//v8::Handle<v8::Object> shellHelper = baseContext_->Global()->Get( v8::String::New( "shellHelper" ) )->ToObject();
while ( 1 ){
char * line = shellReadline( "> " );
if ( ! line || ( strlen(line) == 4 && strstr( line , "exit" ) ) ){
cout << "bye" << endl;
break;
}
string code = line;
if ( code == "exit" ){
break;
}
bool wascmd = false;
{
string cmd = line;
if ( cmd.find( " " ) > 0 )
cmd = cmd.substr( 0 , cmd.find( " " ) );
scope->exec( (string)"__iscmd__ = shellHelper[\"" + cmd + "\"];" , "(shellhelp1)" , false , true , true );
if ( scope->getBoolean( "__iscmd__" ) ){
scope->exec( (string)"shellHelper( \"" + cmd + "\" , \"" + code.substr( cmd.size() ) + "\");" , "(shellhelp2)" , false , true , false );
wascmd = true;
}
}
if ( ! wascmd ){
scope->exec( code.c_str() , "(shell)" , false , true , false );
scope->exec( "shellPrintHelper( __lastres__ );" , "(shell2)" , true , true , false );
}
shellHistoryAdd( line );
}
shellHistoryDone();
}
return 0;
}
int main(int argc, char* argv[]) {
setupSignals();
//v8::V8::SetFlagsFromCommandLine(&argc, argv, true);
v8::Locker l;
v8::HandleScope handle_scope;
v8::Handle<v8::ObjectTemplate> global = v8::ObjectTemplate::New();
installShellUtils( global );
installMongoGlobals( global );
baseContext_ = v8::Context::New(NULL, global);
v8::Context::Scope context_scope(baseContext_);
{ // init mongo code
v8::HandleScope handle_scope;
if ( ! ExecuteString( v8::String::New( jsconcatcode ) , v8::String::New( "(mongo init)" ) , false , true ) )
return -1;
}
string url = "test";
string dbhost;
string port;
string username;
string password;
bool runShell = false;
bool nodb = false;
int argNumber = 1;
for ( ; argNumber < argc; argNumber++) {
const char* str = argv[argNumber];
if (strcmp(str, "--shell") == 0) {
runShell = true;
continue;
}
if (strcmp(str, "--nodb") == 0) {
nodb = true;
continue;
}
if ( strcmp( str , "--port" ) == 0 ){
port = argv[argNumber+1];
argNumber++;
continue;
}
if ( strcmp( str , "--host" ) == 0 ){
dbhost = argv[argNumber+1];
argNumber++;
continue;
}
if ( strcmp( str , "-u" ) == 0 ){
username = argv[argNumber+1];
argNumber++;
continue;
}
if ( strcmp( str , "-p" ) == 0 ){
password = argv[argNumber+1];
argNumber++;
continue;
}
if ( strstr( str , "-p" ) == str ){
password = str;
password = password.substr( 2 );
continue;
}
if ( strcmp(str, "--help") == 0 ||
strcmp(str, "-h" ) == 0 ) {
cout
<< "usage: " << argv[0] << " [options] [db address] [file names]\n"
<< "db address can be:\n"
<< " foo = foo database on local machine\n"
<< " 192.169.0.5/foo = foo database on 192.168.0.5 machine\n"
<< " 192.169.0.5:9999/foo = foo database on 192.168.0.5 machine on port 9999\n"
<< "options\n"
<< " --shell run the shell after executing files\n"
<< " -u <username>\n"
<< " -p<password> - notice no space\n"
<< " --host <host> - server to connect to\n"
<< " --port <port> - port to connect to\n"
<< " --nodb don't connect to mongo program on startup. No 'db address' arg expected.\n"
<< "file names: a list of files to run. will exit after unless --shell is specified\n"
;
return 0;
}
if (strcmp(str, "-f") == 0) {
continue;
}
if (strncmp(str, "--", 2) == 0) {
printf("Warning: unknown flag %s.\nTry --help for options\n", str);
continue;
}
if ( nodb )
break;
const char * last = strstr( str , "/" );
if ( last )
last++;
else
last = str;
if ( ! strstr( last , "." ) ){
url = str;
continue;
}
if ( strstr( last , ".js" ) )
break;
path p( str );
if ( ! boost::filesystem::exists( p ) ){
url = str;
continue;
}
break;
}
if ( !nodb ) { // init mongo code
v8::HandleScope handle_scope;
cout << "url: " << url << endl;
string setup = (string)"db = connect( \"" + fixHost( url , dbhost , port ) + "\")";
if ( ! ExecuteString( v8::String::New( setup.c_str() ) , v8::String::New( "(connect)" ) , false , true ) ){
return -1;
}
if ( username.size() && password.size() ){
stringstream ss;
ss << "if ( ! db.auth( \"" << username << "\" , \"" << password << "\" ) ){ throw 'login failed'; }";
if ( ! ExecuteString( v8::String::New( ss.str().c_str() ) , v8::String::New( "(auth)" ) , true , true ) ){
cout << "login failed" << endl;
return -1;
}
}
}
int numFiles = 0;
for ( ; argNumber < argc; argNumber++) {
const char* str = argv[argNumber];
v8::HandleScope handle_scope;
v8::Handle<v8::String> file_name = v8::String::New(str);
v8::Handle<v8::String> source = ReadFile(str);
if (source.IsEmpty()) {
printf("Error reading '%s'\n", str);
return 1;
}
if (!ExecuteString(source, file_name, false, true)){
cout << "error processing: " << file_name << endl;
return 1;
}
numFiles++;
}
if ( numFiles == 0 )
runShell = true;
if ( runShell ){
shellHistoryInit();
cout << "type \"help\" for help" << endl;
v8::Handle<v8::Object> shellHelper = baseContext_->Global()->Get( v8::String::New( "shellHelper" ) )->ToObject();
while ( 1 ){
char * line = shellReadline( "> " );
if ( ! line || ( strlen(line) == 4 && strstr( line , "exit" ) ) ){
cout << "bye" << endl;
break;
}
string code = line;
if ( code == "exit" ){
break;
}
{
string cmd = line;
if ( cmd.find( " " ) > 0 )
cmd = cmd.substr( 0 , cmd.find( " " ) );
if ( shellHelper->HasRealNamedProperty( v8::String::New( cmd.c_str() ) ) ){
stringstream ss;
ss << "shellHelper( \"" << cmd << "\" , \"" << code.substr( cmd.size() ) << "\" )";
code = ss.str();
}
}
v8::HandleScope handle_scope;
ExecuteString(v8::String::New( code.c_str() ),
v8::String::New("(shell)"),
true,
true);
shellHistoryAdd( line );
}
shellHistoryDone();
}
return 0;
}
