#import #include #import "NSStringAdditions.h" OSStatus GeneratePreviewForURL(void* thisInterface, QLPreviewRequestRef preview, CFURLRef url, CFStringRef contentTypeUTI, CFDictionaryRef options); void CancelPreviewGeneration(void* thisInterface, QLPreviewRequestRef preview); NSString* generateIconData(NSString* fileExtension, NSUInteger width, NSMutableDictionary* allImgProps) { NSString* rawFilename = ![fileExtension isEqualToString:@""] ? fileExtension : @"blank_file_name_transmission"; // we need to do this once per file extension, per size NSString* iconFileName = [NSString stringWithFormat:@"%ldx%@.tiff", width, rawFilename]; if (![allImgProps objectForKey:iconFileName]) { NSImage* icon = [[NSWorkspace sharedWorkspace] iconForFileType:fileExtension]; NSRect const iconFrame = NSMakeRect(0.0, 0.0, width, width); NSImage* renderedIcon = [[NSImage alloc] initWithSize:iconFrame.size]; [renderedIcon lockFocus]; [icon drawInRect:iconFrame fromRect:NSZeroRect operation:NSCompositingOperationCopy fraction:1.0]; [renderedIcon unlockFocus]; NSData* iconData = [renderedIcon TIFFRepresentation]; NSDictionary* imgProps = @{ (NSString*)kQLPreviewPropertyMIMETypeKey : @"image/png", (NSString*)kQLPreviewPropertyAttachmentDataKey : iconData }; [allImgProps setObject:imgProps forKey:iconFileName]; } return [@"cid:" stringByAppendingString:iconFileName]; } OSStatus GeneratePreviewForURL(void* thisInterface, QLPreviewRequestRef preview, CFURLRef url, CFStringRef contentTypeUTI, CFDictionaryRef options) { // Before proceeding make sure the user didn't cancel the request if (QLPreviewRequestIsCancelled(preview)) { return noErr; } //we need this call to ensure NSApp is initialized (not done automatically for plugins) [NSApplication sharedApplication]; //try to parse the torrent file tr_info inf; tr_ctor* ctor = tr_ctorNew(NULL); tr_ctorSetMetainfoFromFile(ctor, [[(__bridge NSURL*)url path] UTF8String]); int const err = tr_torrentParse(ctor, &inf); tr_ctorFree(ctor); if (err) { return noErr; } NSBundle* bundle = [NSBundle bundleWithIdentifier:@"org.m0k.transmission.QuickLookPlugin"]; NSURL* styleURL = [bundle URLForResource:@"style" withExtension:@"css"]; NSString* styleContents = [NSString stringWithContentsOfURL:styleURL encoding:NSUTF8StringEncoding error:NULL]; NSMutableString* htmlString = [NSMutableString string]; [htmlString appendFormat:@"", styleContents]; NSMutableDictionary* allImgProps = [NSMutableDictionary dictionary]; NSString* name = [NSString stringWithUTF8String:inf.name]; NSString* fileTypeString = inf.isFolder ? NSFileTypeForHFSTypeCode(kGenericFolderIcon) : [name pathExtension]; NSUInteger const width = 32; [htmlString appendFormat:@"

%@

", generateIconData(fileTypeString, width, allImgProps), width, width, name]; NSString* fileSizeString = [NSString stringForFileSize:inf.totalSize]; if (inf.isFolder) { NSString* fileCountString; if (inf.fileCount == 1) { fileCountString = NSLocalizedStringFromTableInBundle(@"1 file", nil, bundle, "quicklook file count"); } else { fileCountString = [NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"%@ files", nil, bundle, "quicklook file count"), [NSString formattedUInteger:inf.fileCount]]; } fileSizeString = [NSString stringWithFormat:@"%@, %@", fileCountString, fileSizeString]; } [htmlString appendFormat:@"

