//
//  $Id$
//
//  TableDump.h
//  sequel-pro
//
//  Created by lorenz textor (lorenz@textor.ch) on Wed May 01 2002.
//  Copyright (c) 2002-2003 Lorenz Textor. All rights reserved.
//
//  This program is free software; you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation; either version 2 of the License, or
//  (at your option) any later version.
//
//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with this program; if not, write to the Free Software
//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//
//  More info at <http://code.google.com/p/sequel-pro/>
//  Or mail to <lorenz@textor.ch>

#import <Cocoa/Cocoa.h>
#import <MCPKit_bundled/MCPKit_bundled.h>
#import "CMMCPConnection.h"
#import "CMMCPResult.h"


@interface TableDump : NSObject {

	IBOutlet id tableDocumentInstance;
	IBOutlet id tablesListInstance;
	IBOutlet id tableSourceInstance;
	IBOutlet id tableContentInstance;
	IBOutlet id tableDataInstance;
	IBOutlet id customQueryInstance;

    IBOutlet id tableWindow;
	IBOutlet id tableListView;
	
    IBOutlet id exportDumpView;
    IBOutlet id exportCSVView;
    IBOutlet id exportMultipleCSVView;
    IBOutlet id exportMultipleXMLView;
    IBOutlet id exportDumpTableView;
    IBOutlet id exportMultipleCSVTableView;
    IBOutlet id exportMultipleXMLTableView;
    IBOutlet id exportFieldNamesSwitch;
    IBOutlet id exportFieldsTerminatedField;
    IBOutlet id exportFieldsEnclosedField;
    IBOutlet id exportFieldsEscapedField;
    IBOutlet id exportLinesTerminatedField;
    IBOutlet id exportMultipleFieldNamesSwitch;
    IBOutlet id exportMultipleFieldsTerminatedField;
    IBOutlet id exportMultipleFieldsEnclosedField;
    IBOutlet id exportMultipleFieldsEscapedField;
    IBOutlet id exportMultipleLinesTerminatedField;
	
	// New Export Window
	IBOutlet id exportWindow;
	IBOutlet id exportTabBar;
	IBOutlet id exportToolbar;
	IBOutlet id	exportTableList;
	
	IBOutlet id importCSVView;
	IBOutlet NSPopUpButton *importFormatPopup;
	IBOutlet id importCSVBox;
    IBOutlet id importFieldNamesSwitch;
    IBOutlet id importFieldsTerminatedField;
    IBOutlet id importFieldsEnclosedField;
    IBOutlet id importFieldsEscapedField;
    IBOutlet id importLinesTerminatedField;
	
    IBOutlet id addDropTableSwitch;
    IBOutlet id addCreateTableSwitch;
    IBOutlet id addTableContentSwitch;
    IBOutlet id addErrorsSwitch;
    IBOutlet id errorsSheet;
    IBOutlet id errorsView;
    IBOutlet id singleProgressSheet;
    IBOutlet id singleProgressBar;
    IBOutlet id singleProgressText;
	
    IBOutlet id fieldMappingSheet;
	IBOutlet id fieldMappingPopup;
    IBOutlet id fieldMappingTableView;
    
	IBOutlet id rowUpButton;
    IBOutlet id rowDownButton;
	IBOutlet id recordCountLabel;

	CMMCPConnection *mySQLConnection;

	NSMutableArray *tables;
	NSArray *importArray;
	NSMutableArray *fieldMappingArray;
	NSMutableArray *fieldMappingButtonOptions;
	int currentRow;
	NSString *savePath;
	NSString *openPath;
	NSUserDefaults *prefs;
	BOOL progressCancelled;
}

//IBAction methods
- (IBAction)reloadTables:(id)sender;
- (IBAction)selectTables:(id)sender;
- (IBAction)closeSheet:(id)sender;
- (IBAction)stepRow:(id)sender;
- (IBAction)cancelProgressBar:(id)sender;

//export methods
- (void)export;
- (void)exportFile:(int)tag;
- (void)savePanelDidEnd:(NSSavePanel *)sheet returnCode:(int)returnCode contextInfo:(NSString *)contextInfo;

//import methods
- (void)importFile;
- (IBAction)changeFormat:(id)sender;
- (IBAction)changeTable:(id)sender;
- (void)openPanelDidEnd:(NSOpenPanel *)sheet returnCode:(int)returnCode contextInfo:(NSString *)contextInfo;
- (void)setupFieldMappingArray;
- (void)updateFieldMappingButtonCell;
- (NSArray *)arrayForCSV:(NSString *)csv terminatedBy:(NSString *)terminated
	enclosedBy:(NSString *)enclosed escapedBy:(NSString *)escaped lineEnds:(NSString *)lineEnds;
- (NSArray *)arrayForString:(NSString *)string enclosed:(NSString *)enclosed
	escaped:(NSString *)escaped terminated:(NSString *)terminated;

// Export methods
- (BOOL)dumpSelectedTablesAsSqlToFileHandle:(NSFileHandle *)fileHandle;
- (BOOL)dumpSchemaAsDotToFileHandle:(NSFileHandle *)fileHandle;
- (BOOL)writeCsvForArray:(NSArray *)array orQueryResult:(CMMCPResult *)queryResult
	toFileHandle:(NSFileHandle *)fileHandle
	outputFieldNames:(BOOL)firstLine terminatedBy:(NSString *)terminated
	enclosedBy:(NSString *)enclosed escapedBy:(NSString *)escaped
	lineEnds:(NSString *)lineEnds withNumericColumns:(NSArray *)tableColumnNumericStatus silently:(BOOL)silently;
- (BOOL)writeXmlForArray:(NSArray *)array orQueryResult:(CMMCPResult *)queryResult
	toFileHandle:(NSFileHandle *)fileHandle
	tableName:(NSString *)table withHeader:(BOOL)header silently:(BOOL)silently;
- (NSString *)htmlEscapeString:(NSString *)string;

- (BOOL)exportTables:(NSArray *)selectedTables toFileHandle:(NSFileHandle *)fileHandle usingFormat:(NSString *)type usingMulti:(BOOL)multi;
- (BOOL)exportSelectedTablesToFileHandle:(NSFileHandle *)fileHandle usingFormat:(NSString *)type;

// New Export methods
- (IBAction)switchTab:(id)sender;
- (IBAction)switchInput:(id)sender;

//additional methods
- (void)setConnection:(CMMCPConnection *)theConnection;

// Import/export delegate notifications
- (void)panelSelectionDidChange:(id)sender;

//last but not least
- (id)init;
- (void)dealloc;

@end