-
-
Notifications
You must be signed in to change notification settings - Fork 57
/
ArchiveDropView.m
206 lines (171 loc) · 6.11 KB
/
ArchiveDropView.m
1
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
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
//
// Apple Public Source License
// http://www.opensource.apple.com/apsl/
//
// Created by Vitaly Davidenko on 12/10/06.
// Copyright 2006 Vitaly Davidenko.
//
// Updated and refactored by Rob Rohan on 2007-09-18
#import "ArchiveDropView.h"
#import "Extractor.h"
#import "OutputType.h"
static void logMessage(NSTextView* log, NSColor* color, NSString* message)
{
[log setEditable:YES];
NSMutableDictionary* dict = [NSMutableDictionary dictionaryWithDictionary: [log typingAttributes]];
// [dict setValue:color forKey:NSForegroundColorAttributeName];
[log setTypingAttributes:dict];
[log insertText: message replacementRange: [log selectedRange]];
[log insertText: @"\n" replacementRange: [log selectedRange]];
[log setEditable:NO];
[log displayIfNeeded];
}
@implementation ArchiveDropView
- (id)initWithFrame:(NSRect)frameRect
{
if ((self = [super initWithFrame:frameRect]) != nil) {
// Add initialization code here
[self registerForDraggedTypes:[NSArray arrayWithObjects: NSFilenamesPboardType, nil]];
//set the drop target image
NSImage *newImage = [[NSImage alloc] initByReferencingFile:[[NSBundle mainBundle] pathForImageResource: @"extract_archive.png"]];
[self setImage:newImage];
}
return self;
}
- (void)drawRect:(NSRect)rect
{
NSRect ourBounds = [self bounds];
NSImage *image = [self image];
[super drawRect:rect];
[image drawAtPoint:(ourBounds.origin) fromRect:rect operation:NSCompositingOperationSourceOver fraction:1];
}
- (void)setImage:(NSImage *)newImage
{
_dropImage = newImage;
}
- (NSImage *)image
{
return _dropImage;
}
////////////////////////////////////////////////////////////////
- (void)logError:(NSString*) message
{
logMessage(logOutput, [NSColor redColor], message);
}
- (void)logWarning:(NSString*) message
{
logMessage(logOutput, [NSColor orangeColor], message);
}
- (void)logInfo:(NSString*) message
{
logMessage(logOutput, [NSColor blueColor], message);
}
- (void)logResult:(NSString*) message
{
logMessage(logOutput, [NSColor darkGrayColor], message);
}
- (BOOL)performDragOperation:(id <NSDraggingInfo>)sender
{
[logOutput selectAll:self];
[logOutput insertText:@"" replacementRange:logOutput.selectedRange];
NSPasteboard *pboard = [sender draggingPasteboard];
///////////////////////////////////
// This probably shouldn't be here
//get the user defined index name
NSString * indexFileName = [[userDefaults values] valueForKey:@"WAEIndexName"];
if (indexFileName == nil || [indexFileName length] == 0) {
indexFileName = @"index.html";
}
//get the user selected output type
//HACK alert. I need to figure out a better way to do this. I thought the User
//types from the select box would get an object, but it only returns a string :-/
NSString * outputType = [[userDefaults values] valueForKey:@"WAEOutputType"];
NSXMLDocumentContentKind type = NSXMLDocumentXHTMLKind;
if ( [outputType isEqualToString:@"HTML"] ) {
type = NSXMLDocumentHTMLKind;
} else if ( [outputType isEqualToString:@"XML"] ) {
type = NSXMLDocumentXMLKind;
} else if ( [outputType isEqualToString:@"XHTML"] ) {
type = NSXMLDocumentXHTMLKind;
} else if ( [outputType isEqualToString:@"Text"] ) {
type = NSXMLDocumentTextKind;
}
NSString * URLPrepend = [[userDefaults values] valueForKey:@"WAEURLOffset"];
if (URLPrepend == nil || [URLPrepend length] == 0) {
URLPrepend = @"";
}
///////////////////////////////////
if ( [[pboard types] containsObject:NSFilenamesPboardType] ) {
NSArray *files = [pboard propertyListForType:NSFilenamesPboardType];
NSUInteger numberOfFiles = [files count];
//NSLog(@"%i\n", numberOfFiles);
NSUInteger i;
for (i=0; i<numberOfFiles; i++)
{
NSString* fileName = [files objectAtIndex:i];
[self logInfo:[NSString stringWithFormat: NSLocalizedStringFromTable(@"processing", @"InfoPlist", @"processing file: 1 name"), fileName] ];
if ([fileName hasSuffix:@"webarchive"])
{
NSFileManager * fm = [NSFileManager defaultManager];
NSString * dirPath = [fileName stringByDeletingLastPathComponent];
if ([fm isWritableFileAtPath:dirPath])
{
NSString * archiveName = [[fileName lastPathComponent] stringByDeletingPathExtension];
NSString * outputPath = [dirPath stringByAppendingPathComponent: archiveName];
NSUInteger i = 0;
while([fm fileExistsAtPath:outputPath])
{
[self logWarning:[NSString stringWithFormat: NSLocalizedStringFromTable(@"folder exists", @"InfoPlist", @"folder already exists: 1 name"), outputPath] ];
NSString * dirName = [archiveName stringByAppendingString:@"-%tu"];
outputPath = [dirPath stringByAppendingPathComponent: [NSString stringWithFormat: dirName, i++]];
}
Extractor * extr = [[Extractor alloc] init];
[extr loadWebArchive: fileName];
[extr setEntryFileName: indexFileName];
[extr setContentKind: type];
[extr setURLPrepend: URLPrepend];
NSString * mainResourcePath = [extr extractResources: outputPath];
[self logResult:[NSString stringWithFormat: NSLocalizedStringFromTable(@"extract success", @"InfoPlist", @"extract success 1=folder name 2=main file"), outputPath, mainResourcePath]];
}
}
else
{
[self logError: NSLocalizedStringFromTable(@"not archive", @"InfoPlist", @"")];
}
}
}
return YES;
}
- (NSDragOperation)draggingEntered:(id <NSDraggingInfo>)sender
{
NSPasteboard *pboard;
// NSDragOperation sourceDragMask;
// sourceDragMask = [sender draggingSourceOperationMask];
pboard = [sender draggingPasteboard];
if ( [[pboard types] containsObject:NSFilenamesPboardType] ) {
/*
if (sourceDragMask & NSDragOperationLink) {
return NSDragOperationLink;
} else if (sourceDragMask & NSDragOperationCopy) {
return NSDragOperationCopy;
}
*/
return NSDragOperationCopy;
}
return NSDragOperationNone;
}
- (void)draggingExited:(id <NSDraggingInfo>)sender
{
}
- (void)draggingEnded:(id <NSDraggingInfo>)sender
{
}
- (BOOL)prepareForDragOperation:(id <NSDraggingInfo>)sender
{
return YES;
}
- (void)concludeDragOperation:(id <NSDraggingInfo>)sender
{
//[self setNeedsDisplay:YES];
}
@end