QueryWindowController.m

//
//  QueryWindowController.m
//  MongoHub
//
//  Created by Syd on 10-4-28.
//  Copyright 2010 ThePeppersStudio.COM. All rights reserved.
//

#import "Configure.h"
#import "NSProgressIndicator+Extras.h"
#import "QueryWindowController.h"
#import "DatabasesArrayController.h"
#import "ResultsOutlineViewController.h"
#import "Connection.h"
#import "NSString+Extras.h"
#import "JsonWindowController.h"
#import "MODServer.h"
#import "MODCollection.h"
#import "MODDatabase.h"
#import "MODHelper.h"

@implementation QueryWindowController

@synthesize managedObjectContext;
@synthesize databasesArrayController;
@synthesize findResultsViewController;
@synthesize mongoCollection;
@synthesize conn;

@synthesize criticalTextField;
@synthesize fieldsTextField;
@synthesize skipTextField;
@synthesize limitTextField;
@synthesize sortTextField;
@synthesize totalResultsTextField;
@synthesize findQueryTextField;
@synthesize findResultsOutlineView;
@synthesize findQueryLoaderIndicator;

@synthesize updateCriticalTextField;
@synthesize updateSetTextField;
@synthesize upsetCheckBox;
@synthesize updateResultsTextField;
@synthesize updateQueryTextField;
@synthesize updateQueryLoaderIndicator;

@synthesize removeCriticalTextField;
@synthesize removeResultsTextField;
@synthesize removeQueryTextField;
@synthesize removeQueryLoaderIndicator;

@synthesize insertDataTextView;
@synthesize insertResultsTextField;
@synthesize insertLoaderIndicator;

@synthesize indexTextField;
@synthesize indexesOutlineViewController;
@synthesize indexLoaderIndicator;

@synthesize mapFunctionTextView;
@synthesize reduceFunctionTextView;
@synthesize mrcriticalTextField;
@synthesize mroutputTextField;
@synthesize mrOutlineViewController;
@synthesize mrLoaderIndicator;

@synthesize expCriticalTextField;
@synthesize expFieldsTextField;
@synthesize expSkipTextField;
@synthesize expLimitTextField;
@synthesize expSortTextField;
@synthesize expResultsTextField;
@synthesize expPathTextField;
@synthesize expTypePopUpButton;
@synthesize expQueryTextField;
@synthesize expJsonArrayCheckBox;
@synthesize expProgressIndicator;

@synthesize impIgnoreBlanksCheckBox;
@synthesize impDropCheckBox;
@synthesize impHeaderlineCheckBox;
@synthesize impFieldsTextField;
@synthesize impResultsTextField;
@synthesize impPathTextField;
@synthesize impTypePopUpButton;
@synthesize impJsonArrayCheckBox;
@synthesize impStopOnErrorCheckBox;
@synthesize impProgressIndicator;


- (id)init {
    self = [super initWithWindowNibName:@"QueryWindow"];
    return self;
}

- (void)dealloc {
    [managedObjectContext release];
    [databasesArrayController release];
    [findResultsViewController release];
    [conn release];
    [mongoCollection release];
    
    [criticalTextField release];
    [fieldsTextField release];
    [skipTextField release];
    [limitTextField release];
    [sortTextField release];
    [totalResultsTextField release];
    [findQueryTextField release];
    [findResultsOutlineView release];
    [findQueryLoaderIndicator release];
    
    [updateCriticalTextField release];
    [updateSetTextField release];
    [upsetCheckBox release];
    [updateResultsTextField release];
    [updateQueryTextField release];
    [updateQueryLoaderIndicator release];
    
    [removeCriticalTextField release];
    [removeResultsTextField release];
    [removeQueryTextField release];
    [removeQueryLoaderIndicator release];
    
    [insertDataTextView release];
    [insertResultsTextField release];
    [insertLoaderIndicator release];
    
    [indexTextField release];
    [indexesOutlineViewController release];
    [indexLoaderIndicator release];
    
    [mapFunctionTextView release];
    [reduceFunctionTextView release];
    [mrcriticalTextField release];
    [mroutputTextField release];
    [mrOutlineViewController release];
    [mrLoaderIndicator release];
    
    [expCriticalTextField release];
    [expFieldsTextField release];
    [expSkipTextField release];
    [expLimitTextField release];
    [expSortTextField release];
    [expResultsTextField release];
    [expPathTextField release];
    [expTypePopUpButton release];
    [expQueryTextField release];
    [expJsonArrayCheckBox release];
    [expProgressIndicator release];
    
    [impIgnoreBlanksCheckBox release];
    [impDropCheckBox release];
    [impHeaderlineCheckBox release];
    [impFieldsTextField release];
    [impResultsTextField release];
    [impPathTextField release];
    [impTypePopUpButton release];
    [impJsonArrayCheckBox release];
    [impStopOnErrorCheckBox release];
    [impProgressIndicator release];
    
    [super dealloc];
}

