240 lines
9.3 KiB
Plaintext
240 lines
9.3 KiB
Plaintext
@import AppKit;
|
|
@import CoreFoundation;
|
|
@import QuickLook;
|
|
|
|
#include <string>
|
|
|
|
#include <libtransmission/torrent-metainfo.h>
|
|
|
|
#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:@"<html><style type=\"text/css\">%@</style><body>", 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:@"<h2><img class=\"icon\" src=\"%@\" width=\"%ld\" height=\"%ld\" />%@</h2>",
|
|
generateIconData(fileTypeString, width, allImgProps),
|
|
width,
|
|
width,
|
|
name];
|
|
|
|
NSString* fileSizeString = [NSString stringForFileSize:metainfo.totalSize()];
|
|
if (is_multifile)
|
|
{
|
|
NSString* fileCountString = [NSString
|
|
stringWithFormat:NSLocalizedStringFromTableInBundle(@"%lu files", nil, bundle, "quicklook file count"), n_files];
|
|
fileSizeString = [NSString stringWithFormat:@"%@, %@", fileCountString, fileSizeString];
|
|
}
|
|
[htmlString appendFormat:@"<p>%@</p>", 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:@"<p>%@</p>", creationString];
|
|
}
|
|
|
|
auto const& commentStr = metainfo.comment();
|
|
if (!std::empty(commentStr))
|
|
{
|
|
NSString* comment = @(commentStr.c_str());
|
|
if (![comment isEqualToString:@""])
|
|
[htmlString appendFormat:@"<p>%@</p>", comment];
|
|
}
|
|
|
|
NSMutableArray* lists = [NSMutableArray array];
|
|
|
|
auto const n_webseeds = metainfo.webseedCount();
|
|
if (n_webseeds > 0)
|
|
{
|
|
NSMutableString* listSection = [NSMutableString string];
|
|
[listSection appendString:@"<table>"];
|
|
|
|
NSString* headerTitleString = n_webseeds == 1 ?
|
|
NSLocalizedStringFromTableInBundle(@"1 Web Seed", nil, bundle, "quicklook web seed header") :
|
|
[NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"%lu Web Seeds", nil, bundle, "quicklook web seed header"), n_webseeds];
|
|
[listSection appendFormat:@"<tr><th>%@</th></tr>", headerTitleString];
|
|
|
|
for (size_t i = 0; i < n_webseeds; ++i)
|
|
{
|
|
[listSection appendFormat:@"<tr><td>%s<td></tr>", metainfo.webseed(i).c_str()];
|
|
}
|
|
|
|
[listSection appendString:@"</table>"];
|
|
|
|
[lists addObject:listSection];
|
|
}
|
|
|
|
auto const& announce_list = metainfo.announceList();
|
|
if (!std::empty(announce_list))
|
|
{
|
|
NSMutableString* listSection = [NSMutableString string];
|
|
[listSection appendString:@"<table>"];
|
|
|
|
auto const n = std::size(announce_list);
|
|
NSString* headerTitleString = n == 1 ?
|
|
NSLocalizedStringFromTableInBundle(@"1 Tracker", nil, bundle, "quicklook tracker header") :
|
|
[NSString stringWithFormat:NSLocalizedStringFromTableInBundle(@"%lu Trackers", nil, bundle, "quicklook tracker header"), n];
|
|
[listSection appendFormat:@"<tr><th>%@</th></tr>", headerTitleString];
|
|
|
|
#warning handle tiers?
|
|
for (auto const& tracker : announce_list)
|
|
{
|
|
[listSection appendFormat:@"<tr><td>%s<td></tr>", tracker.announce.c_str()];
|
|
}
|
|
|
|
[listSection appendString:@"</table>"];
|
|
|
|
[lists addObject:listSection];
|
|
}
|
|
|
|
if (is_multifile)
|
|
{
|
|
NSMutableString* listSection = [NSMutableString string];
|
|
[listSection appendString:@"<table>"];
|
|
|
|
NSString* fileTitleString = [NSString
|
|
stringWithFormat:NSLocalizedStringFromTableInBundle(@"%lu Files", nil, bundle, "quicklook file header"), n_files];
|
|
[listSection appendFormat:@"<tr><th>%@</th></tr>", 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:@"<tr><td><img class=\"icon\" src=\"%@\" width=\"%ld\" height=\"%ld\" />%@<td></tr>",
|
|
generateIconData(shortenedFilePath.pathExtension, width, allImgProps),
|
|
width,
|
|
width,
|
|
shortenedFilePathAndSize];
|
|
}
|
|
|
|
[listSection appendString:@"</table>"];
|
|
|
|
[lists addObject:listSection];
|
|
}
|
|
|
|
if (lists.count > 0)
|
|
{
|
|
[htmlString appendFormat:@"<hr/><br>%@", [lists componentsJoinedByString:@"<br>"]];
|
|
}
|
|
|
|
[htmlString appendString:@"</body></html>"];
|
|
|
|
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
|
|
}
|