67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
|
# Python imports
|
||
|
|
|
||
|
|
# Lib imports
|
||
|
|
|
||
|
|
# Application imports
|
||
|
|
from .mixins.code_comment_tags_mixin import CodeCommentTagsMixin
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class Commenter(CodeCommentTagsMixin):
|
||
|
|
def __init__(self):
|
||
|
|
...
|
||
|
|
|
||
|
|
|
||
|
|
def keyboard_tggl_comment(self, buffer):
|
||
|
|
language = buffer.get_language()
|
||
|
|
if language is None: return
|
||
|
|
|
||
|
|
start_tag, end_tag = self.get_comment_tags(language)
|
||
|
|
# Note: Only handling line comment tag- no block comment option
|
||
|
|
if not start_tag and not end_tag: return
|
||
|
|
|
||
|
|
bounds = buffer.get_selection_bounds()
|
||
|
|
if bounds:
|
||
|
|
self._bounds_comment(
|
||
|
|
start_tag, end_tag, bounds, buffer
|
||
|
|
)
|
||
|
|
else:
|
||
|
|
self._line_comment(start_tag, end_tag, buffer)
|
||
|
|
|
||
|
|
|
||
|
|
def _line_comment(self, start_tag, end_tag, buffer):
|
||
|
|
start_itr = buffer.get_iter_at_mark( buffer.get_insert() ).copy()
|
||
|
|
end_itr = start_itr.copy()
|
||
|
|
if not start_itr.starts_line():
|
||
|
|
start_itr.set_line_offset(0)
|
||
|
|
if not end_itr.ends_line():
|
||
|
|
end_itr.forward_to_line_end()
|
||
|
|
|
||
|
|
text = buffer.get_text(start_itr, end_itr, True)
|
||
|
|
text = text.replace(start_tag, "") if text.startswith(start_tag) else start_tag + text
|
||
|
|
|
||
|
|
buffer.begin_user_action()
|
||
|
|
buffer.delete(start_itr, end_itr)
|
||
|
|
buffer.insert(start_itr, text)
|
||
|
|
buffer.end_user_action()
|
||
|
|
|
||
|
|
|
||
|
|
def _bounds_comment(self, start_tag, end_tag, bounds, buffer):
|
||
|
|
start_itr, end_itr = bounds
|
||
|
|
if not start_itr.starts_line():
|
||
|
|
start_itr.set_line_offset(0)
|
||
|
|
if not end_itr.ends_line():
|
||
|
|
end_itr.forward_to_line_end()
|
||
|
|
|
||
|
|
text = buffer.get_text(start_itr, end_itr, True)
|
||
|
|
text = "\n".join(
|
||
|
|
line.replace(start_tag, "") if line.startswith(start_tag) else start_tag + line
|
||
|
|
for line in text.splitlines()
|
||
|
|
)
|
||
|
|
|
||
|
|
buffer.begin_user_action()
|
||
|
|
buffer.delete(start_itr, end_itr)
|
||
|
|
buffer.insert(start_itr, text)
|
||
|
|
buffer.end_user_action()
|
||
|
|
|