parser.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983
  1. /*
  2. * ECAL
  3. *
  4. * Copyright 2020 Matthias Ladkau. All rights reserved.
  5. *
  6. * This Source Code Form is subject to the terms of the MIT
  7. * License, If a copy of the MIT License was not distributed with this
  8. * file, You can obtain one at https://opensource.org/licenses/MIT.
  9. */
  10. package parser
  11. import (
  12. "fmt"
  13. )
  14. /*
  15. Map of AST nodes corresponding to lexer tokens. The map determines how a given
  16. sequence of lexer tokens are organized into an AST.
  17. */
  18. var astNodeMap map[LexTokenID]*ASTNode
  19. func init() {
  20. astNodeMap = map[LexTokenID]*ASTNode{
  21. TokenEOF: {NodeEOF, nil, nil, nil, nil, 0, ndTerm, nil},
  22. // Value tokens
  23. TokenSTRING: {NodeSTRING, nil, nil, nil, nil, 0, ndTerm, nil},
  24. TokenNUMBER: {NodeNUMBER, nil, nil, nil, nil, 0, ndTerm, nil},
  25. TokenIDENTIFIER: {NodeIDENTIFIER, nil, nil, nil, nil, 0, ndIdentifier, nil},
  26. // Constructed tokens
  27. TokenSTATEMENTS: {NodeSTATEMENTS, nil, nil, nil, nil, 0, nil, nil},
  28. TokenFUNCCALL: {NodeFUNCCALL, nil, nil, nil, nil, 0, nil, nil},
  29. TokenCOMPACCESS: {NodeCOMPACCESS, nil, nil, nil, nil, 0, nil, nil},
  30. TokenLIST: {NodeLIST, nil, nil, nil, nil, 0, nil, nil},
  31. TokenMAP: {NodeMAP, nil, nil, nil, nil, 0, nil, nil},
  32. TokenPARAMS: {NodePARAMS, nil, nil, nil, nil, 0, nil, nil},
  33. TokenGUARD: {NodeGUARD, nil, nil, nil, nil, 0, nil, nil},
  34. // Condition operators
  35. TokenGEQ: {NodeGEQ, nil, nil, nil, nil, 60, nil, ldInfix},
  36. TokenLEQ: {NodeLEQ, nil, nil, nil, nil, 60, nil, ldInfix},
  37. TokenNEQ: {NodeNEQ, nil, nil, nil, nil, 60, nil, ldInfix},
  38. TokenEQ: {NodeEQ, nil, nil, nil, nil, 60, nil, ldInfix},
  39. TokenGT: {NodeGT, nil, nil, nil, nil, 60, nil, ldInfix},
  40. TokenLT: {NodeLT, nil, nil, nil, nil, 60, nil, ldInfix},
  41. // Grouping symbols
  42. TokenLPAREN: {"", nil, nil, nil, nil, 150, ndInner, nil},
  43. TokenRPAREN: {"", nil, nil, nil, nil, 0, nil, nil},
  44. TokenLBRACK: {"", nil, nil, nil, nil, 150, ndList, nil},
  45. TokenRBRACK: {"", nil, nil, nil, nil, 0, nil, nil},
  46. TokenLBRACE: {"", nil, nil, nil, nil, 150, ndMap, nil},
  47. TokenRBRACE: {"", nil, nil, nil, nil, 0, nil, nil},
  48. // Separators
  49. TokenDOT: {"", nil, nil, nil, nil, 0, nil, nil},
  50. TokenCOMMA: {"", nil, nil, nil, nil, 0, nil, nil},
  51. TokenSEMICOLON: {"", nil, nil, nil, nil, 0, nil, nil},
  52. // Grouping
  53. TokenCOLON: {NodeKVP, nil, nil, nil, nil, 60, nil, ldInfix},
  54. TokenEQUAL: {NodePRESET, nil, nil, nil, nil, 60, nil, ldInfix},
  55. // Arithmetic operators
  56. TokenPLUS: {NodePLUS, nil, nil, nil, nil, 110, ndPrefix, ldInfix},
  57. TokenMINUS: {NodeMINUS, nil, nil, nil, nil, 110, ndPrefix, ldInfix},
  58. TokenTIMES: {NodeTIMES, nil, nil, nil, nil, 120, nil, ldInfix},
  59. TokenDIV: {NodeDIV, nil, nil, nil, nil, 120, nil, ldInfix},
  60. TokenDIVINT: {NodeDIVINT, nil, nil, nil, nil, 120, nil, ldInfix},
  61. TokenMODINT: {NodeMODINT, nil, nil, nil, nil, 120, nil, ldInfix},
  62. // Assignment statement
  63. TokenASSIGN: {NodeASSIGN, nil, nil, nil, nil, 10, nil, ldInfix},
  64. // Import statement
  65. TokenIMPORT: {NodeIMPORT, nil, nil, nil, nil, 0, ndImport, nil},
  66. TokenAS: {NodeAS, nil, nil, nil, nil, 0, nil, nil},
  67. // Sink definition
  68. TokenSINK: {NodeSINK, nil, nil, nil, nil, 0, ndSkink, nil},
  69. TokenKINDMATCH: {NodeKINDMATCH, nil, nil, nil, nil, 150, ndPrefix, nil},
  70. TokenSCOPEMATCH: {NodeSCOPEMATCH, nil, nil, nil, nil, 150, ndPrefix, nil},
  71. TokenSTATEMATCH: {NodeSTATEMATCH, nil, nil, nil, nil, 150, ndPrefix, nil},
  72. TokenPRIORITY: {NodePRIORITY, nil, nil, nil, nil, 150, ndPrefix, nil},
  73. TokenSUPPRESSES: {NodeSUPPRESSES, nil, nil, nil, nil, 150, ndPrefix, nil},
  74. // Function definition
  75. TokenFUNC: {NodeFUNC, nil, nil, nil, nil, 0, ndFunc, nil},
  76. TokenRETURN: {NodeRETURN, nil, nil, nil, nil, 0, ndReturn, nil},
  77. // Boolean operators
  78. TokenAND: {NodeAND, nil, nil, nil, nil, 40, nil, ldInfix},
  79. TokenOR: {NodeOR, nil, nil, nil, nil, 30, nil, ldInfix},
  80. TokenNOT: {NodeNOT, nil, nil, nil, nil, 20, ndPrefix, nil},
  81. // Condition operators
  82. TokenLIKE: {NodeLIKE, nil, nil, nil, nil, 60, nil, ldInfix},
  83. TokenIN: {NodeIN, nil, nil, nil, nil, 60, nil, ldInfix},
  84. TokenHASPREFIX: {NodeHASPREFIX, nil, nil, nil, nil, 60, nil, ldInfix},
  85. TokenHASSUFFIX: {NodeHASSUFFIX, nil, nil, nil, nil, 60, nil, ldInfix},
  86. TokenNOTIN: {NodeNOTIN, nil, nil, nil, nil, 60, nil, ldInfix},
  87. // Constant terminals
  88. TokenFALSE: {NodeFALSE, nil, nil, nil, nil, 0, ndTerm, nil},
  89. TokenTRUE: {NodeTRUE, nil, nil, nil, nil, 0, ndTerm, nil},
  90. TokenNULL: {NodeNULL, nil, nil, nil, nil, 0, ndTerm, nil},
  91. // Conditional statements
  92. TokenIF: {NodeIF, nil, nil, nil, nil, 0, ndGuard, nil},
  93. TokenELIF: {"", nil, nil, nil, nil, 0, nil, nil},
  94. TokenELSE: {"", nil, nil, nil, nil, 0, nil, nil},
  95. // Loop statement
  96. TokenFOR: {NodeLOOP, nil, nil, nil, nil, 0, ndLoop, nil},
  97. TokenBREAK: {NodeBREAK, nil, nil, nil, nil, 0, ndTerm, nil},
  98. TokenCONTINUE: {NodeCONTINUE, nil, nil, nil, nil, 0, ndTerm, nil},
  99. // Try statement
  100. TokenTRY: {NodeTRY, nil, nil, nil, nil, 0, ndTry, nil},
  101. TokenEXCEPT: {NodeEXCEPT, nil, nil, nil, nil, 0, nil, nil},
  102. TokenFINALLY: {NodeFINALLY, nil, nil, nil, nil, 0, nil, nil},
  103. }
  104. }
  105. // Parser
  106. // ======
  107. /*
  108. Parser data structure
  109. */
  110. type parser struct {
  111. name string // Name to identify the input
  112. node *ASTNode // Current ast node
  113. tokens *LABuffer // Buffer which is connected to the channel which contains lex tokens
  114. rp RuntimeProvider // Runtime provider which creates runtime components
  115. }
  116. /*
  117. Parse parses a given input string and returns an AST.
  118. */
  119. func Parse(name string, input string) (*ASTNode, error) {
  120. return ParseWithRuntime(name, input, nil)
  121. }
  122. /*
  123. ParseWithRuntime parses a given input string and returns an AST decorated with
  124. runtime components.
  125. */
  126. func ParseWithRuntime(name string, input string, rp RuntimeProvider) (*ASTNode, error) {
  127. // Create a new parser with a look-ahead buffer of 3
  128. p := &parser{name, nil, NewLABuffer(Lex(name, input), 3), rp}
  129. // Read and set initial AST node
  130. node, err := p.next()
  131. if err != nil {
  132. return nil, err
  133. }
  134. p.node = node
  135. n, err := p.run(0)
  136. if err == nil && hasMoreStatements(p, n) {
  137. st := astNodeMap[TokenSTATEMENTS].instance(p, nil)
  138. st.Children = append(st.Children, n)
  139. for err == nil && hasMoreStatements(p, n) {
  140. // Skip semicolons
  141. if p.node.Token.ID == TokenSEMICOLON {
  142. skipToken(p, TokenSEMICOLON)
  143. }
  144. n, err = p.run(0)
  145. st.Children = append(st.Children, n)
  146. }
  147. n = st
  148. }
  149. if err == nil && p.node != nil && p.node.Token.ID != TokenEOF {
  150. token := *p.node.Token
  151. err = p.newParserError(ErrUnexpectedEnd, fmt.Sprintf("extra token id:%v (%v)",
  152. token.ID, token), token)
  153. }
  154. return n, err
  155. }
  156. /*
  157. run models the main parser function.
  158. */
  159. func (p *parser) run(rightBinding int) (*ASTNode, error) {
  160. var err error
  161. n := p.node
  162. p.node, err = p.next()
  163. if err != nil {
  164. return nil, err
  165. }
  166. // Start with the null denotation of this statement / expression
  167. if n.nullDenotation == nil {
  168. return nil, p.newParserError(ErrImpossibleNullDenotation,
  169. n.Token.String(), *n.Token)
  170. }
  171. left, err := n.nullDenotation(p, n)
  172. if err != nil {
  173. return nil, err
  174. }
  175. // Collect left denotations as long as the left binding power is greater
  176. // than the initial right one
  177. for rightBinding < p.node.binding {
  178. var nleft *ASTNode
  179. n = p.node
  180. if n.leftDenotation == nil {
  181. if left.Token.Lline < n.Token.Lline {
  182. // If the impossible left denotation is on a new line
  183. // we might be parsing a new statement
  184. return left, nil
  185. }
  186. return nil, p.newParserError(ErrImpossibleLeftDenotation,
  187. n.Token.String(), *n.Token)
  188. }
  189. p.node, err = p.next()
  190. if err != nil {
  191. return nil, err
  192. }
  193. // Get the next left denotation
  194. nleft, err = n.leftDenotation(p, n, left)
  195. left = nleft
  196. if err != nil {
  197. return nil, err
  198. }
  199. }
  200. return left, nil
  201. }
  202. /*
  203. next retrieves the next lexer token.
  204. */
  205. func (p *parser) next() (*ASTNode, error) {
  206. var preComments []MetaData
  207. var postComments []MetaData
  208. token, more := p.tokens.Next()
  209. // Skip over pre comment token
  210. for more && token.ID == TokenPRECOMMENT {
  211. preComments = append(preComments, NewLexTokenInstance(token))
  212. token, more = p.tokens.Next()
  213. }
  214. // Skip over post comment token
  215. for more && token.ID == TokenPOSTCOMMENT {
  216. postComments = append(postComments, NewLexTokenInstance(token))
  217. token, more = p.tokens.Next()
  218. }
  219. if !more {
  220. // Unexpected end of input - the associated token is an empty error token
  221. return nil, p.newParserError(ErrUnexpectedEnd, "", token)
  222. } else if token.ID == TokenError {
  223. // There was a lexer error wrap it in a parser error
  224. return nil, p.newParserError(ErrLexicalError, token.Val, token)
  225. } else if node, ok := astNodeMap[token.ID]; ok {
  226. // We got a normal AST component
  227. ret := node.instance(p, &token)
  228. ret.Meta = append(ret.Meta, preComments...) // Attach pre comments to the next AST node
  229. if len(postComments) > 0 && p.node != nil {
  230. p.node.Meta = append(p.node.Meta, postComments...) // Attach post comments to the previous AST node
  231. }
  232. return ret, nil
  233. }
  234. return nil, p.newParserError(ErrUnknownToken, fmt.Sprintf("id:%v (%v)", token.ID, token), token)
  235. }
  236. // Standard null denotation functions
  237. // ==================================
  238. /*
  239. ndTerm is used for terminals.
  240. */
  241. func ndTerm(p *parser, self *ASTNode) (*ASTNode, error) {
  242. return self, nil
  243. }
  244. /*
  245. ndInner returns the inner expression of an enclosed block and discard the
  246. block token. This method is used for brackets.
  247. */
  248. func ndInner(p *parser, self *ASTNode) (*ASTNode, error) {
  249. // Get the inner expression
  250. exp, err := p.run(0)
  251. if err != nil {
  252. return nil, err
  253. }
  254. // We return here the inner expression - discarding the bracket tokens
  255. return exp, skipToken(p, TokenRPAREN)
  256. }
  257. /*
  258. ndPrefix is used for prefix operators.
  259. */
  260. func ndPrefix(p *parser, self *ASTNode) (*ASTNode, error) {
  261. // Make sure a prefix will only prefix the next item
  262. val, err := p.run(self.binding + 20)
  263. if err != nil {
  264. return nil, err
  265. }
  266. self.Children = append(self.Children, val)
  267. return self, nil
  268. }
  269. // Null denotation functions for specific expressions
  270. // ==================================================
  271. /*
  272. ndImport is used to parse imports.
  273. */
  274. func ndImport(p *parser, self *ASTNode) (*ASTNode, error) {
  275. // Must specify a file path
  276. err := acceptChild(p, self, TokenSTRING)
  277. if err == nil {
  278. // Must specify AS
  279. if err = skipToken(p, TokenAS); err == nil {
  280. // Must specify an identifier
  281. err = acceptChild(p, self, TokenIDENTIFIER)
  282. }
  283. }
  284. return self, err
  285. }
  286. /*
  287. ndSink is used to parse sinks.
  288. */
  289. func ndSkink(p *parser, self *ASTNode) (*ASTNode, error) {
  290. var exp, ret *ASTNode
  291. // Must specify a name
  292. err := acceptChild(p, self, TokenIDENTIFIER)
  293. if err == nil {
  294. // Parse the rest of the parameters as children until we reach the body
  295. for err == nil && IsNotEndAndNotToken(p, TokenLBRACE) {
  296. if exp, err = p.run(150); err == nil {
  297. self.Children = append(self.Children, exp)
  298. // Skip commas
  299. if p.node.Token.ID == TokenCOMMA {
  300. err = skipToken(p, TokenCOMMA)
  301. }
  302. }
  303. }
  304. if err == nil {
  305. // Parse the body
  306. ret, err = parseInnerStatements(p, self)
  307. }
  308. }
  309. return ret, err
  310. }
  311. /*
  312. ndFunc is used to parse function definitions.
  313. */
  314. func ndFunc(p *parser, self *ASTNode) (*ASTNode, error) {
  315. var exp *ASTNode
  316. var err error
  317. // Might specify a function name
  318. if p.node.Token.ID == TokenIDENTIFIER {
  319. err = acceptChild(p, self, TokenIDENTIFIER)
  320. }
  321. // Read in parameters
  322. if err == nil {
  323. err = skipToken(p, TokenLPAREN)
  324. params := astNodeMap[TokenPARAMS].instance(p, nil)
  325. self.Children = append(self.Children, params)
  326. for err == nil && IsNotEndAndNotToken(p, TokenRPAREN) {
  327. // Parse all the expressions inside
  328. if exp, err = p.run(0); err == nil {
  329. params.Children = append(params.Children, exp)
  330. if p.node.Token.ID == TokenCOMMA {
  331. err = skipToken(p, TokenCOMMA)
  332. }
  333. }
  334. }
  335. if err == nil {
  336. err = skipToken(p, TokenRPAREN)
  337. }
  338. }
  339. if err == nil {
  340. // Parse the body
  341. self, err = parseInnerStatements(p, self)
  342. }
  343. return self, err
  344. }
  345. /*
  346. ndReturn is used to parse return statements.
  347. */
  348. func ndReturn(p *parser, self *ASTNode) (*ASTNode, error) {
  349. var err error
  350. if self.Token.Lline == p.node.Token.Lline {
  351. var val *ASTNode
  352. // Consume the next expression only if it is on the same line
  353. val, err = p.run(0)
  354. if err == nil {
  355. self.Children = append(self.Children, val)
  356. }
  357. }
  358. return self, err
  359. }
  360. /*
  361. ndIdentifier is to parse identifiers and function calls.
  362. */
  363. func ndIdentifier(p *parser, self *ASTNode) (*ASTNode, error) {
  364. var parseMore, parseSegment, parseFuncCall, parseCompositionAccess func(parent *ASTNode) error
  365. parseMore = func(current *ASTNode) error {
  366. var err error
  367. if p.node.Token.ID == TokenDOT {
  368. err = parseSegment(current)
  369. } else if p.node.Token.ID == TokenLPAREN {
  370. err = parseFuncCall(current)
  371. } else if p.node.Token.ID == TokenLBRACK && p.node.Token.Lline == self.Token.Lline {
  372. // Composition access needs to be on the same line as the identifier
  373. // as we might otherwise have a list
  374. err = parseCompositionAccess(current)
  375. }
  376. return err
  377. }
  378. parseSegment = func(current *ASTNode) error {
  379. var err error
  380. var next *ASTNode
  381. if err = skipToken(p, TokenDOT); err == nil {
  382. next = p.node
  383. if err = acceptChild(p, current, TokenIDENTIFIER); err == nil {
  384. err = parseMore(next)
  385. }
  386. }
  387. return err
  388. }
  389. parseFuncCall = func(current *ASTNode) error {
  390. var exp *ASTNode
  391. err := skipToken(p, TokenLPAREN)
  392. fc := astNodeMap[TokenFUNCCALL].instance(p, nil)
  393. current.Children = append(current.Children, fc)
  394. // Read in parameters
  395. for err == nil && IsNotEndAndNotToken(p, TokenRPAREN) {
  396. // Parse all the expressions inside the directives
  397. if exp, err = p.run(0); err == nil {
  398. fc.Children = append(fc.Children, exp)
  399. if p.node.Token.ID == TokenCOMMA {
  400. err = skipToken(p, TokenCOMMA)
  401. }
  402. }
  403. }
  404. if err == nil {
  405. err = skipToken(p, TokenRPAREN)
  406. if err == nil {
  407. err = parseMore(current)
  408. }
  409. }
  410. return err
  411. }
  412. parseCompositionAccess = func(current *ASTNode) error {
  413. var exp *ASTNode
  414. err := skipToken(p, TokenLBRACK)
  415. if err == nil {
  416. ca := astNodeMap[TokenCOMPACCESS].instance(p, nil)
  417. current.Children = append(current.Children, ca)
  418. // Parse all the expressions inside the directives
  419. if exp, err = p.run(0); err == nil {
  420. ca.Children = append(ca.Children, exp)
  421. if err = skipToken(p, TokenRBRACK); err == nil {
  422. err = parseMore(current)
  423. }
  424. }
  425. }
  426. return err
  427. }
  428. return self, parseMore(self)
  429. }
  430. /*
  431. ndList is used to collect elements of a list.
  432. */
  433. func ndList(p *parser, self *ASTNode) (*ASTNode, error) {
  434. var err error
  435. var exp *ASTNode
  436. // Create a list token
  437. st := astNodeMap[TokenLIST].instance(p, self.Token)
  438. // Get the inner expression
  439. for err == nil && IsNotEndAndNotToken(p, TokenRBRACK) {
  440. // Parse all the expressions inside
  441. if exp, err = p.run(0); err == nil {
  442. st.Children = append(st.Children, exp)
  443. if p.node.Token.ID == TokenCOMMA {
  444. err = skipToken(p, TokenCOMMA)
  445. }
  446. }
  447. }
  448. if err == nil {
  449. err = skipToken(p, TokenRBRACK)
  450. }
  451. // Must have a closing bracket
  452. return st, err
  453. }
  454. /*
  455. ndMap is used to collect elements of a map.
  456. */
  457. func ndMap(p *parser, self *ASTNode) (*ASTNode, error) {
  458. var err error
  459. var exp *ASTNode
  460. // Create a map token
  461. st := astNodeMap[TokenMAP].instance(p, self.Token)
  462. // Get the inner expression
  463. for err == nil && IsNotEndAndNotToken(p, TokenRBRACE) {
  464. // Parse all the expressions inside
  465. if exp, err = p.run(0); err == nil {
  466. st.Children = append(st.Children, exp)
  467. if p.node.Token.ID == TokenCOMMA {
  468. err = skipToken(p, TokenCOMMA)
  469. }
  470. }
  471. }
  472. if err == nil {
  473. err = skipToken(p, TokenRBRACE)
  474. }
  475. // Must have a closing brace
  476. return st, err
  477. }
  478. /*
  479. ndGuard is used to parse a conditional statement.
  480. */
  481. func ndGuard(p *parser, self *ASTNode) (*ASTNode, error) {
  482. var err error
  483. parseGuardAndStatements := func() error {
  484. // The brace starts statements while parsing the expression of an if statement
  485. nodeMapEntryBak := astNodeMap[TokenLBRACE]
  486. astNodeMap[TokenLBRACE] = &ASTNode{"", nil, nil, nil, nil, 0, parseInnerStatements, nil}
  487. exp, err := p.run(0)
  488. astNodeMap[TokenLBRACE] = nodeMapEntryBak
  489. if err == nil {
  490. g := astNodeMap[TokenGUARD].instance(p, nil)
  491. g.Children = append(g.Children, exp)
  492. self.Children = append(self.Children, g)
  493. _, err = parseInnerStatements(p, self)
  494. }
  495. return err
  496. }
  497. if err = parseGuardAndStatements(); err == nil {
  498. for err == nil && IsNotEndAndToken(p, TokenELIF) {
  499. // Parse an elif
  500. if err = skipToken(p, TokenELIF); err == nil {
  501. err = parseGuardAndStatements()
  502. }
  503. }
  504. if err == nil && p.node.Token.ID == TokenELSE {
  505. // Parse else
  506. if err = skipToken(p, TokenELSE); err == nil {
  507. g := astNodeMap[TokenGUARD].instance(p, nil)
  508. g.Children = append(g.Children, astNodeMap[TokenTRUE].instance(p, nil))
  509. self.Children = append(self.Children, g)
  510. _, err = parseInnerStatements(p, self)
  511. }
  512. }
  513. }
  514. return self, err
  515. }
  516. /*
  517. ndLoop is used to parse a loop statement.
  518. */
  519. func ndLoop(p *parser, self *ASTNode) (*ASTNode, error) {
  520. // The brace starts statements while parsing the expression of a for statement
  521. nodeMapEntryBak := astNodeMap[TokenLBRACE]
  522. astNodeMap[TokenLBRACE] = &ASTNode{"", nil, nil, nil, nil, 0, parseInnerStatements, nil}
  523. exp, err := p.run(0)
  524. astNodeMap[TokenLBRACE] = nodeMapEntryBak
  525. if err == nil {
  526. g := exp
  527. if exp.Token.ID != TokenIN {
  528. g = astNodeMap[TokenGUARD].instance(p, nil)
  529. g.Children = append(g.Children, exp)
  530. }
  531. self.Children = append(self.Children, g)
  532. _, err = parseInnerStatements(p, self)
  533. }
  534. return self, err
  535. }
  536. /*
  537. ndTry is used to parse a try block.
  538. */
  539. func ndTry(p *parser, self *ASTNode) (*ASTNode, error) {
  540. try, err := parseInnerStatements(p, self)
  541. if p.node.Token.ID != TokenFINALLY {
  542. for err == nil && IsNotEndAndToken(p, TokenEXCEPT) {
  543. except := p.node
  544. err = acceptChild(p, try, TokenEXCEPT)
  545. if err == nil {
  546. for err == nil &&
  547. IsNotEndAndNotToken(p, TokenAS) &&
  548. IsNotEndAndNotToken(p, TokenIDENTIFIER) &&
  549. IsNotEndAndNotToken(p, TokenLBRACE) {
  550. if err = acceptChild(p, except, TokenSTRING); err == nil {
  551. // Skip commas
  552. if p.node.Token.ID == TokenCOMMA {
  553. err = skipToken(p, TokenCOMMA)
  554. }
  555. }
  556. }
  557. if err == nil {
  558. if p.node.Token.ID == TokenAS {
  559. as := p.node
  560. err = acceptChild(p, except, TokenAS)
  561. if err == nil {
  562. err = acceptChild(p, as, TokenIDENTIFIER)
  563. }
  564. } else if p.node.Token.ID == TokenIDENTIFIER {
  565. err = acceptChild(p, except, TokenIDENTIFIER)
  566. }
  567. }
  568. if err == nil {
  569. _, err = parseInnerStatements(p, except)
  570. }
  571. }
  572. }
  573. }
  574. if err == nil && p.node.Token.ID == TokenFINALLY {
  575. finally := p.node
  576. if err = acceptChild(p, try, TokenFINALLY); err == nil {
  577. _, err = parseInnerStatements(p, finally)
  578. }
  579. }
  580. return try, err
  581. }
  582. // Standard left denotation functions
  583. // ==================================
  584. /*
  585. ldInfix is used for infix operators.
  586. */
  587. func ldInfix(p *parser, self *ASTNode, left *ASTNode) (*ASTNode, error) {
  588. right, err := p.run(self.binding)
  589. if err != nil {
  590. return nil, err
  591. }
  592. self.Children = append(self.Children, left)
  593. self.Children = append(self.Children, right)
  594. return self, nil
  595. }
  596. // Helper functions
  597. // ================
  598. /*
  599. IsNotEndAndToken checks if the next token is of a specific type or the end has been reached.
  600. */
  601. func IsNotEndAndToken(p *parser, i LexTokenID) bool {
  602. return p.node != nil && p.node.Name != NodeEOF && p.node.Token.ID == i
  603. }
  604. /*
  605. IsNotEndAndNotToken checks if the next token is not of a specific type or the end has been reached.
  606. */
  607. func IsNotEndAndNotToken(p *parser, i LexTokenID) bool {
  608. return p.node != nil && p.node.Name != NodeEOF && p.node.Token.ID != i
  609. }
  610. /*
  611. hasMoreStatements returns true if there are more statements to parse.
  612. */
  613. func hasMoreStatements(p *parser, currentNode *ASTNode) bool {
  614. nextNode := p.node
  615. if nextNode == nil || nextNode.Token.ID == TokenEOF {
  616. return false
  617. } else if nextNode.Token.ID == TokenSEMICOLON {
  618. return true
  619. }
  620. return currentNode != nil && currentNode.Token.Lline < nextNode.Token.Lline
  621. }
  622. /*
  623. skipToken skips over a given token.
  624. */
  625. func skipToken(p *parser, ids ...LexTokenID) error {
  626. var err error
  627. canSkip := func(id LexTokenID) bool {
  628. for _, i := range ids {
  629. if i == id {
  630. return true
  631. }
  632. }
  633. return false
  634. }
  635. if !canSkip(p.node.Token.ID) {
  636. if p.node.Token.ID == TokenEOF {
  637. return p.newParserError(ErrUnexpectedEnd, "", *p.node.Token)
  638. }
  639. return p.newParserError(ErrUnexpectedToken, p.node.Token.Val, *p.node.Token)
  640. }
  641. // This should never return an error unless we skip over EOF or complex tokens
  642. // like values
  643. p.node, err = p.next()
  644. return err
  645. }
  646. /*
  647. acceptChild accepts the current token as a child.
  648. */
  649. func acceptChild(p *parser, self *ASTNode, id LexTokenID) error {
  650. var err error
  651. current := p.node
  652. if p.node, err = p.next(); err == nil {
  653. if current.Token.ID == id {
  654. self.Children = append(self.Children, current)
  655. } else {
  656. err = p.newParserError(ErrUnexpectedToken, current.Token.Val, *current.Token)
  657. }
  658. }
  659. return err
  660. }
  661. /*
  662. parseInnerStatements collects the inner statements of a block statement. It
  663. is assumed that a block statement starts with a left brace '{' and ends with
  664. a right brace '}'.
  665. */
  666. func parseInnerStatements(p *parser, self *ASTNode) (*ASTNode, error) {
  667. // Must start with an opening brace
  668. if err := skipToken(p, TokenLBRACE); err != nil {
  669. return nil, err
  670. }
  671. // Always create a statements node
  672. st := astNodeMap[TokenSTATEMENTS].instance(p, nil)
  673. self.Children = append(self.Children, st)
  674. // Check if there are actually children
  675. if p.node != nil && p.node.Token.ID != TokenRBRACE {
  676. n, err := p.run(0)
  677. if p.node != nil && p.node.Token.ID != TokenEOF {
  678. st.Children = append(st.Children, n)
  679. for hasMoreStatements(p, n) {
  680. if p.node.Token.ID == TokenSEMICOLON {
  681. skipToken(p, TokenSEMICOLON)
  682. } else if p.node.Token.ID == TokenRBRACE {
  683. break
  684. }
  685. n, err = p.run(0)
  686. st.Children = append(st.Children, n)
  687. }
  688. }
  689. if err != nil {
  690. return nil, err
  691. }
  692. }
  693. // Must end with a closing brace
  694. return self, skipToken(p, TokenRBRACE)
  695. }