def act(context, actionObject, operation='entab'): def replacements(match): '''Utility function for replacing items''' return match.group(0).replace(search, replace) spaces = int(actionObject.userInput().stringValue()) if operation == 'entab': target = re.compile(r'^(\t* +\t*)+', re.MULTILINE) search = ' ' * spaces replace = '\t' else: target = re.compile(r'^( *\t+ *)+', re.MULTILINE) search = '\t' replace = ' ' * spaces insertions = tea.new_recipe() ranges = tea.get_ranges(context) if len(ranges) == 1 and ranges[0].length == 0: # No selection, use the document ranges[0] = tea.new_range(0, context.string().length()) for range in ranges: text = tea.get_selection(context, range) # Non-Unix line endings will bork things; convert them text = tea.unix_line_endings(text) text = re.sub(target, replacements, text) if tea.get_line_ending(context) != '\n': text = tea.clean_line_endings(context, text) insertions.addReplacementString_forRange_(text, range) insertions.setUndoActionName_(operation.title()) context.applyTextRecipe_(insertions) return True
def act(controller, bundle, options): context = tea.get_context(controller) snippet = tea.get_option(options, 'snippet', '') maintain_selection = tea.get_option(options, 'maintain_selection', False) text, range = tea.selection_and_range(context) snippet = tea.indent_snippet(context, snippet, range) snippet = tea.clean_line_endings(context, snippet) # Set up target selection sel_loc = snippet.find('$SELECTED_TEXT') cursor_loc = snippet.find('$0') if maintain_selection: select_range = tea.new_range(sel_loc + range.location, range.length) elif cursor_loc != -1: select_range = tea.new_range(snippet.find('$0') + range.location, 0) else: select_range = None snippet = snippet.replace('$SELECTED_TEXT', text) snippet = snippet.replace('$0', '') if select_range is not None: tea.insert_text_and_select(context, snippet, range, select_range) else: tea.insert_text(context, snippet, range)
def didEndSheet_returnCode_contextInfo_(self, sheet, code, info): def replacements(match): '''Utility function for replacing items''' return match.group(0).replace(self.search, self.replace) if code == 1: # Leave sheet open with "processing" spinner self.spinner.startAnimation_(self) spaces = int(self.numSpaces.stringValue()) if self.action == 'entab': target = re.compile(r'^(\t* +\t*)+', re.MULTILINE) self.search = ' ' * spaces self.replace = '\t' else: target = re.compile(r'^( *\t+ *)+', re.MULTILINE) self.search = '\t' self.replace = ' ' * spaces insertions = tea.new_recipe() ranges = tea.get_ranges(self.context) if len(ranges) == 1 and ranges[0].length == 0: # No selection, use the document ranges[0] = tea.new_range(0, self.context.string().length()) for range in ranges: text = tea.get_selection(self.context, range) # Non-Unix line endings will bork things; convert them text = tea.unix_line_endings(text) text = re.sub(target, replacements, text) if tea.get_line_ending(self.context) != '\n': text = tea.clean_line_endings(self.context, text) insertions.addReplacementString_forRange_(text, range) insertions.setUndoActionName_(self.action.title()) self.context.applyTextRecipe_(insertions) self.spinner.stopAnimation_(self) sheet.orderOut_(self)
def act(controller, bundle, options): context = tea.get_context(controller) # Get the options alpha_numeric = tea.get_option(options, 'alpha_numeric', True) extra_characters = tea.get_option(options, 'extra_characters', '_-') bidirectional = tea.get_option(options, 'bidirectional', True) snippet = tea.get_option(options, 'snippet', '<$SELECTED_TEXT>$0</$WORD>') mode = tea.get_option(options, 'mode', '') # Fetch the word range = context.selectedRange() word, new_range = tea.get_word_or_selection(context, range, alpha_numeric, extra_characters, bidirectional) if word == '': # No word, so nothing further to do return False # If we're using $WORD, make sure the word is just a word if snippet.find('$WORD') >= 0: fullword = word word = tea.parse_word(word) if word is None: word = '' else: fullword = word # Process that sucker! if mode == 'zen' and fullword.find(' ') < 0: # Explicitly load zen settings zen_settings = settings_loader.load_settings() zen_core.update_settings(zen_settings) # Set up the config variables zen_core.newline = tea.get_line_ending(context) zen_settings['variables']['indentation'] = tea.get_indentation_string(context) # This allows us to use smart incrementing tab stops in zen snippets point_ix = [0] def place_ins_point(text): if not point_ix[0]: point_ix[0] += 1 return '$0' else: return '' zen_core.insertion_point = place_ins_point # Determine doctype as best we can based on file extension doc_type = tea.get_zen_doctype(context) # Prepare the snippet snippet = zen_core.expand_abbreviation(fullword, doc_type, 'xhtml') elif mode == 'zen' and tea.is_selfclosing(word): # Self-closing, so construct the snippet from scratch snippet = '<' + fullword if fullword == word and not fullword in ['br', 'hr']: snippet += ' $0 />' else: snippet += ' />$0' # Indent the snippet snippet = tea.indent_snippet(context, snippet, new_range) snippet = tea.clean_line_endings(context, snippet) # Special replacement in case we're using $WORD snippet = snippet.replace('$WORD', word) snippet = snippet.replace('$SELECTED_TEXT', fullword) cursor_loc = snippet.find('$0') if cursor_loc != -1: select_range = tea.new_range(cursor_loc + new_range.location, 0) snippet = snippet.replace('$0', '') tea.insert_text_and_select(context, snippet, new_range, select_range) else: tea.insert_text(context, snippet, new_range)
def wrap(self, context, abbr, profile_name='xhtml'): # Set up the config variables zen_settings = settings_loader.load_settings() zen.update_settings(zen_settings) zen.newline = self.safe_str(tea.get_line_ending(context)) zen_settings['variables']['indentation'] = self.safe_str(tea.get_indentation_string(context)) # This allows us to use smart incrementing tab stops in zen snippets point_ix = [0] def place_ins_point(text): if not point_ix[0]: point_ix[0] += 1 return '$0' else: return '' zen.insertion_point = place_ins_point text, rng = tea.selection_and_range(context) if not text: # no selection, find matching tag content = context.string() start, end = html_matcher.match(content, rng.location) if start is None: # nothing to wrap return False def is_space(char): return char.isspace() or char in r'\n\r' # narrow down selection until first non-space character while start < end: if not is_space(content[start]): break start += 1 while end > start: end -= 1 if not is_space(content[end]): end += 1 break rng = tea.new_range(start, end - start) text = tea.get_selection(context, rng) # Fetch the doctype based on file extension doc_type = tea.get_zen_doctype(context) text = self.unindent(context, text) # Damn Python's encodings! Have to convert string to ascii before wrapping # and then back to utf-8 result = zen.wrap_with_abbreviation(self.safe_str(abbr), self.safe_str(text), doc_type, profile_name) result = unicode(result, 'utf-8') result = tea.indent_snippet(context, result, rng) result = tea.clean_line_endings(context, result) cursor_loc = result.find('$0') if cursor_loc != -1: select_range = tea.new_range(cursor_loc + rng.location, 0) result = result.replace('$0', '') tea.insert_text_and_select(context, result, rng, select_range) else: tea.insert_text(context, result, rng)