/** eoutil.m eoutil Tool Copyright (C) 2003 Free Software Foundation, Inc. Author: Stephane Corthesy Date: February 2003 $Revision$ $Date$ This file is part of the GNUstep Database Library. This library is free software; you can redistribute it and/or modify it under the terms of the GNU Library General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This library 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 Library General Public License for more details. You should have received a copy of the GNU Library General Public License along with this library; see the file COPYING.LIB. If not, write to the Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. **/ #include "config.h" RCS_ID("$Id$") #ifdef GNUSTEP #include #include #include #include #include #include #include #include #include #include #include #include #include #else #include #endif #ifndef GNUSTEP #include #endif #include #include /* TODO Implement -ascii option Test multiple models Add new option -dateFormat ISO|Apple (has ISO enough precision for microseconds?) Add new option -nullValue '__NULL__' Question: is it possible to save/restore sequences (for PK)? Replace all autorelease/release/retain by macros Document plist format Document usage */ @interface NSString(eoutil) - (NSString *) lossyASCIIString; @end @implementation NSString(eoutil) - (NSString *) lossyASCIIString { #warning TODO: implement lossyASCIIString return self; } @end static BOOL usage(BOOL fullUsage) { char *usageString; if (fullUsage) usageString = "Command line utility to perform database and EOF oriented tasks.\n" "Syntax: eoutil dump/connect/convert\n" " eoutil dump [-source [sourcefile]] -dest [destfile] [-schemaCreate ] [-schemaDrop ] [-postInstall] [-force] [-connDict ] [-entities ] [-modelGroup ] [-ascii]\n" "\tThis command reads data from and dumps it to . The dump may\n" "\tinclude bits of DML that create bits of database schema. The -source flag is\n" "\toptional if only schema is to be created (i.e. no data is being dumped). The\n" "\tfollowing is a list of possible values for each of the arguments in the command:\n" "\t model -- must be the name of an eomodel or eomodeld file\n" "\t source -- 'plist' (read from stdin unless source file is specified) or\n" "\t 'database' (uses model to connect)\n" "\t sourcefile -- name of file to read plist from\n" "\t dest -- 'plist' or 'script' (both are written to stdout unless destfile is\n" "\t specified) or 'database' (uses model to connect)\n" "\t destfile -- name of file to write plist or script\n" "\t options -- 'database', 'tables', 'primaryKeySupport',\n" "\t 'primaryKeyConstraints' or 'foreignKeyConstraints'\n" "\t postInstall -- looks in the userInfo dictionary for an array of SQL strings\n" "\t to process after any schema creation and data dumping is done\n" "\t force -- Do not quit processing after database error\n" "\t connDict -- A substitute connection dictionary\n" "\t entities -- a subset of the entities in the model (all are used by default)\n" "\t modelGroup -- A list of models to create a model group. (Allows you to use\n" "\t models not in a framework. Model names must be absolute paths.)\n" "\t ascii -- Convert all non-ASCII characters to their nearest ASCII equivalents.\n" "\n" "\tExample: eoutil dump Movies.eomodeld -source plist -dest database -schemaCreate tables primaryKeySupport foreignKeyConstraints < MovieData.plist\n" "\t Creates the tables in the database, reads data in plist format from,\n" "\t stdin stores the data in the database listed in the connection\n" "\t dictionary of model, creates primary key support in the database, and\n" "\t creates foreign key constraints corresponding to the relationship\n" "\t definitions.\n" "\n" " eoutil convert \n" "\tThis command converts the type mapping in a model for the data source specified\n" "\tby the connection dictionary URL. The connection dictionary in the new model is\n" "\tset to connectionDictionary.\n" "\n" "\t model -- must be the name of an eomodel or eomodeld file\n" "\t adaptorName -- Postgres95, FlatFile, etc.\n" "\t connectionDictionary -- string in property list format representing a\n" "\t dictionary, specifying the database name for the\n" "\t data source (required) and username and password (if\n" "\t required by that data source).\n" "\t outFileName -- the name of a directory to write the converted model\n" "\n" "\tExample: eoutil convert Movies.eomodeld Postgres95 '{ databaseName = test; hostName = localhost; userName = postgres; password = postgres; }' M.eomodeld\n" "\t Converts types in Movies.eomodeld file to Postgres95 types and writes\n" "\t the converted model to ./M.eomodel\n" "\n" " eoutil connect ( | ( ))\n" "\tThis command attempts to connect to a data source using the adaptor named\n" "\tadaptorName with connectionDictionary, or using the connection dictionary in\n" "\tmodel. It returns an exit status of 0 if successful and 1 otherwise. This is\n" "\tprimarily useful for scripts.\n" "\n" "\t model -- must be the name of an eomodel or eomodeld file\n" "\t adaptorName -- Postgres95, FlatFile, etc.\n" "\t connectionDictionary -- string in property list format representing a\n" "\t dictionary, specifying the database name for the\n" "\t data source (required) and username and password (if\n" "\t required by that data source).\n" "\n" "\tExample: eoutil connect Postgres95 '{ databaseName = test; hostName = localhost; userName = postgres; password = postgres; }'\n" "\t Attempts to connect to a Postgres database on host localhost.\n" "\n" " eoutil help\n" "\tPrints this text.\n" "\n"; else usageString = "Command line utility to perform database and EOF oriented tasks.\n" "Syntax: eoutil dump/connect/convert/help\n" " eoutil dump [-source [sourcefile]] -dest [destfile] [-schemaCreate ] [-schemaDrop ] [-postInstall] [-force] [-connDict ] [-entities ] [-modelGroup ] [-ascii]\n" " eoutil convert \n" " eoutil connect ( | ( ))\n" " eoutil help\n"; NSLog(@"%s", usageString); return YES; } enum { databaseOption = 1 << 0, tablesOption = 1 << 1, pkSupportOption = 1 << 2, pkConstraintsOption = 1 << 3, fkConstraintsOption = 1 << 4, }; static BOOL dump(NSArray *arguments) { int aCount = [arguments count], i = 0; EOModel *srcModel; NSString *sourceType = nil, *sourceFilename = nil; NSString *destType = nil, *destFilename = nil; int schemaCreateOptions = 0, schemaDropOptions = 0; BOOL postInstall = NO; BOOL force = NO; NSDictionary *aConnectionDictionary = nil; NSMutableArray *entityNameList = nil; NSMutableArray *entitiesList = nil; NSMutableArray *modelList = nil; BOOL convertsNonASCII = NO; NSMutableString *outputString = [NSMutableString string]; NSArray *statements = nil; NSDictionary *dataDictionary = nil; Class exprClass; NSArray *postInstallSQLExpressions = nil; // First, let's parse arguments srcModel = [EOModel modelWithContentsOfFile: [[arguments objectAtIndex: i++] stringByStandardizingPath]]; while (i < aCount) { NSString *anArg = [arguments objectAtIndex: i++]; if ([anArg isEqualToString: @"-source"]) { if (sourceType != nil) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-source'"]; if (i < aCount) sourceType = [arguments objectAtIndex: i++]; else [NSException raise: NSInvalidArgumentException format: @"Missing argument(s) after '-source'"]; if (![sourceType isEqualToString: @"database"]) { if ([sourceType isEqualToString: @"plist"]) { if (i < aCount) { sourceFilename = [[arguments objectAtIndex: i++] stringByStandardizingPath]; if ([sourceFilename hasPrefix: @"-"]) { // We consider it's an option, not a filename sourceFilename = nil; i--; } } } else [NSException raise: NSInvalidArgumentException format: @"Illegal '-source' option: %@", sourceType]; } } else if ([anArg isEqualToString: @"-dest"]) { if (destType != nil) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-dest'"]; if (i < aCount) destType = [arguments objectAtIndex: i++]; else [NSException raise: NSInvalidArgumentException format: @"Missing argument(s) after '-dest'"]; if (![destType isEqualToString: @"database"]) { if ([destType isEqualToString: @"script"] || [destType isEqualToString: @"plist"]) { if (i < aCount) { destFilename = [[arguments objectAtIndex: i++] stringByStandardizingPath]; if ([destFilename hasPrefix: @"-"]) { // We consider it's an option, not a filename destFilename = nil; i--; } } } else [NSException raise: NSInvalidArgumentException format: @"Illegal '-dest' option: %@", destType]; } } else if ([anArg isEqualToString: @"-schemaCreate"]) { if (schemaCreateOptions != 0) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-schemaCreate'"]; while (i < aCount) { anArg = [arguments objectAtIndex: i++]; if ([anArg isEqualToString: @"database"]) schemaCreateOptions |= databaseOption; else if ([anArg isEqualToString: @"tables"]) schemaCreateOptions |= tablesOption; else if ([anArg isEqualToString: @"primaryKeySupport"]) schemaCreateOptions |= pkSupportOption; else if ([anArg isEqualToString: @"primaryKeyConstraints"]) schemaCreateOptions |= pkConstraintsOption; else if ([anArg isEqualToString: @"foreignKeyConstraints"]) schemaCreateOptions |= fkConstraintsOption; else { if (![anArg hasPrefix:@"-"]) [NSException raise: NSInvalidArgumentException format: @"Illegal '-schemaCreate' option: %@", anArg]; i--; break; } } if (schemaCreateOptions == 0) [NSException raise: NSInvalidArgumentException format: @"Missing argument(s) after '-schemaCreate'"]; } else if ([anArg isEqualToString: @"-schemaDrop"]) { if (schemaDropOptions != 0) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-schemaDrop'"]; while (i < aCount) { anArg = [arguments objectAtIndex: i++]; if ([anArg isEqualToString: @"database"]) schemaDropOptions |= databaseOption; else if ([anArg isEqualToString: @"tables"]) schemaDropOptions |= tablesOption; else if ([anArg isEqualToString: @"primaryKeySupport"]) schemaDropOptions |= pkSupportOption; /* else if ([anArg isEqualToString: @"primaryKeyConstraints"]) schemaDropOptions |= pkConstraintsOption; else if ([anArg isEqualToString: @"foreignKeyConstraints"]) schemaDropOptions |= fkConstraintsOption;*/ else { if (![anArg hasPrefix: @"-"]) [NSException raise: NSInvalidArgumentException format: @"Illegal '-schemaDrop' option: %@", anArg]; i--; break; } } if (schemaDropOptions == 0) [NSException raise: NSInvalidArgumentException format: @"Missing argument(s) after '-schemaDrop'"]; } else if ([anArg isEqualToString: @"-postInstall"]) { if (postInstall) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-postInstall'"]; postInstall = YES; } else if ([anArg isEqualToString: @"-force"]) { if (force) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-force'"]; force = YES; } else if ([anArg isEqualToString: @"-ascii"]) { #warning TODO: ascii option is currently not used if (convertsNonASCII) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-ascii'"]; convertsNonASCII = YES; } else if ([anArg isEqualToString: @"-connDict"]) { if (aConnectionDictionary) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-connDict'"]; if (i < aCount) { aConnectionDictionary = [[arguments objectAtIndex: i++] propertyList]; if (![aConnectionDictionary isKindOfClass: [NSDictionary class]]) [NSException raise: NSInvalidArgumentException format: @"Invalid '-connDict' argument; must be a NSDictionary"]; } else [NSException raise: NSInvalidArgumentException format: @"Missing argument after '-connDict'"]; } else if ([anArg isEqualToString: @"-entities"]) { if (entityNameList) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-entities'"]; entityNameList = [NSMutableArray array]; while (i < aCount) { anArg = [arguments objectAtIndex: i++]; if ([anArg hasPrefix: @"-"]) { // We consider it's an option, not an entity name i--; break; } if ([entityNameList containsObject: anArg]) [NSException raise: NSInvalidArgumentException format: @"Entity name argument '%@' occurs more than once", anArg]; [entityNameList addObject: anArg]; } if ([entityNameList count] == 0) [NSException raise: NSInvalidArgumentException format: @"Missing argument(s) after '-entities'"]; } else if ([anArg isEqualToString: @"-modelGroup"]) { #warning TODO: test multiple models if (modelList) [NSException raise: NSInvalidArgumentException format: @"More than one occurence of parameter '-modelGroup'"]; modelList = [NSMutableArray array]; while (i < aCount) { anArg = [[arguments objectAtIndex: i++] stringByStandardizingPath]; if ([anArg hasPrefix: @"-"]) { // We consider it's an option, not a file name i--; break; } if ([modelList containsObject: anArg]) [NSException raise: NSInvalidArgumentException format: @"Model file name argument '%@' occurs more than once", anArg]; if (![anArg isAbsolutePath]) anArg = [[[NSFileManager defaultManager] currentDirectoryPath] stringByAppendingPathComponent:anArg]; // [NSException raise:NSInvalidArgumentException format:@"Model file name argument '%@' is not absolute", anArg]; [modelList addObject: anArg]; } if ([modelList count] == 0) [NSException raise: NSInvalidArgumentException format: @"Missing argument(s) after '-modelGroup'"]; } else // Else, ignore remaining args break; } // Now check parameter consistency if (!destType) [NSException raise: NSInvalidArgumentException format: @"Missing -dest [destfile] argument"]; if (sourceType == nil && schemaCreateOptions == 0 && schemaDropOptions == 0) [NSException raise: NSInvalidArgumentException format: @"Unknown operation"]; if ([destType isEqualToString: @"plist"] && (schemaDropOptions != 0 || schemaCreateOptions != 0)) [NSException raise: NSInvalidArgumentException format: @"Can't declare plist as destination and use schema arguments"]; // Load models { #warning TODO: test this EOModelGroup *defaultModelGroup = [EOModelGroup globalModelGroup]; (void)[defaultModelGroup addModel: srcModel]; if (modelList) { // Load other models in same model group NSEnumerator *anEnum = [modelList objectEnumerator]; NSString *aString; while ((aString = [anEnum nextObject])) { (void)[defaultModelGroup addModelWithFile: aString]; } } [defaultModelGroup loadAllModelObjects]; [EOModelGroup setDefaultGroup: defaultModelGroup]; } // Create entity list if (entityNameList) { NSEnumerator *anEnum = [entityNameList objectEnumerator]; NSString *aString; entitiesList = [NSMutableArray array]; while ((aString = [anEnum nextObject])) { EOEntity *anEntity = [[EOModelGroup defaultGroup] entityNamed: aString]; if (anEntity == nil) [NSException raise: NSInvalidArgumentException format: @"Unknown entity '%@'", aString]; [entitiesList addObject: anEntity]; } } else { NSEnumerator *anEnum = [[srcModel entities] objectEnumerator]; EOEntity *anEntity; entitiesList = [NSMutableArray array]; while ((anEntity = [anEnum nextObject])) [entitiesList addObject:anEntity]; } if (aConnectionDictionary) { [srcModel setConnectionDictionary: aConnectionDictionary]; // Only for srcModel, not for additional models? } exprClass = [[EOAdaptor adaptorWithModel: srcModel] expressionClass]; if (schemaDropOptions != 0 || schemaCreateOptions != 0) { NSDictionary *aDictionary = [NSDictionary dictionaryWithObjectsAndKeys: ((schemaDropOptions & databaseOption) ? @"YES":@"NO"), EODropDatabaseKey, ((schemaDropOptions & tablesOption) ? @"YES":@"NO"), EODropTablesKey, ((schemaDropOptions & pkSupportOption) ? @"YES":@"NO"), EODropPrimaryKeySupportKey, // ((schemaDropOptions & pkConstraintsOption) ? @"YES":@"NO"), EOPrimaryKeyConstraintsKey, // ((schemaDropOptions & fkConstraintsOption) ? @"YES":@"NO"), EOForeignKeyConstraintsKey, ((schemaCreateOptions & databaseOption) ? @"YES":@"NO"), EOCreateDatabaseKey, ((schemaCreateOptions & tablesOption) ? @"YES":@"NO"), EOCreateTablesKey, ((schemaCreateOptions & pkSupportOption) ? @"YES":@"NO"), EOCreatePrimaryKeySupportKey, ((schemaCreateOptions & pkConstraintsOption) ? @"YES":@"NO"), EOPrimaryKeyConstraintsKey, ((schemaCreateOptions & fkConstraintsOption) ? @"YES":@"NO"), EOForeignKeyConstraintsKey, nil]; NSEnumerator *anEnum; EOSQLExpression *aStatement; statements = [exprClass schemaCreationStatementsForEntities: entitiesList options: aDictionary]; if ([destType isEqualToString: @"database"]) { EODatabaseContext *databaseContext; EOEditingContext *anEC = [[EOEditingContext alloc] init]; databaseContext = [EODatabaseContext registeredDatabaseContextForModel: srcModel editingContext: anEC]; [databaseContext lock]; NS_DURING { EODatabaseChannel *databaseChannel = [databaseContext availableChannel]; EOAdaptorChannel *adaptorChannel = [databaseChannel adaptorChannel]; if (![adaptorChannel isOpen]) [adaptorChannel openChannel]; anEnum = [statements objectEnumerator]; while ((aStatement = [anEnum nextObject])) { NS_DURING [adaptorChannel evaluateExpression: aStatement]; NS_HANDLER if(!force) break; NS_ENDHANDLER; } [databaseContext unlock]; [anEC release]; } NS_HANDLER { [databaseContext unlock]; [anEC release]; [localException raise]; } NS_ENDHANDLER; } else { // script anEnum = [statements objectEnumerator]; while ((aStatement = [anEnum nextObject])) [exprClass appendExpression: aStatement toScript: outputString]; } } if (sourceType) { // Dump if ([sourceType isEqualToString: @"plist"]) { if (sourceFilename) { dataDictionary = [[NSDictionary alloc] initWithContentsOfFile: sourceFilename]; NSCAssert1(dataDictionary != nil, @"Unable to read plist from file '%@'", sourceFilename); } #ifndef GNUSTEP else { #warning TODO: test this NSString *errorString = nil; NSData *readData = [[NSFileHandle fileHandleWithStandardInput] readDataToEndOfFile]; dataDictionary = [NSPropertyListSerialization propertyListFromData: readData mutabilityOption: NSPropertyListImmutable format: NULL errorDescription: &errorString]; if (errorString != nil) [NSException raise: NSInvalidArgumentException format: @"Unable to load plist from stdin: %@", errorString]; if (![dataDictionary isKindOfClass:[NSDictionary class]]) [NSException raise: NSInvalidArgumentException format: @"Invalid plist format from stdin: should be a dictionary"]; [dataDictionary retain]; } #endif } else { // database // We get data from database NSEnumerator *anEnum = [entitiesList objectEnumerator]; EOEntity *anEntity; EOEditingContext *aContext = [[EOEditingContext alloc] init]; dataDictionary = [[NSMutableDictionary alloc] init]; while ((anEntity = [anEnum nextObject])) { // Do we also fetch abstract entities?? if (![anEntity isAbstractEntity]) { NSArray *attributes = [anEntity attributes]; NSEnumerator *anAttrEnum = [attributes objectEnumerator]; EOAttribute *anAttribute; NSMutableArray *attributeNames = [NSMutableArray array]; NSArray *rawRows; NSMutableArray *rows; EOFetchSpecification *fetchSpec; NSDictionary *aRawRow; while ((anAttribute = [anAttrEnum nextObject])) { if (![anAttribute isFlattened] && ![anAttribute isDerived]) [attributeNames addObject: [anAttribute name]]; } fetchSpec = [EOFetchSpecification fetchSpecificationWithEntityName: [anEntity name] qualifier: nil sortOrderings: nil]; [fetchSpec setFetchesRawRows: YES]; rows = [NSMutableArray array]; rawRows = [aContext objectsWithFetchSpecification: fetchSpec]; anAttrEnum = [rawRows objectEnumerator]; while ((aRawRow = [anAttrEnum nextObject])) { NSMutableArray *values = [NSMutableArray array]; NSEnumerator *anEnum3 = [attributeNames objectEnumerator]; NSString *aName; while ((aName = [anEnum3 nextObject])) { id aValue = [aRawRow objectForKey: aName]; if ([aValue isEqual: [EONull null]]) [values addObject: @"__NULL__"]; else { if ([aValue isKindOfClass: [NSString class]]) { if (convertsNonASCII) [values addObject: [aValue lossyASCIIString]]; else [values addObject: aValue]; } /* else if ([aValue isKindOfClass:[NSCalendarDate class]]) [values addObject:[aValue description]]; else if ([aValue isKindOfClass:[NSData class]]) [values addObject:[aValue description]]; else if ([aValue isKindOfClass:[NSDecimalNumber class]]) [values addObject:[aValue stringValue]]; else if ([aValue isKindOfClass:[NSNumber class]]) [values addObject:[aValue stringValue]]; */ else [values addObject: aValue]; } } [rows addObject: values]; } [(NSMutableDictionary *)dataDictionary setObject: [NSDictionary dictionaryWithObjectsAndKeys: attributeNames, @"attributeNames", rows, @"rows", nil] forKey: [anEntity name]]; } } [aContext release]; } } // Else, no sourcefile => schemaDrop and/or schemaCreate if (postInstall) { // Look for a key "AdaptorName"PostInstallExpressions which // which have an array of SQL expressions postInstallSQLExpressions = [[srcModel userInfo] objectForKey:[[[EOAdaptor adaptorWithModel: srcModel] name] stringByAppendingString:@"PostInstallExpressions"]]; } if ([destType isEqualToString: @"plist"]) { // Write data as plist NSCAssert(dataDictionary != nil, @"No data to write in plist?!?"); if (destFilename) NSCAssert1([[dataDictionary description] writeToFile:destFilename atomically:NO], @"Unable to write plist to '%@'", destFilename); else [(NSFileHandle *)[NSFileHandle fileHandleWithStandardOutput] writeData:[[dataDictionary description] dataUsingEncoding:NSUTF8StringEncoding]]; } else if ([destType isEqualToString: @"script"]) { // Write schemaDrop and/or schemaCreate script, or data creation SQL script if (dataDictionary) { // Generate SQL to create data from dataDictionary NSEnumerator *entityNameEnum = [dataDictionary keyEnumerator]; NSString *anEntityName; #warning PROBLEM: we cannot create the statement telling to defer constraints! // In the script, we miss: // BEGIN; // SET CONSTRAINTS ALL DEFERRED; // ... // COMMIT while ((anEntityName = [entityNameEnum nextObject])) { EOEntity *anEntity = [[EOModelGroup defaultGroup] entityNamed: anEntityName]; NSEnumerator *aRowEnum = [[[dataDictionary objectForKey: anEntityName] objectForKey: @"rows"] objectEnumerator]; NSMutableDictionary *aDict = [NSMutableDictionary dictionary]; NSArray *aRow, *attrNames = [[dataDictionary objectForKey: anEntityName] objectForKey: @"attributeNames"]; while ((aRow = [aRowEnum nextObject])) { NSEnumerator *anAttributeEnum = [attrNames objectEnumerator]; NSEnumerator *aValueEnum = [aRow objectEnumerator]; NSString *anAttrName; while ((anAttrName = [anAttributeEnum nextObject])) { id aValue = [aValueEnum nextObject]; if (![aValue isKindOfClass: [NSString class]] || ![aValue isEqualToString: @"__NULL__"]) [aDict setObject: aValue forKey: anAttrName]; } [exprClass appendExpression: [exprClass insertStatementForRow: aDict entity: anEntity] toScript: outputString]; [aDict removeAllObjects]; } } } if (postInstallSQLExpressions) { [outputString appendString:[postInstallSQLExpressions componentsJoinedByString:@"\n"]]; } if (destFilename) { if (![outputString writeToFile: destFilename atomically: NO]) [NSException raise: NSInvalidArgumentException format: @"Unable to save file '%@'", destFilename]; } else [(NSFileHandle *)[NSFileHandle fileHandleWithStandardOutput] writeData: [outputString dataUsingEncoding: NSUTF8StringEncoding]]; } else { // database // Execute statements one after the other? if (dataDictionary) { NSEnumerator *entityEnum = [entitiesList objectEnumerator]; EOEntity *anEntity; EOEditingContext *anEC = [[EOEditingContext alloc] init]; while ((anEntity = [entityEnum nextObject])) { NSDictionary *aDictionary = [dataDictionary objectForKey: [anEntity name]]; NSArray *attributeNames = [aDictionary objectForKey: @"attributeNames"]; NSEnumerator *anEnum = [attributeNames objectEnumerator]; NSString *aName; NSMutableSet *newClassProperties = [NSMutableSet set]; EORelationship *aRelationship; if ([anEntity isReadOnly]) [anEntity setReadOnly: NO]; if (![[anEntity className] isEqualToString: @"EOGenericRecord"]) [anEntity setClassName: @"EOGenericRecord"]; // Let's use -[EOEntity attributesToFetch]? while ((aName = [anEnum nextObject])) { EOAttribute *anAttribute = [anEntity attributeNamed: aName]; NSCAssert1(anAttribute != nil, @"'%@' is not a valid attribute name", aName); #warning Remove also derived attributes if ([anAttribute isReadOnly]) [anAttribute setReadOnly: NO]; [newClassProperties addObject: anAttribute]; } NSCAssert1([anEntity setClassProperties:[newClassProperties allObjects]], @"Unable to set new class properties from %@", newClassProperties); anEnum = [[NSArray arrayWithArray: [anEntity relationships]] objectEnumerator]; while ((aRelationship = [anEnum nextObject])) { [anEntity removeRelationship: aRelationship]; } } entityEnum = [entitiesList objectEnumerator]; while ((anEntity = [entityEnum nextObject])) { NSDictionary *aDictionary = [dataDictionary objectForKey: [anEntity name]]; NSEnumerator *rowEnum = [[aDictionary objectForKey: @"rows"] objectEnumerator]; NSArray *aRow; NSArray *attributeNames = [aDictionary objectForKey: @"attributeNames"]; NSEnumerator *attrNameEnum = [attributeNames objectEnumerator]; NSString *anAttrName; while ((aRow = [rowEnum nextObject])) { EOGenericRecord *aRecord = [anEC createAndInsertInstanceOfEntityNamed: [anEntity name]]; int i = 0; attrNameEnum = [attributeNames objectEnumerator]; while ((anAttrName = [attrNameEnum nextObject])) { NSString *aStringValue = [aRow objectAtIndex: i++]; id aValue = nil; if ([aStringValue isEqualToString: @"__NULL__"]) aValue = [EONull null]; else { EOAttribute *anAttribute = [anEntity attributeNamed: anAttrName]; NSString *internalType = [anAttribute valueClassName]; if ([internalType isEqualToString: @"NSString"]) { aValue = aStringValue; } else if ([internalType isEqualToString: @"NSNumber"]) { #if 0 if (![anAttribute valueType] || [@"cil" rangeOfString: [anAttribute valueType]].length != 0) aValue = [NSNumber numberWithInt: [aStringValue intValue]]; else if([@"fd" rangeOfString: [anAttribute valueType]].length != 0) aValue = [NSNumber numberWithDouble: [aStringValue doubleValue]]; else [NSException raise:NSInvalidArgumentException format:@"Unknown valueType '%@' for attribute %@", [anAttribute valueType], anAttribute]; #else aValue = [NSNumber numberWithDouble: [aStringValue doubleValue]]; #endif } else if ([internalType isEqualToString: @"NSDecimalNumber"]) { aValue = [NSDecimalNumber decimalNumberWithString: aStringValue]; NSCAssert1(aValue != nil, @"Invalid decimal number representation: %@", aStringValue); } else if ([internalType isEqualToString:@"NSCalendarDate"]) { // Format is Apple's eoutil format aValue = [[NSCalendarDate alloc] initWithString: aStringValue calendarFormat: @"%b %d %Y %H:%M" /*locale:*/]; if(!aValue) // Format is ISO's aValue = [[NSCalendarDate alloc] initWithString: aStringValue calendarFormat: @"%Y-%m-%d %H:%M:%S %z" /*locale:*/]; if(!aValue) // Format is ISO's + milliseconds aValue = [[NSCalendarDate alloc] initWithString:aStringValue calendarFormat:@"%Y-%m-%d %H:%M:%S.%F %z" /*locale:*/]; NSCAssert1(aValue != nil, @"Unable to parse date from '%@'", aStringValue); } else if ([internalType isEqualToString: @"NSData"]) { if ([aStringValue isEqualToString: @""]) aValue = [NSData data]; else { aValue = [aStringValue propertyList]; NSCAssert1([aValue isKindOfClass: [NSData class]], @"Invalid data: %@", aStringValue); } } else [NSException raise:NSInvalidArgumentException format: @"Unknown attribute (%@) valueClassName: %@", anAttribute, internalType]; } [aRecord takeStoredValue: aValue forKey: anAttrName]; } } } NS_DURING { [anEC saveChanges]; [anEC release]; } NS_HANDLER { [anEC release]; NSLog(@"%@", localException); } NS_ENDHANDLER; } if (postInstallSQLExpressions) { EODatabaseContext *databaseContext; EOEditingContext *anEC = [[EOEditingContext alloc] init]; databaseContext = [EODatabaseContext registeredDatabaseContextForModel: srcModel editingContext: anEC]; [databaseContext lock]; NS_DURING { EODatabaseChannel *databaseChannel = [databaseContext availableChannel]; EOAdaptorChannel *adaptorChannel = [databaseChannel adaptorChannel]; NSString *aString; NSEnumerator *anEnum; if (![adaptorChannel isOpen]) [adaptorChannel openChannel]; anEnum = [postInstallSQLExpressions objectEnumerator]; while ((aString = [anEnum nextObject])) { NS_DURING EOSQLExpression *aStatement = [exprClass expressionForString:aString]; [adaptorChannel evaluateExpression: aStatement]; NS_HANDLER if(!force) break; NS_ENDHANDLER; } [databaseContext unlock]; [anEC release]; } NS_HANDLER { [databaseContext unlock]; [anEC release]; [localException raise]; } NS_ENDHANDLER; } } return YES; } static BOOL convert(NSArray *arguments) { EOModel *aModel; EOAdaptor *newAdaptor; NSDictionary *newConnectionDictionary; NSString *newFilename; if ([arguments count] < 4) return NO; aModel = [EOModel modelWithContentsOfFile: [[arguments objectAtIndex: 0] stringByStandardizingPath]]; newAdaptor = [EOAdaptor adaptorWithName: [arguments objectAtIndex: 1]]; newConnectionDictionary = [[arguments objectAtIndex: 2] propertyList]; newFilename = [[arguments objectAtIndex: 3] stringByStandardizingPath]; [aModel setAdaptorName: [newAdaptor name]]; [aModel setConnectionDictionary: newConnectionDictionary]; [[newAdaptor class] assignExternalInfoForEntireModel: aModel]; [aModel writeToFile: newFilename]; NSLog(@"Converted %@ to %@", [arguments objectAtIndex: 0], newFilename); return YES; } static BOOL dbConnect(NSArray *arguments) { EOAdaptor *anAdaptor = nil; NSDictionary *aConnectionDictionary = nil; EOModel *aModel; switch ([arguments count]) { case 1: aModel = [EOModel modelWithContentsOfFile: [[arguments objectAtIndex: 0] stringByStandardizingPath]]; anAdaptor = [EOAdaptor adaptorWithModel: aModel]; aConnectionDictionary = [aModel connectionDictionary]; default: if (anAdaptor == nil) { anAdaptor = [EOAdaptor adaptorWithName: [arguments objectAtIndex: 0]]; aConnectionDictionary = [[arguments objectAtIndex: 1] propertyList]; [anAdaptor setConnectionDictionary: aConnectionDictionary]; } [anAdaptor assertConnectionDictionaryIsValid]; } return YES; } int main(int arcg, char *argv[], char **envp) { NSAutoreleasePool *localAP = [[NSAutoreleasePool alloc] init]; BOOL noUsage = NO; NSArray *arguments = [[NSProcessInfo processInfo] arguments]; int argc = [arguments count]; BOOL hasError = NO; if (argc >= 2) { NSString *command = [arguments objectAtIndex: 1]; if (argc > 2) arguments = [arguments subarrayWithRange: NSMakeRange(2, argc - 2)]; else arguments = [NSArray array]; NS_DURING { if ([command isEqualToString: @"dump"]) noUsage = dump(arguments); else if ([command isEqualToString: @"convert"]) noUsage = convert(arguments); else if ([command isEqualToString: @"connect"]) noUsage = dbConnect(arguments); else if ([command isEqualToString: @"help"]) noUsage = usage(YES); } NS_HANDLER { noUsage = ![[localException name] isEqualToString: NSInvalidArgumentException]; hasError = YES; NSLog(@"%@", localException); } NS_ENDHANDLER; } if (!noUsage) (void)usage(NO); [localAP release]; return hasError; }