@import AppKit; @import CoreFoundation; @import QuickLook; #include #include #import "NSStringAdditions.h" QL_EXTERN_C_BEGIN OSStatus GeneratePreviewForURL(void* thisInterface, QLPreviewRequestRef preview, CFURLRef url, CFStringRef contentTypeUTI, CFDictionaryRef options); void CancelPreviewGeneration(void* thisInterface, QLPreviewRequestRef preview); QL_EXTERN_C_END 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[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[iconFileName] = imgProps; } 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 = @(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 localizedStringWithFormat:NSLocalizedStringFromTableInBundle(@"%lu files", nil, bundle, "quicklook file count"), 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) ? @(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 = @(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 localizedStringWithFormat:NSLocalizedStringFromTableInBundle(@"%lu Web Seeds", nil, bundle, "quicklook web seed header"), 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 localizedStringWithFormat:NSLocalizedStringFromTableInBundle(@"%lu Trackers", nil, bundle, "quicklook tracker header"), 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 localizedStringWithFormat:NSLocalizedStringFromTableInBundle(@"%lu Files", nil, bundle, "quicklook file header"), n_files]; [listSection appendFormat:@"", fileTitleString]; #warning display folders? for (auto const& [path, size] : metainfo.files().sortedByPath()) { NSString* fullFilePath = @(path.c_str()); NSCAssert([fullFilePath hasPrefix:[name stringByAppendingString:@"/"]], @"Expected file path %@ to begin with %@/", fullFilePath, name); NSString* shortenedFilePath = [fullFilePath substringFromIndex:name.length + 1]; NSString* shortenedFilePathAndSize = [NSString stringWithFormat:@"%@ - %@", shortenedFilePath, [NSString stringForFileSize:size]]; NSUInteger const width = 16; [listSection appendFormat:@"", generateIconData(shortenedFilePath.pathExtension, width, allImgProps), width, width, shortenedFilePathAndSize]; } [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 }