Jam/command_prompt.py

from kivy.uix.label import Label
from kivy.core.window import Window
from kivy.graphics import Color,Rectangle
import glob
import os.path

from tools import common_substr

class Command_prompt(Label):
    
    def __init__(self,app,**kwargs):
        # app is used to share information between widgets
        self.app=app

        # insert mode
        self.insert=False

        # width of a letter
        self.char_width=9

        # cursor position
        self.cursor=0

        # the text, with no color information
        self.raw_text=""

        # array of command with arguments
        self.argv=[]

        # init Label
        super(Command_prompt,self).__init__(**kwargs)

        self.keyboard = Window.request_keyboard(None,self,"text")
        self.keyboard.bind(on_textinput=self.on_textinput,on_key_down=self.on_key_down)

        self.draw()


    def draw(self):
        with self.canvas.before:
            # background
            Color(0,0,0)
            Rectangle(pos=self.pos,size=self.size)
            # cursor
            Color(1,1,1)
            Rectangle(pos=(self.pos[0]+self.cursor*self.char_width,self.pos[1]),size=(self.char_width,self.height))

        # make text under cursor black
        if self.cursor<len(self.raw_text):
            self.text=self.raw_text[:self.cursor]+"[color=000]"+self.raw_text[self.cursor]+"[/color]"+self.raw_text[self.cursor+1:]
        else:
            self.text=self.raw_text


    def on_key_down(self, keyboard, keycode, text, modifiers):
        #print(keycode,text,modifiers)
        if self.insert:
            # process modifiers
            mods=self.process_modifiers(modifiers)

            # clear status bar
            self.app.status_bar.text=""

            # process command
            if keycode[1]=="enter":
                self.insert=False
                self.parse_argv()
                self.set_text("")

            # move
            elif keycode[1]=="left":
                self.move_cursor_relative(-1)
            elif keycode[1]=="right":
                self.move_cursor_relative(1)
            elif mods==['c'] and text=="a":
                self.move_cursor(1)
            elif mods==['c'] and text=="e":
                self.move_cursor(len(self.raw_text))

            # delete
            elif keycode[1]=="backspace":
                self.backspace()
            elif keycode[1]=="delete":
                self.delete_forward()
            elif mods==['c'] and text=="k":
                self.delete_all_forward()

            # tab completion
            elif keycode[1]=="tab":
                self.tab_complete()

    def on_textinput(self,window,text):
        #print(text)
        if self.insert:
            self.append_text_at_cursor(text)
        elif text==':':
            self.append_text_at_cursor(text)
            self.insert=True


    # get modifiers from list
    # returns an alphabetically ordered array of characters
    # 'a' is alt
    # 'c' is ctrl
    # 'm' is meta
    # 's' is shift
    # ignore all others
    def process_modifiers(self,modifiers):
        out=[]
        for mod in modifiers:
            if mod=="alt":
                out.append('a')
            elif mod=="ctrl":
                out.append('c')
            elif mod=="meta":
                out.append('m')
            elif mod=="shift":
                out.append('s')
        out.sort()
        return out
                


    # append to text at position of cursor
    def append_text_at_cursor(self,text):
        if self.cursor==len(self.raw_text):
            self.raw_text+=text
        else:
            self.raw_text=self.raw_text[:self.cursor]+text+self.raw_text[self.cursor:]
        self.cursor+=len(text)
        self.draw()

    # delete before cursor
    def backspace(self):
        # do not delete leading ':'
        if self.cursor==1:
            return
        if self.cursor==len(self.raw_text):
            self.raw_text=self.raw_text[:self.cursor-1]
        else:
            self.raw_text=self.raw_text[:self.cursor-1]+self.raw_text[self.cursor:]
        self.cursor-=1
        self.draw()
    # delete after cursor
    def delete_forward(self):
        if self.cursor==len(self.raw_text):
            return
        else:
            self.raw_text=self.raw_text[:self.cursor]+self.raw_text[self.cursor+1:]
        self.draw()
    # delete until end of line(self):
    def delete_all_forward(self):
        if self.cursor==len(self.raw_text):
            return
        else:
            self.raw_text=self.raw_text[:self.cursor]
        self.draw()
        

    # set text in prompt
    def set_text(self,text):
        self.raw_text=text
        self.cursor=len(text)
        self.draw()


    # move cursor n steps to right (left if negative)
    def move_cursor_relative(self,n):
        if n==0 or (n>0 and self.cursor==len(self.raw_text)) or (n<0 and self.cursor==0):
            return
        self.cursor=max(1,min(len(self.raw_text),self.cursor+n))
        self.draw()

    # move cursor to absolute position
    def move_cursor(self,n):
        self.cursor=max(1,min(len(self.raw_text),n))
        self.draw()


    # parse text as argv
    def parse_argv(self):
        # init
        self.argv=[""]

        # return position of cursor
        cursor=(0,0)

        # whether inside quotes
        single_quoted=False
        double_quoted=False
        # whether after '\'
        backslashed=False

        # start after ':'
        # proceed one character at a time
        for i in range(1,len(self.raw_text)):
            char=self.raw_text[i]

            # split argument
            if single_quoted==False and double_quoted==False and backslashed==False and char==' ':
                # new argv
                self.argv.append("")
            
            # quotes or '\'
            elif double_quoted==False and backslashed==False and char=='\'':
                single_quoted=not single_quoted
            elif single_quoted==False and backslashed==False and char=='"':
                double_quoted=not double_quoted
            elif single_quoted==False and backslashed==False and char=='\\':
                backslashed=True

            # write character
            else:
                self.argv[len(self.argv)-1]+=char
                # reset backslash
                backslashed=False

            # record position of cursor
            if self.cursor==i+1:
                cursor=(len(self.argv)-1,len(self.argv[len(self.argv)-1]))

        
        return cursor


    # tab completion
    def tab_complete(self):
        # parse argv
        cursor=self.parse_argv()

        # write and edit commands
        if self.argv[0]=="w" or self.argv[0]=="e":
            # check that cursor is in first argument
            if cursor[0]==1:
                # complete filesystem path
                self.append_text_at_cursor(self.complete_path(self.argv[1],cursor[1]))

    # complete filesystem path
    def complete_path(self,base,end):
        paths=glob.glob(glob.escape(base[:end])+"*")
        print(glob.escape(base[:end])+"*", paths)
        if len(paths)==0:
            return ""
        elif len(paths)==1:
            # append '/' to directories
            if os.path.isdir(paths[0]):
                return paths[0][end:]+"/"
            else:
                return paths[0][end:]
        else:
            # display in status bar
            self.app.status_bar.raw_text=""
            for path in paths:
                self.app.status_bar.raw_text+="\""+path+"\" "
            self.app.status_bar.draw()
            return common_substr(paths)[end:]