- (NSString *)formatedQueryWithReplace:(BOOL)replace
{
    NSString *query = @"";
    NSString *value;
    
    value = [[criticalTextField stringValue] stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];
    if ([value length] > 0) {
        if ([value hasPrefix:@"{"]) {
            NSString *innerValue;
            
            innerValue = [[value substringFromIndex:1] stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];
            if ([innerValue hasPrefix:@"\"$oid\""] || [innerValue hasPrefix:@"'$iod'"]) {
                query = [NSString stringWithFormat:@"{\"_id\": %@ }",value];
            } else {
                query = value;
            }
        } else if ([value hasPrefix:@"\"$oid\""] || [value hasPrefix:@"'$iod'"]) {
            query = [NSString stringWithFormat:@"{\"_id\": {%@}}",value];
        } else {
            query = [NSString stringWithFormat:@"{\"_id\": %@}",value];
        }
    }
    if (replace) {
        [criticalTextField setStringValue:query];
    }
    return query;
}

- (void)windowDidLoad {
    [super windowDidLoad];
    NSString *title = [[NSString alloc] initWithFormat:@"Query in %@", mongoCollection.absoluteCollectionName];
    [self.window setTitle:title];
    [title release];
}

- (void)windowWillClose:(NSNotification *)notification {
    [self release];
}

- (IBAction)findQuery:(id)sender
{
    int limit = [limitTextField intValue];
    NSMutableArray *fields;
    NSString *criteria;
    
    if (limit <= 0) {
        limit = 30;
    }
    criteria = [self formatedQueryWithReplace:YES];
    fields = [[NSMutableArray alloc] init];
    for (NSString *field in [[fieldsTextField stringValue] componentsSeparatedByString:@","]) {
        field = [field stringByTrimmingWhitespace];
        if ([field length] > 0) {
            [fields addObject:field];
        }
    }
    [findQueryLoaderIndicator start];
    [mongoCollection findWithCriteria:criteria fields:fields skip:[skipTextField intValue] limit:limit sort:[sortTextField stringValue] callback:^(NSArray *documents, MODQuery *mongoQuery) {
        if (mongoQuery.error) {
            [findQueryLoaderIndicator stop];
            NSRunAlertPanel(@"Error", [mongoQuery.error localizedDescription], @"OK", nil, nil);
        } else {
            [findResultsViewController.results removeAllObjects];
            [findResultsViewController.results addObjectsFromArray:[MODHelper convertForOutlineWithObjects:documents]];
            [findResultsViewController.myOutlineView reloadData];
            [mongoCollection countWithCriteria:criteria callback:^(int64_t count, MODQuery *mongoQuery) {
                [findQueryLoaderIndicator stop];
                [totalResultsTextField setStringValue:[NSString stringWithFormat:@"Total Results: %lld (%0.2fs)", count, [[mongoQuery.userInfo objectForKey:@"timequery"] duration]]];
            }];
        }
    }];
    [fields release];
}

- (IBAction)expandFindResults:(id)sender
{
    [findResultsOutlineView expandItem:nil expandChildren:YES];
}

- (IBAction)collapseFindResults:(id)sender
{
    [findResultsOutlineView collapseItem:nil collapseChildren:YES];
}

- (IBAction)updateQuery:(id)sender
{
    NSString *criteria = [updateCriticalTextField stringValue];
    
    [updateQueryLoaderIndicator start];
    [mongoCollection countWithCriteria:criteria callback:^(int64_t count, MODQuery *mongoQuery) {
        [updateResultsTextField setStringValue:[NSString stringWithFormat:@"Affected Rows: %lld", count]];
    }];
    [mongoCollection updateWithCriteria:criteria update:[updateSetTextField stringValue] upsert:[upsetCheckBox state] multiUpdate:YES callback:^(MODQuery *mongoQuery) {
        [updateQueryLoaderIndicator stop];
    }];
}

- (IBAction)removeQuery:(id)sender
{
    [removeQueryLoaderIndicator start];
    NSString *criteria = [removeCriticalTextField stringValue];
    
    [mongoCollection countWithCriteria:criteria callback:^(int64_t count, MODQuery *mongoQuery) {
        [updateResultsTextField setStringValue:[NSString stringWithFormat:@"Affected Rows: %lld", count]];
    }];
    [mongoCollection removeWithCriteria:criteria callback:^(MODQuery *mongoQuery) {
        [removeQueryLoaderIndicator stop];
    }];
}

- (IBAction) insertQuery:(id)sender
{
    [insertLoaderIndicator start];
    NSString *insertData = [insertDataTextView string];
    [mongoCollection insertWithDocuments:[NSArray arrayWithObjects:insertData, nil] callback:^(MODQuery *mongoQuery) {
        [insertResultsTextField setStringValue:@"Completed!"];
        [insertLoaderIndicator stop];
    }];
}

