//
// $Id$
//
// SPTextViewAdditions.m
// sequel-pro
//
// Created by Hans-Jörg Bibiko on April 05, 2009
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
//
// More info at
#import "SPAlertSheets.h"
#import "SPTooltip.h"
#import "SPBundleHTMLOutputController.h"
#import "SPCustomQuery.h"
@implementation NSTextView (SPTextViewAdditions)
/*
* Returns the range of the current word.
* finds: [| := caret] |word wo|rd word|
* If | is in between whitespaces nothing will be selected.
*/
- (NSRange)getRangeForCurrentWord
{
NSRange curRange = [self selectedRange];
if (curRange.length)
return curRange;
NSInteger curLocation = curRange.location;
NSInteger start = curLocation;
NSInteger end = curLocation;
NSUInteger strLen = [[self string] length];
NSMutableCharacterSet *wordCharSet = [NSMutableCharacterSet alphanumericCharacterSet];
[wordCharSet addCharactersInString:@"_."];
[wordCharSet removeCharactersInString:@"`"];
if(start) {
start--;
while([wordCharSet characterIsMember:[[self string] characterAtIndex:start]]) {
start--;
if(start < 0) break;
}
start++;
}
while(end < strLen && [wordCharSet characterIsMember:[[self string] characterAtIndex:end]]) {
end++;
}
return(NSMakeRange(start, end-start));
}
/*
* Select current word.
* finds: [| := caret] |word wo|rd word|
* If | is in between whitespaces nothing will be selected.
*/
- (IBAction)selectCurrentWord:(id)sender
{
[self setSelectedRange:[self getRangeForCurrentWord]];
}
/*
* Select current line.
*/
- (IBAction)selectCurrentLine:(id)sender
{
NSRange lineRange = [[self string] lineRangeForRange:[self selectedRange]];
if(lineRange.location != NSNotFound && lineRange.length)
[self setSelectedRange:lineRange];
else
NSBeep();
}
/*
*
*/
- (IBAction)selectEnclosingBrackets:(id)sender
{
NSUInteger caretPosition = [self selectedRange].location;
NSUInteger stringLength = [[self string] length];
unichar co, cc;
if(caretPosition == 0 || caretPosition >= stringLength) return;
NSInteger pcnt = 0;
NSInteger bcnt = 0;
NSInteger scnt = 0;
NSInteger i;
// look for the first non-balanced closing bracket
for(i=caretPosition; i=0; i--) {
if([[self string] characterAtIndex:i] == co) {
if(!bracketCounter) {
start = i;
break;
}
bracketCounter--;
}
if([[self string] characterAtIndex:i] == cc) {
bracketCounter++;
}
}
if(start < 0 ) return;
bracketCounter = 0;
for(i=caretPosition; i [[self string] length])
{
// caret is at the end of a text field
// transpose last two characters
[self moveLeftAndModifySelection:self];
[self moveLeftAndModifySelection:self];
workingRange = [self selectedRange];
}
else if(curRange.location == 0)
{
// caret is at the beginning of the text field
// do nothing
workingRange.length = 0;
}
else
{
// caret is in between two characters
// reverse adjacent characters
NSRange twoCharRange = NSMakeRange(curRange.location-1, 2);
[self setSelectedRange:twoCharRange];
workingRange = twoCharRange;
}
}
@catch(id ae)
{ workingRange.length = 0; }
// reverse string : TODO not yet combining diacritics safe!
NSUInteger len = workingRange.length;
if (len > 1)
{
NSMutableString *reversedStr = [NSMutableString stringWithCapacity:len];
while (len > 0)
[reversedStr appendString:
[NSString stringWithFormat:@"%C", [[self string] characterAtIndex:--len+workingRange.location]]];
[self insertText:reversedStr];
[self setSelectedRange:curRange];
}
}
/**
* Inserts the preference's NULL value set by the user
*/
- (IBAction)insertNULLvalue:(id)sender
{
id prefs = [NSUserDefaults standardUserDefaults];
if([self respondsToSelector:@selector(insertText:)])
if([prefs objectForKey:SPNullValue] && [[prefs objectForKey:SPNullValue] length])
[self insertText:[prefs objectForKey:SPNullValue]];
else
[self insertText:@"NULL"];
}
/**
* Move selected lines or current line one line up
*/
- (IBAction)moveSelectionLineUp:(id)sender;
{
NSRange currentSelection = [self selectedRange];
NSRange lineRange = [[self string] lineRangeForRange:currentSelection];
if(lineRange.location > 0) {
NSRange beforeLineRange = [[self string] lineRangeForRange:NSMakeRange(lineRange.location-1, 0)];
NSRange insertPoint = NSMakeRange(beforeLineRange.location, 0);
NSString *currentLine = [[self string] substringWithRange:lineRange];
BOOL lastLine = NO;
if([currentLine characterAtIndex:[currentLine length]-1] != '\n') {
currentLine = [NSString stringWithFormat:@"%@\n", currentLine];
lastLine = YES;
}
[self setSelectedRange:lineRange];
[self insertText:@""];
[self setSelectedRange:insertPoint];
[self insertText:currentLine];
if(lastLine) {
[self setSelectedRange:NSMakeRange([[self string] length]-1,1)];
[self insertText:@""];
}
if(currentSelection.length)
insertPoint.length+=[currentLine length];
[self setSelectedRange:insertPoint];
}
}
/**
* Move selected lines or current line one line down
*/
- (IBAction)moveSelectionLineDown:(id)sender
{
NSRange currentSelection = [self selectedRange];
NSRange lineRange = [[self string] lineRangeForRange:currentSelection];
if(NSMaxRange(lineRange) < [[self string] length]) {
NSRange afterLineRange = [[self string] lineRangeForRange:NSMakeRange(NSMaxRange(lineRange), 0)];
NSRange insertPoint = NSMakeRange(lineRange.location + afterLineRange.length, 0);
NSString *currentLine = [[self string] substringWithRange:lineRange];
[self setSelectedRange:lineRange];
[self insertText:@""];
[self setSelectedRange:insertPoint];
if([[self string] characterAtIndex:insertPoint.location-1] != '\n') {
[self insertText:@"\n"];
insertPoint.location++;
currentLine = [currentLine substringToIndex:[currentLine length]-1];
}
[self insertText:currentLine];
if(currentSelection.length)
insertPoint.length+=[currentLine length];
[self setSelectedRange:insertPoint];
}
}
/**
* Increase the textView's font size by 1
*/
- (void)makeTextSizeLarger
{
NSFont *aFont = [self font];
BOOL editableStatus = [self isEditable];
[self setEditable:YES];
[self setFont:[[NSFontManager sharedFontManager] convertFont:aFont toSize:[aFont pointSize]+1]];
[self setEditable:editableStatus];
}
/*
* Decrease the textView's font size by 1 but not smaller than 4pt
*/
- (void)makeTextSizeSmaller
{
NSFont *aFont = [self font];
NSInteger newSize = ([aFont pointSize]-1 < 4) ? [aFont pointSize] : [aFont pointSize]-1;
BOOL editableStatus = [self isEditable];
[self setEditable:YES];
[self setFont:[[NSFontManager sharedFontManager] convertFont:aFont toSize:newSize]];
[self setEditable:editableStatus];
}
- (IBAction)executeBundleItemForInputField:(id)sender
{
NSInteger idx = [sender tag] - 1000000;
NSString *infoPath = nil;
NSArray *bundleItems = [[NSApp delegate] bundleItemsForScope:SPBundleScopeInputField];
if(idx >=0 && idx < [bundleItems count]) {
infoPath = [[bundleItems objectAtIndex:idx] objectForKey:SPBundleInternPathToFileKey];
} else {
if([sender tag] == 0 && [[sender toolTip] length]) {
infoPath = [sender toolTip];
}
}
if(!infoPath) {
NSBeep();
return;
}
NSError *readError = nil;
NSString *convError = nil;
NSPropertyListFormat format;
NSDictionary *cmdData = nil;
NSData *pData = [NSData dataWithContentsOfFile:infoPath options:NSUncachedRead error:&readError];
cmdData = [[NSPropertyListSerialization propertyListFromData:pData
mutabilityOption:NSPropertyListImmutable format:&format errorDescription:&convError] retain];
if(!cmdData || readError != nil || [convError length] || !(format == NSPropertyListXMLFormat_v1_0 || format == NSPropertyListBinaryFormat_v1_0)) {
NSLog(@"“%@” file couldn't be read.", infoPath);
NSBeep();
if (cmdData) [cmdData release];
return;
} else {
if([cmdData objectForKey:SPBundleFileCommandKey] && [[cmdData objectForKey:SPBundleFileCommandKey] length]) {
NSString *cmd = [cmdData objectForKey:SPBundleFileCommandKey];
NSString *inputAction = @"";
NSString *inputFallBackAction = @"";
NSError *err = nil;
NSString *uuid = [NSString stringWithNewUUID];
NSString *bundleInputFilePath = [NSString stringWithFormat:@"%@_%@", SPBundleTaskInputFilePath, uuid];
NSRange currentWordRange, currentSelectionRange, currentLineRange, currentQueryRange;
[[NSFileManager defaultManager] removeItemAtPath:bundleInputFilePath error:nil];
BOOL selfIsQueryEditor = ([[[self class] description] isEqualToString:@"SPTextView"] && [[self delegate] respondsToSelector:@selector(currentQueryRange)]);
if([cmdData objectForKey:SPBundleFileInputSourceKey])
inputAction = [[cmdData objectForKey:SPBundleFileInputSourceKey] lowercaseString];
if([cmdData objectForKey:SPBundleFileInputSourceFallBackKey])
inputFallBackAction = [[cmdData objectForKey:SPBundleFileInputSourceFallBackKey] lowercaseString];
currentSelectionRange = [self selectedRange];
currentWordRange = [self getRangeForCurrentWord];
currentLineRange = [[self string] lineRangeForRange:NSMakeRange([self selectedRange].location, 0)];
if(selfIsQueryEditor) {
currentQueryRange = [[self delegate] currentQueryRange];
} else {
currentQueryRange = currentLineRange;
}
if(!currentQueryRange.length)
currentQueryRange = currentSelectionRange;
NSRange replaceRange = currentSelectionRange;
if([inputAction isEqualToString:SPBundleInputSourceSelectedText]) {
if(!currentSelectionRange.length) {
if([inputFallBackAction isEqualToString:SPBundleInputSourceCurrentWord])
replaceRange = currentWordRange;
else if([inputFallBackAction isEqualToString:SPBundleInputSourceCurrentLine])
replaceRange = currentLineRange;
else if([inputFallBackAction isEqualToString:SPBundleInputSourceCurrentQuery])
replaceRange = currentQueryRange;
else if([inputFallBackAction isEqualToString:SPBundleInputSourceEntireContent])
replaceRange = NSMakeRange(0,[[self string] length]);
} else {
replaceRange = currentSelectionRange;
}
}
else if([inputAction isEqualToString:SPBundleInputSourceEntireContent]) {
replaceRange = NSMakeRange(0, [[self string] length]);
}
NSMutableDictionary *env = [NSMutableDictionary dictionary];
[env setObject:[infoPath stringByDeletingLastPathComponent] forKey:SPBundleShellVariableBundlePath];
[env setObject:bundleInputFilePath forKey:SPBundleShellVariableInputFilePath];
[env setObject:SPBundleScopeInputField forKey:SPBundleShellVariableBundleScope];
id tableSource = [self delegate];
if([[[tableSource class] description] isEqualToString:@"SPFieldEditorController"]) {
NSDictionary *editedFieldInfo = [tableSource editedFieldInfo];
[env setObject:[editedFieldInfo objectForKey:@"colName"] forKey:SPBundleShellVariableCurrentEditedColumnName];
if([editedFieldInfo objectForKey:@"tableName"])
[env setObject:[editedFieldInfo objectForKey:@"tableName"] forKey:SPBundleShellVariableCurrentEditedTable];
[env setObject:[editedFieldInfo objectForKey:@"usedQuery"] forKey:SPBundleShellVariableUsedQueryForTable];
[env setObject:[editedFieldInfo objectForKey:@"tableSource"] forKey:SPBundleShellVariableDataTableSource];
}
else if([[[tableSource class] description] isEqualToString:@"SPCopyTable"]) {
NSInteger editedCol = [tableSource editedColumn];
if(editedCol > -1) {
NSString *colName = [[[[tableSource tableColumns] objectAtIndex:editedCol] headerCell] stringValue];
if([[[[tableSource dataSource] class] description] hasSuffix:@"Content"]) {
[env setObject:[colName description] forKey:SPBundleShellVariableCurrentEditedColumnName];
[env setObject:@"content" forKey:SPBundleShellVariableDataTableSource];
} else {
NSArray *defs = [[tableSource delegate] dataColumnDefinitions];
for(NSDictionary* col in defs) {
if([[col objectForKey:@"name"] isEqualToString:colName]) {
[env setObject:[col objectForKey:@"org_name"] forKey:SPBundleShellVariableCurrentEditedColumnName];
[env setObject:[col objectForKey:@"org_table"] forKey:SPBundleShellVariableCurrentEditedTable];
break;
}
}
[env setObject:@"query" forKey:SPBundleShellVariableDataTableSource];
}
if([[tableSource delegate] respondsToSelector:@selector(usedQuery)] && [[tableSource delegate] usedQuery])
[env setObject:[[tableSource delegate] usedQuery] forKey:SPBundleShellVariableUsedQueryForTable];
}
}
if(selfIsQueryEditor && [[self delegate] currentQueryRange].length)
[env setObject:[[self string] substringWithRange:[[self delegate] currentQueryRange]] forKey:SPBundleShellVariableCurrentQuery];
if(currentSelectionRange.length)
[env setObject:[[self string] substringWithRange:currentSelectionRange] forKey:SPBundleShellVariableSelectedText];
if(currentWordRange.length)
[env setObject:[[self string] substringWithRange:currentWordRange] forKey:SPBundleShellVariableCurrentWord];
if(currentLineRange.length)
[env setObject:[[self string] substringWithRange:currentLineRange] forKey:SPBundleShellVariableCurrentLine];
[env setObject:NSStringFromRange(replaceRange) forKey:SPBundleShellVariableSelectedTextRange];
NSError *inputFileError = nil;
NSString *input = [NSString stringWithString:[[self string] substringWithRange:replaceRange]];
[input writeToFile:bundleInputFilePath
atomically:YES
encoding:NSUTF8StringEncoding
error:&inputFileError];
if(inputFileError != nil) {
NSString *errorMessage = [inputFileError localizedDescription];
SPBeginAlertSheet(NSLocalizedString(@"Bundle Error", @"bundle error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, [self window], self, nil, nil,
[NSString stringWithFormat:@"%@ “%@”:\n%@", NSLocalizedString(@"Error for", @"error for message"), [cmdData objectForKey:@"name"], errorMessage]);
if (cmdData) [cmdData release];
return;
}
NSString *output = [cmd runBashCommandWithEnvironment:env
atCurrentDirectoryPath:nil
callerInstance:[[NSApp delegate] frontDocument]
contextInfo:[NSDictionary dictionaryWithObjectsAndKeys:
([cmdData objectForKey:SPBundleFileNameKey])?:@"-", @"name",
NSLocalizedString(@"Input Field", @"input field menu item label"), @"scope",
uuid, SPBundleFileInternalexecutionUUID,
nil]
error:&err];
[[NSFileManager defaultManager] removeItemAtPath:bundleInputFilePath error:nil];
NSString *action = SPBundleOutputActionNone;
if([cmdData objectForKey:SPBundleFileOutputActionKey] && [[cmdData objectForKey:SPBundleFileOutputActionKey] length])
action = [[cmdData objectForKey:SPBundleFileOutputActionKey] lowercaseString];
// Redirect due exit code
if(err != nil) {
if([err code] == SPBundleRedirectActionNone) {
action = SPBundleOutputActionNone;
err = nil;
}
else if([err code] == SPBundleRedirectActionReplaceSection) {
action = SPBundleOutputActionReplaceSelection;
err = nil;
}
else if([err code] == SPBundleRedirectActionReplaceContent) {
action = SPBundleOutputActionReplaceContent;
err = nil;
}
else if([err code] == SPBundleRedirectActionInsertAsText) {
action = SPBundleOutputActionInsertAsText;
err = nil;
}
else if([err code] == SPBundleRedirectActionInsertAsSnippet) {
action = SPBundleOutputActionInsertAsSnippet;
err = nil;
}
else if([err code] == SPBundleRedirectActionShowAsHTML) {
action = SPBundleOutputActionShowAsHTML;
err = nil;
}
else if([err code] == SPBundleRedirectActionShowAsTextTooltip) {
action = SPBundleOutputActionShowAsTextTooltip;
err = nil;
}
else if([err code] == SPBundleRedirectActionShowAsHTMLTooltip) {
action = SPBundleOutputActionShowAsHTMLTooltip;
err = nil;
}
}
if(err == nil && output) {
if(![action isEqualToString:SPBundleOutputActionNone]) {
if([action isEqualToString:SPBundleOutputActionShowAsTextTooltip]) {
[SPTooltip showWithObject:output];
}
else if([action isEqualToString:SPBundleOutputActionShowAsHTMLTooltip]) {
[SPTooltip showWithObject:output ofType:@"html"];
}
else if([action isEqualToString:SPBundleOutputActionShowAsHTML]) {
BOOL correspondingWindowFound = NO;
for(id win in [NSApp windows]) {
if([[win delegate] isKindOfClass:[SPBundleHTMLOutputController class]]) {
if([[[win delegate] windowUUID] isEqualToString:[cmdData objectForKey:SPBundleFileUUIDKey]]) {
correspondingWindowFound = YES;
[[win delegate] displayHTMLContent:output withOptions:nil];
break;
}
}
}
if(!correspondingWindowFound) {
SPBundleHTMLOutputController *c = [[SPBundleHTMLOutputController alloc] init];
[c setWindowUUID:[cmdData objectForKey:SPBundleFileUUIDKey]];
[c displayHTMLContent:output withOptions:nil];
[[NSApp delegate] addHTMLOutputController:c];
}
}
if([self isEditable]) {
if([action isEqualToString:SPBundleOutputActionInsertAsText]) {
[self insertText:output];
}
else if([action isEqualToString:SPBundleOutputActionInsertAsSnippet]) {
if([self respondsToSelector:@selector(insertAsSnippet:atRange:)])
[self insertAsSnippet:output atRange:replaceRange];
else
[SPTooltip showWithObject:NSLocalizedString(@"Input Field doesn't support insertion of snippets.", @"input field doesn't support insertion of snippets.")];
}
else if([action isEqualToString:SPBundleOutputActionReplaceContent]) {
if([[self string] length])
[self setSelectedRange:NSMakeRange(0, [[self string] length])];
[self insertText:output];
}
else if([action isEqualToString:SPBundleOutputActionReplaceSelection]) {
NSRange safeRange = NSIntersectionRange(replaceRange, NSMakeRange(0, [[self string] length]));
[self shouldChangeTextInRange:safeRange replacementString:output];
[self replaceCharactersInRange:safeRange withString:output];
}
} else {
[SPTooltip showWithObject:NSLocalizedString(@"Input Field is not editable.", @"input field is not editable.")];
}
}
} else if([err code] != 9) { // Suppress an error message if command was killed
NSString *errorMessage = [err localizedDescription];
SPBeginAlertSheet(NSLocalizedString(@"BASH Error", @"bash error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, [self window], self, nil, nil,
[NSString stringWithFormat:@"%@ “%@”:\n%@", NSLocalizedString(@"Error for", @"error for message"), [cmdData objectForKey:@"name"], errorMessage]);
}
}
if (cmdData) [cmdData release];
}
}
/**
* Add Bundle menu items.
*/
- (NSMenu *)menuForEvent:(NSEvent *)event
{
NSMenu *menu = [[self class] defaultMenu];
// Remove 'Bundles' sub menu and separator
NSMenuItem *bItem = [menu itemWithTag:10000000];
if(bItem) {
NSInteger sepIndex = [menu indexOfItem:bItem]-1;
[menu removeItemAtIndex:sepIndex];
[menu removeItem:bItem];
}
if([[[[[[NSApp delegate] frontDocumentWindow] delegate] selectedTableDocument] connectionID] isEqualToString:@"_"]) return menu;
[[NSApp delegate] reloadBundles:self];
NSArray *bundleCategories = [[NSApp delegate] bundleCategoriesForScope:SPBundleScopeInputField];
NSArray *bundleItems = [[NSApp delegate] bundleItemsForScope:SPBundleScopeInputField];
// Add 'Bundles' sub menu
if(bundleItems && [bundleItems count]) {
[menu addItem:[NSMenuItem separatorItem]];
NSMenu *bundleMenu = [[[NSMenu alloc] init] autorelease];
NSMenuItem *bundleSubMenuItem = [[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Bundles", @"bundles menu item label") action:nil keyEquivalent:@""];
[bundleSubMenuItem setTag:10000000];
[menu addItem:bundleSubMenuItem];
[menu setSubmenu:bundleMenu forItem:bundleSubMenuItem];
NSMutableArray *categorySubMenus = [NSMutableArray array];
NSMutableArray *categoryMenus = [NSMutableArray array];
if([bundleCategories count]) {
for(NSString* title in bundleCategories) {
[categorySubMenus addObject:[[[NSMenuItem alloc] initWithTitle:title action:nil keyEquivalent:@""] autorelease]];
[categoryMenus addObject:[[[NSMenu alloc] init] autorelease]];
[bundleMenu addItem:[categorySubMenus lastObject]];
[bundleMenu setSubmenu:[categoryMenus lastObject] forItem:[categorySubMenus lastObject]];
}
}
NSInteger i = 0;
for(NSDictionary *item in bundleItems) {
NSString *keyEq;
if([item objectForKey:SPBundleFileKeyEquivalentKey])
keyEq = [[item objectForKey:SPBundleFileKeyEquivalentKey] objectAtIndex:0];
else
keyEq = @"";
NSMenuItem *mItem = [[[NSMenuItem alloc] initWithTitle:[item objectForKey:SPBundleInternLabelKey] action:@selector(executeBundleItemForInputField:) keyEquivalent:keyEq] autorelease];
if([keyEq length])
[mItem setKeyEquivalentModifierMask:[[[item objectForKey:SPBundleFileKeyEquivalentKey] objectAtIndex:1] intValue]];
[mItem setTarget:[[NSApp mainWindow] firstResponder]];
if([item objectForKey:SPBundleFileTooltipKey])
[mItem setToolTip:[item objectForKey:SPBundleFileTooltipKey]];
[mItem setTag:1000000 + i++];
if([item objectForKey:SPBundleFileCategoryKey]) {
[[categoryMenus objectAtIndex:[bundleCategories indexOfObject:[item objectForKey:SPBundleFileCategoryKey]]] addItem:mItem];
} else {
[bundleMenu addItem:mItem];
}
}
[bundleSubMenuItem release];
}
return menu;
}
#pragma mark -
#pragma mark multi-touch trackpad support
/**
* Trackpad two-finger zooming gesture for in/decreasing the font size
*/
- (void) magnifyWithEvent:(NSEvent *)anEvent
{
//Avoid font resizing for NSTextViews in SPCopyTable or NSTableView
if([[[[self delegate] class] description] isEqualToString:@"SPCopyTable"]
|| [[[[self delegate] class] description] isEqualToString:@"NSTableView"]) return;
if([anEvent deltaZ]>5.0)
[self makeTextSizeLarger];
else if([anEvent deltaZ]<-5.0)
[self makeTextSizeSmaller];
}
@end