FilesExpand file tree

 master

/

lua.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

57 lines (44 loc) · 1.54 KB

 master

/

lua.py

Copy path

Top

File metadata and controls

• Code
• Blame

57 lines (44 loc) · 1.54 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

'''

Language parser for Lua

'''

from .rubylike import RubylikeReader, RubylikeStateMachine

class LuaReader(RubylikeReader):

# pylint: disable=R0903

ext = ['lua']

language_names = ['lua']

@staticmethod

def generate_tokens(source_code, addition='', token_class=None):

return RubylikeReader.generate_tokens(

source_code,

r"|\-\-\[\[.*?\]\]" +

r"|\[\=*\[.*?\]\=*\]" +

r"|\-\-.*?$" +

addition)

def get_comment_from_token(self, token):

if token.startswith("--"):

return token

def __init__(self, context):

super(LuaReader, self).__init__(context)

self.parallel_states = [LuaStateMachine(context)]

class LuaStateMachine(RubylikeStateMachine):

FUNC_KEYWORD = 'function'

def __init__(self, context):

super(LuaStateMachine, self).__init__(context)

self.probable_function_name = None

def _state_global(self, token):

if token == "=":

self.next(self._assigning)

return

self.probable_function_name = token

super(LuaStateMachine, self)._state_global(token)

def _assigning(self, token):

if token == self.FUNC_KEYWORD:

self._state = self._anonymous_def

return

self.next(self._state_global, token)

def _anonymous_def(self, token):

if token != '(':

self.next(self._def, token)

return

self.context.push_new_function(self.probable_function_name)

self.next(self._def_parameters)