This is a test program for mruby mirb. Currently, I'm sure the only work in "Renesas GR-PEACH".
Dependencies: mbed-src mruby-mbed
main.cpp@1:340b448861c1, 2015-04-13 (annotated)
- Committer:
- mzta
- Date:
- Mon Apr 13 05:24:00 2015 +0000
- Revision:
- 1:340b448861c1
- Parent:
- 0:e7eb8eab3aa9
- update mbed-src.; - code refactoring.; - add SPI, SPISlave, I2C class to mruby-mbed (Incomplete).
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
mzta | 0:e7eb8eab3aa9 | 1 | /* |
mzta | 0:e7eb8eab3aa9 | 2 | ** mirb - Embeddable Interactive Ruby Shell |
mzta | 0:e7eb8eab3aa9 | 3 | ** |
mzta | 0:e7eb8eab3aa9 | 4 | ** This program takes code from the user in |
mzta | 0:e7eb8eab3aa9 | 5 | ** an interactive way and executes it |
mzta | 0:e7eb8eab3aa9 | 6 | ** immediately. It's a REPL... |
mzta | 0:e7eb8eab3aa9 | 7 | */ |
mzta | 0:e7eb8eab3aa9 | 8 | |
mzta | 0:e7eb8eab3aa9 | 9 | #include "mbed.h" |
mzta | 0:e7eb8eab3aa9 | 10 | |
mzta | 0:e7eb8eab3aa9 | 11 | Serial pc(USBTX, USBRX); |
mzta | 0:e7eb8eab3aa9 | 12 | |
mzta | 0:e7eb8eab3aa9 | 13 | #include <stdlib.h> |
mzta | 0:e7eb8eab3aa9 | 14 | #include <string.h> |
mzta | 0:e7eb8eab3aa9 | 15 | #include <stdio.h> |
mzta | 0:e7eb8eab3aa9 | 16 | #include <ctype.h> |
mzta | 0:e7eb8eab3aa9 | 17 | |
mzta | 0:e7eb8eab3aa9 | 18 | #ifdef ENABLE_READLINE |
mzta | 0:e7eb8eab3aa9 | 19 | #include <readline/readline.h> |
mzta | 0:e7eb8eab3aa9 | 20 | #include <readline/history.h> |
mzta | 0:e7eb8eab3aa9 | 21 | #define MIRB_ADD_HISTORY(line) add_history(line) |
mzta | 0:e7eb8eab3aa9 | 22 | #define MIRB_READLINE(ch) readline(ch) |
mzta | 0:e7eb8eab3aa9 | 23 | #define MIRB_WRITE_HISTORY(path) write_history(path) |
mzta | 0:e7eb8eab3aa9 | 24 | #define MIRB_READ_HISTORY(path) read_history(path) |
mzta | 0:e7eb8eab3aa9 | 25 | #define MIRB_USING_HISTORY() using_history() |
mzta | 0:e7eb8eab3aa9 | 26 | #elif defined(ENABLE_LINENOISE) |
mzta | 0:e7eb8eab3aa9 | 27 | #define ENABLE_READLINE |
mzta | 0:e7eb8eab3aa9 | 28 | #include <linenoise.h> |
mzta | 0:e7eb8eab3aa9 | 29 | #define MIRB_ADD_HISTORY(line) linenoiseHistoryAdd(line) |
mzta | 0:e7eb8eab3aa9 | 30 | #define MIRB_READLINE(ch) linenoise(ch) |
mzta | 0:e7eb8eab3aa9 | 31 | #define MIRB_WRITE_HISTORY(path) linenoiseHistorySave(path) |
mzta | 0:e7eb8eab3aa9 | 32 | #define MIRB_READ_HISTORY(path) linenoiseHistoryLoad(history_path) |
mzta | 0:e7eb8eab3aa9 | 33 | #define MIRB_USING_HISTORY() |
mzta | 0:e7eb8eab3aa9 | 34 | #endif |
mzta | 0:e7eb8eab3aa9 | 35 | |
mzta | 0:e7eb8eab3aa9 | 36 | #include "mruby.h" |
mzta | 0:e7eb8eab3aa9 | 37 | #include "mruby/array.h" |
mzta | 0:e7eb8eab3aa9 | 38 | #include "mruby/proc.h" |
mzta | 0:e7eb8eab3aa9 | 39 | #include "mruby/compile.h" |
mzta | 0:e7eb8eab3aa9 | 40 | #include "mruby/string.h" |
mzta | 0:e7eb8eab3aa9 | 41 | |
mzta | 0:e7eb8eab3aa9 | 42 | #ifdef ENABLE_READLINE |
mzta | 0:e7eb8eab3aa9 | 43 | |
mzta | 0:e7eb8eab3aa9 | 44 | static const char history_file_name[] = ".mirb_history"; |
mzta | 0:e7eb8eab3aa9 | 45 | |
mzta | 0:e7eb8eab3aa9 | 46 | static char * |
mzta | 0:e7eb8eab3aa9 | 47 | get_history_path(mrb_state *mrb) |
mzta | 0:e7eb8eab3aa9 | 48 | { |
mzta | 0:e7eb8eab3aa9 | 49 | char *path = NULL; |
mzta | 0:e7eb8eab3aa9 | 50 | const char *home = getenv("HOME"); |
mzta | 0:e7eb8eab3aa9 | 51 | |
mzta | 0:e7eb8eab3aa9 | 52 | #ifdef _WIN32 |
mzta | 0:e7eb8eab3aa9 | 53 | if (home != NULL) { |
mzta | 0:e7eb8eab3aa9 | 54 | home = getenv("USERPROFILE"); |
mzta | 0:e7eb8eab3aa9 | 55 | } |
mzta | 0:e7eb8eab3aa9 | 56 | #endif |
mzta | 0:e7eb8eab3aa9 | 57 | |
mzta | 0:e7eb8eab3aa9 | 58 | if (home != NULL) { |
mzta | 0:e7eb8eab3aa9 | 59 | int len = snprintf(NULL, 0, "%s/%s", home, history_file_name); |
mzta | 0:e7eb8eab3aa9 | 60 | if (len >= 0) { |
mzta | 0:e7eb8eab3aa9 | 61 | size_t size = len + 1; |
mzta | 0:e7eb8eab3aa9 | 62 | path = (char *)mrb_malloc_simple(mrb, size); |
mzta | 0:e7eb8eab3aa9 | 63 | if (path != NULL) { |
mzta | 0:e7eb8eab3aa9 | 64 | int n = snprintf(path, size, "%s/%s", home, history_file_name); |
mzta | 0:e7eb8eab3aa9 | 65 | if (n != len) { |
mzta | 0:e7eb8eab3aa9 | 66 | mrb_free(mrb, path); |
mzta | 0:e7eb8eab3aa9 | 67 | path = NULL; |
mzta | 0:e7eb8eab3aa9 | 68 | } |
mzta | 0:e7eb8eab3aa9 | 69 | } |
mzta | 0:e7eb8eab3aa9 | 70 | } |
mzta | 0:e7eb8eab3aa9 | 71 | } |
mzta | 0:e7eb8eab3aa9 | 72 | |
mzta | 0:e7eb8eab3aa9 | 73 | return path; |
mzta | 0:e7eb8eab3aa9 | 74 | } |
mzta | 0:e7eb8eab3aa9 | 75 | |
mzta | 0:e7eb8eab3aa9 | 76 | #endif |
mzta | 0:e7eb8eab3aa9 | 77 | |
mzta | 0:e7eb8eab3aa9 | 78 | static void |
mzta | 0:e7eb8eab3aa9 | 79 | p(mrb_state *mrb, mrb_value obj, int prompt) |
mzta | 0:e7eb8eab3aa9 | 80 | { |
mzta | 0:e7eb8eab3aa9 | 81 | obj = mrb_funcall(mrb, obj, "inspect", 0); |
mzta | 0:e7eb8eab3aa9 | 82 | if (prompt) { |
mzta | 0:e7eb8eab3aa9 | 83 | if (!mrb->exc) { |
mzta | 0:e7eb8eab3aa9 | 84 | pc.printf(" => "); |
mzta | 0:e7eb8eab3aa9 | 85 | } |
mzta | 0:e7eb8eab3aa9 | 86 | else { |
mzta | 0:e7eb8eab3aa9 | 87 | obj = mrb_funcall(mrb, mrb_obj_value(mrb->exc), "inspect", 0); |
mzta | 0:e7eb8eab3aa9 | 88 | } |
mzta | 0:e7eb8eab3aa9 | 89 | } |
mzta | 0:e7eb8eab3aa9 | 90 | fwrite(RSTRING_PTR(obj), RSTRING_LEN(obj), 1, stdout); |
mzta | 0:e7eb8eab3aa9 | 91 | putc('\n', stdout); |
mzta | 0:e7eb8eab3aa9 | 92 | } |
mzta | 0:e7eb8eab3aa9 | 93 | |
mzta | 0:e7eb8eab3aa9 | 94 | /* Guess if the user might want to enter more |
mzta | 0:e7eb8eab3aa9 | 95 | * or if he wants an evaluation of his code now */ |
mzta | 0:e7eb8eab3aa9 | 96 | static mrb_bool |
mzta | 0:e7eb8eab3aa9 | 97 | is_code_block_open(struct mrb_parser_state *parser) |
mzta | 0:e7eb8eab3aa9 | 98 | { |
mzta | 0:e7eb8eab3aa9 | 99 | mrb_bool code_block_open = FALSE; |
mzta | 0:e7eb8eab3aa9 | 100 | |
mzta | 0:e7eb8eab3aa9 | 101 | /* check for heredoc */ |
mzta | 0:e7eb8eab3aa9 | 102 | if (parser->parsing_heredoc != NULL) return TRUE; |
mzta | 0:e7eb8eab3aa9 | 103 | if (parser->heredoc_end_now) { |
mzta | 0:e7eb8eab3aa9 | 104 | parser->heredoc_end_now = FALSE; |
mzta | 0:e7eb8eab3aa9 | 105 | return FALSE; |
mzta | 0:e7eb8eab3aa9 | 106 | } |
mzta | 0:e7eb8eab3aa9 | 107 | |
mzta | 0:e7eb8eab3aa9 | 108 | /* check for unterminated string */ |
mzta | 0:e7eb8eab3aa9 | 109 | if (parser->lex_strterm) return TRUE; |
mzta | 0:e7eb8eab3aa9 | 110 | |
mzta | 0:e7eb8eab3aa9 | 111 | /* check if parser error are available */ |
mzta | 0:e7eb8eab3aa9 | 112 | if (0 < parser->nerr) { |
mzta | 0:e7eb8eab3aa9 | 113 | const char unexpected_end[] = "syntax error, unexpected $end"; |
mzta | 0:e7eb8eab3aa9 | 114 | const char *message = parser->error_buffer[0].message; |
mzta | 0:e7eb8eab3aa9 | 115 | |
mzta | 0:e7eb8eab3aa9 | 116 | /* a parser error occur, we have to check if */ |
mzta | 0:e7eb8eab3aa9 | 117 | /* we need to read one more line or if there is */ |
mzta | 0:e7eb8eab3aa9 | 118 | /* a different issue which we have to show to */ |
mzta | 0:e7eb8eab3aa9 | 119 | /* the user */ |
mzta | 0:e7eb8eab3aa9 | 120 | |
mzta | 0:e7eb8eab3aa9 | 121 | if (strncmp(message, unexpected_end, sizeof(unexpected_end) - 1) == 0) { |
mzta | 0:e7eb8eab3aa9 | 122 | code_block_open = TRUE; |
mzta | 0:e7eb8eab3aa9 | 123 | } |
mzta | 0:e7eb8eab3aa9 | 124 | else if (strcmp(message, "syntax error, unexpected keyword_end") == 0) { |
mzta | 0:e7eb8eab3aa9 | 125 | code_block_open = FALSE; |
mzta | 0:e7eb8eab3aa9 | 126 | } |
mzta | 0:e7eb8eab3aa9 | 127 | else if (strcmp(message, "syntax error, unexpected tREGEXP_BEG") == 0) { |
mzta | 0:e7eb8eab3aa9 | 128 | code_block_open = FALSE; |
mzta | 0:e7eb8eab3aa9 | 129 | } |
mzta | 0:e7eb8eab3aa9 | 130 | return code_block_open; |
mzta | 0:e7eb8eab3aa9 | 131 | } |
mzta | 0:e7eb8eab3aa9 | 132 | |
mzta | 0:e7eb8eab3aa9 | 133 | switch (parser->lstate) { |
mzta | 0:e7eb8eab3aa9 | 134 | |
mzta | 0:e7eb8eab3aa9 | 135 | /* all states which need more code */ |
mzta | 0:e7eb8eab3aa9 | 136 | |
mzta | 0:e7eb8eab3aa9 | 137 | case EXPR_BEG: |
mzta | 0:e7eb8eab3aa9 | 138 | /* beginning of a statement, */ |
mzta | 0:e7eb8eab3aa9 | 139 | /* that means previous line ended */ |
mzta | 0:e7eb8eab3aa9 | 140 | code_block_open = FALSE; |
mzta | 0:e7eb8eab3aa9 | 141 | break; |
mzta | 0:e7eb8eab3aa9 | 142 | case EXPR_DOT: |
mzta | 0:e7eb8eab3aa9 | 143 | /* a message dot was the last token, */ |
mzta | 0:e7eb8eab3aa9 | 144 | /* there has to come more */ |
mzta | 0:e7eb8eab3aa9 | 145 | code_block_open = TRUE; |
mzta | 0:e7eb8eab3aa9 | 146 | break; |
mzta | 0:e7eb8eab3aa9 | 147 | case EXPR_CLASS: |
mzta | 0:e7eb8eab3aa9 | 148 | /* a class keyword is not enough! */ |
mzta | 0:e7eb8eab3aa9 | 149 | /* we need also a name of the class */ |
mzta | 0:e7eb8eab3aa9 | 150 | code_block_open = TRUE; |
mzta | 0:e7eb8eab3aa9 | 151 | break; |
mzta | 0:e7eb8eab3aa9 | 152 | case EXPR_FNAME: |
mzta | 0:e7eb8eab3aa9 | 153 | /* a method name is necessary */ |
mzta | 0:e7eb8eab3aa9 | 154 | code_block_open = TRUE; |
mzta | 0:e7eb8eab3aa9 | 155 | break; |
mzta | 0:e7eb8eab3aa9 | 156 | case EXPR_VALUE: |
mzta | 0:e7eb8eab3aa9 | 157 | /* if, elsif, etc. without condition */ |
mzta | 0:e7eb8eab3aa9 | 158 | code_block_open = TRUE; |
mzta | 0:e7eb8eab3aa9 | 159 | break; |
mzta | 0:e7eb8eab3aa9 | 160 | |
mzta | 0:e7eb8eab3aa9 | 161 | /* now all the states which are closed */ |
mzta | 0:e7eb8eab3aa9 | 162 | |
mzta | 0:e7eb8eab3aa9 | 163 | case EXPR_ARG: |
mzta | 0:e7eb8eab3aa9 | 164 | /* an argument is the last token */ |
mzta | 0:e7eb8eab3aa9 | 165 | code_block_open = FALSE; |
mzta | 0:e7eb8eab3aa9 | 166 | break; |
mzta | 0:e7eb8eab3aa9 | 167 | |
mzta | 0:e7eb8eab3aa9 | 168 | /* all states which are unsure */ |
mzta | 0:e7eb8eab3aa9 | 169 | |
mzta | 0:e7eb8eab3aa9 | 170 | case EXPR_CMDARG: |
mzta | 0:e7eb8eab3aa9 | 171 | break; |
mzta | 0:e7eb8eab3aa9 | 172 | case EXPR_END: |
mzta | 0:e7eb8eab3aa9 | 173 | /* an expression was ended */ |
mzta | 0:e7eb8eab3aa9 | 174 | break; |
mzta | 0:e7eb8eab3aa9 | 175 | case EXPR_ENDARG: |
mzta | 0:e7eb8eab3aa9 | 176 | /* closing parenthese */ |
mzta | 0:e7eb8eab3aa9 | 177 | break; |
mzta | 0:e7eb8eab3aa9 | 178 | case EXPR_ENDFN: |
mzta | 0:e7eb8eab3aa9 | 179 | /* definition end */ |
mzta | 0:e7eb8eab3aa9 | 180 | break; |
mzta | 0:e7eb8eab3aa9 | 181 | case EXPR_MID: |
mzta | 0:e7eb8eab3aa9 | 182 | /* jump keyword like break, return, ... */ |
mzta | 0:e7eb8eab3aa9 | 183 | break; |
mzta | 0:e7eb8eab3aa9 | 184 | case EXPR_MAX_STATE: |
mzta | 0:e7eb8eab3aa9 | 185 | /* don't know what to do with this token */ |
mzta | 0:e7eb8eab3aa9 | 186 | break; |
mzta | 0:e7eb8eab3aa9 | 187 | default: |
mzta | 0:e7eb8eab3aa9 | 188 | /* this state is unexpected! */ |
mzta | 0:e7eb8eab3aa9 | 189 | break; |
mzta | 0:e7eb8eab3aa9 | 190 | } |
mzta | 0:e7eb8eab3aa9 | 191 | |
mzta | 0:e7eb8eab3aa9 | 192 | return code_block_open; |
mzta | 0:e7eb8eab3aa9 | 193 | } |
mzta | 0:e7eb8eab3aa9 | 194 | |
mzta | 0:e7eb8eab3aa9 | 195 | |
mzta | 0:e7eb8eab3aa9 | 196 | #if defined(__cplusplus) |
mzta | 0:e7eb8eab3aa9 | 197 | extern "C" { |
mzta | 0:e7eb8eab3aa9 | 198 | #endif |
mzta | 0:e7eb8eab3aa9 | 199 | void mrb_show_version(mrb_state *); |
mzta | 0:e7eb8eab3aa9 | 200 | void mrb_show_copyright(mrb_state *); |
mzta | 0:e7eb8eab3aa9 | 201 | #if defined(__cplusplus) |
mzta | 0:e7eb8eab3aa9 | 202 | } |
mzta | 0:e7eb8eab3aa9 | 203 | #endif |
mzta | 0:e7eb8eab3aa9 | 204 | |
mzta | 0:e7eb8eab3aa9 | 205 | struct _args { |
mzta | 0:e7eb8eab3aa9 | 206 | mrb_bool verbose : 1; |
mzta | 0:e7eb8eab3aa9 | 207 | int argc; |
mzta | 0:e7eb8eab3aa9 | 208 | char** argv; |
mzta | 0:e7eb8eab3aa9 | 209 | }; |
mzta | 0:e7eb8eab3aa9 | 210 | |
mzta | 0:e7eb8eab3aa9 | 211 | static void |
mzta | 0:e7eb8eab3aa9 | 212 | usage(const char *name) |
mzta | 0:e7eb8eab3aa9 | 213 | { |
mzta | 0:e7eb8eab3aa9 | 214 | static const char *const usage_msg[] = { |
mzta | 0:e7eb8eab3aa9 | 215 | "switches:", |
mzta | 0:e7eb8eab3aa9 | 216 | "-v print version number, then run in verbose mode", |
mzta | 0:e7eb8eab3aa9 | 217 | "--verbose run in verbose mode", |
mzta | 0:e7eb8eab3aa9 | 218 | "--version print the version", |
mzta | 0:e7eb8eab3aa9 | 219 | "--copyright print the copyright", |
mzta | 0:e7eb8eab3aa9 | 220 | NULL |
mzta | 0:e7eb8eab3aa9 | 221 | }; |
mzta | 0:e7eb8eab3aa9 | 222 | const char *const *p = usage_msg; |
mzta | 0:e7eb8eab3aa9 | 223 | |
mzta | 0:e7eb8eab3aa9 | 224 | pc.printf("Usage: %s [switches]\n", name); |
mzta | 0:e7eb8eab3aa9 | 225 | while (*p) |
mzta | 0:e7eb8eab3aa9 | 226 | pc.printf(" %s\n", *p++); |
mzta | 0:e7eb8eab3aa9 | 227 | } |
mzta | 0:e7eb8eab3aa9 | 228 | |
mzta | 0:e7eb8eab3aa9 | 229 | static int |
mzta | 0:e7eb8eab3aa9 | 230 | parse_args(mrb_state *mrb, int argc, char **argv, struct _args *args) |
mzta | 0:e7eb8eab3aa9 | 231 | { |
mzta | 0:e7eb8eab3aa9 | 232 | static const struct _args args_zero = { 0 }; |
mzta | 0:e7eb8eab3aa9 | 233 | |
mzta | 0:e7eb8eab3aa9 | 234 | *args = args_zero; |
mzta | 0:e7eb8eab3aa9 | 235 | |
mzta | 0:e7eb8eab3aa9 | 236 | for (argc--,argv++; argc > 0; argc--,argv++) { |
mzta | 0:e7eb8eab3aa9 | 237 | char *item; |
mzta | 0:e7eb8eab3aa9 | 238 | if (argv[0][0] != '-') break; |
mzta | 0:e7eb8eab3aa9 | 239 | |
mzta | 0:e7eb8eab3aa9 | 240 | item = argv[0] + 1; |
mzta | 0:e7eb8eab3aa9 | 241 | switch (*item++) { |
mzta | 0:e7eb8eab3aa9 | 242 | case 'v': |
mzta | 0:e7eb8eab3aa9 | 243 | if (!args->verbose) mrb_show_version(mrb); |
mzta | 0:e7eb8eab3aa9 | 244 | args->verbose = TRUE; |
mzta | 0:e7eb8eab3aa9 | 245 | break; |
mzta | 0:e7eb8eab3aa9 | 246 | case '-': |
mzta | 0:e7eb8eab3aa9 | 247 | if (strcmp((*argv) + 2, "version") == 0) { |
mzta | 0:e7eb8eab3aa9 | 248 | mrb_show_version(mrb); |
mzta | 0:e7eb8eab3aa9 | 249 | exit(EXIT_SUCCESS); |
mzta | 0:e7eb8eab3aa9 | 250 | } |
mzta | 0:e7eb8eab3aa9 | 251 | else if (strcmp((*argv) + 2, "verbose") == 0) { |
mzta | 0:e7eb8eab3aa9 | 252 | args->verbose = TRUE; |
mzta | 0:e7eb8eab3aa9 | 253 | break; |
mzta | 0:e7eb8eab3aa9 | 254 | } |
mzta | 0:e7eb8eab3aa9 | 255 | else if (strcmp((*argv) + 2, "copyright") == 0) { |
mzta | 0:e7eb8eab3aa9 | 256 | mrb_show_copyright(mrb); |
mzta | 0:e7eb8eab3aa9 | 257 | exit(EXIT_SUCCESS); |
mzta | 0:e7eb8eab3aa9 | 258 | } |
mzta | 0:e7eb8eab3aa9 | 259 | default: |
mzta | 0:e7eb8eab3aa9 | 260 | return EXIT_FAILURE; |
mzta | 0:e7eb8eab3aa9 | 261 | } |
mzta | 0:e7eb8eab3aa9 | 262 | } |
mzta | 0:e7eb8eab3aa9 | 263 | return EXIT_SUCCESS; |
mzta | 0:e7eb8eab3aa9 | 264 | } |
mzta | 0:e7eb8eab3aa9 | 265 | |
mzta | 0:e7eb8eab3aa9 | 266 | static void |
mzta | 0:e7eb8eab3aa9 | 267 | cleanup(mrb_state *mrb, struct _args *args) |
mzta | 0:e7eb8eab3aa9 | 268 | { |
mzta | 0:e7eb8eab3aa9 | 269 | mrb_close(mrb); |
mzta | 0:e7eb8eab3aa9 | 270 | } |
mzta | 0:e7eb8eab3aa9 | 271 | |
mzta | 0:e7eb8eab3aa9 | 272 | /* Print a short remark for the user */ |
mzta | 0:e7eb8eab3aa9 | 273 | static void |
mzta | 0:e7eb8eab3aa9 | 274 | print_hint(void) |
mzta | 0:e7eb8eab3aa9 | 275 | { |
mzta | 0:e7eb8eab3aa9 | 276 | pc.printf("mirb - Embeddable Interactive Ruby Shell\n\n"); |
mzta | 0:e7eb8eab3aa9 | 277 | } |
mzta | 0:e7eb8eab3aa9 | 278 | |
mzta | 0:e7eb8eab3aa9 | 279 | #ifndef ENABLE_READLINE |
mzta | 0:e7eb8eab3aa9 | 280 | /* Print the command line prompt of the REPL */ |
mzta | 0:e7eb8eab3aa9 | 281 | static void |
mzta | 0:e7eb8eab3aa9 | 282 | print_cmdline(int code_block_open) |
mzta | 0:e7eb8eab3aa9 | 283 | { |
mzta | 0:e7eb8eab3aa9 | 284 | if (code_block_open) { |
mzta | 0:e7eb8eab3aa9 | 285 | pc.printf("* "); |
mzta | 0:e7eb8eab3aa9 | 286 | } |
mzta | 0:e7eb8eab3aa9 | 287 | else { |
mzta | 0:e7eb8eab3aa9 | 288 | pc.printf("> "); |
mzta | 0:e7eb8eab3aa9 | 289 | } |
mzta | 0:e7eb8eab3aa9 | 290 | } |
mzta | 0:e7eb8eab3aa9 | 291 | #endif |
mzta | 0:e7eb8eab3aa9 | 292 | |
mzta | 0:e7eb8eab3aa9 | 293 | #if defined(__cplusplus) |
mzta | 0:e7eb8eab3aa9 | 294 | extern "C" { |
mzta | 0:e7eb8eab3aa9 | 295 | #endif |
mzta | 0:e7eb8eab3aa9 | 296 | void mrb_codedump_all(mrb_state*, struct RProc*); |
mzta | 0:e7eb8eab3aa9 | 297 | #if defined(__cplusplus) |
mzta | 0:e7eb8eab3aa9 | 298 | } |
mzta | 0:e7eb8eab3aa9 | 299 | #endif |
mzta | 0:e7eb8eab3aa9 | 300 | |
mzta | 0:e7eb8eab3aa9 | 301 | static int |
mzta | 0:e7eb8eab3aa9 | 302 | check_keyword(const char *buf, const char *word) |
mzta | 0:e7eb8eab3aa9 | 303 | { |
mzta | 0:e7eb8eab3aa9 | 304 | const char *p = buf; |
mzta | 0:e7eb8eab3aa9 | 305 | size_t len = strlen(word); |
mzta | 0:e7eb8eab3aa9 | 306 | |
mzta | 0:e7eb8eab3aa9 | 307 | /* skip preceding spaces */ |
mzta | 0:e7eb8eab3aa9 | 308 | while (*p && isspace((unsigned char)*p)) { |
mzta | 0:e7eb8eab3aa9 | 309 | p++; |
mzta | 0:e7eb8eab3aa9 | 310 | } |
mzta | 0:e7eb8eab3aa9 | 311 | /* check keyword */ |
mzta | 0:e7eb8eab3aa9 | 312 | if (strncmp(p, word, len) != 0) { |
mzta | 0:e7eb8eab3aa9 | 313 | return 0; |
mzta | 0:e7eb8eab3aa9 | 314 | } |
mzta | 0:e7eb8eab3aa9 | 315 | p += len; |
mzta | 0:e7eb8eab3aa9 | 316 | /* skip trailing spaces */ |
mzta | 0:e7eb8eab3aa9 | 317 | while (*p) { |
mzta | 0:e7eb8eab3aa9 | 318 | if (!isspace((unsigned char)*p)) return 0; |
mzta | 0:e7eb8eab3aa9 | 319 | p++; |
mzta | 0:e7eb8eab3aa9 | 320 | } |
mzta | 0:e7eb8eab3aa9 | 321 | return 1; |
mzta | 0:e7eb8eab3aa9 | 322 | } |
mzta | 0:e7eb8eab3aa9 | 323 | |
mzta | 0:e7eb8eab3aa9 | 324 | char *mirb_argv[] = { |
mzta | 0:e7eb8eab3aa9 | 325 | "mirb" |
mzta | 0:e7eb8eab3aa9 | 326 | }; |
mzta | 0:e7eb8eab3aa9 | 327 | |
mzta | 0:e7eb8eab3aa9 | 328 | int |
mzta | 0:e7eb8eab3aa9 | 329 | main(void) |
mzta | 0:e7eb8eab3aa9 | 330 | { |
mzta | 0:e7eb8eab3aa9 | 331 | int argc = 1; |
mzta | 0:e7eb8eab3aa9 | 332 | char **argv = mirb_argv; |
mzta | 0:e7eb8eab3aa9 | 333 | char ruby_code[1024] = { 0 }; |
mzta | 0:e7eb8eab3aa9 | 334 | char last_code_line[1024] = { 0 }; |
mzta | 0:e7eb8eab3aa9 | 335 | #ifndef ENABLE_READLINE |
mzta | 0:e7eb8eab3aa9 | 336 | int last_char; |
mzta | 0:e7eb8eab3aa9 | 337 | int char_index; |
mzta | 0:e7eb8eab3aa9 | 338 | #else |
mzta | 0:e7eb8eab3aa9 | 339 | char *history_path; |
mzta | 0:e7eb8eab3aa9 | 340 | #endif |
mzta | 0:e7eb8eab3aa9 | 341 | mrbc_context *cxt; |
mzta | 0:e7eb8eab3aa9 | 342 | struct mrb_parser_state *parser; |
mzta | 0:e7eb8eab3aa9 | 343 | mrb_state *mrb; |
mzta | 0:e7eb8eab3aa9 | 344 | mrb_value result; |
mzta | 0:e7eb8eab3aa9 | 345 | struct _args args; |
mzta | 0:e7eb8eab3aa9 | 346 | int n; |
mzta | 0:e7eb8eab3aa9 | 347 | mrb_bool code_block_open = FALSE; |
mzta | 0:e7eb8eab3aa9 | 348 | int ai; |
mzta | 0:e7eb8eab3aa9 | 349 | unsigned int stack_keep = 0; |
mzta | 0:e7eb8eab3aa9 | 350 | |
mzta | 0:e7eb8eab3aa9 | 351 | /* new interpreter instance */ |
mzta | 0:e7eb8eab3aa9 | 352 | mrb = mrb_open(); |
mzta | 0:e7eb8eab3aa9 | 353 | if (mrb == NULL) { |
mzta | 0:e7eb8eab3aa9 | 354 | pc.printf("Invalid mrb interpreter, exiting mirb\n"); |
mzta | 0:e7eb8eab3aa9 | 355 | return EXIT_FAILURE; |
mzta | 0:e7eb8eab3aa9 | 356 | } |
mzta | 0:e7eb8eab3aa9 | 357 | mrb_define_global_const(mrb, "ARGV", mrb_ary_new_capa(mrb, 0)); |
mzta | 0:e7eb8eab3aa9 | 358 | |
mzta | 0:e7eb8eab3aa9 | 359 | n = parse_args(mrb, argc, argv, &args); |
mzta | 0:e7eb8eab3aa9 | 360 | if (n == EXIT_FAILURE) { |
mzta | 0:e7eb8eab3aa9 | 361 | cleanup(mrb, &args); |
mzta | 0:e7eb8eab3aa9 | 362 | usage(argv[0]); |
mzta | 0:e7eb8eab3aa9 | 363 | return n; |
mzta | 0:e7eb8eab3aa9 | 364 | } |
mzta | 0:e7eb8eab3aa9 | 365 | |
mzta | 0:e7eb8eab3aa9 | 366 | #ifdef ENABLE_READLINE |
mzta | 0:e7eb8eab3aa9 | 367 | history_path = get_history_path(mrb); |
mzta | 0:e7eb8eab3aa9 | 368 | if (history_path == NULL) { |
mzta | 0:e7eb8eab3aa9 | 369 | pc.printf("failed to get history path\n"); |
mzta | 0:e7eb8eab3aa9 | 370 | mrb_close(mrb); |
mzta | 0:e7eb8eab3aa9 | 371 | return EXIT_FAILURE; |
mzta | 0:e7eb8eab3aa9 | 372 | } |
mzta | 0:e7eb8eab3aa9 | 373 | |
mzta | 0:e7eb8eab3aa9 | 374 | MIRB_USING_HISTORY(); |
mzta | 0:e7eb8eab3aa9 | 375 | MIRB_READ_HISTORY(history_path); |
mzta | 0:e7eb8eab3aa9 | 376 | #endif |
mzta | 0:e7eb8eab3aa9 | 377 | |
mzta | 0:e7eb8eab3aa9 | 378 | print_hint(); |
mzta | 0:e7eb8eab3aa9 | 379 | |
mzta | 0:e7eb8eab3aa9 | 380 | cxt = mrbc_context_new(mrb); |
mzta | 0:e7eb8eab3aa9 | 381 | cxt->capture_errors = TRUE; |
mzta | 0:e7eb8eab3aa9 | 382 | cxt->lineno = 1; |
mzta | 0:e7eb8eab3aa9 | 383 | mrbc_filename(mrb, cxt, "(mirb)"); |
mzta | 0:e7eb8eab3aa9 | 384 | if (args.verbose) cxt->dump_result = TRUE; |
mzta | 0:e7eb8eab3aa9 | 385 | |
mzta | 0:e7eb8eab3aa9 | 386 | ai = mrb_gc_arena_save(mrb); |
mzta | 0:e7eb8eab3aa9 | 387 | |
mzta | 0:e7eb8eab3aa9 | 388 | while (TRUE) { |
mzta | 0:e7eb8eab3aa9 | 389 | #ifndef ENABLE_READLINE |
mzta | 0:e7eb8eab3aa9 | 390 | print_cmdline(code_block_open); |
mzta | 0:e7eb8eab3aa9 | 391 | |
mzta | 0:e7eb8eab3aa9 | 392 | char_index = 0; |
mzta | 0:e7eb8eab3aa9 | 393 | while ((last_char = pc.getc()) != '\n') { |
mzta | 0:e7eb8eab3aa9 | 394 | pc.printf("%c", last_char); |
mzta | 0:e7eb8eab3aa9 | 395 | if (last_char == EOF) break; |
mzta | 0:e7eb8eab3aa9 | 396 | if (char_index > sizeof(last_code_line)-2) { |
mzta | 0:e7eb8eab3aa9 | 397 | pc.printf("input string too long\n"); |
mzta | 0:e7eb8eab3aa9 | 398 | continue; |
mzta | 0:e7eb8eab3aa9 | 399 | } |
mzta | 0:e7eb8eab3aa9 | 400 | if (last_char == '\x08') { |
mzta | 0:e7eb8eab3aa9 | 401 | if (char_index > 0) char_index--; |
mzta | 0:e7eb8eab3aa9 | 402 | continue; |
mzta | 0:e7eb8eab3aa9 | 403 | } else { |
mzta | 0:e7eb8eab3aa9 | 404 | last_code_line[char_index++] = last_char; |
mzta | 0:e7eb8eab3aa9 | 405 | } |
mzta | 0:e7eb8eab3aa9 | 406 | } |
mzta | 0:e7eb8eab3aa9 | 407 | pc.printf("\n"); |
mzta | 0:e7eb8eab3aa9 | 408 | |
mzta | 0:e7eb8eab3aa9 | 409 | if (last_char == EOF) { |
mzta | 0:e7eb8eab3aa9 | 410 | pc.printf("\n"); |
mzta | 0:e7eb8eab3aa9 | 411 | break; |
mzta | 0:e7eb8eab3aa9 | 412 | } |
mzta | 0:e7eb8eab3aa9 | 413 | |
mzta | 0:e7eb8eab3aa9 | 414 | last_code_line[char_index++] = '\n'; |
mzta | 0:e7eb8eab3aa9 | 415 | last_code_line[char_index] = '\0'; |
mzta | 0:e7eb8eab3aa9 | 416 | #else |
mzta | 0:e7eb8eab3aa9 | 417 | char* line = MIRB_READLINE(code_block_open ? "* " : "> "); |
mzta | 0:e7eb8eab3aa9 | 418 | if (line == NULL) { |
mzta | 0:e7eb8eab3aa9 | 419 | printf("\n"); |
mzta | 0:e7eb8eab3aa9 | 420 | break; |
mzta | 0:e7eb8eab3aa9 | 421 | } |
mzta | 0:e7eb8eab3aa9 | 422 | if (strlen(line) > sizeof(last_code_line)-2) { |
mzta | 0:e7eb8eab3aa9 | 423 | pc.printf("input string too long\n"); |
mzta | 0:e7eb8eab3aa9 | 424 | continue; |
mzta | 0:e7eb8eab3aa9 | 425 | } |
mzta | 0:e7eb8eab3aa9 | 426 | strcpy(last_code_line, line); |
mzta | 0:e7eb8eab3aa9 | 427 | strcat(last_code_line, "\n"); |
mzta | 0:e7eb8eab3aa9 | 428 | MIRB_ADD_HISTORY(line); |
mzta | 0:e7eb8eab3aa9 | 429 | free(line); |
mzta | 0:e7eb8eab3aa9 | 430 | #endif |
mzta | 0:e7eb8eab3aa9 | 431 | |
mzta | 0:e7eb8eab3aa9 | 432 | if (code_block_open) { |
mzta | 0:e7eb8eab3aa9 | 433 | if (strlen(ruby_code)+strlen(last_code_line) > sizeof(ruby_code)-1) { |
mzta | 0:e7eb8eab3aa9 | 434 | pc.printf("concatenated input string too long\n"); |
mzta | 0:e7eb8eab3aa9 | 435 | continue; |
mzta | 0:e7eb8eab3aa9 | 436 | } |
mzta | 0:e7eb8eab3aa9 | 437 | strcat(ruby_code, last_code_line); |
mzta | 0:e7eb8eab3aa9 | 438 | } |
mzta | 0:e7eb8eab3aa9 | 439 | else { |
mzta | 0:e7eb8eab3aa9 | 440 | if (check_keyword(last_code_line, "quit") || check_keyword(last_code_line, "exit")) { |
mzta | 0:e7eb8eab3aa9 | 441 | break; |
mzta | 0:e7eb8eab3aa9 | 442 | } |
mzta | 0:e7eb8eab3aa9 | 443 | strcpy(ruby_code, last_code_line); |
mzta | 0:e7eb8eab3aa9 | 444 | } |
mzta | 0:e7eb8eab3aa9 | 445 | |
mzta | 0:e7eb8eab3aa9 | 446 | /* parse code */ |
mzta | 0:e7eb8eab3aa9 | 447 | parser = mrb_parser_new(mrb); |
mzta | 0:e7eb8eab3aa9 | 448 | if (parser == NULL) { |
mzta | 0:e7eb8eab3aa9 | 449 | pc.printf("create parser state error\n"); |
mzta | 0:e7eb8eab3aa9 | 450 | break; |
mzta | 0:e7eb8eab3aa9 | 451 | } |
mzta | 0:e7eb8eab3aa9 | 452 | parser->s = ruby_code; |
mzta | 0:e7eb8eab3aa9 | 453 | parser->send = ruby_code + strlen(ruby_code); |
mzta | 0:e7eb8eab3aa9 | 454 | parser->lineno = cxt->lineno; |
mzta | 0:e7eb8eab3aa9 | 455 | mrb_parser_parse(parser, cxt); |
mzta | 0:e7eb8eab3aa9 | 456 | code_block_open = is_code_block_open(parser); |
mzta | 0:e7eb8eab3aa9 | 457 | |
mzta | 0:e7eb8eab3aa9 | 458 | if (code_block_open) { |
mzta | 0:e7eb8eab3aa9 | 459 | /* no evaluation of code */ |
mzta | 0:e7eb8eab3aa9 | 460 | } |
mzta | 0:e7eb8eab3aa9 | 461 | else { |
mzta | 0:e7eb8eab3aa9 | 462 | if (0 < parser->nerr) { |
mzta | 0:e7eb8eab3aa9 | 463 | /* syntax error */ |
mzta | 0:e7eb8eab3aa9 | 464 | pc.printf("line %d: %s\n", parser->error_buffer[0].lineno, parser->error_buffer[0].message); |
mzta | 0:e7eb8eab3aa9 | 465 | } |
mzta | 0:e7eb8eab3aa9 | 466 | else { |
mzta | 0:e7eb8eab3aa9 | 467 | /* generate bytecode */ |
mzta | 0:e7eb8eab3aa9 | 468 | struct RProc *proc = mrb_generate_code(mrb, parser); |
mzta | 0:e7eb8eab3aa9 | 469 | if (proc == NULL) { |
mzta | 0:e7eb8eab3aa9 | 470 | pc.printf("codegen error\n"); |
mzta | 0:e7eb8eab3aa9 | 471 | mrb_parser_free(parser); |
mzta | 0:e7eb8eab3aa9 | 472 | break; |
mzta | 0:e7eb8eab3aa9 | 473 | } |
mzta | 0:e7eb8eab3aa9 | 474 | |
mzta | 0:e7eb8eab3aa9 | 475 | if (args.verbose) { |
mzta | 0:e7eb8eab3aa9 | 476 | mrb_codedump_all(mrb, proc); |
mzta | 0:e7eb8eab3aa9 | 477 | } |
mzta | 0:e7eb8eab3aa9 | 478 | /* pass a proc for evaulation */ |
mzta | 0:e7eb8eab3aa9 | 479 | /* evaluate the bytecode */ |
mzta | 0:e7eb8eab3aa9 | 480 | result = mrb_context_run(mrb, |
mzta | 0:e7eb8eab3aa9 | 481 | proc, |
mzta | 0:e7eb8eab3aa9 | 482 | mrb_top_self(mrb), |
mzta | 0:e7eb8eab3aa9 | 483 | stack_keep); |
mzta | 0:e7eb8eab3aa9 | 484 | stack_keep = proc->body.irep->nlocals; |
mzta | 0:e7eb8eab3aa9 | 485 | /* did an exception occur? */ |
mzta | 0:e7eb8eab3aa9 | 486 | if (mrb->exc) { |
mzta | 0:e7eb8eab3aa9 | 487 | p(mrb, mrb_obj_value(mrb->exc), 0); |
mzta | 0:e7eb8eab3aa9 | 488 | mrb->exc = 0; |
mzta | 0:e7eb8eab3aa9 | 489 | } |
mzta | 0:e7eb8eab3aa9 | 490 | else { |
mzta | 0:e7eb8eab3aa9 | 491 | /* no */ |
mzta | 0:e7eb8eab3aa9 | 492 | if (!mrb_respond_to(mrb, result, mrb_intern_lit(mrb, "inspect"))){ |
mzta | 0:e7eb8eab3aa9 | 493 | result = mrb_any_to_s(mrb, result); |
mzta | 0:e7eb8eab3aa9 | 494 | } |
mzta | 0:e7eb8eab3aa9 | 495 | p(mrb, result, 1); |
mzta | 0:e7eb8eab3aa9 | 496 | } |
mzta | 0:e7eb8eab3aa9 | 497 | } |
mzta | 0:e7eb8eab3aa9 | 498 | ruby_code[0] = '\0'; |
mzta | 0:e7eb8eab3aa9 | 499 | last_code_line[0] = '\0'; |
mzta | 0:e7eb8eab3aa9 | 500 | mrb_gc_arena_restore(mrb, ai); |
mzta | 0:e7eb8eab3aa9 | 501 | } |
mzta | 0:e7eb8eab3aa9 | 502 | mrb_parser_free(parser); |
mzta | 0:e7eb8eab3aa9 | 503 | cxt->lineno++; |
mzta | 0:e7eb8eab3aa9 | 504 | } |
mzta | 0:e7eb8eab3aa9 | 505 | |
mzta | 0:e7eb8eab3aa9 | 506 | #ifdef ENABLE_READLINE |
mzta | 0:e7eb8eab3aa9 | 507 | MIRB_WRITE_HISTORY(history_path); |
mzta | 0:e7eb8eab3aa9 | 508 | mrb_free(mrb, history_path); |
mzta | 0:e7eb8eab3aa9 | 509 | #endif |
mzta | 0:e7eb8eab3aa9 | 510 | |
mzta | 0:e7eb8eab3aa9 | 511 | mrbc_context_free(mrb, cxt); |
mzta | 0:e7eb8eab3aa9 | 512 | mrb_close(mrb); |
mzta | 0:e7eb8eab3aa9 | 513 | |
mzta | 0:e7eb8eab3aa9 | 514 | return 0; |
mzta | 0:e7eb8eab3aa9 | 515 | } |