// This file Copyright © Transmission authors and contributors. // It may be used under the MIT (SPDX: MIT) license. // License text can be found in the licenses/ folder. #import "PreviewProvider.h" #include #include #include #include #include #import "NSStringAdditions.h" 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(UTType* type, NSUInteger width, NSMutableDictionary* allImgProps) { // We need to do this once per file type, per image size NSString* iconFileName = [NSString stringWithFormat:@"%@.%ld.tiff", type.identifier, width]; if (!allImgProps[iconFileName]) { NSImage* icon = [NSWorkspace.sharedWorkspace iconForContentType:type]; NSRect const iconFrame = NSMakeRect(0.0, 0.0, width, width); NSImage* renderedIcon = [[NSImage alloc] initWithSize:iconFrame.size]; [renderedIcon lockFocus]; [icon drawInRect:iconFrame]; [renderedIcon unlockFocus]; NSData* iconData = renderedIcon.TIFFRepresentation; QLPreviewReplyAttachment* imageAttachment = [[QLPreviewReplyAttachment alloc] initWithData:iconData contentType:UTTypePNG]; allImgProps[iconFileName] = imageAttachment; } return [@"cid:" stringByAppendingString:iconFileName]; } @implementation PreviewProvider - (void)providePreviewForFileRequest:(QLFilePreviewRequest*)request completionHandler:(void (^)(QLPreviewReply* _Nullable reply, NSError* _Nullable error))handler { QLPreviewReply* reply = [[QLPreviewReply alloc] initWithDataOfContentType:UTTypeHTML contentSize:CGSizeMake(1200, 800) dataCreationBlock:^NSData* _Nullable(QLPreviewReply* _Nonnull replyToUpdate, NSError* __autoreleasing _Nullable* _Nullable error) { NSString* previewHTML = [self generateHTMLPreviewFor:request.fileURL andReply:replyToUpdate]; return [previewHTML dataUsingEncoding:NSUTF8StringEncoding]; }]; handler(reply, nil); } - (NSString*)generateHTMLPreviewFor:(NSURL*)url andReply:(QLPreviewReply* _Nonnull)replyToUpdate { // Try to parse the torrent file auto metainfo = tr_torrent_metainfo{}; if (!metainfo.parse_torrent_file(url.fileSystemRepresentation)) { return nil; } NSBundle* bundle = NSBundle.mainBundle; NSURL* styleURL = [bundle URLForResource:@"style" withExtension:@"css"]; NSString* styleContents = [NSString stringWithContentsOfURL:styleURL encoding:NSUTF8StringEncoding error:NULL]; NSMutableString* htmlString = [NSMutableString string]; [htmlString appendFormat:@"", styleContents]; NSMutableDictionary* attachments = [NSMutableDictionary dictionary]; NSString* name = @(metainfo.name().c_str()); auto const n_files = metainfo.file_count(); auto const is_multifile = n_files > 1; UTType* fileType = is_multifile ? UTTypeFolder : [UTType typeWithFilenameExtension:name.pathExtension]; NSUInteger const width = 32; [htmlString appendFormat:@"

%@

", generateIconData(fileType, width, attachments), 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* pathPart = @(it->first.c_str()); UTType* pathType = nil; NSString* fileSize = nil; if (level < last) { // This node is a directory. pathType = UTTypeFolder; fileSize = @""; } else { // This node is a leaf file. pathType = [UTType typeWithFilenameExtension:pathPart.pathExtension]; fileSize = [NSString stringForFileSize:size]; } [listSection appendFormat:@"", level * kIconWidth, generateIconData(pathType, kIconWidth, attachments), kIconWidth, kIconWidth, pathPart, fileSize]; } curNode = &it->second; level++; } } [listSection appendString:@"

%@
%@	%@

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

%@", [lists componentsJoinedByString:@"
"]]; } [htmlString appendString:@""]; replyToUpdate.attachments = attachments; return htmlString; } @end