#import #import #include #include #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 auto metainfo = tr_torrent_metainfo{}; if (!metainfo.parseTorrentFile([[(__bridge NSURL*)url path] UTF8String])) { 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:metainfo.name().c_str()]; auto const n_files = metainfo.fileCount(); auto const is_multifile = n_files > 1; NSString* fileTypeString = is_multifile ? NSFileTypeForHFSTypeCode(kGenericFolderIcon) : [name pathExtension]; NSUInteger const width = 32; [htmlString appendFormat:@"

%@

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

", fileSizeString]; auto const date_created = metainfo.dateCreated(); NSString* dateCreatedString = date_created > 0 ? [NSDateFormatter localizedStringFromDate:[NSDate dateWithTimeIntervalSince1970:date_created] dateStyle:NSDateFormatterLongStyle timeStyle:NSDateFormatterShortStyle] : nil; auto const& creator = metainfo.creator(); NSString* creatorString = !std::empty(creator) ? [NSString stringWithUTF8String:creator.c_str()] : 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]; } auto const& commentStr = metainfo.comment(); if (!std::empty(commentStr)) { NSString* comment = [NSString stringWithUTF8String:commentStr.c_str()]; if (![comment isEqualToString:@""]) [htmlString appendFormat:@"

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

%@
%s

"]; [lists addObject:listSection]; } auto const& announce_list = metainfo.announceList(); if (!std::empty(announce_list)) { NSMutableString* listSection = [NSMutableString string]; [listSection appendString:@""]; auto const n = std::size(announce_list); NSString* headerTitleString = n == 1 ? NSLocalizedStringFromTableInBundle(@"1 Tracker", nil, bundle, "quicklook tracker header") : [NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"%@ Trackers", nil, bundle, "quicklook tracker header"), [NSString formattedUInteger:n]]; [listSection appendFormat:@"", headerTitleString]; #warning handle tiers? for (auto const& tracker : announce_list) { [listSection appendFormat:@"", tracker.announce.c_str()]; } [listSection appendString:@"

%@
%s

"]; [lists addObject:listSection]; } if (is_multifile) { NSMutableString* listSection = [NSMutableString string]; [listSection appendString:@""]; NSString* fileTitleString = [NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"%@ Files", nil, bundle, "quicklook file header"), [NSString formattedUInteger:n_files]]; [listSection appendFormat:@"", fileTitleString]; #warning display size? #warning display folders? for (tr_file_index_t i = 0; i < n_files; ++i) { NSString* fullFilePath = [NSString stringWithUTF8String:metainfo.fileSubpath(i).c_str()]; 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:@""]; 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 }