forked from facelessuser/ExportHtml
-
Notifications
You must be signed in to change notification settings - Fork 0
/
HtmlAnnotations.py
263 lines (211 loc) · 9.08 KB
/
HtmlAnnotations.py
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
import sublime
import sublime_plugin
from ExportHtml.lib.notify import error
PACKAGE_SETTINGS = "ExportHtml.sublime-settings"
def get_highlight_style():
style_flag = 0
settings = sublime.load_settings(PACKAGE_SETTINGS)
scope = settings.get("annotation_highlight_scope", "comment")
style = settings.get("annotation_highlight_style", "outline")
if style == "outline":
style_flag |= sublime.DRAW_OUTLINED
return scope, style_flag
def clean_invalid_regions(view, annotations):
deletions = 0
for x in range(0, int(annotations["count"])):
key_name = "html_annotation_%d" % x
regions = view.get_regions(key_name)
if len(regions) and not regions[0].empty():
annotations["annotations"]["html_annotation_%d" % x]["region"] = [regions[0].begin(), regions[0].end()]
if deletions:
new_key = "html_annotation_%d" % (x - deletions)
annotations["annotations"][new_key] = annotations["annotations"][key_name]
del annotations["annotations"][key_name]
new_region = annotations["annotations"][new_key]["region"]
view.erase_regions(key_name)
scope, style = get_highlight_style()
view.add_regions(
new_key,
[sublime.Region(new_region[0], new_region[1])],
scope,
"",
style
)
else:
del annotations["annotations"]["html_annotation_%d" % x]
annotations["count"] -= 1
deletions += 1
if len(regions):
view.erase_regions(key_name)
view.settings().set("annotation_comments", annotations)
def get_annotations(view):
annotations = view.settings().get("annotation_comments", {"count": 0, "annotations": {}})
clean_invalid_regions(view, annotations)
return annotations
def clear_annotations(view):
annotations = view.settings().get("annotation_comments", {"count": 0, "annotations": {}})
for x in range(0, int(annotations["count"])):
view.erase_regions("html_annotation_%d" % x)
view.settings().set("annotation_comments", {"count": 0, "annotations": {}})
def delete_annotations(view):
annotations = view.settings().get("annotation_comments", {"count": 0, "annotations": {}})
for sel in view.sel():
for x in range(0, int(annotations["count"])):
region = annotations["annotations"]["html_annotation_%d" % x]["region"]
annotation = sublime.Region(int(region[0]), int(region[1]))
if annotation.contains(sel):
view.erase_regions("html_annotation_%d" % x)
break
clean_invalid_regions(view, annotations)
def get_annotation_comment(view):
comment = None
annotations = view.settings().get("annotation_comments", {"count": 0, "annotations": {}})
if len(view.sel()):
sel = view.sel()[0]
for x in range(0, int(annotations["count"])):
region = annotations["annotations"]["html_annotation_%d" % x]["region"]
annotation = sublime.Region(int(region[0]), int(region[1]))
if annotation.contains(sel):
comment = annotations["annotations"]["html_annotation_%d" % x]["comment"]
return comment
def is_selection_in_annotation(view, first_only=False):
mode = view.settings().get("annotation_mode", False)
selection = False
if mode:
annotations = view.settings().get("annotation_comments", {"count": 0, "annotations": {}})
for sel in view.sel():
for x in range(0, int(annotations["count"])):
region = annotations["annotations"]["html_annotation_%d" % x]["region"]
annotation = sublime.Region(int(region[0]), int(region[1]))
if annotation.contains(sel):
selection = True
break
if first_only:
break
return mode and selection
def annotations_exist(view):
mode = view.settings().get("annotation_mode", False)
found = False
if mode:
annotations = view.settings().get("annotation_comments", {"count": 0, "annotations": {}})
if int(annotations["count"]):
found = True
return mode and found
def is_selected(view):
mode = view.settings().get("annotation_mode", False)
selected = not view.sel()[0].empty()
return mode and selected
class ShowAnnotationCommentCommand(sublime_plugin.TextCommand):
def is_visible(self):
return is_selection_in_annotation(self.view)
def run(self, edit):
comment = get_annotation_comment(self.view)
if comment is not None:
sublime.message_dialog("Annotation Comment:\n\n%s" % comment)
sublime.set_clipboard(comment)
class ClearAnnotationsCommand(sublime_plugin.TextCommand):
def is_visible(self):
return annotations_exist(self.view)
def run(self, edit):
clear_annotations(self.view)
class DeleteAnnotationsCommand(sublime_plugin.TextCommand):
def is_visible(self):
return is_selection_in_annotation(self.view)
def run(self, edit):
delete_annotations(self.view)
class EnableAnnotationModeCommand(sublime_plugin.TextCommand):
def is_visible(self):
return not self.view.settings().get("annotation_mode", False)
def run(self, edit):
self.view.run_command("toggle_annotation_html_mode")
class DisableAnnotationModeCommand(sublime_plugin.TextCommand):
def is_visible(self):
return self.view.settings().get("annotation_mode", False)
def run(self, edit):
self.view.run_command("toggle_annotation_html_mode")
class ToggleAnnotationHtmlModeCommand(sublime_plugin.TextCommand):
def is_enabled(self):
return not self.view.settings().get('is_widget')
def run(self, edit):
mode = False if self.view.settings().get("annotation_mode", False) else True
self.view.settings().set("annotation_mode", mode)
if mode:
self.view.settings().set("annotation_read_mode", self.view.is_read_only())
self.view.set_read_only(True)
self.view.set_status("html_annotation_mode", "Annotation Mode: ON")
else:
clear_annotations(self.view)
self.view.set_read_only(self.view.settings().get("annotation_read_mode", False))
self.view.erase_status("html_annotation_mode")
class AddAnnotationCommand(sublime_plugin.TextCommand):
def is_visible(self):
return is_selected(self.view)
def run(self, edit):
AnnotateHtml(self.view).run()
class EditAnnotationCommand(sublime_plugin.TextCommand):
def is_visible(self):
return is_selection_in_annotation(self.view, first_only=True)
def run(self, edit):
AnnotateHtml(self.view).run()
class AnnotateHtml(object):
def __init__(self, view):
self.view = view
def subset_annotation_adjust(self):
subset = None
comment = ""
parent = None
intersect = False
for k, v in self.annotations["annotations"].items():
region = sublime.Region(int(v["region"][0]), int(v["region"][1]))
if region.contains(self.sel):
subset = region
comment = v["comment"]
parent = k
break
elif region.intersects(self.sel):
intersect = True
break
if subset is not None:
self.sel = subset
return comment, parent, intersect
def add_annotation(self, s, view_id, subset):
window = sublime.active_window()
view = window.active_view() if window is not None else None
if s != "" and view is not None and view_id == view.id():
if subset is None:
idx = self.annotations["count"]
key_name = ("html_annotation_%d" % idx)
else:
key_name = subset
self.annotations["annotations"][key_name] = {
"region": [self.sel.begin(), self.sel.end()],
"comment": s
}
if subset is None:
self.annotations["count"] += 1
self.view.settings().set("annotation_comments", self.annotations)
scope, style = get_highlight_style()
self.view.add_regions(
key_name,
[self.sel],
scope,
"",
style
)
def annotation_panel(self, default_comment, subset):
view_id = self.view.id()
self.view.window().show_input_panel(
("Annotate region (%d, %d)" % (self.sel.begin(), self.sel.end())),
default_comment,
lambda x: self.add_annotation(x, view_id=view_id, subset=subset),
None,
None
)
def run(self):
self.sel = self.view.sel()[0]
self.annotations = get_annotations(self.view)
comment, subset, intersects = self.subset_annotation_adjust()
if not intersects:
self.annotation_panel(comment, subset)
else:
error("Cannot have intersecting annotation regions!")