fix issue #41, add square brackets support for metatable syntax. fix using metatable with "with" syntax issue.

3 files changed, 33 insertions, 11 deletions

diff --git a/src/yuescript/yue_ast.h b/src/yuescript/yue_ast.h
index 535a239..b673eb3 100644
--- a/src/yuescript/yue_ast.h
+++ b/src/yuescript/yue_ast.h
@@ -431,7 +431,7 @@ AST_NODE(meta_variable_pair)
 AST_END(meta_variable_pair)
 
 AST_NODE(meta_normal_pair)
-        ast_sel<true, Name_t> key;
+        ast_sel<true, Name_t, Exp_t> key;
         ast_sel<true, Exp_t, TableBlock_t> value;
         AST_MEMBER(meta_normal_pair, &key, &value)
 AST_END(meta_normal_pair)
diff --git a/src/yuescript/yue_compiler.cpp b/src/yuescript/yue_compiler.cpp
index 043c705..a958462 100644
--- a/src/yuescript/yue_compiler.cpp
+++ b/src/yuescript/yue_compiler.cpp
@@ -12,6 +12,7 @@ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLI
 #include <stack>
 #include <vector>
 #include <memory>
+#include <set>
 
 #include "yuescript/yue_parser.h"
 #include "yuescript/yue_compiler.h"
@@ -58,7 +59,7 @@ inline std::string s(std::string_view sv) {
         return std::string(sv);
 }
 
-const std::string_view version = "0.7.0"sv;
+const std::string_view version = "0.7.1"sv;
 const std::string_view extension = "yue"sv;
 
 class YueCompilerImpl {
@@ -1549,11 +1550,20 @@ private:
                                                 }
                                                 case id<meta_normal_pair_t>(): {
                                                         auto mp = static_cast<meta_normal_pair_t*>(item);
-                                                        auto key = _parser.toString(mp->key);
-                                                        _buf << "__"sv << key;
-                                                        auto newKey = toAst<KeyName_t>(clearBuf(), item);
                                                         auto newPair = item->new_ptr<normal_pair_t>();
-                                                        newPair->key.set(newKey);
+                                                        switch (mp->key->getId()) {
+                                                                case id<Name_t>(): {
+                                                                        auto key = _parser.toString(mp->key);
+                                                                        _buf << "__"sv << key;
+                                                                        auto newKey = toAst<KeyName_t>(clearBuf(), mp->key);
+                                                                        newPair->key.set(newKey);
+                                                                        break;
+                                                                }
+                                                                case id<Exp_t>():
+                                                                        newPair->key.set(mp->key);
+                                                                        break;
+                                                                default: YUEE("AST node mismatch", mp->key); break;
+                                                        }
                                                         newPair->value.set(mp->value);
                                                         subMetaDestruct->values.push_back(newPair);
                                                         break;
@@ -3167,6 +3177,9 @@ private:
                 if (opIt == chainList.end()) return false;
                 auto x = chainList.front();
                 auto chain = x->new_ptr<ChainValue_t>();
+                if (opIt == chainList.begin() && ast_is<ColonChainItem_t, DotChainItem_t>(x)) {
+                        chain->items.push_back(toAst<Callable_t>(_withVars.top(), x));
+                }
                 for (auto it = chainList.begin(); it != opIt; ++it) {
                         chain->items.push_back(*it);
                 }
@@ -5157,11 +5170,20 @@ private:
                                 case id<meta_normal_pair_t>(): {
                                         isMetamethod = true;
                                         auto mp = static_cast<meta_normal_pair_t*>(pair);
-                                        auto key = _parser.toString(mp->key);
-                                        _buf << "__"sv << key;
-                                        auto newKey = toAst<KeyName_t>(clearBuf(), pair);
                                         auto newPair = pair->new_ptr<normal_pair_t>();
-                                        newPair->key.set(newKey);
+                                        switch (mp->key->getId()) {
+                                                case id<Name_t>(): {
+                                                        auto key = _parser.toString(mp->key);
+                                                        _buf << "__"sv << key;
+                                                        auto newKey = toAst<KeyName_t>(clearBuf(), mp->key);
+                                                        newPair->key.set(newKey);
+                                                        break;
+                                                }
+                                                case id<Exp_t>():
+                                                        newPair->key.set(mp->key);
+                                                        break;
+                                                default: YUEE("AST node mismatch", mp->key); break;
+                                        }
                                         newPair->value.set(mp->value);
                                         metatable->pairs.push_back(newPair);
                                         break;
diff --git a/src/yuescript/yue_parser.cpp b/src/yuescript/yue_parser.cpp
index c72c005..ec6c997 100644
--- a/src/yuescript/yue_parser.cpp
+++ b/src/yuescript/yue_parser.cpp
@@ -525,7 +525,7 @@ YueParser::YueParser() {
 
         meta_variable_pair = sym(':') >> Variable >> expr('#');
 
-        meta_normal_pair = Space >> Name >> expr("#:") >>
+        meta_normal_pair = (Space >> Name | sym('[') >> Exp >> sym(']')) >> expr("#:") >>
                 (Exp | TableBlock | +(SpaceBreak) >> Exp);
 
         KeyValue = variable_pair | normal_pair | meta_variable_pair | meta_normal_pair;