- (IBAction) indexQuery:(id)sender
{
    [mongoCollection indexListWithcallback:^(NSArray *indexes, MODQuery *mongoQuery) {
        [indexesOutlineViewController.results removeAllObjects];
        [indexesOutlineViewController.results addObjectsFromArray:[MODHelper convertForOutlineWithObjects:indexes]];
        [indexesOutlineViewController.myOutlineView reloadData];
    }];
}

- (IBAction) ensureIndex:(id)sender
{
//    [indexLoaderIndicator start];
//    NSString *indexData = [indexTextField stringValue];
//    [mongoCollection ensureIndexInDB:mongoCollection.databaseName 
//                  collection:mongoCollection.collectionName 
//                        user:user 
//                    password:password 
//                   indexData:indexData];
//    [self indexQuery:nil];
//    [indexLoaderIndicator stop];
}


- (IBAction) reIndex:(id)sender
{
//    [indexLoaderIndicator start];
//    [mongoCollection reIndexInDB:mongoCollection.databaseName 
//              collection:mongoCollection.collectionName 
//                    user:user 
//                password:password];
//    [self indexQuery:nil];
//    [indexLoaderIndicator stop];
}


- (IBAction) dropIndex:(id)sender
{
    NSLog(@"%@", indexesOutlineViewController.selectedItem);
//    [indexLoaderIndicator start];
//    NSString *user=nil;
//    NSString *password=nil;
//    NSString *indexName = [indexTextField stringValue];
//    [mongoCollection dropIndexInDB:mongoCollection.databaseName 
//                collection:mongoCollection.collectionName 
//                      user:user 
//                  password:password 
//                 indexName:indexName];
//    [self indexQuery:nil];
//    [indexLoaderIndicator stop];
}

- (IBAction) mapReduce:(id)sender
{
    [NSThread detachNewThreadSelector:@selector(doMapReduce) toTarget:self withObject:nil];
}

- (void)doMapReduce {
//    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
//    [mrLoaderIndicator start];
//    NSString *user=nil;
//    NSString *password=nil;
//    Database *db = [databasesArrayController dbInfo:conn name:mongoCollection.databaseName];
//    if (db) {
//        user = db.user;
//        password = db.password;
//    }
//    NSString *mapFunction = [mapFunctionTextView string];
//    NSString *reduceFunction = [reduceFunctionTextView string];
//    NSString *critical = [mrcriticalTextField stringValue];
//    NSString *output = [mroutputTextField stringValue];
//    NSMutableArray *results = [[NSMutableArray alloc] initWithArray:[mongoCollection mapReduceInDB:mongoCollection.databaseName 
//                                                                                collection:mongoCollection.collectionName 
//                                                                                      user:user 
//                                                                                  password:password 
//                                                                                     mapJs:mapFunction 
//                                                                                  reduceJs:reduceFunction 
//                                                                                  critical:critical 
//                                                                                    output:output]];
//    mrOutlineViewController.results = results;
//    [mrOutlineViewController.myOutlineView reloadData];
//    [results release];
//    [mrLoaderIndicator stop];
//    [pool drain];
//    [NSThread exit];
}

- (IBAction) export:(id)sender
{
    if ([[expPathTextField stringValue] length] == 0) {
        NSRunAlertPanel(@"Error", @"Please choose export path", @"OK", nil, nil);
        return;
    }
    if ([[expFieldsTextField stringValue] length] == 0 && [[expTypePopUpButton selectedItem] tag]==1)
    {
        NSRunAlertPanel(@"Error", @"You need to specify fields", @"OK", nil, nil);
        return;
    }
    [NSThread detachNewThreadSelector:@selector(doExport) toTarget:self withObject:nil];
}

