Просмотр исходного кода

add mongo-hacker

Coloured output, useful API enhancements.
weiyi 12 лет назад
Родитель
Сommit
7f2e2c0417
3 измененных файлов с 1080 добавлено и 2 удалено
  1. 3 0
      .gitmodules
  2. 1 0
      mongo/mongo-hacker
  3. 1076 2
      mongo/mongorc.js

+ 3 - 0
.gitmodules

@@ -76,3 +76,6 @@
 [submodule "vim/bundle/vim-autoclose"]
 [submodule "vim/bundle/vim-autoclose"]
 	path = vim/bundle/vim-autoclose
 	path = vim/bundle/vim-autoclose
 	url = https://github.com/Townk/vim-autoclose.git
 	url = https://github.com/Townk/vim-autoclose.git
+[submodule "mongo/mongo-hacker"]
+	path = mongo/mongo-hacker
+	url = https://github.com/TylerBrock/mongo-hacker.git

+ 1 - 0
mongo/mongo-hacker

@@ -0,0 +1 @@
+Subproject commit 5dad0b37265acadda0497baaba7f83e2166d2260

+ 1076 - 2
mongo/mongorc.js

@@ -1,4 +1,1078 @@
+/*
+ *
+ * Mongo-Hacker
+ * MongoDB Shell Enhancements for Hackers
+ *
+ * Tyler J. Brock - 2013
+ *
+ * http://tylerbrock.github.com/mongo-hacker
+ *
+ */
+
+if (_isWindows()) {
+    print("\nSorry! MongoDB Shell Enhancements for Hackers isn't compatible with Windows.\n");
+}
+
+if (typeof db !== 'undefined') {
+    var current_version = parseFloat(db.serverBuildInfo().version).toFixed(2)
+
+    if (current_version < 2.2) {
+        print("Sorry! MongoDB Shell Enhancements for Hackers is only compatible with Mongo 2.2+\n");
+    }
+}
+
+mongo_hacker_config = {
+    verbose_shell:  true,      // additional verbosity
+    index_paranoia: true,      // querytime explain
+    enhance_api:    true,      // additonal api extensions
+    indent:         2,         // number of spaces for indent
+    uuid_type:      'default', // 'java', 'c#', 'python' or 'default'
+    banner_message: 'Shell Enhanced by MongoHacker Ver. ', //banner message to show on mongo shell load
+    version:        '0.0.3',    // current mongo-hacker version
+    show_banner:     true,      // show mongo-hacker version banner on startup
+
+    // Shell Color Settings
+    // [<color>, <bold>, <underline>]
+    colors: {
+        'number':     [ 'blue', false, false ],
+        'null':       [ 'red', false, false ],
+        'undefined':  [ 'magenta', false, false ],
+        'objectid':   [ 'green', false, false ],
+        'string':     [ 'green', false, false ],
+        'function':   [ 'magenta', false, false ],
+        'date':       [ 'blue', false, false ]
+    }
+}
+
+if (mongo_hacker_config['show_banner']) {
+    print(mongo_hacker_config['banner_message'] + mongo_hacker_config['version']);
+}
+
+function hasDollar(fields){
+    for (k in fields){
+        if(k.indexOf('$') !== -1){
+            return true;
+        };
+    };
+    return false;
+}
+
+// Aggregate extension to support alternate API
+DBCollection.prototype.aggregate = function( ops ){
+    if(arguments.length >= 1 && (hasDollar(ops) || hasDollar(ops[0]))){
+        var arr = ops;
+
+        if (!ops.length) {
+            arr = [];
+            for (var i=0; i<arguments.length; i++) {
+                arr.push(arguments[i]);
+            }
+        }
+
+        var res = this.runCommand("aggregate", {pipeline: arr});
+        if (!res.ok) {
+            printStackTrace();
+            throw "aggregate failed: " + tojson(res);
+        }
+        return res;
+    } else {
+       return new AggHelper( this ).match( ops || {} );
+    }
+};
+
+// Aggregation Framework Helper
+AggHelper = function( collection, fields ){
+    this.collection = collection;
+    this.pipeline = [];
+};
+
+AggHelper.prototype.execute = function(){
+    var res = this.collection.runCommand("aggregate", {pipeline: this.pipeline});
+    if (!res.ok) {
+        printStackTrace();
+        throw "aggregate failed: " + tojson(res);
+    }
+    return res;
+};
+
+AggHelper.prototype.shellPrint = function(){
+    this.execute().result.forEach(function(result){
+        printjson(result);
+    });
+};
+
+AggHelper.prototype.project = function( fields ){
+    if(!fields){
+        throw "project needs fields";
+    }
+    this.pipeline.push({"$project" : fields});
+    return this;
+};
+
+AggHelper.prototype.match = function( criteria ){
+    if(!criteria){
+        throw "match needs a query object";
+    }
+    this.pipeline.push({"$match" : criteria});
+    return this;
+};
+
+AggHelper.prototype.limit = function( limit ){
+    if(!limit){
+        throw "limit needs an integer indicating the max number of documents to limit";
+    }
+    this.pipeline.push({"$limit" : limit});
+    return this;
+};
+
+AggHelper.prototype.skip = function( skip ){
+    if(!skip){
+        throw "skip needs an integer indicating the number of documents to skip";
+    }
+    this.pipeline.push({"$skip" : skip});
+    return this;
+};
+
+AggHelper.prototype.unwind = function( field ){
+    if(!field){
+        throw "unwind needs a string indicating the key of an array field to unwind";
+    }
+    this.pipeline.push({"$unwind" : "$" + field});
+    return this;
+};
+
+AggHelper.prototype.group = function( group_expression ){
+    if(!group_expression){
+        throw "group needs an group expression";
+    }
+    this.pipeline.push({"$group" : group_expression});
+    return this;
+};
+
+AggHelper.prototype.sort = function( sort ){
+    this.pipeline.push({"$sort" : sort});
+    return this;
+};
+//----------------------------------------------------------------------------
+// API Additions
+//----------------------------------------------------------------------------
+DBCollection.prototype.filter = function( filter ) {
+    return new DBQuery(
+        this._mongo,
+        this._db,
+        this,
+        this._fullName,
+        this._massageObject( filter ) 
+    );
+};
+
+DBQuery.prototype.select = function( fields ){
+    this._fields = fields;
+    return this;
+};
+
+DBQuery.prototype.one = function(){
+    return this.limit(1)[0];
+};
+
+DBQuery.prototype.first = function(field){
+    var field = field || "$natural";
+    var sortBy = {};
+    sortBy[field] = 1;
+    return this.sort(sortBy).one();
+}
+
+DBQuery.prototype.reverse = function( field ){
+    var field = field || "$natural";
+    var sortBy = {};
+    sortBy[field] = -1;
+    return this.sort(sortBy);
+}
+
+DBQuery.prototype.last = function( field ){
+    var field = field || "$natural";
+    return this.reverse(field).one();
+}
+
+DB.prototype.rename = function(newName) {
+    if(newName == this.getName() || newName.length === 0)
+        return;
+
+    this.copyDatabase(this.getName(), newName, "localhost");
+    this.dropDatabase();
+    db = this.getSiblingDB(newName);
+};
+
+//----------------------------------------------------------------------------
+// API Modifications (additions and changes)
+//----------------------------------------------------------------------------
+
+// Add upsert method which has upsert set as true and multi as false
+DBQuery.prototype.upsert = function( upsert ){
+    assert( upsert , "need an upsert object" );
+
+    this._validate(upsert);
+    this._db._initExtraInfo();
+    this._mongo.update( this._ns , this._query , upsert , true , false );
+    this._db._getExtraInfo("Upserted");
+};
+
+// Updates are always multi and never an upsert
+DBQuery.prototype.update = function( update ){
+    assert( update , "need an update object" );
+
+    this._checkMulti();
+    this._validate(update);
+    this._db._initExtraInfo();
+    this._mongo.update( this._ns , this._query , update , false , true );
+    this._db._getExtraInfo("Updated");
+};
+
+// Replace one document
+DBQuery.prototype.replace = function( replacement ){
+   assert( replacement , "need an update object" );
+
+   this._validate(replacement);
+   this._db._initExtraInfo();
+   this._mongo.update( this._ns , this._query , replacement , false , false );
+   this._db._getExtraInfo("Replaced");
+};
+
+// Remove is always multi
+DBQuery.prototype.remove = function(){
+    for ( var k in this._query ){
+        if ( k == "_id" && typeof( this._query[k] ) == "undefined" ){
+            throw "can't have _id set to undefined in a remove expression";
+        }
+    }
+
+    this._checkMulti();
+    this._db._initExtraInfo();
+    this._mongo.remove( this._ns , this._query , false );
+    this._db._getExtraInfo("Removed");
+};
+shellHelper.find = function (query) {
+    assert(typeof query == "string");
+
+    var args = query.split( /\s+/ );
+    query = args[0];
+    args = args.splice(1);
+
+    if (query !== "") {
+        var regexp = new RegExp(query, "i");
+        var result = db.runCommand("listCommands");
+        for (var command in result.commands) {
+            var commandObj = result.commands[command];
+            var help = commandObj.help;
+            if (commandObj.help.indexOf('\n') != -1 ) {
+                help = commandObj.help.substring(0, commandObj.help.lastIndexOf('\n'));
+            }
+            if (regexp.test(command) || regexp.test(help)) {
+                var numSpaces = 30 - command.length;
+                print(colorize(command, 'green'), Array(numSpaces).join(" "), "-", help);
+            }
+        }
+    }
+};
+
+//----------------------------------------------------------------------------
+// Color Functions
+//----------------------------------------------------------------------------
+__ansi = {
+    csi: String.fromCharCode(0x1B) + '[',
+    reset: '0',
+    text_prop: 'm',
+    foreground: '3',
+    bright: '1',
+    underline: '4',
+
+    colors: {
+        red: '1',
+        green: '2',
+        yellow: '3',
+        blue: '4',
+        magenta: '5',
+        cyan: '6'
+    }
+};
+
+function controlCode( parameters ) {
+    if ( parameters === undefined ) {
+        parameters = "";
+    }
+    else if (typeof(parameters) == 'object' && (parameters instanceof Array)) {
+        parameters = parameters.join(';');
+    }
+
+    return __ansi.csi + String(parameters) + String(__ansi.text_prop);
+};
+
+function applyColorCode( string, properties ) {
+    return controlCode(properties) + String(string) + controlCode();
+};
+
+function colorize( string, color, bright, underline ) {
+    var params = [];
+    var code = __ansi.foreground + __ansi.colors[color];
+
+    params.push(code);
+
+    if ( bright === true ) params.push(__ansi.bright);
+    if ( underline === true ) params.push(__ansi.underline);
+
+    return applyColorCode( string, params );
+};
+__indent = Array(mongo_hacker_config.indent + 1).join(' ');
+
+ObjectId.prototype.toString = function() {
+    return this.str;
+};
+
+ObjectId.prototype.tojson = function(indent, nolint) {
+    return tojson(this);
+};
+
+Date.prototype.tojson = function() {
+
+    var UTC = Date.printAsUTC ? 'UTC' : '';
+
+    var year = this['get'+UTC+'FullYear']().zeroPad(4);
+    var month = (this['get'+UTC+'Month']() + 1).zeroPad(2);
+    var date = this['get'+UTC+'Date']().zeroPad(2);
+    var hour = this['get'+UTC+'Hours']().zeroPad(2);
+    var minute = this['get'+UTC+'Minutes']().zeroPad(2);
+    var sec = this['get'+UTC+'Seconds']().zeroPad(2);
+
+    if (this['get'+UTC+'Milliseconds']())
+        sec += '.' + this['get'+UTC+'Milliseconds']().zeroPad(3);
+
+    var ofs = 'Z';
+    if (!Date.printAsUTC) {
+        var ofsmin = this.getTimezoneOffset();
+        if (ofsmin !== 0){
+            ofs = ofsmin > 0 ? '-' : '+'; // This is correct
+            ofs += (ofsmin/60).zeroPad(2);
+            ofs += (ofsmin%60).zeroPad(2);
+        }
+    }
+
+    var isodate =  colorize('"' + [year, month, date].join('-') + 'T' + hour +':' + minute + ':' + sec + ofs + '"', "cyan");
+    return 'ISODate(' + isodate + ')';
+};
+
+Array.tojson = function( a , indent , nolint ){
+    var lineEnding = nolint ? " " : "\n";
+
+    if (!indent)
+        indent = "";
+
+    if ( nolint )
+        indent = "";
+
+    if (a.length === 0) {
+        return "[ ]";
+    }
+
+    var s = "[" + lineEnding;
+    indent += __indent;
+    for ( var i=0; i<a.length; i++){
+        s += indent + tojson( a[i], indent , nolint );
+        if ( i < a.length - 1 ){
+            s += "," + lineEnding;
+        }
+    }
+    if ( a.length === 0 ) {
+        s += indent;
+    }
+
+    indent = indent.substring(__indent.length);
+    s += lineEnding+indent+"]";
+    return s;
+};
+
+NumberLong.prototype.tojson = function() {
+    return 'NumberLong(' + colorize('"' + this.toString().match(/-?\d+/)[0] + '"', "red") + ')';
+};
+
+NumberInt.prototype.tojson = function() {
+    return 'NumberInt(' + colorize('"' + this.toString().match(/-?\d+/)[0] + '"', "red") + ')';
+};
+
+BinData.prototype.tojson = function(indent , nolint) {
+    if (this.subtype() === 3) {
+        return 'UUID(' + colorize('"' + uuidToString(this) + '"', "cyan") + ', ' + colorize('"' + mongo_hacker_config['uuid_type'] + '"', "cyan") + ')'
+    } else if (this.subtype() === 4) {
+        return 'UUID(' + colorize('"' + uuidToString(this, "default") + '"', "cyan") + ')'
+    } else {
+        return 'BinData(' + colorize(this.subtype(), "red") + ', ' + colorize('"' + this.base64() + '"', "green", true) + ')';
+    }
+};
+
+DBQuery.prototype.shellPrint = function(){
+    try {
+        var start = new Date().getTime();
+        var n = 0;
+        while ( this.hasNext() && n < DBQuery.shellBatchSize ){
+            var s = this._prettyShell ? tojson( this.next() ) : tojson( this.next() , "" , true );
+            print( s );
+            n++;
+        }
+
+        var output = [];
+
+        if (typeof _verboseShell !== 'undefined' && _verboseShell) {
+            var time = new Date().getTime() - start;
+            var slowms = getSlowms();
+            var fetched = "Fetched " + n + " record(s) in ";
+            if (time > slowms) {
+                fetched += colorize(time + "ms", "red", true);
+            } else {
+                fetched += colorize(time + "ms", "green", true);
+            }
+            output.push(fetched);
+        }
+
+        var paranoia = mongo_hacker_config.index_paranoia;
+
+        if (typeof paranoia !== 'undefined' && paranoia) {
+            var explain = this.clone();
+            explain._ensureSpecial();
+            explain._query.$explain = true;
+            explain._limit = Math.abs(n._limit) * -1;
+            var result = explain.next();
+            var type = result.cursor;
+
+            if (type !== undefined) {
+                var index_use = "Index[";
+                if (type == "BasicCursor") {
+                    index_use += colorize( "none", "red", true);
+                } else {
+                    index_use += colorize( result.cursor.substring(12), "green", true );
+                }
+                index_use += "]";
+                output.push(index_use);
+            }
+        }
+
+        if ( this.hasNext() ) {
+            ___it___  = this;
+            output.push("More[" + colorize("true", "green", true) + "]");
+        }
+        print(output.join(" -- "));
+    }
+    catch ( e ){
+        print( e );
+    }
+};
+
+tojsonObject = function( x, indent, nolint ) {
+    var lineEnding = nolint ? " " : "\n";
+    var tabSpace = nolint ? "" : __indent;
+
+    assert.eq( ( typeof x ) , "object" , "tojsonObject needs object, not [" + ( typeof x ) + "]" );
+
+    if (!indent) 
+        indent = "";
+
+    if ( typeof( x.tojson ) == "function" && x.tojson != tojson ) {
+        return x.tojson(indent,nolint);
+    }
+
+    if ( x.constructor && typeof( x.constructor.tojson ) == "function" && x.constructor.tojson != tojson ) {
+        return x.constructor.tojson( x, indent , nolint );
+    }
+
+    if ( x.toString() == "[object MaxKey]" )
+        return "{ $maxKey : 1 }";
+    if ( x.toString() == "[object MinKey]" )
+        return "{ $minKey : 1 }";
+
+    var s = "{" + lineEnding;
+
+    // push one level of indent
+    indent += tabSpace;
+
+    var total = 0;
+    for ( var k in x ) total++;
+    if ( total === 0 ) {
+        s += indent + lineEnding;
+    }
+
+    var keys = x;
+    if ( typeof( x._simpleKeys ) == "function" )
+        keys = x._simpleKeys();
+    var num = 1;
+    for ( var key in keys ){
+
+        var val = x[key];
+        if ( val == DB.prototype || val == DBCollection.prototype )
+            continue;
+
+        s += indent + colorize("\"" + key + "\"", "yellow") + ": " + tojson( val, indent , nolint );
+        if (num != total) {
+            s += ",";
+            num++;
+        }
+        s += lineEnding;
+    }
+
+    // pop one level of indent
+    indent = indent.substring(__indent.length);
+    return s + indent + "}";
+};
+
+
+tojson = function( x, indent , nolint ) {
+    if ( x === null )
+        return colorize("null", "red", true);
+
+    if ( x === undefined )
+        return colorize("undefined", "magenta", true);
+
+    if ( x.isObjectId ) {
+        return 'ObjectId(' + colorize('"' + x.str + '"', "green", false, true) + ')';
+    }
+
+    if (!indent)
+        indent = "";
+
+    var s;
+    switch ( typeof x ) {
+    case "string": {
+        s = "\"";
+        for ( var i=0; i<x.length; i++ ){
+            switch (x[i]){
+                case '"': s += '\\"'; break;
+                case '\\': s += '\\\\'; break;
+                case '\b': s += '\\b'; break;
+                case '\f': s += '\\f'; break;
+                case '\n': s += '\\n'; break;
+                case '\r': s += '\\r'; break;
+                case '\t': s += '\\t'; break;
+
+                default: {
+                    var code = x.charCodeAt(i);
+                    if (code < 0x20){
+                        s += (code < 0x10 ? '\\u000' : '\\u00') + code.toString(16);
+                    } else {
+                        s += x[i];
+                    }
+                }
+            }
+        }
+        s += "\"";
+        return colorize(s, "green", true);
+    }
+    case "number":
+        return colorize(x, "red");
+    case "boolean":
+        return colorize("" + x, "blue");
+    case "object": {
+        s = tojsonObject( x, indent , nolint );
+        if ( ( nolint === null || nolint === true ) && s.length < 80 && ( indent === null || indent.length === 0 ) ){
+            s = s.replace( /[\s\r\n ]+/gm , " " );
+        }
+        return s;
+    }
+    case "function":
+        return colorize(x.toString(), "magenta");
+    default:
+        throw "tojson can't handle type " + ( typeof x );
+    }
+
+};
+
+
+DBQuery.prototype._validate = function( o ){
+    var firstKey = null;
+    for (var k in o) { firstKey = k; break; }
+
+    if (firstKey !== null && firstKey[0] == '$') {
+        // for mods we only validate partially, for example keys may have dots
+        this._validateObject( o );
+    } else {
+        // we're basically inserting a brand new object, do full validation
+        this._validateForStorage( o );
+    }
+};
+
+DBQuery.prototype._validateObject = function( o ){
+    if (typeof(o) != "object")
+        throw "attempted to save a " + typeof(o) + " value.  document expected.";
+
+    if ( o._ensureSpecial && o._checkModify )
+        throw "can't save a DBQuery object";
+};
+
+DBQuery.prototype._validateForStorage = function( o ){
+    this._validateObject( o );
+    for ( var k in o ){
+        if ( k.indexOf( "." ) >= 0 ) {
+            throw "can't have . in field names [" + k + "]" ;
+        }
+
+        if ( k.indexOf( "$" ) === 0 && ! DBCollection._allowedFields[k] ) {
+            throw "field names cannot start with $ [" + k + "]";
+        }
+
+        if ( o[k] !== null && typeof( o[k] ) === "object" ) {
+            this._validateForStorage( o[k] );
+        }
+    }
+};
+
+DBQuery.prototype._checkMulti = function(){
+  if(this._limit > 0 || this._skip > 0){
+    var ids = this.clone().select({_id: 1}).map(function(o){return o._id;});
+    this._query['_id'] = {'$in': ids};
+    return true;
+  } else {
+    return false;
+  }
+};
+
+DBQuery.prototype.ugly = function(){
+    this._prettyShell = false;
+    return this;
+}
+function runMatch(cmd, args, regexp) {
+    clearRawMongoProgramOutput();
+    if (args) {
+        run(cmd, args);
+    } else {
+        run(cmd);
+    }
+    var output = rawMongoProgramOutput();
+    return output.match(regexp);
+};
+
+function getEnv(env_var) {
+    var env_regex = new RegExp(env_var + '=(.*)');
+    return runMatch('env', '', env_regex)[1];
+};
+
+function getVersion() {
+    var regexp = /version: (\d).(\d).(\d)/;
+    return runMatch('mongo', '--version', regexp).slice(1, 4);
+};
+
+function isMongos() {
+    return db.isMaster().msg == 'isdbgrid';
+};
+
+function getSlowms(){
+    if(!isMongos()){
+        return db.getProfilingStatus().slowms;
+    } else {
+        return 100;
+    }
+};
+// Override group because map/reduce style is deprecated
+DBCollection.prototype.agg_group = function( name, group_field, operation, op_value, filter ) {
+    var ops = [];
+    var group_op = { $group: { _id: '$' + group_field } };
+
+    if (filter !== undefined) {
+        ops.push({ '$match': filter });
+    }
+
+    group_op['$group'][name] = { };
+    group_op['$group'][name]['$' + operation] = op_value;
+    ops.push(group_op);
+
+    return this.aggregate(ops);
+};
+
+// Function that groups and counts by group after applying filter
+DBCollection.prototype.gcount = function( group_field, filter ) {
+    return this.agg_group('count', group_field, 'sum', 1, filter);
+};
+
+// Function that groups and sums sum_field after applying filter
+DBCollection.prototype.gsum = function( group_field, sum_field, filter ) {
+    return this.agg_group('sum', group_field, 'sum', '$' + sum_field, filter);
+};
+
+// Function that groups and averages avg_feld after applying filter
+DBCollection.prototype.gavg = function( group_field, avg_field, filter ) {
+    return this.agg_group('avg', group_field, 'avg', '$' + avg_field, filter);
+};
+// Improve the default prompt with hostname, process type, and version
 prompt = function() {
 prompt = function() {
-  var host = db.serverStatus().host;
-  return "[" + host + "] db: " + db + " docs: " + db.stats().objects + " > ";
+    var serverstatus = db.serverStatus();
+    var host = serverstatus.host.split('.')[0];
+    var process = serverstatus.process;
+    var version = db.serverBuildInfo().version;
+    var repl_set = db._adminCommand({"replSetGetStatus": 1}).ok !== 0;
+    var rs_state = '';
+    if(repl_set) {
+        members = rs.status().members;
+        for(var i = 0; i<members.length; i++){
+            if(members[i].self === true){
+                rs_state = '[' + members[i].stateStr + ']';
+            }
+        };
+    }
+    var state = isMongos() ? '[mongos]' : rs_state;
+    return host + '(' + process + '-' + version + ')' + state + ' ' + db + '> ';
+};
+
+// Better show dbs
+shellHelper.show = function (what) {
+    assert(typeof what == "string");
+
+    var args = what.split( /\s+/ );
+    what = args[0]
+    args = args.splice(1)
+
+    if (what == "profile") {
+        if (db.system.profile.count() == 0) {
+            print("db.system.profile is empty");
+            print("Use db.setProfilingLevel(2) will enable profiling");
+            print("Use db.system.profile.find() to show raw profile entries");
+        }
+        else {
+            print();
+            db.system.profile.find({ millis: { $gt: 0} }).sort({ $natural: -1 }).limit(5).forEach(
+                function (x) {
+                    print("" + x.op + "\t" + x.ns + " " + x.millis + "ms " + String(x.ts).substring(0, 24));
+                    var l = "";
+                    for ( var z in x ){
+                        if ( z == "op" || z == "ns" || z == "millis" || z == "ts" )
+                            continue;
+
+                        var val = x[z];
+                        var mytype = typeof(val);
+
+                        if ( mytype == "string" ||
+                             mytype == "number" )
+                            l += z + ":" + val + " ";
+                        else if ( mytype == "object" )
+                            l += z + ":" + tojson(val ) + " ";
+                        else if ( mytype == "boolean" )
+                            l += z + " ";
+                        else
+                            l += z + ":" + val + " ";
+
+                    }
+                    print( l );
+                    print("\n");
+                }
+            )
+        }
+        return "";
+    }
+
+    if (what == "users") {
+        db.system.users.find().forEach(printjson);
+        return "";
+    }
+
+    if (what == "collections" || what == "tables") {
+        db.getCollectionNames().forEach(function (x) { print(x) });
+        return "";
+    }
+
+    if (what == "dbs" || what == "databases") {
+        var dbs = db.getMongo().getDBs();
+        var dbinfo = [];
+        var maxNameLength = 0;
+        var maxGbDigits = 0;
+
+        dbs.databases.forEach(function (x){
+            var sizeStr = (x.sizeOnDisk / 1024 / 1024 / 1024).toFixed(3);
+            var nameLength = x.name.length;
+            var gbDigits = sizeStr.indexOf(".");
+
+            if( nameLength > maxNameLength) maxNameLength = nameLength;
+            if( gbDigits > maxGbDigits ) maxGbDigits = gbDigits;
+
+            dbinfo.push({
+                name:      x.name,
+                size:      x.sizeOnDisk,
+                size_str:  sizeStr,
+                name_size: nameLength,
+                gb_digits: gbDigits
+            });
+        });
+
+        dbinfo.sort(function (a,b) { a.name - b.name });
+        dbinfo.forEach(function (db) {
+            var namePadding = maxNameLength - db.name_size;
+            var sizePadding = maxGbDigits   - db.gb_digits;
+            var padding = Array(namePadding + sizePadding + 3).join(" ");
+            if (db.size > 1) {
+                print(colorize(db.name, "green", true) + padding + db.size_str + "GB");
+            } else {
+                print(colorize(db.name, "green", true) + padding + "(empty)");
+            }
+        });
+
+        return "";
+    }
+
+    if (what == "log" ) {
+        var n = "global";
+        if ( args.length > 0 )
+            n = args[0]
+
+        var res = db.adminCommand( { getLog : n } );
+        if ( ! res.ok ) {
+            print("Error while trying to show " + n + " log: " + res.errmsg);
+            return "";
+        }
+        for ( var i=0; i<res.log.length; i++){
+            print( res.log[i] )
+        }
+        return ""
+    }
+
+    if (what == "logs" ) {
+        var res = db.adminCommand( { getLog : "*" } )
+        if ( ! res.ok ) {
+            print("Error while trying to show logs: " + res.errmsg);
+            return "";
+        }
+        for ( var i=0; i<res.names.length; i++){
+            print( res.names[i] )
+        }
+        return ""
+    }
+
+    if (what == "startupWarnings" ) {
+        var dbDeclared, ex;
+        try {
+            // !!db essentially casts db to a boolean
+            // Will throw a reference exception if db hasn't been declared.
+            dbDeclared = !!db;
+        } catch (ex) {
+            dbDeclared = false;
+        }
+        if (dbDeclared) {
+            var res = db.adminCommand( { getLog : "startupWarnings" } );
+            if ( res.ok ) {
+                if (res.log.length == 0) {
+                    return "";
+                }
+                print( "Server has startup warnings: " );
+                for ( var i=0; i<res.log.length; i++){
+                    print( res.log[i] )
+                }
+                return "";
+            } else if (res.errmsg == "unauthorized") {
+                // Don't print of startupWarnings command failed due to auth
+                return "";
+            } else {
+                print("Error while trying to show server startup warnings: " + res.errmsg);
+                return "";
+            }
+        } else {
+            print("Cannot show startupWarnings, \"db\" is not set");
+            return "";
+        }
+    }
+
+    throw "don't know how to show [" + what + "]";
+
+}
+printShardingStatus = function( configDB , verbose ){
+    if (configDB === undefined)
+        configDB = db.getSisterDB('config')
+
+    var version = configDB.getCollection( "version" ).findOne();
+    if ( version == null ){
+        print( "printShardingStatus: this db does not have sharding enabled. be sure you are connecting to a mongos from the shell and not to a mongod." );
+        return;
+    }
+
+    var raw = "";
+    var output = function(s){
+        raw += s + "\n";
+    }
+    output( "--- Sharding Status --- " );
+    output( "  sharding version: " + tojson( configDB.getCollection( "version" ).findOne(), "  " ) );
+
+    output( "  shards:" );
+    configDB.shards.find().sort( { _id : 1 } ).forEach(
+        function(z){
+            output( "    " + tojsononeline( z ) );
+        }
+    );
+
+    output( "  databases:" );
+    configDB.databases.find().sort( { name : 1 } ).forEach(
+        function(db){
+            output( "    " + tojsononeline(db,"",true) );
+
+            if (db.partitioned){
+                configDB.collections.find( { _id : new RegExp( "^" +
+                    RegExp.escape(db._id) + "\\." ) } ).
+                    sort( { _id : 1 } ).forEach( function( coll ){
+                        if ( coll.dropped == false ){
+                            output( "    " + coll._id );
+                            output( "      shard key: " + tojson(coll.key, 0, true) );
+                            output( "      chunks:" );
+
+                            res = configDB.chunks.aggregate(
+                                { "$match": { ns: coll._id } },
+                                { "$group": { _id: "$shard", nChunks: { "$sum": 1 } } }
+                            ).result
+
+                            var totalChunks = 0;
+                            res.forEach( function(z){
+                                totalChunks += z.nChunks;
+                                output( "        " + z._id + ": " + z.nChunks );
+                            } )
+
+                            if ( totalChunks < 20 || verbose ){
+                                configDB.chunks.find( { "ns" : coll._id } ).sort( { min : 1 } ).forEach(
+                                    function(chunk){
+                                        output( "        " +
+                                            tojson( chunk.min, 0, true) + " -> " +
+                                            tojson( chunk.max, 0, true ) +
+                                            " on: " + colorize(chunk.shard, "cyan") + " " +
+                                            ( chunk.jumbo ? "jumbo " : "" )
+                                        );
+                                    }
+                                );
+                            }
+                            else {
+                                output( "\t\t\ttoo many chunks to print, use verbose if you want to force print" );
+                            }
+
+                            configDB.tags.find( { ns : coll._id } ).sort( { min : 1 } ).forEach(
+                                function( tag ) {
+                                    output( "        tag: " + tag.tag + "  " + tojson( tag.min ) + " -> " + tojson( tag.max ) );
+                                }
+                            )
+                        }
+                    }
+                )
+            }
+        }
+    );
+
+    print( raw );
+}
+function listDbs(){
+  return db.adminCommand("listDatabases").databases.map(function(d){return d.name});
+}
+
+this.__proto__.constructor.autocomplete = listDbs;function base64ToHex(base64) {
+    var base64Digits = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=";
+    var hexDigits = "0123456789abcdef";
+    var hex = "";
+    for (var i = 0; i < 24; ) {
+        var e1 = base64Digits.indexOf(base64[i++]);
+        var e2 = base64Digits.indexOf(base64[i++]);
+        var e3 = base64Digits.indexOf(base64[i++]);
+        var e4 = base64Digits.indexOf(base64[i++]);
+        var c1 = (e1 << 2) | (e2 >> 4);
+        var c2 = ((e2 & 15) << 4) | (e3 >> 2);
+        var c3 = ((e3 & 3) << 6) | e4;
+        hex += hexDigits[c1 >> 4];
+        hex += hexDigits[c1 & 15];
+        if (e3 != 64) {
+            hex += hexDigits[c2 >> 4];
+            hex += hexDigits[c2 & 15];
+        }
+        if (e4 != 64) {
+            hex += hexDigits[c3 >> 4];
+            hex += hexDigits[c3 & 15];
+        }
+    }
+    return hex;
+}
+
+function hexToBase64(hex) {
+    var base64Digits = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+    var base64 = "";
+    var group;
+    for (var i = 0; i < 30; i += 6) {
+        group = parseInt(hex.substr(i, 6), 16);
+        base64 += base64Digits[(group >> 18) & 0x3f];
+        base64 += base64Digits[(group >> 12) & 0x3f];
+        base64 += base64Digits[(group >> 6) & 0x3f];
+        base64 += base64Digits[group & 0x3f];
+    }
+    group = parseInt(hex.substr(30, 2), 16);
+    base64 += base64Digits[(group >> 2) & 0x3f];
+    base64 += base64Digits[(group << 4) & 0x3f];
+    base64 += "==";
+    return base64;
+}
+
+var platformSpecificUuidModifications = {
+    "java": function (hex) {
+        var msb = hex.substr(0, 16);
+        var lsb = hex.substr(16, 16);
+        msb = msb.substr(14, 2) + msb.substr(12, 2) + msb.substr(10, 2) + msb.substr(8, 2)
+            + msb.substr(6, 2) + msb.substr(4, 2) + msb.substr(2, 2) + msb.substr(0, 2);
+        lsb = lsb.substr(14, 2) + lsb.substr(12, 2) + lsb.substr(10, 2) + lsb.substr(8, 2)
+            + lsb.substr(6, 2) + lsb.substr(4, 2) + lsb.substr(2, 2) + lsb.substr(0, 2);
+        return msb + lsb;
+    },
+    "c#": function (hex) {
+        return hex.substr(6, 2) + hex.substr(4, 2) + hex.substr(2, 2) + hex.substr(0, 2)
+            + hex.substr(10, 2) + hex.substr(8, 2) + hex.substr(14, 2) + hex.substr(12, 2)
+            + hex.substr(16, 16);
+    },
+    "python": function (hex) {
+        return hex;
+    },
+    "default": function (hex) {
+        return hex;
+    }
+};
+
+function UUID(uuid, type) {
+    var hex = uuid.replace(/[{}-]/g, "");
+    var typeNum = 4;
+    if (type != undefined) {
+        typeNum = 3;
+        hex = platformSpecificUuidModifications[type](hex);
+    }
+    return new BinData(typeNum, hexToBase64(hex));
+}
+
+function uuidToString(uuid, uuidType) {
+    var uuidType = uuidType || mongo_hacker_config['uuid_type'];
+    var hex = platformSpecificUuidModifications[uuidType](base64ToHex(uuid.base64()));
+    return hex.substr(0, 8) + '-' + hex.substr(8, 4) + '-' + hex.substr(12, 4)
+        + '-' + hex.substr(16, 4) + '-' + hex.substr(20, 12);
 }
 }
+setVerboseShell(true);
+
+DBQuery.prototype._prettyShell = true
+
+DB.prototype._getExtraInfo = function(action) {
+    if ( typeof _verboseShell === 'undefined' || !_verboseShell ) {
+        __callLastError = true;
+        return;
+    }
+
+    // explicit w:1 so that replset getLastErrorDefaults aren't used here which would be bad.
+    var startTime = new Date().getTime();
+    var res = this.getLastErrorCmd(1);
+    if (res) {
+        if (res.err !== undefined && res.err !== null) {
+            // error occurred, display it
+            print(res.err);
+            return;
+        }
+
+        var info = action + " ";
+        // hack for inserted because res.n is 0
+        info += action != "Inserted" ? res.n : 1;
+        if (res.n > 0 && res.updatedExisting !== undefined) info += " " + (res.updatedExisting ? "existing" : "new");
+        info += " record(s) in ";
+        var time = new Date().getTime() - startTime;
+        var slowms = getSlowms();
+        if (time > slowms) {
+            info += colorize(time + "ms", "red", true);
+        } else {
+            info += colorize(time + "ms", "green", true);
+        }
+        print(info);
+    }
+};