transmission/macosx/QuickLookPlugin/GeneratePreviewForURL.mm

242 lines
9.6 KiB
Plaintext
Raw Normal View History

#import <QuickLook/QuickLook.h>
#include <libtransmission/transmission.h>
#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:NSCompositeCopy 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:@"<html><style type=\"text/css\">%@</style><body>", 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:@"<h2><img class=\"icon\" src=\"%@\" width=\"%ld\" height=\"%ld\" />%@</h2>",
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:@"<p>%@</p>", 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:@"<p>%@</p>", creationString];
}
if (inf.comment)
{
NSString* comment = [NSString stringWithUTF8String:inf.comment];
if (![comment isEqualToString:@""])
[htmlString appendFormat:@"<p>%@</p>", comment];
}
NSMutableArray* lists = [NSMutableArray array];
if (inf.webseedCount > 0)
{
NSMutableString* listSection = [NSMutableString string];
[listSection appendString:@"<table>"];
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:@"<tr><th>%@</th></tr>", headerTitleString];
for (int i = 0; i < inf.webseedCount; ++i)
{
[listSection appendFormat:@"<tr><td>%s<td></tr>", inf.webseeds[i]];
}
[listSection appendString:@"</table>"];
[lists addObject:listSection];
}
if (inf.trackerCount > 0)
{
NSMutableString* listSection = [NSMutableString string];
[listSection appendString:@"<table>"];
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:@"<tr><th>%@</th></tr>", headerTitleString];
#warning handle tiers?
for (int i = 0; i < inf.trackerCount; ++i)
{
[listSection appendFormat:@"<tr><td>%s<td></tr>", inf.trackers[i].announce];
}
[listSection appendString:@"</table>"];
[lists addObject:listSection];
}
if (inf.isFolder)
{
NSMutableString* listSection = [NSMutableString string];
[listSection appendString:@"<table>"];
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:@"<tr><th>%@</th></tr>", 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:@"<tr><td><img class=\"icon\" src=\"%@\" width=\"%ld\" height=\"%ld\" />%@<td></tr>",
generateIconData([shortenedFilePath pathExtension], width, allImgProps),
width,
width,
shortenedFilePath];
}
[listSection appendString:@"</table>"];
[lists addObject:listSection];
}
if ([lists count] > 0)
{
[htmlString appendFormat:@"<hr/><br>%@", [lists componentsJoinedByString:@"<br>"]];
}
[htmlString appendString:@"</body></html>"];
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
}