• SQL file Quicklook generator

- truncate SQL files larger than 500k
- changed encoding detection: try utf-8, if it fails let Cocoa to detect the enc, if it fails fall back to latin1

1 files changed, 86 insertions, 62 deletions

diff --git a/Source/GeneratePreviewForURL.m b/Source/GeneratePreviewForURL.m
index 8d1783ec..697ee552 100644
--- a/Source/GeneratePreviewForURL.m
+++ b/Source/GeneratePreviewForURL.m
@@ -30,7 +30,6 @@
 #import <Cocoa/Cocoa.h>
 #import "SPEditorTokens.h"
 
-
 /* -----------------------------------------------------------------------------
   Generate a preview for file
 
@@ -194,80 +193,105 @@ OSStatus GeneratePreviewForURL(void *thisInterface, QLPreviewRequestRef preview,
 			return noErr;
 		}
 
-		// compose the html
+		NSError *readError = nil;
+
+		NSStringEncoding sqlEncoding = NSUTF8StringEncoding;
+
 		if(fileAttributes)
 		{
+
 			NSNumber *filesize = [fileAttributes objectForKey:NSFileSize];
-			// catch large files since Finder blocks
-			if([filesize unsignedLongValue] > 3000000) {
-				html = [[NSMutableString alloc] initWithString:[NSString stringWithFormat:template,
-					[[iconImage TIFFRepresentationUsingCompression:NSTIFFCompressionJPEG factor:0.01] base64EncodingWithLineLength:0],
-					[NSString stringForByteSize:[[fileAttributes objectForKey:NSFileSize] longLongValue]],
-					@"... SQL ..."
-				]];
-			} else {
-				
-				// perform syntax highlighting
-				NSString *sqlText = [NSString stringWithContentsOfFile:[myURL path] encoding:NSUTF8StringEncoding error:nil];
-				NSMutableString *sqlHTML = [NSMutableString string];
-
-				NSRange textRange = NSMakeRange(0, [sqlText length]);
-				NSString *tokenColor;
-				size_t token;
-				NSRange tokenRange;
-
-				// initialise flex
-				yyuoffset = 0; yyuleng = 0;
-				yy_switch_to_buffer(yy_scan_string([sqlText UTF8String]));
-
-				// now loop through all the tokens
-				while (token=yylex()){
-					switch (token) {
-						case SPT_SINGLE_QUOTED_TEXT:
-						case SPT_DOUBLE_QUOTED_TEXT:
-						    tokenColor = @"#A7221C";
-						    break;
-						case SPT_BACKTICK_QUOTED_TEXT:
-						    tokenColor = @"#001892";
-						    break;
-						case SPT_RESERVED_WORD:
-						    tokenColor = @"#0041F6";
-						    break;
-						case SPT_NUMERIC:
-							tokenColor = @"#67350F";
-							break;
-						case SPT_COMMENT:
-						    tokenColor = @"#265C10";
-						    break;
-						case SPT_VARIABLE:
-						    tokenColor = @"#6C6C6C";
-						    break;
-						case SPT_WHITESPACE:
-						    tokenColor = @"black";
-						    break;
-						default:
-						    tokenColor = @"black";
-					}
-
-					tokenRange = NSMakeRange(yyuoffset, yyuleng);
-					[sqlHTML appendFormat:@"<font color='%@'>%@</font>", tokenColor, [sqlText substringWithRange:tokenRange]];
 
+			// compose the html and perform syntax highlighting
+
+			// read the file and try to get a proper encoding
+			NSString *sqlText = [NSString stringWithContentsOfFile:[myURL path] encoding:sqlEncoding error:&readError];
+			
+			if(readError != nil) {
+				// cocoa tries to detect the encoding
+				sqlText = [NSString stringWithContentsOfFile:[myURL path] usedEncoding:&sqlEncoding error:&readError];
+				// fall back to latin1 if no sqlText couldn't read
+				if(sqlText == nil) {
+					sqlEncoding = NSISOLatin1StringEncoding;
+					sqlText = [NSString stringWithContentsOfFile:[myURL path] encoding:sqlEncoding error:&readError];
 				}
+			}
 
-				html = [[NSMutableString alloc] initWithString:[NSString stringWithFormat:template,
-					[[iconImage TIFFRepresentationUsingCompression:NSTIFFCompressionJPEG factor:0.01] base64EncodingWithLineLength:0],
-					[NSString stringForByteSize:[[fileAttributes objectForKey:NSFileSize] longLongValue]],
-					sqlHTML
-				]];
+			// bail if nothing could be read
+			if(!sqlText) {
+				[html release];
+				[pool release];
+				return noErr;
+			}
 
+			// truncate large files since Finder blocks
+			NSString *truncatedString = @"";
+			if([filesize unsignedLongValue] > 500000) {
+				sqlText = [sqlText substringToIndex:500000-1];
+				truncatedString = @"\n ✂ ...";
 			}
-		} else {
+
+			NSMutableString *sqlHTML = [NSMutableString string];
+
+			NSRange textRange = NSMakeRange(0, [sqlText length]);
+			NSString *tokenColor;
+			size_t token;
+			NSRange tokenRange;
+
+			// initialise flex
+			yyuoffset = 0; yyuleng = 0;
+			yy_switch_to_buffer(yy_scan_string([sqlText UTF8String]));
+
+			// now loop through all the tokens
+			while (token=yylex()){
+				switch (token) {
+					case SPT_SINGLE_QUOTED_TEXT:
+					case SPT_DOUBLE_QUOTED_TEXT:
+					    tokenColor = @"#A7221C";
+					    break;
+					case SPT_BACKTICK_QUOTED_TEXT:
+					    tokenColor = @"#001892";
+					    break;
+					case SPT_RESERVED_WORD:
+					    tokenColor = @"#0041F6";
+					    break;
+					case SPT_NUMERIC:
+						tokenColor = @"#67350F";
+						break;
+					case SPT_COMMENT:
+					    tokenColor = @"#265C10";
+					    break;
+					case SPT_VARIABLE:
+					    tokenColor = @"#6C6C6C";
+					    break;
+					case SPT_WHITESPACE:
+					    tokenColor = @"black";
+					    break;
+					default:
+					    tokenColor = @"black";
+				}
+
+				tokenRange = NSMakeRange(yyuoffset, yyuleng);
+				[sqlHTML appendFormat:@"<font color='%@'>%@</font>", tokenColor, [sqlText substringWithRange:tokenRange]];
+
+			}
+			[sqlHTML appendString:truncatedString];
+
 			html = [[NSMutableString alloc] initWithString:[NSString stringWithFormat:template,
 				[[iconImage TIFFRepresentationUsingCompression:NSTIFFCompressionJPEG factor:0.01] base64EncodingWithLineLength:0],
 				[NSString stringForByteSize:[[fileAttributes objectForKey:NSFileSize] longLongValue]],
-				[NSString stringWithContentsOfFile:[myURL path] encoding:NSUTF8StringEncoding error:nil]
+				sqlHTML
 			]];
+
+		} else {
+
+			// No file attributes were read, bail for safety reasons
+			[html release];
+			[pool release];
+			return noErr;
+
 		}
+
 	}
 
 	CFDictionaryRef properties = (CFDictionaryRef)[NSDictionary dictionary];