- (void)doExport
{
//    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
//    std::auto_ptr<std::ofstream> fileStream;
//    std::ofstream * s = new std::ofstream( [[expPathTextField stringValue] UTF8String] , std::ios_base::out );
//    fileStream.reset( s );
//    std::ostream *outPtr = &std::cout;
//    outPtr = s;
//    if ( ! s->good() ) {
//        NSRunAlertPanel(@"Error", [NSString stringWithFormat:@"Couldn't open [%@]", [expPathTextField stringValue]], @"OK", nil, nil);
//    }
//    std::ostream &out = *outPtr;
//    bool _jsonArray = false;
//    if ([expJsonArrayCheckBox state] == 1) {
//        _jsonArray = true;
//    }
//    unsigned int exportType = [[expTypePopUpButton selectedItem] tag];
//    [expResultsTextField setStringValue:@"Start exporting"];
//    NSString *user=nil;
//    NSString *password=nil;
//    Database *db = [databasesArrayController dbInfo:conn name:mongoCollection.databaseName];
//    if (db) {
//        user = db.user;
//        password = db.password;
//    }
//    NSString *critical = [expCriticalTextField stringValue];
//    NSString *fields = [expFieldsTextField stringValue];
//    NSString *sort = [expSortTextField stringValue];
//    NSNumber *skip = [NSNumber numberWithInt:[expSkipTextField intValue]];
//    NSNumber *limit = [NSNumber numberWithInt:[expLimitTextField intValue]];
//    MODQuery *query = [mongoCollection countWithCriteria:critical];
//    [query waitUntilFinished];
//    long long int total = [[query.parameters objectForKey:@"count"] longLongValue];
//    if (total == 0) {
//        [expResultsTextField setStringValue:@"No data to export!"];
//        return;
//    }
//    
//    if ( exportType == 1 ) {
//        out << [fields UTF8String] << std::endl;
//    }else if (_jsonArray) {
//        out << '[';
//    }
//
//    
//    [expProgressIndicator setUsesThreadedAnimation:YES];
//    [expProgressIndicator startAnimation: self];
//    [expProgressIndicator setDoubleValue:0];
//    std::auto_ptr<mongo::DBClientCursor> cursor = [mongoCollection findCursorInDB:mongoCollection.databaseName 
//                                                               collection:mongoCollection.collectionName 
//                                                                     user:user 
//                                                                 password:password 
//                                                                 critical:critical 
//                                                                   fields:fields 
//                                                                     skip:skip 
//                                                                    limit:limit
//                                                                     sort:sort];
//    unsigned int i = 1;
//    while( cursor->more() )
//    {
//        mongo::BSONObj obj = cursor->next();
//        if ( exportType == 1 ) {
//            NSArray *keys = [[NSArray alloc] initWithArray:[fields componentsSeparatedByString:@","]];
//            unsigned int fieldIndex = 0;
//            for (NSString *str in keys) {
//                if (fieldIndex > 0) {
//                    out << ",";
//                }
//                const mongo::BSONElement & e = obj.getFieldDotted([str UTF8String]);
//                if ( ! e.eoo() ) {
//                    out << e.jsonString( mongo::TenGen , false );
//                }
//                fieldIndex ++;
//            }
//            [keys release];
//            out << std::endl;
//        }else {
//            if (_jsonArray && i != 1)
//                out << ',';
//            out << obj.jsonString();
//            if (!_jsonArray)
//            {
//                out << std::endl;
//            }
//        }
//        [expProgressIndicator setDoubleValue:(double)i/total*100];
//        i ++;
//    }
//    if ( exportType == 1 && _jsonArray) 
//        out << ']' << std::endl;
//    [expProgressIndicator stopAnimation: self];
//    [expResultsTextField setStringValue:[NSString stringWithFormat:@"Exported %d records.", total]];
//    [pool drain];
//    [NSThread exit];
}

- (IBAction) import:(id)sender
{
    if ([[impPathTextField stringValue] length] == 0) {
        NSRunAlertPanel(@"Error", @"Please choose import file", @"OK", nil, nil);
        return;
    }
    if ([[expFieldsTextField stringValue] length] == 0 && [[expTypePopUpButton selectedItem] tag]==1)
    {
        NSRunAlertPanel(@"Error", @"You need to specify fields", @"OK", nil, nil);
        return;
    }
    [NSThread detachNewThreadSelector:@selector(doImport) toTarget:self withObject:nil];
}

