From d1a1c1592e610526c4a0432f93bd9ea6ae96d6e8 Mon Sep 17 00:00:00 2001
From: mryouse
Date: Sat, 18 Jun 2022 02:45:04 +0000
Subject: refactor: make neb a module

---
 neb/typeclass.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 neb/typeclass.py

(limited to 'neb/typeclass.py')

diff --git a/neb/typeclass.py b/neb/typeclass.py
new file mode 100644
index 0000000..eae412c
--- /dev/null
+++ b/neb/typeclass.py
@@ -0,0 +1,34 @@
+from enum import Enum, auto
+
+class TypeEnum(Enum):
+    ANY = auto()
+    STRING = auto()
+    INT = auto()
+    FLOAT = auto()
+    NUMBER = auto()
+    LIST = auto()
+    LITERAL = auto()
+    BOOL = auto()
+
+    def __str__(self):
+        return f":{self.name.lower()}"
+
+HIERARCHY = { TypeEnum.ANY: None,
+                    TypeEnum.LITERAL: TypeEnum.ANY,
+                    TypeEnum.LIST: TypeEnum.ANY,
+                    TypeEnum.STRING: TypeEnum.LITERAL,
+                    TypeEnum.BOOL: TypeEnum.LITERAL,
+                    TypeEnum.NUMBER: TypeEnum.LITERAL,
+                    TypeEnum.INT: TypeEnum.NUMBER,
+                    TypeEnum.FLOAT: TypeEnum.NUMBER }
+
+def is_subtype_of(candidate, expected):
+    if candidate == expected:
+        return True
+    parent = HIERARCHY[candidate]
+    while parent is not None:
+        if parent == expected:
+            return True
+        parent = HIERARCHY[parent]
+    return False
+
-- 
cgit v1.2.3