Viewing: raw.githubusercontent.com

[ Web Proxy ]

Viewing: https://raw.githubusercontent.com/fabgithub/quicklook-csv/master/GeneratePreviewForURL.m

// // GeneratePreviewForURL.m // QuickLookCSV // // Created by Pascal Pfiffner on 03.07.2009. // This sourcecode is released under the Apache License, Version 2.0 // http://www.apache.org/licenses/LICENSE-2.0.html // #include #include #include #import #import "CSVDocument.h" #import "CSVRowObject.h" #define MAX_ROWS 500 static char* htmlReadableFileEncoding(NSStringEncoding stringEncoding); static char* humanReadableFileEncoding(NSStringEncoding stringEncoding); static char* formatFilesize(float bytes); /** * Generates a preview for the given file. * * This function parses the CSV and generates an HTML string that can be presented by QuickLook. */ OSStatus GeneratePreviewForURL(void *thisInterface, QLPreviewRequestRef preview, CFURLRef url, CFStringRef contentTypeUTI, CFDictionaryRef options) { @autoreleasepool { NSError *theErr = nil; NSURL *myURL = (__bridge NSURL *)url; // Load document data using NSStrings house methods // For huge files, maybe guess file encoding using `file --brief --mime` and use NSFileHandle? Not for now... NSStringEncoding stringEncoding; NSString *fileString = [NSString stringWithContentsOfURL:myURL usedEncoding:&stringEncoding error:&theErr]; // We could not open the file, probably unknown encoding; try ISO-8859-1 if (!fileString) { stringEncoding = NSISOLatin1StringEncoding; fileString = [NSString stringWithContentsOfURL:myURL encoding:stringEncoding error:&theErr]; // Still no success, give up if (!fileString) { if (nil != theErr) { NSLog(@"Error opening the file: %@", theErr); } return noErr; } } // Parse the data if still interested in the preview if (false == QLPreviewRequestIsCancelled(preview)) { CSVDocument *csvDoc = [CSVDocument new]; csvDoc.autoDetectSeparator = YES; NSUInteger numRowsParsed = [csvDoc numRowsFromCSVString:fileString maxRows:MAX_ROWS error:NULL]; // Create HTML of the data if still interested in the preview if (false == QLPreviewRequestIsCancelled(preview)) { NSBundle *myBundle = [NSBundle bundleForClass:[CSVDocument class]]; NSString *cssPath = [myBundle pathForResource:@"Style" ofType:@"css"]; NSString *css = [[NSString alloc] initWithContentsOfFile:cssPath encoding:NSUTF8StringEncoding error:NULL]; NSString *path = [myURL path]; NSDictionary *fileAttributes = [[NSFileManager defaultManager] attributesOfItemAtPath:path error:nil]; // compose the html NSMutableString *html = [[NSMutableString alloc] initWithString:@"\n"]; [html appendString:@"\n"]; [html appendFormat:@"\n", htmlReadableFileEncoding(stringEncoding)]; [html appendString:@"

%lu %@, %@ %@

", (unsigned long)[csvDoc.columnKeys count], (1 == [csvDoc.columnKeys count]) ? NSLocalizedString(@"column", nil) : NSLocalizedString(@"columns", nil), numRows, (1 == numRowsParsed) ? NSLocalizedString(@"row", nil) : NSLocalizedString(@"rows", nil) ]; [html appendFormat:@"

%s, %@-%@, %s

", formatFilesize([fileAttributes[NSFileSize] floatValue]), NSLocalizedString(separatorDesc, nil), NSLocalizedString(@"Separated", nil), humanReadableFileEncoding(stringEncoding) ]; // add the table rows BOOL altRow = NO; for (CSVRowObject *row in csvDoc.rows) { [html appendFormat:@"\n"]; altRow = !altRow; } [html appendString:@"

", altRow ? @" class=\"alt_row\"" : @""]; [html appendString:[row columns:csvDoc.columnKeys combinedByString:@"

"]]; [html appendString:@"

\n"]; // not all rows were parsed, show hint if (numRowsParsed > MAX_ROWS) { NSString *rowsHint = [NSString stringWithFormat:NSLocalizedString(@"Only the first %i rows are being displayed", nil), MAX_ROWS]; [html appendFormat:@"

", rowsHint]; } [html appendString:@""]; // feed the HTML CFDictionaryRef properties = (__bridge CFDictionaryRef)@{}; QLPreviewRequestSetDataRepresentation(preview, (__bridge CFDataRef)[html dataUsingEncoding:stringEncoding], kUTTypeHTML, properties ); } } } return noErr; } void CancelPreviewGeneration(void* thisInterface, QLPreviewRequestRef preview) { } #pragma mark - Output Utilities /** * To be used for the generated HTML. */ static char* htmlReadableFileEncoding(NSStringEncoding stringEncoding) { if (NSUTF8StringEncoding == stringEncoding || NSUnicodeStringEncoding == stringEncoding) { return "utf-8"; } if (NSASCIIStringEncoding == stringEncoding) { return "ascii"; } if (NSISOLatin1StringEncoding == stringEncoding) { return "iso-8859-1"; } if (NSMacOSRomanStringEncoding == stringEncoding) { return "x-mac-roman"; } if (NSUTF16BigEndianStringEncoding == stringEncoding || NSUTF16LittleEndianStringEncoding == stringEncoding) { return "utf-16"; } if (NSUTF32StringEncoding == stringEncoding || NSUTF32BigEndianStringEncoding == stringEncoding || NSUTF32LittleEndianStringEncoding == stringEncoding) { return "utf-32"; } return "utf-8"; } static char* humanReadableFileEncoding(NSStringEncoding stringEncoding) { if (NSUTF8StringEncoding == stringEncoding || NSUnicodeStringEncoding == stringEncoding) { return "UTF-8"; } if (NSASCIIStringEncoding == stringEncoding) { return "ASCII-text"; } if (NSISOLatin1StringEncoding == stringEncoding) { return "ISO-8859-1"; } if (NSMacOSRomanStringEncoding == stringEncoding) { return "Mac-Roman"; } if (NSUTF16BigEndianStringEncoding == stringEncoding || NSUTF16LittleEndianStringEncoding == stringEncoding) { return "UTF-16"; } if (NSUTF32StringEncoding == stringEncoding || NSUTF32BigEndianStringEncoding == stringEncoding || NSUTF32LittleEndianStringEncoding == stringEncoding) { return "UTF-32"; } return "UTF-8"; } static char* formatFilesize(float bytes) { if (bytes < 1) { return ""; } char *format[] = { "%.0f", "%.0f", "%.2f", "%.2f", "%.2f", "%.2f" }; char *unit[] = { "Byte", "KB", "MB", "GB", "TB", "PB" }; int i = 0; while (bytes > 1000) { bytes /= 1000; // Since OS X 10.7 (or 10.6?) Apple uses "kilobyte" and no longer "Kilobyte" or "kibibyte" i++; } char formatString[10]; static char result[9]; // longest string can be "999 Byte" or "999.99 GB" sprintf(formatString, "%s %s", format[i], unit[i]); sprintf(result, formatString, bytes); return result; }

Web Proxy Viewer | New URL | Original Page