- (void)doImport
{
//    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
//    [impProgressIndicator setUsesThreadedAnimation:YES];
//    [impProgressIndicator startAnimation: self];
//    [impProgressIndicator setDoubleValue:0];
//    
//    long long fileSize = 0;
//    std::istream * in = &std::cin;
//    std::ifstream file( [[impPathTextField stringValue] UTF8String] , std::ios_base::in);
//    in = &file;
//    fileSize = boost::filesystem::file_size( [[impPathTextField stringValue] UTF8String] );
//    bool _ignoreBlanks = false;
//    bool _headerLine = false;
//    bool _jsonArray = false;
//    bool _stopOnError = false;
//    if ([impHeaderlineCheckBox state] == 1)
//    {
//        _headerLine = true;
//    }
//    if ([impJsonArrayCheckBox state] == 1) {
//        _jsonArray = true;
//    }
//    if ([impStopOnErrorCheckBox state] == 1) {
//        _stopOnError = true;
//    }
//    unsigned int _type = [[impTypePopUpButton selectedItem] tag];
//    std::string _sep;
//    if (_type == 1)
//    {
//        _sep = ",";
//    }else if(_type == 2){
//        _sep = "\t";
//    }
//    std::vector<std::string> _fields;
//    if (!_headerLine && [[impFieldsTextField stringValue] length] > 0)
//    {
//        
//        NSArray *keys = [[NSArray alloc] initWithArray:[[impFieldsTextField stringValue] componentsSeparatedByString:@","]];
//        for (NSString *str in keys) {
//            _fields.push_back([str UTF8String]);
//        }
//        [keys release];
//    }
//    
//    if (_type!=0 && !_headerLine && _fields.empty())
//    {
//        NSRunAlertPanel(@"Error", @"Please check headerline", @"OK", nil, nil);
//        return;
//    }
//    
//    
//    NSString *user=nil;
//    NSString *password=nil;
//    Database *db = [databasesArrayController dbInfo:conn name:mongoCollection.databaseName];
//    if (db) {
//        user = db.user;
//        password = db.password;
//    }
//    
//    if ([impDropCheckBox state] == 1)
//    {
//        [[mongoCollection dropCollectionWithName:mongoCollection.collectionName databaseName:mongoCollection.databaseName userName:user password:password] waitUntilFinished];
//    }
//    
//    if ([impIgnoreBlanksCheckBox state] == 1)
//    {
//        _ignoreBlanks = true;
//    }
//    
//    int errors = 0;
//    int num = 0;
//    const int BUF_SIZE = 1024 * 1024 * 4;
//    boost::scoped_array<char> line(new char[BUF_SIZE+2]);
//    char * buf = line.get();
//    while ( _jsonArray || in->rdstate() == 0 ) {
//        if (_jsonArray) {
//            if (buf == line.get()) { //first pass
//                in->read(buf, BUF_SIZE);
//                if (!(in->rdstate() & std::ios_base::eofbit))
//                {
//                    NSRunAlertPanel(@"Error", @"JSONArray file too large", @"OK", nil, nil);
//                    return;
//                }
//                buf[ in->gcount() ] = '\0';
//            }
//        }else {
//            buf = line.get();
//            in->getline( buf , BUF_SIZE );
//        }
//        if (!((!(in->rdstate() & std::ios_base::badbit)) && (!(in->rdstate() & std::ios_base::failbit) || (in->rdstate() & std::ios_base::eofbit))))
//        {
//            NSRunAlertPanel(@"Error", @"unknown error reading file", @"OK", nil, nil);
//            return;
//        }
//        
//        if (strncmp("\xEF\xBB\xBF", buf, 3) == 0) { // UTF-8 BOM (notepad is stupid)
//            buf += 3;
//        }
//        
//        if (_jsonArray) {
//            while (buf[0] != '{' && buf[0] != '\0') {
//                buf++;
//            }
//            if (buf[0] == '\0')
//                break;
//        }else {
//            while (std::isspace( buf[0] )) {
//                buf++;
//            }
//            if (buf[0] == '\0')
//                continue;
//        }
//        
//        try {
//            mongo::BSONObj o;
//            if (_jsonArray) {
//                int jslen;
//                o = mongo::fromjson(buf, &jslen);
//                buf += jslen;
//            }else {
//                o = [self parseCSVLine:buf type:_type sep:_sep.c_str() headerLine:_headerLine ignoreBlanks:_ignoreBlanks fields:_fields];NSLog(@"%@", [NSString stringWithUTF8String:o.jsonString( mongo::TenGen , false ).c_str()]);
//            }
//            if ( _headerLine ) {
//                _headerLine = false;
//            }else{
//                [mongoCollection insertInDB:mongoCollection.databaseName 
//                             collection:mongoCollection.collectionName 
//                                   user:user 
//                               password:password 
//                             insertData:[NSString stringWithUTF8String:o.jsonString( mongo::TenGen , false ).c_str()]
//                 ];
//            }
//            
//            num++;
//        }catch ( std::exception& e ) {
//            std::cout << "exception:" << e.what() << std::endl;
//            std::cout << buf << std::endl;
//            errors++;
//            
//            if (_stopOnError || _jsonArray)
//                break;
//        }
//    }
//    
//    [impProgressIndicator stopAnimation: self];
//    [impResultsTextField setStringValue:[NSString stringWithFormat:@"Imported %d records, %d failed", num, errors]];
//    [pool drain];
//    [NSThread exit];
}

- (IBAction)removeRecord:(id)sender
{
    if ([findResultsViewController.myOutlineView selectedRow] != -1)
    {
        id currentItem = [findResultsViewController.myOutlineView itemAtRow:[findResultsViewController.myOutlineView selectedRow]];
        //NSLog(@"%@", [findResultsViewController rootForItem:currentItem]);
        [removeQueryLoaderIndicator start];
        NSString *user=nil;
        NSString *password=nil;
        Database *db = [databasesArrayController dbInfo:conn name:mongoCollection.databaseName];
        if (db) {
            user = db.user;
            password = db.password;
        }
        NSString *critical;
        if ([[currentItem objectForKey:@"type"] isEqualToString:@"ObjectId"]) {
            critical = [NSString stringWithFormat:@"{\"_id\":ObjectId(\"%@\")}", [currentItem objectForKey:@"value"]];
        }else if ([[currentItem objectForKey:@"type"] isEqualToString:@"String"]) {
            critical = [NSString stringWithFormat:@"{\"_id\":\"%@\"}", [currentItem objectForKey:@"value"]];
        }else {
            critical = [NSString stringWithFormat:@"{\"_id\":%@}", [currentItem objectForKey:@"value"]];
        }
        [mongoCollection removeWithCriteria:critical callback:^(MODQuery *mongoQuery) {
            [removeQueryLoaderIndicator stop];
            [self findQuery:nil];
        }];
    }
}