", fileSizeString]; NSString* dateCreatedString = inf.dateCreated > 0 ? [NSDateFormatter localizedStringFromDate:[NSDate dateWithTimeIntervalSince1970:inf.dateCreated] dateStyle:NSDateFormatterLongStyle timeStyle:NSDateFormatterShortStyle] : nil; NSString* creatorString = inf.creator ? [NSString stringWithUTF8String:inf.creator] : nil; if ([creatorString isEqualToString:@""]) { creatorString = nil; } NSString* creationString = nil; if (dateCreatedString && creatorString) { creationString = [NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"Created on %@ with %@", nil, bundle, "quicklook creation info"), dateCreatedString, creatorString]; } else if (dateCreatedString) { creationString = [NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"Created on %@", nil, bundle, "quicklook creation info"), dateCreatedString]; } else if (creatorString) { creationString = [NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"Created with %@", nil, bundle, "quicklook creation info"), creatorString]; } if (creationString) { [htmlString appendFormat:@"

", creationString]; } if (inf.comment) { NSString* comment = [NSString stringWithUTF8String:inf.comment]; if (![comment isEqualToString:@""]) [htmlString appendFormat:@"

", comment]; } NSMutableArray* lists = [NSMutableArray array]; if (inf.webseedCount > 0) { NSMutableString* listSection = [NSMutableString string]; [listSection appendString:@""]; NSString* headerTitleString = inf.webseedCount == 1 ? NSLocalizedStringFromTableInBundle(@"1 Web Seed", nil, bundle, "quicklook web seed header") : [NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"%@ Web Seeds", nil, bundle, "quicklook web seed header"), [NSString formattedUInteger:inf.webseedCount]]; [listSection appendFormat:@"", headerTitleString]; for (int i = 0; i < inf.webseedCount; ++i) { [listSection appendFormat:@"", inf.webseeds[i]]; } [listSection appendString:@"

%@
%s

"]; [lists addObject:listSection]; } if (inf.trackerCount > 0) { NSMutableString* listSection = [NSMutableString string]; [listSection appendString:@""]; NSString* headerTitleString = inf.trackerCount == 1 ? NSLocalizedStringFromTableInBundle(@"1 Tracker", nil, bundle, "quicklook tracker header") : [NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"%@ Trackers", nil, bundle, "quicklook tracker header"), [NSString formattedUInteger:inf.trackerCount]]; [listSection appendFormat:@"", headerTitleString]; #warning handle tiers? for (int i = 0; i < inf.trackerCount; ++i) { [listSection appendFormat:@"", inf.trackers[i].announce]; } [listSection appendString:@"

%@
%s

"]; [lists addObject:listSection]; } if (inf.isFolder) { NSMutableString* listSection = [NSMutableString string]; [listSection appendString:@""]; NSString* fileTitleString = inf.fileCount == 1 ? NSLocalizedStringFromTableInBundle(@"1 File", nil, bundle, "quicklook file header") : [NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"%@ Files", nil, bundle, "quicklook file header"), [NSString formattedUInteger:inf.fileCount]]; [listSection appendFormat:@"", fileTitleString]; #warning display size? #warning display folders? for (int i = 0; i < inf.fileCount; ++i) { NSString* fullFilePath = [NSString stringWithUTF8String:inf.files[i].name]; NSCAssert([fullFilePath hasPrefix:[name stringByAppendingString:@"/"]], @"Expected file path %@ to begin with %@/", fullFilePath, name); NSString* shortenedFilePath = [fullFilePath substringFromIndex:[name length] + 1]; NSUInteger const width = 16; [listSection appendFormat:@"", generateIconData([shortenedFilePath pathExtension], width, allImgProps), width, width, shortenedFilePath]; } [listSection appendString:@"

%@
%@

"]; [lists addObject:listSection]; } if ([lists count] > 0) { [htmlString appendFormat:@"

%@", [lists componentsJoinedByString:@"
"]]; } [htmlString appendString:@""]; tr_metainfoFree(&inf); NSDictionary* props = @{ (NSString*)kQLPreviewPropertyTextEncodingNameKey : @"UTF-8", (NSString*)kQLPreviewPropertyMIMETypeKey : @"text/html", (NSString*)kQLPreviewPropertyAttachmentsKey : allImgProps }; QLPreviewRequestSetDataRepresentation( preview, (__bridge CFDataRef)[htmlString dataUsingEncoding:NSUTF8StringEncoding], kUTTypeHTML, (__bridge CFDictionaryRef)props); return noErr; } void CancelPreviewGeneration(void* thisInterface, QLPreviewRequestRef preview) { // Implement only if supported }