js.cpp 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326
  1. /*
  2. * Copyright (c) 2020, Andreas Kling <kling@serenityos.org>
  3. * All rights reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions are met:
  7. *
  8. * 1. Redistributions of source code must retain the above copyright notice, this
  9. * list of conditions and the following disclaimer.
  10. *
  11. * 2. Redistributions in binary form must reproduce the above copyright notice,
  12. * this list of conditions and the following disclaimer in the documentation
  13. * and/or other materials provided with the distribution.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  16. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  18. * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
  19. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  21. * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  22. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  23. * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25. */
  26. #include <AK/ByteBuffer.h>
  27. #include <AK/NonnullOwnPtr.h>
  28. #include <AK/StringBuilder.h>
  29. #include <LibCore/ArgsParser.h>
  30. #include <LibCore/File.h>
  31. #include <LibJS/AST.h>
  32. #include <LibJS/Interpreter.h>
  33. #include <LibJS/Parser.h>
  34. #include <LibJS/Runtime/Array.h>
  35. #include <LibJS/Runtime/Date.h>
  36. #include <LibJS/Runtime/Function.h>
  37. #include <LibJS/Runtime/GlobalObject.h>
  38. #include <LibJS/Runtime/Object.h>
  39. #include <LibJS/Runtime/PrimitiveString.h>
  40. #include <LibJS/Runtime/Value.h>
  41. #include <LibLine/Editor.h>
  42. #include <stdio.h>
  43. class ReplObject : public JS::GlobalObject {
  44. public:
  45. ReplObject();
  46. virtual ~ReplObject() override;
  47. private:
  48. virtual const char* class_name() const override { return "ReplObject"; }
  49. static JS::Value exit_interpreter(JS::Interpreter&);
  50. static JS::Value repl_help(JS::Interpreter&);
  51. static JS::Value load_file(JS::Interpreter&);
  52. };
  53. bool dump_ast = false;
  54. static OwnPtr<Line::Editor> editor;
  55. String read_next_piece()
  56. {
  57. StringBuilder piece;
  58. int level = 0;
  59. StringBuilder prompt_builder;
  60. do {
  61. prompt_builder.clear();
  62. prompt_builder.append("> ");
  63. for (auto i = 0; i < level; ++i)
  64. prompt_builder.append(" ");
  65. String line = editor->get_line(prompt_builder.build());
  66. piece.append(line);
  67. auto lexer = JS::Lexer(line);
  68. for (JS::Token token = lexer.next(); token.type() != JS::TokenType::Eof; token = lexer.next()) {
  69. switch (token.type()) {
  70. case JS::TokenType::BracketOpen:
  71. case JS::TokenType::CurlyOpen:
  72. case JS::TokenType::ParenOpen:
  73. level++;
  74. break;
  75. case JS::TokenType::BracketClose:
  76. case JS::TokenType::CurlyClose:
  77. case JS::TokenType::ParenClose:
  78. level--;
  79. break;
  80. default:
  81. break;
  82. }
  83. }
  84. } while (level > 0);
  85. return piece.to_string();
  86. }
  87. static void print_value(JS::Value value, HashTable<JS::Object*>& seen_objects);
  88. static void print_array(const JS::Array* array, HashTable<JS::Object*>& seen_objects)
  89. {
  90. fputs("[ ", stdout);
  91. for (size_t i = 0; i < array->elements().size(); ++i) {
  92. print_value(array->elements()[i], seen_objects);
  93. if (i != array->elements().size() - 1)
  94. fputs(", ", stdout);
  95. }
  96. fputs(" ]", stdout);
  97. }
  98. static void print_object(const JS::Object* object, HashTable<JS::Object*>& seen_objects)
  99. {
  100. fputs("{ ", stdout);
  101. size_t index = 0;
  102. for (auto& it : object->own_properties()) {
  103. printf("\"\033[33;1m%s\033[0m\": ", it.key.characters());
  104. print_value(it.value, seen_objects);
  105. if (index != object->own_properties().size() - 1)
  106. fputs(", ", stdout);
  107. ++index;
  108. }
  109. fputs(" }", stdout);
  110. }
  111. static void print_function(const JS::Object* function, HashTable<JS::Object*>&)
  112. {
  113. printf("\033[34;1m[%s]\033[0m", function->class_name());
  114. }
  115. static void print_date(const JS::Object* date, HashTable<JS::Object*>&)
  116. {
  117. printf("\033[34;1mDate %s\033[0m", static_cast<const JS::Date*>(date)->string().characters());
  118. }
  119. void print_value(JS::Value value, HashTable<JS::Object*>& seen_objects)
  120. {
  121. if (value.is_object()) {
  122. if (seen_objects.contains(value.as_object())) {
  123. // FIXME: Maybe we should only do this for circular references,
  124. // not for all reoccurring objects.
  125. printf("<already printed Object %p>", value.as_object());
  126. return;
  127. }
  128. seen_objects.set(value.as_object());
  129. }
  130. if (value.is_array())
  131. return print_array(static_cast<const JS::Array*>(value.as_object()), seen_objects);
  132. if (value.is_object()) {
  133. auto* object = value.as_object();
  134. if (object->is_function())
  135. return print_function(object, seen_objects);
  136. if (object->is_date())
  137. return print_date(object, seen_objects);
  138. return print_object(object, seen_objects);
  139. }
  140. if (value.is_string())
  141. printf("\033[31;1m");
  142. else if (value.is_number())
  143. printf("\033[35;1m");
  144. else if (value.is_boolean())
  145. printf("\033[32;1m");
  146. else if (value.is_null())
  147. printf("\033[33;1m");
  148. else if (value.is_undefined())
  149. printf("\033[34;1m");
  150. if (value.is_string())
  151. putchar('"');
  152. printf("%s", value.to_string().characters());
  153. if (value.is_string())
  154. putchar('"');
  155. printf("\033[0m");
  156. }
  157. static void print(JS::Value value)
  158. {
  159. HashTable<JS::Object*> seen_objects;
  160. print_value(value, seen_objects);
  161. putchar('\n');
  162. }
  163. bool file_has_shebang(AK::ByteBuffer file_contents)
  164. {
  165. if (file_contents.size() >= 2 && file_contents[0] == '#' && file_contents[1] == '!')
  166. return true;
  167. return false;
  168. }
  169. StringView strip_shebang(AK::ByteBuffer file_contents)
  170. {
  171. size_t i = 0;
  172. for (i = 2; i < file_contents.size(); ++i) {
  173. if (file_contents[i] == '\n')
  174. break;
  175. }
  176. return StringView((const char*)file_contents.data() + i, file_contents.size() - i);
  177. }
  178. ReplObject::ReplObject()
  179. {
  180. put_native_function("exit", exit_interpreter);
  181. put_native_function("help", repl_help);
  182. put_native_function("load", load_file);
  183. }
  184. ReplObject::~ReplObject()
  185. {
  186. }
  187. JS::Value ReplObject::exit_interpreter(JS::Interpreter& interpreter)
  188. {
  189. if (interpreter.call_frame().arguments.is_empty())
  190. exit(0);
  191. int exit_code = interpreter.call_frame().arguments[0].to_number().as_double();
  192. exit(exit_code);
  193. return JS::js_undefined();
  194. }
  195. JS::Value ReplObject::repl_help(JS::Interpreter& interpreter)
  196. {
  197. StringBuilder help_text;
  198. help_text.append("REPL commands:\n");
  199. help_text.append(" exit(code): exit the REPL with specified code. Defaults to 0.\n");
  200. help_text.append(" help(): display this menu\n");
  201. help_text.append(" load(files): Accepts file names as params to load into running session. For example repl.load(\"js/1.js\", \"js/2.js\", \"js/3.js\")\n");
  202. String result = help_text.to_string();
  203. return js_string(interpreter.heap(), result);
  204. }
  205. JS::Value ReplObject::load_file(JS::Interpreter& interpreter)
  206. {
  207. if (interpreter.call_frame().arguments.is_empty())
  208. return JS::Value(false);
  209. Vector<JS::Value> files = interpreter.call_frame().arguments;
  210. for (JS::Value file : files) {
  211. String file_name = file.as_string()->string();
  212. auto js_file = Core::File::construct(file_name);
  213. if (!js_file->open(Core::IODevice::ReadOnly)) {
  214. fprintf(stderr, "Failed to open %s: %s\n", file_name.characters(), js_file->error_string());
  215. }
  216. auto file_contents = js_file->read_all();
  217. StringView source;
  218. if (file_has_shebang(file_contents)) {
  219. source = strip_shebang(file_contents);
  220. } else {
  221. source = file_contents;
  222. }
  223. auto program = JS::Parser(JS::Lexer(source)).parse_program();
  224. if (dump_ast)
  225. program->dump(0);
  226. auto result = interpreter.run(*program);
  227. print(result);
  228. }
  229. return JS::Value(true);
  230. }
  231. void repl(JS::Interpreter& interpreter)
  232. {
  233. while (true) {
  234. String piece = read_next_piece();
  235. if (piece.is_empty())
  236. continue;
  237. auto program = JS::Parser(JS::Lexer(piece)).parse_program();
  238. if (dump_ast)
  239. program->dump(0);
  240. auto result = interpreter.run(*program);
  241. print(result);
  242. }
  243. }
  244. int main(int argc, char** argv)
  245. {
  246. bool gc_on_every_allocation = false;
  247. bool print_last_result = false;
  248. const char* script_path = nullptr;
  249. Core::ArgsParser args_parser;
  250. args_parser.add_option(dump_ast, "Dump the AST", "dump-ast", 'A');
  251. args_parser.add_option(print_last_result, "Print last result", "print-last-result", 'l');
  252. args_parser.add_option(gc_on_every_allocation, "GC on every allocation", "gc-on-every-allocation", 'g');
  253. args_parser.add_positional_argument(script_path, "Path to script file", "script", Core::ArgsParser::Required::No);
  254. args_parser.parse(argc, argv);
  255. if (script_path == nullptr) {
  256. auto interpreter = JS::Interpreter::create<ReplObject>();
  257. interpreter->heap().set_should_collect_on_every_allocation(gc_on_every_allocation);
  258. interpreter->global_object().put("global", &interpreter->global_object());
  259. editor = make<Line::Editor>();
  260. editor->initialize();
  261. repl(*interpreter);
  262. } else {
  263. auto interpreter = JS::Interpreter::create<JS::GlobalObject>();
  264. interpreter->heap().set_should_collect_on_every_allocation(gc_on_every_allocation);
  265. interpreter->global_object().put("global", &interpreter->global_object());
  266. auto file = Core::File::construct(script_path);
  267. if (!file->open(Core::IODevice::ReadOnly)) {
  268. fprintf(stderr, "Failed to open %s: %s\n", script_path, file->error_string());
  269. return 1;
  270. }
  271. auto file_contents = file->read_all();
  272. StringView source;
  273. if (file_has_shebang(file_contents)) {
  274. source = strip_shebang(file_contents);
  275. } else {
  276. source = file_contents;
  277. }
  278. auto program = JS::Parser(JS::Lexer(source)).parse_program();
  279. if (dump_ast)
  280. program->dump(0);
  281. auto result = interpreter->run(*program);
  282. if (print_last_result)
  283. printf("%s\n", result.to_string().characters());
  284. }
  285. return 0;
  286. }