- (void)controlTextDidChange:(NSNotification *)nd
{
	NSTextField *ed = [nd object];
    
	if (ed == criticalTextField || ed == fieldsTextField || ed == sortTextField || ed == skipTextField || ed == limitTextField) {
        [self findQueryComposer:nil];
    } else if (ed == updateCriticalTextField || ed == updateSetTextField) {
        [self updateQueryComposer:nil];
    } else if (ed == removeCriticalTextField) {
        [self removeQueryComposer:nil];
    } else if (ed == expCriticalTextField || ed == expFieldsTextField || ed == expSortTextField || ed == expSkipTextField || ed == expLimitTextField) {
        [self exportQueryComposer:nil];
    }

}

- (IBAction) findQueryComposer:(id)sender
{
    NSString *critical;
    critical = [self formatedQueryWithReplace:NO];
    
    NSString *jsFields;
    if ([[fieldsTextField stringValue] length] > 0) {
        NSArray *keys = [[NSArray alloc] initWithArray:[[fieldsTextField stringValue] componentsSeparatedByString:@","]];
        NSMutableArray *tmpstr = [[NSMutableArray alloc] initWithCapacity:[keys count]];
        for (NSString *str in keys) {
            [tmpstr addObject:[NSString stringWithFormat:@"%@:1", str]];
        }
        jsFields = [[NSString alloc] initWithFormat:@", {%@}", [tmpstr componentsJoinedByString:@","] ];
        [keys release];
        [tmpstr release];
    }else {
        jsFields = [[NSString alloc] initWithString:@""];
    }
    
    NSString *sort;
    if ([[sortTextField stringValue] length] > 0) {
        sort = [[NSString alloc] initWithFormat:@".sort(%@)", [sortTextField stringValue]];
    }else {
        sort = [[NSString alloc] initWithString:@""];
    }
    
    NSString *skip = [[NSString alloc] initWithFormat:@".skip(%d)", [skipTextField intValue]];
    NSString *limit = [[NSString alloc] initWithFormat:@".limit(%d)", [limitTextField intValue]];
    NSString *col = [NSString stringWithFormat:@"%@.%@", mongoCollection.databaseName, mongoCollection.collectionName];
    
    NSString *query = [NSString stringWithFormat:@"db.%@.find(%@%@)%@%@%@", col, critical, jsFields, sort, skip, limit];
    [jsFields release];
    [sort release];
    [skip release];
    [limit release];
    [findQueryTextField setStringValue:query];
}

- (IBAction)updateQueryComposer:(id)sender
{
    NSString *col = [NSString stringWithFormat:@"%@.%@", mongoCollection.databaseName, mongoCollection.collectionName];
    NSString *critical;
    if ([[updateCriticalTextField stringValue] length] > 0) {
        critical = [[NSString alloc] initWithString:[updateCriticalTextField stringValue]];
    }else {
        critical = [[NSString alloc] initWithString:@""];
    }
    NSString *sets;
    if ([[updateSetTextField stringValue] length] > 0) {
        //sets = [[NSString alloc] initWithFormat:@", {$set:%@}", [updateSetTextField stringValue]];
        sets = [[NSString alloc] initWithFormat:@", %@", [updateSetTextField stringValue]];
    }else {
        sets = [[NSString alloc] initWithString:@""];
    }
    NSString *upset;
    if ([upsetCheckBox state] == 1) {
        upset = [[NSString alloc] initWithString:@", true"];
    }else {
        upset = [[NSString alloc] initWithString:@", false"];
    }

    NSString *query = [NSString stringWithFormat:@"db.%@.update(%@%@%@)", col, critical, sets, upset];
    [critical release];
    [sets release];
    [upset release];
    [updateQueryTextField setStringValue:query];
}

- (IBAction)removeQueryComposer:(id)sender
{
    NSString *col = [NSString stringWithFormat:@"%@.%@", mongoCollection.databaseName, mongoCollection.collectionName];
    NSString *critical;
    if ([[removeCriticalTextField stringValue] length] > 0) {
        critical = [[NSString alloc] initWithString:[removeCriticalTextField stringValue]];
    }else {
        critical = [[NSString alloc] initWithString:@""];
    }
    NSString *query = [NSString stringWithFormat:@"db.%@.remove(%@)", col, critical];
    [critical release];
    [removeQueryTextField setStringValue:query];
}

