Lexer.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734
  1. /*
  2. * Copyright (c) 2020, Stephan Unverwerth <s.unverwerth@serenityos.org>
  3. * Copyright (c) 2020-2021, Linus Groh <linusg@serenityos.org>
  4. *
  5. * SPDX-License-Identifier: BSD-2-Clause
  6. */
  7. #include "Lexer.h"
  8. #include <AK/CharacterTypes.h>
  9. #include <AK/Debug.h>
  10. #include <AK/HashMap.h>
  11. #include <stdio.h>
  12. namespace JS {
  13. HashMap<String, TokenType> Lexer::s_keywords;
  14. HashMap<String, TokenType> Lexer::s_three_char_tokens;
  15. HashMap<String, TokenType> Lexer::s_two_char_tokens;
  16. HashMap<char, TokenType> Lexer::s_single_char_tokens;
  17. Lexer::Lexer(StringView source, StringView filename, size_t line_number, size_t line_column)
  18. : m_source(source)
  19. , m_current_token(TokenType::Eof, {}, StringView(nullptr), StringView(nullptr), filename, 0, 0, 0)
  20. , m_filename(filename)
  21. , m_line_number(line_number)
  22. , m_line_column(line_column)
  23. {
  24. if (s_keywords.is_empty()) {
  25. s_keywords.set("await", TokenType::Await);
  26. s_keywords.set("break", TokenType::Break);
  27. s_keywords.set("case", TokenType::Case);
  28. s_keywords.set("catch", TokenType::Catch);
  29. s_keywords.set("class", TokenType::Class);
  30. s_keywords.set("const", TokenType::Const);
  31. s_keywords.set("continue", TokenType::Continue);
  32. s_keywords.set("debugger", TokenType::Debugger);
  33. s_keywords.set("default", TokenType::Default);
  34. s_keywords.set("delete", TokenType::Delete);
  35. s_keywords.set("do", TokenType::Do);
  36. s_keywords.set("else", TokenType::Else);
  37. s_keywords.set("enum", TokenType::Enum);
  38. s_keywords.set("export", TokenType::Export);
  39. s_keywords.set("extends", TokenType::Extends);
  40. s_keywords.set("false", TokenType::BoolLiteral);
  41. s_keywords.set("finally", TokenType::Finally);
  42. s_keywords.set("for", TokenType::For);
  43. s_keywords.set("function", TokenType::Function);
  44. s_keywords.set("if", TokenType::If);
  45. s_keywords.set("import", TokenType::Import);
  46. s_keywords.set("in", TokenType::In);
  47. s_keywords.set("instanceof", TokenType::Instanceof);
  48. s_keywords.set("let", TokenType::Let);
  49. s_keywords.set("new", TokenType::New);
  50. s_keywords.set("null", TokenType::NullLiteral);
  51. s_keywords.set("return", TokenType::Return);
  52. s_keywords.set("super", TokenType::Super);
  53. s_keywords.set("switch", TokenType::Switch);
  54. s_keywords.set("this", TokenType::This);
  55. s_keywords.set("throw", TokenType::Throw);
  56. s_keywords.set("true", TokenType::BoolLiteral);
  57. s_keywords.set("try", TokenType::Try);
  58. s_keywords.set("typeof", TokenType::Typeof);
  59. s_keywords.set("var", TokenType::Var);
  60. s_keywords.set("void", TokenType::Void);
  61. s_keywords.set("while", TokenType::While);
  62. s_keywords.set("with", TokenType::With);
  63. s_keywords.set("yield", TokenType::Yield);
  64. }
  65. if (s_three_char_tokens.is_empty()) {
  66. s_three_char_tokens.set("===", TokenType::EqualsEqualsEquals);
  67. s_three_char_tokens.set("!==", TokenType::ExclamationMarkEqualsEquals);
  68. s_three_char_tokens.set("**=", TokenType::DoubleAsteriskEquals);
  69. s_three_char_tokens.set("<<=", TokenType::ShiftLeftEquals);
  70. s_three_char_tokens.set(">>=", TokenType::ShiftRightEquals);
  71. s_three_char_tokens.set("&&=", TokenType::DoubleAmpersandEquals);
  72. s_three_char_tokens.set("||=", TokenType::DoublePipeEquals);
  73. s_three_char_tokens.set("\?\?=", TokenType::DoubleQuestionMarkEquals);
  74. s_three_char_tokens.set(">>>", TokenType::UnsignedShiftRight);
  75. s_three_char_tokens.set("...", TokenType::TripleDot);
  76. }
  77. if (s_two_char_tokens.is_empty()) {
  78. s_two_char_tokens.set("=>", TokenType::Arrow);
  79. s_two_char_tokens.set("+=", TokenType::PlusEquals);
  80. s_two_char_tokens.set("-=", TokenType::MinusEquals);
  81. s_two_char_tokens.set("*=", TokenType::AsteriskEquals);
  82. s_two_char_tokens.set("/=", TokenType::SlashEquals);
  83. s_two_char_tokens.set("%=", TokenType::PercentEquals);
  84. s_two_char_tokens.set("&=", TokenType::AmpersandEquals);
  85. s_two_char_tokens.set("|=", TokenType::PipeEquals);
  86. s_two_char_tokens.set("^=", TokenType::CaretEquals);
  87. s_two_char_tokens.set("&&", TokenType::DoubleAmpersand);
  88. s_two_char_tokens.set("||", TokenType::DoublePipe);
  89. s_two_char_tokens.set("??", TokenType::DoubleQuestionMark);
  90. s_two_char_tokens.set("**", TokenType::DoubleAsterisk);
  91. s_two_char_tokens.set("==", TokenType::EqualsEquals);
  92. s_two_char_tokens.set("<=", TokenType::LessThanEquals);
  93. s_two_char_tokens.set(">=", TokenType::GreaterThanEquals);
  94. s_two_char_tokens.set("!=", TokenType::ExclamationMarkEquals);
  95. s_two_char_tokens.set("--", TokenType::MinusMinus);
  96. s_two_char_tokens.set("++", TokenType::PlusPlus);
  97. s_two_char_tokens.set("<<", TokenType::ShiftLeft);
  98. s_two_char_tokens.set(">>", TokenType::ShiftRight);
  99. s_two_char_tokens.set("?.", TokenType::QuestionMarkPeriod);
  100. }
  101. if (s_single_char_tokens.is_empty()) {
  102. s_single_char_tokens.set('&', TokenType::Ampersand);
  103. s_single_char_tokens.set('*', TokenType::Asterisk);
  104. s_single_char_tokens.set('[', TokenType::BracketOpen);
  105. s_single_char_tokens.set(']', TokenType::BracketClose);
  106. s_single_char_tokens.set('^', TokenType::Caret);
  107. s_single_char_tokens.set(':', TokenType::Colon);
  108. s_single_char_tokens.set(',', TokenType::Comma);
  109. s_single_char_tokens.set('{', TokenType::CurlyOpen);
  110. s_single_char_tokens.set('}', TokenType::CurlyClose);
  111. s_single_char_tokens.set('=', TokenType::Equals);
  112. s_single_char_tokens.set('!', TokenType::ExclamationMark);
  113. s_single_char_tokens.set('-', TokenType::Minus);
  114. s_single_char_tokens.set('(', TokenType::ParenOpen);
  115. s_single_char_tokens.set(')', TokenType::ParenClose);
  116. s_single_char_tokens.set('%', TokenType::Percent);
  117. s_single_char_tokens.set('.', TokenType::Period);
  118. s_single_char_tokens.set('|', TokenType::Pipe);
  119. s_single_char_tokens.set('+', TokenType::Plus);
  120. s_single_char_tokens.set('?', TokenType::QuestionMark);
  121. s_single_char_tokens.set(';', TokenType::Semicolon);
  122. s_single_char_tokens.set('/', TokenType::Slash);
  123. s_single_char_tokens.set('~', TokenType::Tilde);
  124. s_single_char_tokens.set('<', TokenType::LessThan);
  125. s_single_char_tokens.set('>', TokenType::GreaterThan);
  126. }
  127. consume();
  128. }
  129. void Lexer::consume()
  130. {
  131. auto did_reach_eof = [this] {
  132. if (m_position != m_source.length())
  133. return false;
  134. m_eof = true;
  135. m_current_char = '\0';
  136. m_position++;
  137. m_line_column++;
  138. return true;
  139. };
  140. if (m_position > m_source.length())
  141. return;
  142. if (did_reach_eof())
  143. return;
  144. if (is_line_terminator()) {
  145. if constexpr (LEXER_DEBUG) {
  146. String type;
  147. if (m_current_char == '\n')
  148. type = "LINE FEED";
  149. else if (m_current_char == '\r')
  150. type = "CARRIAGE RETURN";
  151. else if (m_source[m_position + 1] == (char)0xa8)
  152. type = "LINE SEPARATOR";
  153. else
  154. type = "PARAGRAPH SEPARATOR";
  155. dbgln("Found a line terminator: {}", type);
  156. }
  157. // This is a three-char line terminator, we need to increase m_position some more.
  158. // We might reach EOF and need to check again.
  159. if (m_current_char != '\n' && m_current_char != '\r') {
  160. m_position += 2;
  161. if (did_reach_eof())
  162. return;
  163. }
  164. // If the previous character is \r and the current one \n we already updated line number
  165. // and column - don't do it again. From https://tc39.es/ecma262/#sec-line-terminators:
  166. // The sequence <CR><LF> is commonly used as a line terminator.
  167. // It should be considered a single SourceCharacter for the purpose of reporting line numbers.
  168. auto second_char_of_crlf = m_position > 1 && m_source[m_position - 2] == '\r' && m_current_char == '\n';
  169. if (!second_char_of_crlf) {
  170. m_line_number++;
  171. m_line_column = 1;
  172. dbgln_if(LEXER_DEBUG, "Incremented line number, now at: line {}, column 1", m_line_number);
  173. } else {
  174. dbgln_if(LEXER_DEBUG, "Previous was CR, this is LF - not incrementing line number again.");
  175. }
  176. } else {
  177. m_line_column++;
  178. }
  179. m_current_char = m_source[m_position++];
  180. }
  181. bool Lexer::consume_decimal_number()
  182. {
  183. if (!is_ascii_digit(m_current_char))
  184. return false;
  185. while (is_ascii_digit(m_current_char) || match_numeric_literal_separator_followed_by(is_ascii_digit)) {
  186. consume();
  187. }
  188. return true;
  189. }
  190. bool Lexer::consume_exponent()
  191. {
  192. consume();
  193. if (m_current_char == '-' || m_current_char == '+')
  194. consume();
  195. if (!is_ascii_digit(m_current_char))
  196. return false;
  197. return consume_decimal_number();
  198. }
  199. static constexpr bool is_octal_digit(char ch)
  200. {
  201. return ch >= '0' && ch <= '7';
  202. }
  203. bool Lexer::consume_octal_number()
  204. {
  205. consume();
  206. if (!is_octal_digit(m_current_char))
  207. return false;
  208. while (is_octal_digit(m_current_char) || match_numeric_literal_separator_followed_by(is_octal_digit))
  209. consume();
  210. return true;
  211. }
  212. bool Lexer::consume_hexadecimal_number()
  213. {
  214. consume();
  215. if (!is_ascii_hex_digit(m_current_char))
  216. return false;
  217. while (is_ascii_hex_digit(m_current_char) || match_numeric_literal_separator_followed_by(is_ascii_hex_digit))
  218. consume();
  219. return true;
  220. }
  221. static constexpr bool is_binary_digit(char ch)
  222. {
  223. return ch == '0' || ch == '1';
  224. }
  225. bool Lexer::consume_binary_number()
  226. {
  227. consume();
  228. if (!is_binary_digit(m_current_char))
  229. return false;
  230. while (is_binary_digit(m_current_char) || match_numeric_literal_separator_followed_by(is_binary_digit))
  231. consume();
  232. return true;
  233. }
  234. template<typename Callback>
  235. bool Lexer::match_numeric_literal_separator_followed_by(Callback callback) const
  236. {
  237. if (m_position >= m_source.length())
  238. return false;
  239. return m_current_char == '_'
  240. && callback(m_source[m_position]);
  241. }
  242. bool Lexer::match(char a, char b) const
  243. {
  244. if (m_position >= m_source.length())
  245. return false;
  246. return m_current_char == a
  247. && m_source[m_position] == b;
  248. }
  249. bool Lexer::match(char a, char b, char c) const
  250. {
  251. if (m_position + 1 >= m_source.length())
  252. return false;
  253. return m_current_char == a
  254. && m_source[m_position] == b
  255. && m_source[m_position + 1] == c;
  256. }
  257. bool Lexer::match(char a, char b, char c, char d) const
  258. {
  259. if (m_position + 2 >= m_source.length())
  260. return false;
  261. return m_current_char == a
  262. && m_source[m_position] == b
  263. && m_source[m_position + 1] == c
  264. && m_source[m_position + 2] == d;
  265. }
  266. bool Lexer::is_eof() const
  267. {
  268. return m_eof;
  269. }
  270. bool Lexer::is_line_terminator() const
  271. {
  272. if (m_current_char == '\n' || m_current_char == '\r')
  273. return true;
  274. if (m_position > 0 && m_position + 1 < m_source.length()) {
  275. auto three_chars_view = m_source.substring_view(m_position - 1, 3);
  276. return (three_chars_view == LINE_SEPARATOR) || (three_chars_view == PARAGRAPH_SEPARATOR);
  277. }
  278. return false;
  279. }
  280. bool Lexer::is_identifier_start() const
  281. {
  282. return is_ascii_alpha(m_current_char) || m_current_char == '_' || m_current_char == '$';
  283. }
  284. bool Lexer::is_identifier_middle() const
  285. {
  286. return is_identifier_start() || is_ascii_digit(m_current_char);
  287. }
  288. bool Lexer::is_line_comment_start(bool line_has_token_yet) const
  289. {
  290. return match('/', '/')
  291. || (m_allow_html_comments && match('<', '!', '-', '-'))
  292. // "-->" is considered a line comment start if the current line is only whitespace and/or
  293. // other block comment(s); or in other words: the current line does not have a token or
  294. // ongoing line comment yet
  295. || (m_allow_html_comments && !line_has_token_yet && match('-', '-', '>'))
  296. // https://tc39.es/proposal-hashbang/out.html#sec-updated-syntax
  297. || (match('#', '!') && m_position == 1);
  298. }
  299. bool Lexer::is_block_comment_start() const
  300. {
  301. return match('/', '*');
  302. }
  303. bool Lexer::is_block_comment_end() const
  304. {
  305. return match('*', '/');
  306. }
  307. bool Lexer::is_numeric_literal_start() const
  308. {
  309. return is_ascii_digit(m_current_char) || (m_current_char == '.' && m_position < m_source.length() && is_ascii_digit(m_source[m_position]));
  310. }
  311. bool Lexer::slash_means_division() const
  312. {
  313. auto type = m_current_token.type();
  314. return type == TokenType::BigIntLiteral
  315. || type == TokenType::BoolLiteral
  316. || type == TokenType::BracketClose
  317. || type == TokenType::CurlyClose
  318. || type == TokenType::Identifier
  319. || type == TokenType::In
  320. || type == TokenType::Instanceof
  321. || type == TokenType::MinusMinus
  322. || type == TokenType::NullLiteral
  323. || type == TokenType::NumericLiteral
  324. || type == TokenType::ParenClose
  325. || type == TokenType::PlusPlus
  326. || type == TokenType::RegexLiteral
  327. || type == TokenType::StringLiteral
  328. || type == TokenType::TemplateLiteralEnd
  329. || type == TokenType::This;
  330. }
  331. Token Lexer::next()
  332. {
  333. size_t trivia_start = m_position;
  334. auto in_template = !m_template_states.is_empty();
  335. bool line_has_token_yet = m_line_column > 1;
  336. bool unterminated_comment = false;
  337. if (!in_template || m_template_states.last().in_expr) {
  338. // consume whitespace and comments
  339. while (true) {
  340. if (is_line_terminator()) {
  341. line_has_token_yet = false;
  342. do {
  343. consume();
  344. } while (is_line_terminator());
  345. } else if (is_ascii_space(m_current_char)) {
  346. do {
  347. consume();
  348. } while (is_ascii_space(m_current_char));
  349. } else if (is_line_comment_start(line_has_token_yet)) {
  350. consume();
  351. do {
  352. consume();
  353. } while (!is_eof() && !is_line_terminator());
  354. } else if (is_block_comment_start()) {
  355. consume();
  356. do {
  357. consume();
  358. } while (!is_eof() && !is_block_comment_end());
  359. if (is_eof())
  360. unterminated_comment = true;
  361. consume(); // consume *
  362. if (is_eof())
  363. unterminated_comment = true;
  364. consume(); // consume /
  365. } else {
  366. break;
  367. }
  368. }
  369. }
  370. size_t value_start = m_position;
  371. size_t value_start_line_number = m_line_number;
  372. size_t value_start_column_number = m_line_column;
  373. auto token_type = TokenType::Invalid;
  374. auto did_consume_whitespace_or_comments = trivia_start != value_start;
  375. // This is being used to communicate info about invalid tokens to the parser, which then
  376. // can turn that into more specific error messages - instead of us having to make up a
  377. // bunch of Invalid* tokens (bad numeric literals, unterminated comments etc.)
  378. String token_message;
  379. if (m_current_token.type() == TokenType::RegexLiteral && !is_eof() && is_ascii_alpha(m_current_char) && !did_consume_whitespace_or_comments) {
  380. token_type = TokenType::RegexFlags;
  381. while (!is_eof() && is_ascii_alpha(m_current_char))
  382. consume();
  383. } else if (m_current_char == '`') {
  384. consume();
  385. if (!in_template) {
  386. token_type = TokenType::TemplateLiteralStart;
  387. m_template_states.append({ false, 0 });
  388. } else {
  389. if (m_template_states.last().in_expr) {
  390. m_template_states.append({ false, 0 });
  391. token_type = TokenType::TemplateLiteralStart;
  392. } else {
  393. m_template_states.take_last();
  394. token_type = TokenType::TemplateLiteralEnd;
  395. }
  396. }
  397. } else if (in_template && m_template_states.last().in_expr && m_template_states.last().open_bracket_count == 0 && m_current_char == '}') {
  398. consume();
  399. token_type = TokenType::TemplateLiteralExprEnd;
  400. m_template_states.last().in_expr = false;
  401. } else if (in_template && !m_template_states.last().in_expr) {
  402. if (is_eof()) {
  403. token_type = TokenType::UnterminatedTemplateLiteral;
  404. m_template_states.take_last();
  405. } else if (match('$', '{')) {
  406. token_type = TokenType::TemplateLiteralExprStart;
  407. consume();
  408. consume();
  409. m_template_states.last().in_expr = true;
  410. } else {
  411. while (!match('$', '{') && m_current_char != '`' && !is_eof()) {
  412. if (match('\\', '$') || match('\\', '`'))
  413. consume();
  414. consume();
  415. }
  416. if (is_eof() && !m_template_states.is_empty())
  417. token_type = TokenType::UnterminatedTemplateLiteral;
  418. else
  419. token_type = TokenType::TemplateLiteralString;
  420. }
  421. } else if (is_identifier_start()) {
  422. // identifier or keyword
  423. do {
  424. consume();
  425. } while (is_identifier_middle());
  426. StringView value = m_source.substring_view(value_start - 1, m_position - value_start);
  427. auto it = s_keywords.find(value.hash(), [&](auto& entry) { return entry.key == value; });
  428. if (it == s_keywords.end()) {
  429. token_type = TokenType::Identifier;
  430. } else {
  431. token_type = it->value;
  432. }
  433. } else if (is_numeric_literal_start()) {
  434. token_type = TokenType::NumericLiteral;
  435. bool is_invalid_numeric_literal = false;
  436. if (m_current_char == '0') {
  437. consume();
  438. if (m_current_char == '.') {
  439. // decimal
  440. consume();
  441. while (is_ascii_digit(m_current_char) || match_numeric_literal_separator_followed_by(is_ascii_digit))
  442. consume();
  443. if (m_current_char == 'e' || m_current_char == 'E')
  444. is_invalid_numeric_literal = !consume_exponent();
  445. } else if (m_current_char == 'e' || m_current_char == 'E') {
  446. is_invalid_numeric_literal = !consume_exponent();
  447. } else if (m_current_char == 'o' || m_current_char == 'O') {
  448. // octal
  449. is_invalid_numeric_literal = !consume_octal_number();
  450. if (m_current_char == 'n') {
  451. consume();
  452. token_type = TokenType::BigIntLiteral;
  453. }
  454. } else if (m_current_char == 'b' || m_current_char == 'B') {
  455. // binary
  456. is_invalid_numeric_literal = !consume_binary_number();
  457. if (m_current_char == 'n') {
  458. consume();
  459. token_type = TokenType::BigIntLiteral;
  460. }
  461. } else if (m_current_char == 'x' || m_current_char == 'X') {
  462. // hexadecimal
  463. is_invalid_numeric_literal = !consume_hexadecimal_number();
  464. if (m_current_char == 'n') {
  465. consume();
  466. token_type = TokenType::BigIntLiteral;
  467. }
  468. } else if (m_current_char == 'n') {
  469. consume();
  470. token_type = TokenType::BigIntLiteral;
  471. } else if (is_ascii_digit(m_current_char)) {
  472. // octal without '0o' prefix. Forbidden in 'strict mode'
  473. do {
  474. consume();
  475. } while (is_ascii_digit(m_current_char) || match_numeric_literal_separator_followed_by(is_ascii_digit));
  476. }
  477. } else {
  478. // 1...9 or period
  479. while (is_ascii_digit(m_current_char) || match_numeric_literal_separator_followed_by(is_ascii_digit))
  480. consume();
  481. if (m_current_char == 'n') {
  482. consume();
  483. token_type = TokenType::BigIntLiteral;
  484. } else {
  485. if (m_current_char == '.') {
  486. consume();
  487. while (is_ascii_digit(m_current_char) || match_numeric_literal_separator_followed_by(is_ascii_digit))
  488. consume();
  489. }
  490. if (m_current_char == 'e' || m_current_char == 'E')
  491. is_invalid_numeric_literal = !consume_exponent();
  492. }
  493. }
  494. if (is_invalid_numeric_literal) {
  495. token_type = TokenType::Invalid;
  496. token_message = "Invalid numeric literal";
  497. }
  498. } else if (m_current_char == '"' || m_current_char == '\'') {
  499. char stop_char = m_current_char;
  500. consume();
  501. // Note: LS/PS line terminators are allowed in string literals.
  502. while (m_current_char != stop_char && m_current_char != '\r' && m_current_char != '\n' && !is_eof()) {
  503. if (m_current_char == '\\') {
  504. consume();
  505. if (m_current_char == '\r' && m_position < m_source.length() && m_source[m_position] == '\n') {
  506. consume();
  507. }
  508. }
  509. consume();
  510. }
  511. if (m_current_char != stop_char) {
  512. token_type = TokenType::UnterminatedStringLiteral;
  513. } else {
  514. consume();
  515. token_type = TokenType::StringLiteral;
  516. }
  517. } else if (m_current_char == '/' && !slash_means_division()) {
  518. consume();
  519. token_type = consume_regex_literal();
  520. } else if (m_eof) {
  521. if (unterminated_comment) {
  522. token_type = TokenType::Invalid;
  523. token_message = "Unterminated multi-line comment";
  524. } else {
  525. token_type = TokenType::Eof;
  526. }
  527. } else {
  528. // There is only one four-char operator: >>>=
  529. bool found_four_char_token = false;
  530. if (match('>', '>', '>', '=')) {
  531. found_four_char_token = true;
  532. consume();
  533. consume();
  534. consume();
  535. consume();
  536. token_type = TokenType::UnsignedShiftRightEquals;
  537. }
  538. bool found_three_char_token = false;
  539. if (!found_four_char_token && m_position + 1 < m_source.length()) {
  540. auto three_chars_view = m_source.substring_view(m_position - 1, 3);
  541. auto it = s_three_char_tokens.find(three_chars_view.hash(), [&](auto& entry) { return entry.key == three_chars_view; });
  542. if (it != s_three_char_tokens.end()) {
  543. found_three_char_token = true;
  544. consume();
  545. consume();
  546. consume();
  547. token_type = it->value;
  548. }
  549. }
  550. bool found_two_char_token = false;
  551. if (!found_four_char_token && !found_three_char_token && m_position < m_source.length()) {
  552. auto two_chars_view = m_source.substring_view(m_position - 1, 2);
  553. auto it = s_two_char_tokens.find(two_chars_view.hash(), [&](auto& entry) { return entry.key == two_chars_view; });
  554. if (it != s_two_char_tokens.end()) {
  555. // OptionalChainingPunctuator :: ?. [lookahead ∉ DecimalDigit]
  556. if (!(it->value == TokenType::QuestionMarkPeriod && m_position + 1 < m_source.length() && is_ascii_digit(m_source[m_position + 1]))) {
  557. found_two_char_token = true;
  558. consume();
  559. consume();
  560. token_type = it->value;
  561. }
  562. }
  563. }
  564. bool found_one_char_token = false;
  565. if (!found_four_char_token && !found_three_char_token && !found_two_char_token) {
  566. auto it = s_single_char_tokens.find(m_current_char);
  567. if (it != s_single_char_tokens.end()) {
  568. found_one_char_token = true;
  569. consume();
  570. token_type = it->value;
  571. }
  572. }
  573. if (!found_four_char_token && !found_three_char_token && !found_two_char_token && !found_one_char_token) {
  574. consume();
  575. token_type = TokenType::Invalid;
  576. }
  577. }
  578. if (!m_template_states.is_empty() && m_template_states.last().in_expr) {
  579. if (token_type == TokenType::CurlyOpen) {
  580. m_template_states.last().open_bracket_count++;
  581. } else if (token_type == TokenType::CurlyClose) {
  582. m_template_states.last().open_bracket_count--;
  583. }
  584. }
  585. m_current_token = Token(
  586. token_type,
  587. token_message,
  588. m_source.substring_view(trivia_start - 1, value_start - trivia_start),
  589. m_source.substring_view(value_start - 1, m_position - value_start),
  590. m_filename,
  591. value_start_line_number,
  592. value_start_column_number,
  593. m_position);
  594. if constexpr (LEXER_DEBUG) {
  595. dbgln("------------------------------");
  596. dbgln("Token: {}", m_current_token.name());
  597. dbgln("Trivia: _{}_", m_current_token.trivia());
  598. dbgln("Value: _{}_", m_current_token.value());
  599. dbgln("Line: {}, Column: {}", m_current_token.line_number(), m_current_token.line_column());
  600. dbgln("------------------------------");
  601. }
  602. return m_current_token;
  603. }
  604. Token Lexer::force_slash_as_regex()
  605. {
  606. VERIFY(m_current_token.type() == TokenType::Slash || m_current_token.type() == TokenType::SlashEquals);
  607. bool has_equals = m_current_token.type() == TokenType::SlashEquals;
  608. VERIFY(m_position > 0);
  609. size_t value_start = m_position - 1;
  610. if (has_equals) {
  611. VERIFY(m_source[value_start - 1] == '=');
  612. --value_start;
  613. --m_position;
  614. m_current_char = '=';
  615. }
  616. TokenType token_type = consume_regex_literal();
  617. m_current_token = Token(
  618. token_type,
  619. "",
  620. m_current_token.trivia(),
  621. m_source.substring_view(value_start - 1, m_position - value_start),
  622. m_filename,
  623. m_current_token.line_number(),
  624. m_current_token.line_column(),
  625. m_position);
  626. if constexpr (LEXER_DEBUG) {
  627. dbgln("------------------------------");
  628. dbgln("Token: {}", m_current_token.name());
  629. dbgln("Trivia: _{}_", m_current_token.trivia());
  630. dbgln("Value: _{}_", m_current_token.value());
  631. dbgln("Line: {}, Column: {}", m_current_token.line_number(), m_current_token.line_column());
  632. dbgln("------------------------------");
  633. }
  634. return m_current_token;
  635. }
  636. TokenType Lexer::consume_regex_literal()
  637. {
  638. TokenType token_type = TokenType::RegexLiteral;
  639. while (!is_eof()) {
  640. if (is_line_terminator() || (!m_regex_is_in_character_class && m_current_char == '/')) {
  641. break;
  642. } else if (m_current_char == '[') {
  643. m_regex_is_in_character_class = true;
  644. } else if (m_current_char == ']') {
  645. m_regex_is_in_character_class = false;
  646. } else if (!m_regex_is_in_character_class && m_current_char == '/') {
  647. break;
  648. }
  649. if (match('\\', '/') || match('\\', '[') || match('\\', '\\') || (m_regex_is_in_character_class && match('\\', ']')))
  650. consume();
  651. consume();
  652. }
  653. if (m_current_char == '/') {
  654. consume();
  655. return TokenType::RegexLiteral;
  656. } else {
  657. return TokenType::UnterminatedRegexLiteral;
  658. }
  659. return token_type;
  660. }
  661. }