support enum as string

TODO.md

@@ -21,7 +21,7 @@
 * [done] lambda捕获列表支持move
 * [done] 内置/注册函数支持赋值给callable【目前不支持内置函数返回类型，因为有些内置函数是模板化的，其类型根据输入参数确定】
 * [done] 支持enum
-  - enum的字面量格式应该是什么样的？
+  - enum的字面量格式如下:
     - [done] enum type:enum_item
 * [done] 优化内置方法的错误提示
 * [done] 支持变量引用（不涉及类型系统）
@@ -31,7 +31,7 @@
 * [done] 支持比较运算符 < , <=, >, >=
 * 支持新的数据类型 Map
 * 支持函数重载
-* 支持注册函数的包管理
+* [done] 支持注册函数的包管理
 * [done] Output to json format
 * for语句支持变量定义
 * [done] 重构ValueCategory处理系统(bug: lambda捕获的ref变量不应该和callable保持一致的ValueCategory，它应该总是LValue) 

include/wamon/type.h

@@ -185,6 +185,9 @@ inline bool IsStructOrEnumType(const std::unique_ptr<Type>& type) {
   return type->IsBasicType() && !IsBuiltInType(type);
 }
 
+class PackageUnit;
+bool IsEnumType(const std::unique_ptr<Type>& type, const PackageUnit&);
+
 namespace detail {
 
 inline std::vector<std::string> GetBuiltInTypesWithoutVoid() {

src/operator.cc

@@ -230,6 +230,14 @@ static void register_buildin_operator_handles(std::unordered_map<std::string, Op
       tmp->SetValue(AsListVariable(v1)->get_string_only_for_byte_list());
       return tmp;
     }
+    // enum item to string
+    if (IsEnumType(from_type, interpreter.GetPackageUnit()) && IsStringType(to_type)) {
+      auto enum_def = interpreter.GetPackageUnit().FindEnum(from_type->GetTypeInfo());
+      assert(enum_def != nullptr);
+      const auto& enum_item = AsEnumVariable(v1)->GetEnumItem();
+      std::string enum_str = enum_def->GetEnumName() + ":" + enum_item;
+      return std::make_shared<StringVariable>(enum_str, Variable::ValueCategory::RValue, "");
+    }
     // struct to struct trait
     auto v = VariableFactory(to_type, Variable::ValueCategory::RValue, "", interpreter);
     v->ConstructByFields({v1});

src/type.cc

@@ -184,4 +184,11 @@ void CheckCanConstructBy(const PackageUnit& pu, const std::unique_ptr<Type>& var
 
 }  // namespace detail
 
+bool IsEnumType(const std::unique_ptr<Type>& type, const PackageUnit& pu) {
+  if (IsStructOrEnumType(type) == false) {
+    return false;
+  }
+  return pu.FindEnum(type->GetTypeInfo()) != nullptr;
+}
+
 }  // namespace wamon

src/type_checker.cc

@@ -337,6 +337,9 @@ std::unique_ptr<Type> CheckAndGetAsResultType(std::unique_ptr<Type> lt, std::uni
   if (TypeFactory<std::vector<unsigned char>>::Get()->GetTypeInfo() == lt->GetTypeInfo() && IsStringType(rt)) {
     return rt;
   }
+  if (IsEnumType(lt, pu) && IsStringType(rt)) {
+    return rt;
+  }
   std::string reason;
   if (CheckTraitConstraint(pu, rt, lt, reason)) {
     return rt;

test/interpreter_test.cc

@@ -564,6 +564,16 @@ TEST(interpreter, operator) {
       let v : double = 3.9;
       return v as int;
     }
+
+    enum Color {
+      Red;
+      Blue;
+    }
+
+    func as_test_5() -> string {
+      let v : Color = enum Color:Blue;
+      return v as string;
+    }
   )";
   wamon::PackageUnit pu;
   auto tokens = scan.Scan(str);
@@ -629,6 +639,10 @@ TEST(interpreter, operator) {
   EXPECT_EQ(ret->GetTypeInfo(), "int");
   EXPECT_EQ(wamon::AsIntVariable(ret)->GetValue(), 3);
 
+  ret = interpreter.CallFunctionByName("main$as_test_5", {});
+  EXPECT_EQ(ret->GetTypeInfo(), "string");
+  EXPECT_EQ(wamon::AsStringVariable(ret)->GetValue(), "main$Color:Blue");
+
   ret = interpreter.ExecExpression(tc, "main", "0X30 as int");
   EXPECT_EQ(ret->GetTypeInfo(), "int");
   EXPECT_EQ(wamon::AsIntVariable(ret)->GetValue(), 48);