LaTeXTools/latexFillAll.py at master · btstream/LaTeXTools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
from __future__ import print_function

import sublime
import sublime_plugin
import re

if sublime.version() < '3000':
    # we are on ST2 and Python 2.X
    _ST3 = False
    from latex_cite_completions import OLD_STYLE_CITE_REGEX, NEW_STYLE_CITE_REGEX, match
    from latex_ref_completions import OLD_STYLE_REF_REGEX, NEW_STYLE_REF_REGEX
    from latex_input_completions import TEX_INPUT_FILE_REGEX
else:
    _ST3 = True
    from .latex_cite_completions import OLD_STYLE_CITE_REGEX, NEW_STYLE_CITE_REGEX, match
    from .latex_ref_completions import OLD_STYLE_REF_REGEX, NEW_STYLE_REF_REGEX
    from .latex_input_completions import TEX_INPUT_FILE_REGEX

# used to flag whether command is triggered for cite
TRIGGER_CITE = False

def get_Region(a, b):

    if _ST3:
        return sublime.Region(a, b)
    else:
        return sublime.Region(long(a), long(b))

def get_current_word(view, point):

    line_prefix = view.substr(get_Region(view.line(point).a, point))[::-1]
    line_suffix = view.substr(get_Region(point, view.line(point).b))

    nc_current_word = ''

    # prefix is the characters before caret
    prefix = re.match(r'([^{}]*)\{', line_prefix).group(1)
    suffix = re.match(r'([^{}]*)\}', line_suffix).group(1)

    return prefix[::-1], suffix, nc_current_word

class LatexFillAllCommand(sublime_plugin.TextCommand):

    def run(self, edit):

        global TRIGGER_CITE

        view = self.view
        point = view.sel()[0].b

        # Current lines, used to detemine whether is input, include, cite, or includegraphics
        line = view.substr(get_Region(view.line(point).a, point))[::-1]

        # if \cite or \ref
        if match(OLD_STYLE_CITE_REGEX, line) != None or match(NEW_STYLE_CITE_REGEX, line) != None or \
            match(OLD_STYLE_REF_REGEX, line) != None or match(NEW_STYLE_REF_REGEX, line) != None:

            view.run_command('latex_ref_cite')

        # if \input, \include or \includegraphics
        if TEX_INPUT_FILE_REGEX.match(line) != None:

            prefix, suffix, nc_current_word = get_current_word(view, point)
            current_word = prefix + suffix
            if current_word != '':
                startpoint = point - len(prefix)
                endpoint = point + len(suffix)
                view.run_command('latex_tools_replace', {'a': startpoint, 'b': endpoint, 'replacement': ''})
                view.run_command('latex_fill_input')
            else:
                view.run_command("latex_fill_input")

class OnLatexFillAllReplacement(sublime_plugin.EventListener):

    # This trigger is used to delete the last "}"
    # character inserted by latex_cite command
    # when modifing the keyword between two commas.
    def on_selection_modified(self, view):

        global TRIGGER_CITE

        # If selection is modifed by fill all commands
        if TRIGGER_CITE:
            caret = view.sel()[0].b
            last_char = view.substr(get_Region(caret-1, caret))
            if last_char == '}':
                TRIGGER_CITE = False # Turn off triggers
                view.run_command('latex_tools_replace', {'a': caret-1, 'b': caret, 'replacement': ''})