- (IBAction) exportQueryComposer:(id)sender
{
    NSString *critical;
    if ([[expCriticalTextField stringValue] length] > 0) {
        critical = [[NSString alloc] initWithString:[expCriticalTextField stringValue]];
    }else {
        critical = [[NSString alloc] initWithString:@""];
    }
    
    NSString *jsFields;
    if ([[expFieldsTextField stringValue] length] > 0) {
        NSArray *keys = [[NSArray alloc] initWithArray:[[expFieldsTextField stringValue] componentsSeparatedByString:@","]];
        NSMutableArray *tmpstr = [[NSMutableArray alloc] initWithCapacity:[keys count]];
        for (NSString *str in keys) {
            [tmpstr addObject:[NSString stringWithFormat:@"%@:1", str]];
        }
        jsFields = [[NSString alloc] initWithFormat:@", {%@}", [tmpstr componentsJoinedByString:@","] ];
        [keys release];
        [tmpstr release];
    }else {
        jsFields = [[NSString alloc] initWithString:@""];
    }
    
    NSString *sort;
    if ([[expSortTextField stringValue] length] > 0) {
        sort = [[NSString alloc] initWithFormat:@".sort(%@)"];
    }else {
        sort = [[NSString alloc] initWithString:@""];
    }
    
    NSString *skip = [[NSString alloc] initWithFormat:@".skip(%d)", [expSkipTextField intValue]];
    NSString *limit = [[NSString alloc] initWithFormat:@".limit(%d)", [expLimitTextField intValue]];
    NSString *col = [NSString stringWithFormat:@"%@.%@", mongoCollection.databaseName, mongoCollection.collectionName];
    
    NSString *query = [NSString stringWithFormat:@"db.%@.find(%@%@)%@%@%@", col, critical, jsFields, sort, skip, limit];
    [critical release];
    [jsFields release];
    [sort release];
    [skip release];
    [limit release];
    [expQueryTextField setStringValue:query];
}

- (void)showEditWindow:(id)sender
{
    switch([findResultsViewController.myOutlineView selectedRow])
	{
		case -1:
			break;
		default:{
            [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(findQuery:) name:kJsonWindowSaved object:nil];
            [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(jsonWindowWillClose:) name:kJsonWindowWillClose object:nil];
            id currentItem = [findResultsViewController.myOutlineView itemAtRow:[findResultsViewController.myOutlineView selectedRow]];
            //NSLog(@"%@", [findResultsViewController rootForItem:currentItem]);
            JsonWindowController *jsonWindowController = [[JsonWindowController alloc] init];
            jsonWindowController.managedObjectContext = self.managedObjectContext;
            jsonWindowController.conn = conn;
            jsonWindowController.mongoCollection = mongoCollection;
            jsonWindowController.jsonDict = [findResultsViewController rootForItem:currentItem];
            [jsonWindowController showWindow:sender];
			break;
        }
	}
}

- (void)jsonWindowWillClose:(id)sender
{
    [[NSNotificationCenter defaultCenter] removeObserver:self];
}

- (IBAction)chooseExportPath:(id)sender
{
    NSSavePanel *tvarNSSavePanelObj	= [NSSavePanel savePanel];
    int tvarInt	= [tvarNSSavePanelObj runModal];
    if(tvarInt == NSOKButton){
     	NSLog(@"doSaveAs we have an OK button");
        //NSString * tvarDirectory = [tvarNSSavePanelObj directory];
        //NSLog(@"doSaveAs directory = %@",tvarDirectory);
        NSString * tvarFilename = [[tvarNSSavePanelObj URL] path];
        NSLog(@"doSaveAs filename = %@",tvarFilename);
        [expPathTextField setStringValue:tvarFilename];
    } else if(tvarInt == NSCancelButton) {
     	NSLog(@"doSaveAs we have a Cancel button");
     	return;
    } else {
     	NSLog(@"doSaveAs tvarInt not equal 1 or zero = %3d",tvarInt);
     	return;
    } // end if
}

- (IBAction)chooseImportPath:(id)sender
{
    NSOpenPanel *tvarNSOpenPanelObj	= [NSOpenPanel openPanel];
    NSInteger tvarNSInteger	= [tvarNSOpenPanelObj runModal];
    if(tvarNSInteger == NSOKButton){
     	NSLog(@"doOpen we have an OK button");
        //NSString * tvarDirectory = [tvarNSOpenPanelObj directory];
        //NSLog(@"doOpen directory = %@",tvarDirectory);
        NSString * tvarFilename = [[tvarNSOpenPanelObj URL] path];
        NSLog(@"doOpen filename = %@",tvarFilename);
        [impPathTextField setStringValue:tvarFilename];
    } else if(tvarNSInteger == NSCancelButton) {
     	NSLog(@"doOpen we have a Cancel button");
     	return;
    } else {
     	NSLog(@"doOpen tvarInt not equal 1 or zero = %ld",(long int)tvarNSInteger);
     	return;
    } // end if
}

