William Bowling is sharing code with you
Bitbucket is a code hosting site. Unlimited public and private repositories. Free for small teams.
Don't show this again1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 | /*
* Adium is the legal property of its developers, whose names are listed in the copyright file included
* with this source distribution.
*
* 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.
*/
#import "BGICLogImportController.h"
#import "AIXMLAppender.h"
#import "AILoggerPlugin.h"
#import "AICoreComponentLoader.h"
#import <Adium/AIXMLElement.h>
#import <Adium/AIHTMLDecoder.h>
#import <Adium/AILoginControllerProtocol.h>
#import <AIUtilities/NSCalendarDate+ISO8601Unparsing.h>
#import <AIUtilities/AIFileManagerAdditions.h>
#import <AIUtilities/AIStringAdditions.h>
// InstantMessage and other iChat transcript classes are from Spiny Software's Logorrhea, used with permission.
#import "InstantMessage.h"
#import "Presentity.h"
//#define LOG_TO_TEST
#define TEST_LOGGING_LOCATION [@"~/Desktop/testLog" stringByExpandingTildeInPath]
@implementation BGICLogImportController
/* an iChat log importer needs to be set up with a destination pair, specified as @"service.account" to match current representation(s) in PATH_LOGS */
-(id)initWithDestination:(NSString *)newSAPair
{
self = [super init];
if(self)
importServiceAccountPair = [newSAPair copy];
return self;
}
-(void)dealloc
{
[importServiceAccountPair release];
[super dealloc];
}
-(BOOL)createNewLogForPath:(NSString *)fullPath
{
AIHTMLDecoder *xhtmlDecoder = [[AIHTMLDecoder alloc] initWithHeaders:NO
fontTags:YES
closeFontTags:YES
colorTags:YES
styleTags:YES
encodeNonASCII:YES
encodeSpaces:NO
attachmentsAsText:YES
onlyIncludeOutgoingImages:NO
simpleTagsOnly:NO
bodyBackground:NO
allowJavascriptURLs:YES];
[xhtmlDecoder setGeneratesStrictXHTML:YES];
[xhtmlDecoder setUsesAttachmentTextEquivalents:NO];
// read the raw file into an array for working against, two different formats have been employed by iChat, based on available classes
fullPath = [[NSFileManager defaultManager] pathByResolvingAlias:fullPath];
NSArray *rawChat;
@try
{
rawChat = ([[fullPath pathExtension] isEqual:@"ichat"] ?
[NSKeyedUnarchiver unarchiveObjectWithFile:fullPath] :
[NSUnarchiver unarchiveObjectWithFile:fullPath]);
}
@catch (NSException *releaseException)
{
NSLog(@"Could not open iChat log at %@: %@", fullPath, releaseException);
rawChat = nil;
}
if (!rawChat) return NO;
NSString *preceedingPath = nil;
#ifndef LOG_TO_TEST
preceedingPath = [[[adium.loginController userDirectory] stringByAppendingPathComponent:PATH_LOGS] stringByExpandingTildeInPath];
#else
preceedingPath = TEST_LOGGING_LOCATION;
#endif
/*
[rawChat objectAtIndex:0],
TEST_ACCOUNT, // this has to be matched somehow, I can't see a way offhand through as iChat stores the originating party
*/
NSString *parentPath = [preceedingPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@/%@",
importServiceAccountPair, // see commented code above for a brittle methodology
[[[rawChat objectAtIndex:3] objectAtIndex:0] senderID] // this is improperly brittle and imprecise
]];
// create a new xml parser for logs
NSString *documentPath = [parentPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@ (%@).chatlog",
[[[rawChat objectAtIndex:3] objectAtIndex:0] senderID],
[[[[rawChat objectAtIndex:2] objectAtIndex:0] date] dateWithCalendarFormat:@"%Y-%m-%dT%H.%M.%S%z"
timeZone:nil]]];
AIXMLElement *rootElement = [[[AIXMLElement alloc] initWithName:@"chat"] autorelease];
[rootElement setAttributeNames:[NSArray arrayWithObjects:@"xmlns", @"account", @"service", nil]
values:[NSArray arrayWithObjects:XML_LOGGING_NAMESPACE, [[[rawChat objectAtIndex:3] objectAtIndex:0] senderID], [rawChat objectAtIndex:0], nil]];
AIXMLAppender *appender = [AIXMLAppender documentWithPath:documentPath rootElement:rootElement];
NSString *imagesPath = [appender.path stringByDeletingLastPathComponent];
// sequentially add the messages from the iChat transcript sans attributed text features
for(NSInteger i = 0; i < [[rawChat objectAtIndex:2] count]; i++)
{
NSMutableArray *attributeKeys = [NSMutableArray arrayWithObjects:@"sender", @"time", nil];
NSMutableArray *attributeValues = [NSMutableArray arrayWithObjects:
([[(InstantMessage *)[[rawChat objectAtIndex:2] objectAtIndex:i] sender] senderID] != nil ? [[(InstantMessage *)[[rawChat objectAtIndex:2] objectAtIndex:i] sender] senderID] : @""),
[[[(InstantMessage *)[[rawChat objectAtIndex:2] objectAtIndex:i] date] dateWithCalendarFormat:nil timeZone:nil] ISO8601DateString],
nil];
NSMutableString *chatContents = [[[xhtmlDecoder encodeHTML:[[[rawChat objectAtIndex:2] objectAtIndex:i] text] imagesPath:imagesPath] mutableCopy] autorelease];
NSString *elementName = ![[[(InstantMessage *)[[rawChat objectAtIndex:2] objectAtIndex:i] sender] senderID] isEqual:@""] ? @"message" : @"event";
AIXMLElement *elm = [[[AIXMLElement alloc] initWithName:elementName] autorelease];
[elm addEscapedObject:chatContents];
if ([attributeValues count] == 2) {
[elm setAttributeNames:attributeKeys
values:attributeValues];
}
[appender appendElement:elm];
}
if ([[NSFileManager defaultManager] fileExistsAtPath:documentPath]) {
[(AILoggerPlugin *)[[adium componentLoader] pluginWithClassName:@"AILoggerPlugin"] markLogDirtyAtPath:documentPath];
return YES;
} else {
return NO;
}
}
@end
|