@import AppKit; @import CoreFoundation; @import QuickLook; #include #include #include #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 static NSUInteger const kIconWidth = 16; namespace { class FileTreeNode { public: FileTreeNode() = default; ~FileTreeNode() = default; auto MaybeCreateChild(std::string_view child_name) { return children_.try_emplace(std::string{ child_name }); } private: FileTreeNode(FileTreeNode const&) = delete; FileTreeNode& operator=(FileTreeNode&) = delete; std::unordered_map children_; }; } // namespace 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.parse_torrent_file(((__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.file_count(); 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.total_size()]; 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.date_created(); 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.webseed_count(); 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.announce_list(); 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]; // TODO: 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]; FileTreeNode root{}; for (auto const& [path, size] : metainfo.files().sorted_by_path()) { FileTreeNode* curNode = &root; size_t level = 0; auto subpath = std::string_view{ path }; auto path_vec = std::vector{}; auto token = std::string_view{}; while (tr_strv_sep(&subpath, &token, '/')) { path_vec.emplace_back(token); } size_t const last = path_vec.size() - 1; for (auto const& part : path_vec) { auto [it, inserted] = curNode->MaybeCreateChild(part); if (inserted) { NSString* prefix = @""; for (size_t i = 0; i < level; ++i) { prefix = [prefix stringByAppendingString:@" "]; } NSString* pathPart = @(it->first.c_str()); NSString* pathExt = nil; NSString* fileSize = nil; if (level < last) { // This node is a directory. pathExt = NSFileTypeForHFSTypeCode(kGenericFolderIcon); fileSize = @""; } else { // This node is a leaf file. pathExt = pathPart.pathExtension; fileSize = [NSString stringForFileSize:size]; } [listSection appendFormat:@"", prefix, generateIconData(pathExt, kIconWidth, allImgProps), kIconWidth, kIconWidth, pathPart, fileSize]; } curNode = &it->second; level++; } } [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 }