টাইপ চেকিং বেসিকস

class SymbolTable:
    """L28-এর সিম্বল টেবিল -- এখানে শুধু আইডেন্টিফায়ারের টাইপ সরবরাহ করতে ব্যবহৃত।"""
    def __init__(self):
        self._table = {}
    def insert(self, name, type_):
        self._table[name] = {"type": type_}
    def lookup(self, name):
        if name not in self._table:
            raise ValueError(f"সেমান্টিক এরর: '{name}' ডিক্লেয়ার করা হয়নি")
        return self._table[name]


NUMERIC = {"INT", "FLOAT"}

def type_check(node, symtab):
    """AST-এর উপর বটম-আপ টাইপ চেক -- প্রতিটি নোডের টাইপ ফেরত দেয়, অথবা TypeError রেইজ করে।"""
    tag = node[0]

    if tag == "lit":
        value = node[1]
        if isinstance(value, bool):
            return "BOOL"
        if isinstance(value, int):
            return "INT"
        if isinstance(value, float):
            return "FLOAT"
        if isinstance(value, str):
            return "STRING"
        raise TypeError(f"অজানা লিটারেল: {value!r}")

    if tag == "id":
        name = node[1]
        return symtab.lookup(name)["type"]

    if tag == "binop":
        op = node[1]
        left_type = type_check(node[2], symtab)
        right_type = type_check(node[3], symtab)

        if op in ("+", "-", "*", "/"):
            if op == "+" and left_type == "STRING" and right_type == "STRING":
                return "STRING"  # স্ট্রিং কনক্যাটেনেশন আলাদাভাবে বৈধ
            if left_type in NUMERIC and right_type in NUMERIC:
                return "FLOAT" if "FLOAT" in (left_type, right_type) else "INT"
            raise TypeError(
                f"টাইপ এরর: '{op}' অপারেটর {left_type} ও {right_type}-এর মধ্যে প্রযোজ্য নয় "
                f"(ব্যর্থ নোড: {node!r})"
            )

        if op in ("<", ">", "==", "<=", ">="):
            if left_type != right_type:
                raise TypeError(
                    f"টাইপ এরর: তুলনা অপারেটর '{op}' ভিন্ন টাইপ {left_type} ও {right_type}-এর মধ্যে "
                    f"প্রযোজ্য নয় (ব্যর্থ নোড: {node!r})"
                )
            return "BOOL"

        raise TypeError(f"অজানা অপারেটর: {op!r}")

    raise TypeError(f"অজানা নোড টাইপ: {tag!r}")


symtab = SymbolTable()
symtab.insert("x", "INT")

test_cases = [
    ("(3 + 4) * 2  -- সব INT",
     ("binop", "*", ("binop", "+", ("lit", 3), ("lit", 4)), ("lit", 2))),
    ("x + 5  -- সিম্বল টেবিল থেকে x-এর টাইপ (INT)",
     ("binop", "+", ("id", "x"), ("lit", 5))),
    ('"hi" + "there"  -- STRING কনক্যাটেনেশন',
     ("binop", "+", ("lit", "hi"), ("lit", "there"))),
]

print("--- বৈধ এক্সপ্রেশন ---")
for label, tree in test_cases:
    result_type = type_check(tree, symtab)
    print(f"{label:38s} -> {result_type}")

print('\n--- অবৈধ এক্সপ্রেশন: "hello" + 5 ---')
bad_tree = ("binop", "+", ("lit", "hello"), ("lit", 5))
try:
    type_check(bad_tree, symtab)
    print("(এটা প্রিন্ট হওয়ার কথা না -- এরর ধরা পড়া উচিত ছিল)")
except TypeError as e:
    print(f"সঠিকভাবে REJECT হয়েছে -> {e}")