1 files changed, 29 insertions, 9 deletions

diff --git a/src/yuescript/yue_parser.cpp b/src/yuescript/yue_parser.cpp
index f383275..f72407d 100644
--- a/src/yuescript/yue_parser.cpp
+++ b/src/yuescript/yue_parser.cpp
@@ -853,13 +853,6 @@ YueParser::YueParser() {
         ChainAssign = Seperator >> Exp >> +(space >> '=' >> space >> Exp >> space >> and_('=')) >> space >> Assign;
 
         StatementAppendix = (IfLine | WhileLine | CompInner) >> space;
-        StatementSep = and_(
-                *space_break >> check_indent_match >> space >> (
-                        set("($'\"") |
-                        "[[" |
-                        "[="
-                )
-        );
         Statement =
                 Seperator >>
                 -(
@@ -876,8 +869,9 @@ YueParser::YueParser() {
                         StatementAppendix >> empty_block_error
                 ) >>
                 space >>
-                -StatementAppendix >>
-                -StatementSep;
+                -StatementAppendix;
+
+        StatementSep = white >> (set("('\"") | "[[" | "[=");
 
         Body = in_block | Statement;
 
@@ -905,6 +899,32 @@ YueParser::YueParser() {
 }
 // clang-format on
 
+bool YueParser::startWith(std::string_view codes, rule& r) {
+        std::unique_ptr<input> converted;
+        if (codes.substr(0, 3) == "\xEF\xBB\xBF"sv) {
+                codes = codes.substr(3);
+        }
+        try {
+                if (!codes.empty()) {
+                        converted = std::make_unique<input>(_converter.from_bytes(&codes.front(), &codes.back() + 1));
+                } else {
+                        converted = std::make_unique<input>();
+                }
+        } catch (const std::range_error&) {
+                return false;
+        }
+        error_list errors;
+        try {
+                State state;
+                return ::yue::start_with(*converted, r, errors, &state);
+        } catch (const ParserError&) {
+                return false;
+        } catch (const std::logic_error&) {
+                return false;
+        }
+        return true;
+}
+
 ParseInfo YueParser::parse(std::string_view codes, rule& r) {
         ParseInfo res;
         if (codes.substr(0, 3) == "\xEF\xBB\xBF"sv) {