mirror of
https://github.com/W3SLAV/micropython.git
synced 2025-06-19 20:15:33 -04:00
97790455fe
Full CPython compatibility with this requires actually parsing the input so far collected, and if it fails parsing due to lack of tokens, then continue collecting input. It's not worth doing it this way. Not having compatibility at this level does not hurt the goals of Micro Python.
80 lines
2.3 KiB
C
80 lines
2.3 KiB
C
#include "misc.h"
|
|
#include "mpconfig.h"
|
|
#include "repl.h"
|
|
|
|
#if MICROPY_ENABLE_REPL_HELPERS
|
|
|
|
bool str_startswith_word(const char *str, const char *head) {
|
|
int i;
|
|
for (i = 0; str[i] && head[i]; i++) {
|
|
if (str[i] != head[i]) {
|
|
return false;
|
|
}
|
|
}
|
|
return head[i] == '\0' && (str[i] == '\0' || !unichar_isalpha(str[i]));
|
|
}
|
|
|
|
bool mp_repl_continue_with_input(const char *input) {
|
|
// check for blank input
|
|
if (input[0] == '\0') {
|
|
return false;
|
|
}
|
|
|
|
// check if input starts with a certain keyword
|
|
bool starts_with_compound_keyword =
|
|
input[0] == '@'
|
|
|| str_startswith_word(input, "if")
|
|
|| str_startswith_word(input, "while")
|
|
|| str_startswith_word(input, "for")
|
|
|| str_startswith_word(input, "try")
|
|
|| str_startswith_word(input, "with")
|
|
|| str_startswith_word(input, "def")
|
|
|| str_startswith_word(input, "class")
|
|
;
|
|
|
|
// check for unmatched open bracket or triple quote
|
|
// TODO don't look at triple quotes inside single quotes
|
|
int n_paren = 0;
|
|
int n_brack = 0;
|
|
int n_brace = 0;
|
|
int in_triple_quote = 0;
|
|
const char *i;
|
|
for (i = input; *i; i++) {
|
|
switch (*i) {
|
|
case '(': n_paren += 1; break;
|
|
case ')': n_paren -= 1; break;
|
|
case '[': n_brack += 1; break;
|
|
case ']': n_brack -= 1; break;
|
|
case '{': n_brace += 1; break;
|
|
case '}': n_brace -= 1; break;
|
|
case '\'':
|
|
if (in_triple_quote != '"' && i[1] == '\'' && i[2] == '\'') {
|
|
i += 2;
|
|
in_triple_quote = '\'' - in_triple_quote;
|
|
}
|
|
break;
|
|
case '"':
|
|
if (in_triple_quote != '\'' && i[1] == '"' && i[2] == '"') {
|
|
i += 2;
|
|
in_triple_quote = '"' - in_triple_quote;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
// continue if unmatched brackets or quotes
|
|
if (n_paren > 0 || n_brack > 0 || n_brace > 0 || in_triple_quote != 0) {
|
|
return true;
|
|
}
|
|
|
|
// continue if compound keyword and last line was not empty
|
|
if (starts_with_compound_keyword && i[-1] != '\n') {
|
|
return true;
|
|
}
|
|
|
|
// otherwise, don't continue
|
|
return false;
|
|
}
|
|
|
|
#endif // MICROPY_ENABLE_REPL_HELPERS
|