Implementation of SymbolTable::removeSymbol(std::string) function.

n8lang · Dec 26, 2024 · efb2a3f · efb2a3f
1 parent 1b3ca00
commit efb2a3f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/n8/core/SymbolTable.cpp b/src/n8/core/SymbolTable.cpp
@@ -79,6 +79,16 @@ void SymbolTable::setSymbol(
     else this->table[name] = std::move(value);
 }
 
+void SymbolTable::removeSymbol(std::string name) {
+    if(this->parent && this->parent->hasSymbol(name)) {
+        this->parent->removeSymbol(name);
+        return;
+    }
+
+    if(this->hasSymbol(name) && !this->table[name].hasLock())
+        this->table.erase(name);
+}
+
 void SymbolTable::removeSymbol(std::shared_ptr<Token> name) {
     std::string symbol = name->getImage();
     if(this->parent && this->parent->hasSymbol(symbol)) {