def OnData(self, x_cord, y_cord, drag_result):
"""Gets and processes the dropped data
@postcondition: dropped data is processed
"""
self.window.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
if self.window.HasCapture():
self.window.ReleaseMouse()
try:
data = self.GetData()
except wx.PyAssertionError:
wx.PostEvent(self.window.GetTopLevelParent(), \
ed_event.StatusEvent(ed_event.edEVT_STATUS, -1,
_("Unable to accept dropped file "
"or text")))
data = False
drag_result = wx.DragCancel
if data:
files = self._data['fdata'].GetFilenames()
text = self._data['tdata'].GetText()
if len(files) > 0 and self._data['fcallb'] is not None:
self._data['fcallb'](files)
elif len(text) > 0:
if self._data['tcallb'] is not None:
self._data['tcallb'](text)
else:
self.window.DoDropText(x_cord, y_cord, text)
self.InitObjects()
return drag_result
def OnData(self, x_cord, y_cord, drag_result):
"""Gets and processes the dropped data
@postcondition: dropped data is processed
"""
self.window.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
if self.window.HasCapture():
self.window.ReleaseMouse()
try:
data = self.GetData()
except wx.PyAssertionError:
wx.PostEvent(self.window.GetTopLevelParent(), \
ed_event.StatusEvent(ed_event.edEVT_STATUS, -1,
_("Unable to open dropped file or "
"text")))
data = False
drag_result = wx.DragCancel
if data:
files = self._data['fdata'].GetFilenames()
text = self._data['tdata'].GetText()
if len(files) > 0 and self._data['fcallb'] is not None:
self._data['fcallb'](files)
elif (len(text) > 0):
if SetClipboardText(text):
win = self.window
pos = win.PositionFromPointClose(x_cord, y_cord)
if pos != wx.stc.STC_INVALID_POSITION:
win.SetSelection(pos, pos)
win.Paste()
else:
drag_result = wx.DragCancel
self.InitObjects()
return drag_result
def ProcessKey(self, key_code, ctrldown=False,
cmddown=False, shiftdown=False, altdown=False):
"""Processes keys and decided whether to interpret them as vim commands
or normal insert text
@param key_code: Raw key code
@keyword cmddown: Command/Ctrl key is down
@keyword shiftdown: Shift Key is down
@keyword altdown : Alt key is down
"""
if ctrldown or cmddown or altdown:
return False
# Mode may change after processing the key, so we need to remember
# whether this was a command or not
f_cmd = self.IsNormalMode() or self.IsVisualMode()
self._ProcessKey(key_code)
# Update status bar
if self.IsNormalMode():
if self.stc.GetTopLevelParent():
evt = ed_event.StatusEvent(ed_event.edEVT_STATUS,
self.stc.GetId(),
u"NORMAL %s" % self.buffer,
ed_glob.SB_BUFF)
wx.PostEvent(self.stc.GetTopLevelParent(), evt)
if f_cmd:
return True
else:
# If we're in insert mode we must return False
# so the text gets inserted into the editor
return False
def _SetMode(self, newmode, msg):
"""Set the keyhandlers mode
@param newmode: New mode name to change to
"""
self.buffer = u'' # Clear buffer from last mode
self.mode = newmode
# Update status bar
evt = ed_event.StatusEvent(ed_event.edEVT_STATUS, self.stc.GetId(),
msg, ed_glob.SB_BUFF)
wx.PostEvent(self.stc.GetTopLevelParent(), evt)
def SetMode(self, newmode):
"""Set the keyhandlers mode
@param newmode: New mode name to change to
"""
self.mode = newmode
self.last = self.cmdcache = u'' # Clear any partially procesed keys
if self.mode == ViKeyHandler.NORMAL:
self.stc.SetCaretWidth(10)
msg = u'NORMAL'
elif self.mode == ViKeyHandler.INSERT:
self.stc.SetCaretWidth(1)
msg = u'INSERT'
evt = ed_event.StatusEvent(ed_event.edEVT_STATUS, self.stc.GetId(),
msg, ed_glob.SB_BUFF)
wx.PostEvent(self.stc.GetTopLevelParent(), evt)
def ClearMode(self):
"""Clear any key input modes to normal input mode"""
evt = ed_event.StatusEvent(ed_event.edEVT_STATUS, self.stc.GetId(), '',
ed_glob.SB_BUFF)
wx.PostEvent(self.stc.GetTopLevelParent(), evt)
def ProcessKey(self,
key_code,
ctrldown=False,
cmddown=False,
shiftdown=False,
altdown=False):
"""Processes vi commands
@param key_code: Raw key code
@keyword cmddown: Command/Ctrl key is down
@keyword shiftdown: Shift Key is down
@keyword altdown : Alt key is down
@todo: complete rewrite, this was initially intended as a quick hack
put together for testing but now has implemented everything.
"""
if self.mode == ViKeyHandler.INSERT or ctrldown or cmddown or altdown:
return False
# Add key to cache
self.cmdcache = self.cmdcache + unichr(key_code)
# Don't process key if the cache is empty
if not len(self.cmdcache):
return False
# Check for repeat last action command
if self.cmdcache != u'.':
cmd = self.cmdcache
else:
cmd = self.last
# Gather common needed data
cpos = self.stc.GetCurrentPos()
cline = self.stc.LineFromPosition(cpos)
mw = self.stc.GetTopLevelParent()
# Check for change from NORMAL mode to Command mode
if u':' in cmd:
self.cmdcache = u''
mw.ShowCommandCtrl(ed_cmdbar.ID_CMD_CTRL)
# Single key commands
if len(cmd) == 1 and (cmd in 'AHILmM0^$nia/?:'):
if cmd in u'A$': # Insert at EOL
self.stc.GotoPos(self.stc.GetLineEndPosition(cline))
elif cmd == u'H': # Go first visible line # todo allow num
self.stc.GotoIndentPos(self.stc.GetFirstVisibleLine())
elif cmd in u'I^': # Insert at line start / Jump line start
self.stc.GotoIndentPos(cline)
elif cmd == u'0': # Jump to line start column 0
self.stc.GotoPos(self.stc.PositionFromLine(cline))
elif cmd == u'L': # Goto start of last visible line # todo allow num
self.stc.GotoIndentPos(self.stc.GetLastVisibleLine())
elif cmd == u'M': # Goto middle line of display
self.stc.GotoIndentPos(self.stc.GetMiddleVisibleLine())
elif cmd == u'm': # Mark line
if self.stc.MarkerGet(cline):
self.stc.Bookmark(ed_glob.ID_DEL_BM)
else:
self.stc.Bookmark(ed_glob.ID_ADD_BM)
elif cmd == u'a': # insert mode after current pos
self.stc.GotoPos(cpos + 1)
elif cmd in u'/?':
if mw is not None:
mw.ShowCommandCtrl(ed_cmdbar.ID_SEARCH_CTRL)
# Return to insert mode
if cmd in u'aAiI':
self.SetMode(ViKeyHandler.INSERT)
# Save last command and clear cache
self.last = cmd
self.cmdcache = u''
# Repeatable 1 key commands
elif re.match(VI_SINGLE_REPEAT, cmd):
rcmd = cmd[-1]
repeat = cmd[0:-1]
if repeat == u'':
repeat = 1
else:
repeat = int(repeat)
args = list()
kargs = dict()
cmd_map = {
u'b': self.stc.WordPartLeft,
u'B': self.stc.WordLeft,
u'e': self.stc.WordPartRightEnd,
u'E': self.stc.WordRightEnd,
u'h': self.stc.CharLeft,
u'j': self.stc.LineDown,
u'k': self.stc.LineUp,
u'l': self.stc.CharRight,
u'o': self.stc.AddLine,
u'O': self.stc.AddLine,
u'p': self.stc.Paste,
u'P': self.stc.Paste,
u's': self.stc.Cut,
u'u': self.stc.Undo,
u'w': self.stc.WordPartRight,
u'W': self.stc.WordRight,
u'x': self.stc.Cut,
u'X': self.stc.Cut,
u'{': self.stc.ParaUp,
u'}': self.stc.ParaDown,
u'~': self.stc.InvertCase
}
# Command is a Put command
if rcmd in u'pP':
success = False
newline = False
# Check the clipboard
if wx.TheClipboard.Open():
td = wx.TextDataObject()
success = wx.TheClipboard.GetData(td)
wx.TheClipboard.Close()
# We got the text from the clipboard so put it in the buffer
if success:
text = td.GetText()
if text[-1] == '\n':
if cline == self.stc.GetLineCount(
) - 1 and rcmd == u'p':
self.stc.NewLine()
else:
if rcmd == u'P':
self.stc.GotoLine(cline)
else:
self.stc.GotoLine(cline + 1)
newline = True
elif rcmd == u'p' and \
self.stc.LineFromPosition(cpos + 1) == cline:
self.stc.CharRight()
elif rcmd in u'sxX~':
if rcmd in u'sx~':
tmp = self.stc.GetTextRange(cpos, cpos + repeat)
tmp = tmp.split(self.stc.GetEOLChar())
end = cpos + len(tmp[0])
else:
tmp = self.stc.GetTextRange(cpos - repeat, cpos)
tmp = tmp.split(self.stc.GetEOLChar())
end = cpos - len(tmp[-1])
tmp = end
end = cpos
cpos = tmp
if cpos == self.stc.GetLineEndPosition(cline):
self.stc.SetSelection(cpos - 1, cpos)
else:
self.stc.SetSelection(cpos, end)
repeat = 1
elif rcmd == u'O':
kargs['before'] = True
# Start an undo action so all changes can be rolled back at once
self.stc.BeginUndoAction()
if rcmd in u'CD': # Cut line right
self.stc.SetSelection(
cpos, self.stc.GetLineEndPosition(cline + (repeat - 1)))
self.stc.Cut()
elif rcmd == u'J':
self.stc.SetTargetStart(cpos)
if repeat == 1:
repeat = 2
self.stc.SetTargetEnd(
self.stc.PositionFromLine(cline + repeat - 1))
self.stc.LinesJoin()
elif rcmd == u'G':
# Goto line or to end depending on context
if repeat == 1 and '1' not in cmd:
repeat = self.stc.GetLineCount()
self.stc.GotoLine(repeat - 1)
elif rcmd == u'+':
# Goto indent position at x lines from current line
self.stc.GotoIndentPos(cline + repeat)
elif rcmd == u'-':
# Goto indent position at x lines before current line
self.stc.GotoIndentPos(cline - repeat)
elif rcmd == u'|':
# Goto column number at current line
self.stc.GotoColumn(repeat - 1)
else:
if not cmd_map.has_key(rcmd):
return True
run = cmd_map[rcmd]
for count in xrange(repeat):
run(*args, **kargs)
# Do a Put command that is to be repeated
if rcmd == u'p':
if newline:
self.stc.GotoIndentPos(cline + repeat)
else:
self.stc.GotoPos(cpos + 1)
elif rcmd == u'P':
if newline:
self.stc.GotoIndentPos(cline)
else:
self.stc.GotoPos(cpos)
# elif rcmd == u'u':
# self.GotoPos(cpos)
elif rcmd in u'CoOs':
self.SetMode(ViKeyHandler.INSERT)
self.stc.EndUndoAction()
# Save this action and clear command cache
self.last = cmd
self.cmdcache = u''
# 2 key commands
elif re.match(VI_DOUBLE_P1, cmd) or \
re.match(VI_DOUBLE_P2, cmd) or \
re.match(re.compile('[cdy]0'), cmd):
# Handle repetitions
if re.match(re.compile('[cdy]0'), cmd):
rcmd = cmd
else:
rcmd = re.sub(NUM_PAT, u'', cmd)
# Get the number of repetitions from the command string
repeat = re.subn(re.compile(VI_DCMD_RIGHT), u'', cmd, 2)[0]
if repeat == u'':
repeat = 1
else:
repeat = int(repeat)
if rcmd[-1] not in u'bBeEGhHlLMwW$|{}0':
# Go to start of line
self.stc.SetCurrentPos(self.stc.GetLineStartPosition(cline))
if repeat != 1 or rcmd not in u'>><<':
self.stc.SetSelectionStart(self.stc.GetCurrentPos())
self.stc.SetSelectionEnd(
self.stc.PositionFromLine(cline + repeat))
else:
self.stc.SetAnchor(self.stc.GetCurrentPos())
mcmd = {
u'b': self.stc.WordPartLeftExtend,
u'B': self.stc.WordLeftExtend,
u'e': self.stc.WordPartRightEndExtend,
u'E': self.stc.WordRightEndExtend,
u'h': self.stc.CharLeftExtend,
u'l': self.stc.CharRightExtend,
u'w': self.stc.WordPartRightExtend,
u'W': self.stc.WordRightExtend,
u'{': self.stc.ParaUpExtend,
u'}': self.stc.ParaDownExtend
}
# Select to end of line
if u'$' in rcmd:
pos = self.stc.GetLineEndPosition(cline + repeat - \
len(self.stc.GetEOLChar()))
self.stc.SetCurrentPos(pos)
elif u'G' in rcmd:
# Check if command is valid
if repeat == 0:
self.cmdcache = u''
return True
# Select to end of file
if repeat == 1 and u'1' not in cmd: # Default eof
self.stc.SetAnchor(
self.stc.GetLineEndPosition(cline - 1))
repeat = self.stc.GetLength()
elif repeat < cline + 1:
self.stc.SetAnchor(self.stc.PositionFromLine(cline +
1))
repeat = self.stc.PositionFromLine(repeat - 1)
cline = self.stc.LineFromPosition(repeat) - 1
elif repeat > cline:
self.stc.SetAnchor(
self.stc.GetLineEndPosition(cline - 1))
if cline == 0:
repeat = self.stc.PositionFromLine(repeat)
else:
repeat = self.stc.GetLineEndPosition(repeat - 1)
else:
self.stc.SetAnchor(self.stc.PositionFromLine(cline))
repeat = self.stc.PositionFromLine(cline + 1)
self.stc.SetCurrentPos(repeat)
elif rcmd[-1] in u'HM':
fline = self.stc.GetFirstVisibleLine()
lline = self.stc.GetLastVisibleLine()
if u'M' in rcmd:
# Get the line in middle of visible lines
repeat = self.stc.GetMiddleVisibleLine() + 1
elif fline + repeat > lline:
repeat = lline
else:
repeat = fline + repeat
if repeat > cline:
# Move selection lines after middle line
self.stc.SetAnchor(self.stc.PositionFromLine(cline))
self.stc.SetCurrentPos(
self.stc.PositionFromLine(repeat))
else:
# Move select lines before middle line
self.stc.SetAnchor(
self.stc.PositionFromLine(repeat - 1))
self.stc.SetCurrentPos(
self.stc.PositionFromLine(cline + 1))
elif u'L' in rcmd:
fline = self.stc.GetFirstVisibleLine()
lline = self.stc.GetLastVisibleLine()
if lline - repeat < fline:
repeat = fline
else:
repeat = lline - repeat
if repeat < cline:
self.stc.SetAnchor(self.stc.PositionFromLine(cline))
self.stc.SetCurrentPos(
self.stc.PositionFromLine(repeat))
else:
self.stc.SetAnchor(self.stc.PositionFromLine(cline))
self.stc.SetCurrentPos(
self.stc.PositionFromLine(repeat + 2))
elif u'|' in rcmd:
if repeat == 1 and u'1' not in cmd:
repeat = 0
self.stc.SetCurrentCol(repeat)
elif rcmd[-1] == u'0':
self.stc.SetCurrentCol(0)
else:
doit = mcmd[rcmd[-1]]
for x in xrange(repeat):
doit()
# Begin Undo action so all changes can be rolled back in one undo
self.stc.BeginUndoAction()
if re.match(re.compile('c|c' + VI_DCMD_RIGHT), rcmd):
# Cut to selection and enter insert mode
if rcmd == u'cc':
self.stc.SetSelectionEnd(self.stc.GetSelectionEnd() - \
len(self.stc.GetEOLChar()))
self.stc.Cut()
self.SetMode(ViKeyHandler.INSERT)
elif re.match(re.compile('d|d' + VI_DCMD_RIGHT), rcmd):
# Cut selection and stay in normal mode
self.stc.Cut()
elif re.match(re.compile('y|y' + VI_DCMD_RIGHT), rcmd):
# Copy selection
self.stc.Copy()
self.stc.GotoPos(cpos)
elif rcmd == u'<<':
# Unindent selection
self.stc.BackTab()
elif rcmd == u'>>':
# Indent selection
self.stc.Tab()
else:
pass
self.stc.EndUndoAction()
if rcmd in '<<>>' or rcmd[-1] == u'G':
self.stc.GotoIndentPos(cline)
# Save the completed command and clear cache
self.last = cmd
self.cmdcache = u''
elif re.match(VI_GCMDS, cmd):
rcmd = cmd[-1]
if rcmd == u'g':
self.stc.GotoLine(0)
elif rcmd == u'f':
pass # TODO: gf (Goto file at cursor)
self.last = cmd
self.cmdcache = u''
else:
pass
# Update status bar
if mw and self.mode == ViKeyHandler.NORMAL:
evt = ed_event.StatusEvent(ed_event.edEVT_STATUS, self.stc.GetId(),
'NORMAL %s' % self.cmdcache,
ed_glob.SB_BUFF)
wx.PostEvent(self.stc.GetTopLevelParent(), evt)
return True
def ProcessKey(self, key_code):
"""Processes vi commands
@todo: complete rewrite, this was initially intended as a quick hack
put together for testing but now has implemented everything.
"""
if self.mode == ViKeyHandler.INSERT:
return False
self.cmdcache = self.cmdcache + unichr(key_code)
if not len(self.cmdcache):
return False
if self.cmdcache != u'.':
cmd = self.cmdcache
else:
cmd = self.last
cpos = self.stc.GetCurrentPos()
cline = self.stc.LineFromPosition(cpos)
mw = self.stc.GetTopLevelParent()
if u':' in cmd:
self.cmdcache = u''
mw.ShowCommandCtrl()
# Single key commands
if len(cmd) == 1 and (cmd in 'AHILmM0^$nia/?:'):
if cmd in u'A$': # Insert at EOL
self.stc.GotoPos(self.stc.GetLineEndPosition(cline))
elif cmd == u'H': # Go first visible line # todo allow num
self.stc.GotoIndentPos(self.stc.GetFirstVisibleLine())
elif cmd in u'I^': # Insert at line start / Jump line start
self.stc.GotoIndentPos(cline)
elif cmd == u'0': # Jump to line start column 0
self.stc.GotoPos(self.stc.PositionFromLine(cline))
elif cmd == u'L': # Goto start of last visible line # todo allow num
self.stc.GotoIndentPos(self.stc.GetLastVisibleLine())
elif cmd == u'M': # Goto middle line of display
self.stc.GotoIndentPos(self.stc.GetMiddleVisibleLine())
elif cmd == u'm': # Mark line
if self.stc.MarkerGet(cline):
self.stc.Bookmark(ed_glob.ID_DEL_BM)
else:
self.stc.Bookmark(ed_glob.ID_ADD_BM)
elif cmd == u'a': # insert mode after current pos
self.stc.GotoPos(cpos + 1)
elif cmd in u'/?':
if mw is not None:
evt = wx.MenuEvent(wx.wxEVT_COMMAND_MENU_SELECTED,
ed_glob.ID_QUICK_FIND)
wx.PostEvent(mw, evt)
if cmd in u'aAiI':
self.SetMode(ViKeyHandler.INSERT)
self.last = cmd
self.cmdcache = u''
# Repeatable 1 key commands
elif re.match(VI_SINGLE_REPEAT, cmd):
rcmd = cmd[-1]
repeat = cmd[0:-1]
if repeat == u'':
repeat = 1
else:
repeat = int(repeat)
args = list()
kargs = dict()
cmd_map = {
u'b': self.stc.WordPartLeft,
u'B': self.stc.WordLeft,
u'e': self.stc.WordPartRightEnd,
u'E': self.stc.WordRightEnd,
u'h': self.stc.CharLeft,
u'j': self.stc.LineDown,
u'k': self.stc.LineUp,
u'l': self.stc.CharRight,
u'o': self.stc.AddLine,
u'O': self.stc.AddLine,
u'p': self.stc.Paste,
u'P': self.stc.Paste,
u's': self.stc.Cut,
u'u': self.stc.Undo,
u'w': self.stc.WordPartRight,
u'W': self.stc.WordRight,
u'x': self.stc.Cut,
u'X': self.stc.Cut,
u'{': self.stc.ParaUp,
u'}': self.stc.ParaDown,
u'~': self.stc.InvertCase
}
if rcmd in u'pP':
success = False
newline = False
if wx.TheClipboard.Open():
td = wx.TextDataObject()
success = wx.TheClipboard.GetData(td)
wx.TheClipboard.Close()
if success:
text = td.GetText()
if text[-1] == '\n':
if cline == self.stc.GetLineCount(
) - 1 and rcmd == u'p':
self.stc.NewLine()
else:
if rcmd == u'P':
self.stc.GotoLine(cline)
else:
self.stc.GotoLine(cline + 1)
newline = True
elif rcmd == u'p' and \
self.stc.LineFromPosition(cpos + 1) == cline:
self.stc.CharRight()
elif rcmd in u'sxX~':
if rcmd in u'sx~':
tmp = self.stc.GetTextRange(cpos, cpos + repeat)
tmp = tmp.split(self.stc.GetEOLChar())
end = cpos + len(tmp[0])
else:
tmp = self.stc.GetTextRange(cpos - repeat, cpos)
tmp = tmp.split(self.stc.GetEOLChar())
end = cpos - len(tmp[-1])
tmp = end
end = cpos
cpos = tmp
if cpos == self.stc.GetLineEndPosition(cline):
self.stc.SetSelection(cpos - 1, cpos)
else:
self.stc.SetSelection(cpos, end)
repeat = 1
elif rcmd == u'O':
kargs['before'] = True
self.stc.BeginUndoAction()
if rcmd in u'CD': # Cut line right
self.stc.SetSelection(
cpos, self.stc.GetLineEndPosition(cline + (repeat - 1)))
self.stc.Cut()
elif rcmd == u'J':
self.stc.SetTargetStart(cpos)
if repeat == 1:
repeat = 2
self.stc.SetTargetEnd(
self.stc.PositionFromLine(cline + repeat - 1))
self.stc.LinesJoin()
elif rcmd == u'G':
if repeat == 1 and '1' not in cmd:
repeat = self.stc.GetLineCount()
self.stc.GotoLine(repeat - 1)
elif rcmd == u'+':
self.stc.GotoIndentPos(cline + repeat)
elif rcmd == u'-':
self.stc.GotoIndentPos(cline - repeat)
elif rcmd == u'|':
self.stc.GotoColumn(repeat - 1)
else:
if not cmd_map.has_key(rcmd):
return True
run = cmd_map[rcmd]
for count in xrange(repeat):
run(*args, **kargs)
if rcmd == u'p':
if newline:
self.stc.GotoIndentPos(cline + repeat)
else:
self.stc.GotoPos(cpos + 1)
elif rcmd == u'P':
if newline:
self.stc.GotoIndentPos(cline)
else:
self.stc.GotoPos(cpos)
# elif rcmd == u'u':
# self.GotoPos(cpos)
elif rcmd in u'CoOs':
self.SetMode(ViKeyHandler.INSERT)
self.stc.EndUndoAction()
self.last = cmd
self.cmdcache = u''
# 2 key commands
elif re.match(VI_DOUBLE_P1, cmd) or \
re.match(VI_DOUBLE_P2, cmd) or \
re.match(re.compile('[cdy]0'), cmd):
if re.match(re.compile('[cdy]0'), cmd):
rcmd = cmd
else:
rcmd = re.sub(NUM_PAT, u'', cmd)
repeat = re.subn(re.compile(VI_DCMD_RIGHT), u'', cmd, 2)[0]
if repeat == u'':
repeat = 1
else:
repeat = int(repeat)
if rcmd[-1] not in u'bBeEGhHlLMwW$|{}0':
self.stc.GotoLine(cline)
if repeat != 1 or rcmd not in u'>><<':
self.stc.SetSelectionStart(self.stc.GetCurrentPos())
self.stc.SetSelectionEnd(
self.stc.PositionFromLine(cline + repeat))
else:
self.stc.SetAnchor(self.stc.GetCurrentPos())
mcmd = {
u'b': self.stc.WordPartLeftExtend,
u'B': self.stc.WordLeftExtend,
u'e': self.stc.WordPartRightEndExtend,
u'E': self.stc.WordRightEndExtend,
u'h': self.stc.CharLeftExtend,
u'l': self.stc.CharRightExtend,
u'w': self.stc.WordPartRightExtend,
u'W': self.stc.WordRightExtend,
u'{': self.stc.ParaUpExtend,
u'}': self.stc.ParaDownExtend
}
if u'$' in rcmd:
pos = self.stc.GetLineEndPosition(cline + repeat - \
len(self.stc.GetEOLChar()))
self.stc.SetCurrentPos(pos)
elif u'G' in rcmd:
if repeat == 0: # invalid cmd
self.cmdcache = u''
return True
if repeat == 1 and u'1' not in cmd: # Default eof
self.stc.SetAnchor(
self.stc.GetLineEndPosition(cline - 1))
repeat = self.stc.GetLength()
elif repeat < cline + 1:
self.stc.SetAnchor(self.stc.PositionFromLine(cline +
1))
repeat = self.stc.PositionFromLine(repeat - 1)
cline = self.stc.LineFromPosition(repeat) - 1
elif repeat > cline:
self.stc.SetAnchor(
self.stc.GetLineEndPosition(cline - 1))
if cline == 0:
repeat = self.stc.PositionFromLine(repeat)
else:
repeat = self.stc.GetLineEndPosition(repeat - 1)
else:
self.stc.SetAnchor(self.stc.PositionFromLine(cline))
repeat = self.stc.PositionFromLine(cline + 1)
self.stc.SetCurrentPos(repeat)
elif rcmd[-1] in u'HM':
fline = self.stc.GetFirstVisibleLine()
lline = self.stc.GetLastVisibleLine()
if u'M' in rcmd:
repeat = self.stc.GetMiddleVisibleLine() + 1
elif fline + repeat > lline:
repeat = lline
else:
repeat = fline + repeat
if repeat > cline:
self.stc.SetAnchor(self.stc.PositionFromLine(cline))
self.stc.SetCurrentPos(
self.stc.PositionFromLine(repeat))
else:
self.stc.SetAnchor(
self.stc.PositionFromLine(repeat - 1))
self.stc.SetCurrentPos(
self.stc.PositionFromLine(cline + 1))
elif u'L' in rcmd:
fline = self.stc.GetFirstVisibleLine()
lline = self.stc.GetLastVisibleLine()
if lline - repeat < fline:
repeat = fline
else:
repeat = lline - repeat
if repeat < cline:
self.stc.SetAnchor(self.stc.PositionFromLine(cline))
self.stc.SetCurrentPos(
self.stc.PositionFromLine(repeat))
else:
self.stc.SetAnchor(self.stc.PositionFromLine(cline))
self.stc.SetCurrentPos(
self.stc.PositionFromLine(repeat + 2))
elif u'|' in rcmd:
if repeat == 1 and u'1' not in cmd:
repeat = 0
self.stc.SetCurrentCol(repeat)
elif rcmd[-1] == u'0':
self.stc.SetCurrentCol(0)
else:
doit = mcmd[rcmd[-1]]
for x in xrange(repeat):
doit()
self.stc.BeginUndoAction()
if re.match(re.compile('c|c' + VI_DCMD_RIGHT), rcmd):
if rcmd == u'cc':
self.stc.SetSelectionEnd(self.stc.GetSelectionEnd() - \
len(self.stc.GetEOLChar()))
self.stc.Cut()
self.SetMode(ViKeyHandler.INSERT)
elif re.match(re.compile('d|d' + VI_DCMD_RIGHT), rcmd):
self.stc.Cut()
elif re.match(re.compile('y|y' + VI_DCMD_RIGHT), rcmd):
self.stc.Copy()
self.stc.GotoPos(cpos)
elif rcmd == u'<<':
self.stc.BackTab()
elif rcmd == u'>>':
self.stc.Tab()
else:
pass
self.stc.EndUndoAction()
if rcmd in '<<>>' or rcmd[-1] == u'G':
self.stc.GotoIndentPos(cline)
self.last = cmd
self.cmdcache = u''
elif re.match(VI_GCMDS, cmd):
rcmd = cmd[-1]
if rcmd == u'g':
self.stc.GotoLine(0)
elif rcmd == u'f':
pass # TODO: gf (Goto file at cursor)
self.last = cmd
self.cmdcache = u''
else:
pass
# Update status bar
if mw and self.mode == ViKeyHandler.NORMAL:
evt = ed_event.StatusEvent(ed_event.edEVT_STATUS, self.stc.GetId(),
'NORMAL %s' % self.cmdcache,
ed_glob.SB_BUFF)
wx.PostEvent(self.stc.GetTopLevelParent(), evt)
return True