//- (mongo::BSONObj)parseCSVLine:(char *) line type:(int)_type sep:(const char *)_sep headerLine:(bool)_headerLine ignoreBlanks:(bool)_ignoreBlanks fields:(std::vector<std::string>&)_fields
//{
//    if ( _type == 0 ) {
//        char * end = ( line + strlen( line ) ) - 1;
//        while ( std::isspace(*end) ) {
//            *end = 0;
//            end--;
//        }
//        return mongo::fromjson( line );
//    }
//    mongo::BSONObjBuilder b;
//    
//    unsigned int pos=0;
//    while ( line[0] ) {
//        std::string name;
//        if ( pos < _fields.size() ) {
//            name = _fields[pos];
//        }else {
//            std::stringstream ss;
//            ss << "field" << pos;
//            name = ss.str();
//        }
//        pos++;
//        
//        bool done = false;
//        std::string data;
//        char * end;
//        if ( _type == 1 && line[0] == '"' ) {
//            line++; //skip first '"'
//            
//            while (true) {
//                end = strchr( line , '"' );NSLog(@"%s", line);
//                if (!end) {
//                    data += line;
//                    done = true;
//                    break;
//                } else if (end[1] == '"') {
//                    // two '"'s get appended as one
//                    data.append(line, end-line+1); //include '"'
//                    line = end+2; //skip both '"'s
//                } else if (end[-1] == '\\') {
//                    // "\\\"" gets appended as '"'
//                    data.append(line, end-line-1); //exclude '\\'
//                    data.append("\"");
//                    line = end+1; //skip the '"'
//                } else {
//                    data.append(line, end-line);
//                    line = end+2; //skip '"' and ','
//                    break;
//                }
//            }
//        } else {
//            end = strstr( line , _sep );NSLog(@"end: %s", end);
//            if ( ! end ) {
//                done = true;
//                data = std::string( line );
//            } else {
//                data = std::string( line , end - line );
//                line = end+1;
//            }
//        }
//        
//        if ( _headerLine ) {
//            while ( std::isspace( data[0] ) )
//                data = data.substr( 1 );
//            _fields.push_back( data );
//        }else{
//            if ( !b.appendAsNumber( name , data ) && !(_ignoreBlanks && data.size() == 0) ){
//                b.append( name , data );
//            }
//        }
//        
//        if ( done )
//            break;
//    }
//    return b.obj();
//}

- (IBAction)segmentedControlAction:(id)sender
{
    NSString *identifier;
    
    identifier = [[NSString alloc] initWithFormat:@"%ld", [segmentedControl selectedSegment]];
    [tabView selectTabViewItemWithIdentifier:identifier];
    [identifier release];
}

@end

@implementation QueryWindowController(MODCollectionDelegate)

- (void)mongoCollection:(MODCollection *)collection queryResultFetched:(NSArray *)result withMongoQuery:(MODQuery *)mongoQuery errorMessage:(NSString *)errorMessage
{
    [findQueryLoaderIndicator stop];
    if (collection == mongoCollection) {
        if (errorMessage) {
            NSRunAlertPanel(@"Error", errorMessage, @"OK", nil, nil);
        } else {
            [findResultsViewController.results removeAllObjects];
            [findResultsViewController.results addObjectsFromArray:result];
            [findResultsViewController.myOutlineView reloadData];
        }
    }
}

- (void)mongoCollection:(MODCollection *)collection queryCountWithValue:(long long)value withMongoQuery:(MODQuery *)mongoQuery errorMessage:(NSString *)errorMessage
{
    if (collection == mongoCollection) {
        if ([mongoQuery.userInfo objectForKey:@"title"]) {
            if ([mongoQuery.userInfo objectForKey:@"timequery"]) {
                [[mongoQuery.userInfo objectForKey:@"textfield"] setStringValue:[NSString stringWithFormat:[mongoQuery.userInfo objectForKey:@"title"], value, [[mongoQuery.userInfo objectForKey:@"timequery"] duration]]];
            } else {
                [[mongoQuery.userInfo objectForKey:@"textfield"] setStringValue:[NSString stringWithFormat:[mongoQuery.userInfo objectForKey:@"title"], value]];
            }
        }
    }
}

- (void)mongoCollection:(MODCollection *)collection updateDonwWithMongoQuery:(MODQuery *)mongoQuery errorMessage:(NSString *)errorMessage
{
    if (collection == mongoCollection) {
        [findQueryLoaderIndicator stop];
        if (errorMessage) {
            NSRunAlertPanel(@"Error", errorMessage, @"OK", nil, nil);
        }
    }
}

@end