Update utils.zig

Author: michal92299

hacker-parser/utils.zig

@@ -1,6 +1,8 @@
 const std = @import("std");
 const parse = @import("parse.zig");
+
 pub const HACKER_DIR_SUFFIX = "/.hackeros/hacker-lang";
+
 pub fn deinitParseResult(res: *parse.ParseResult, allocator: std.mem.Allocator) void {
     {
         var it = res.deps.keyIterator();
@@ -24,6 +26,14 @@ pub fn deinitParseResult(res: *parse.ParseResult, allocator: std.mem.Allocator)
         }
         res.vars_dict.deinit();
     }
+    {
+        var it = res.local_vars.iterator();
+        while (it.next()) |entry| {
+            allocator.free(entry.key_ptr.*);
+            allocator.free(entry.value_ptr.*);
+        }
+        res.local_vars.deinit();
+    }
     {
         for (res.cmds.items) |item| {
             allocator.free(item);
@@ -43,11 +53,22 @@ pub fn deinitParseResult(res: *parse.ParseResult, allocator: std.mem.Allocator)
         res.binaries.deinit();
     }
     {
-        for (res.plugins.items) |item| {
-            allocator.free(item);
+        for (res.plugins.items) |p| {
+            allocator.free(p.path);
         }
         res.plugins.deinit();
     }
+    {
+        var it = res.functions.iterator();
+        while (it.next()) |entry| {
+            for (entry.value_ptr.items) |item| {
+                allocator.free(item);
+            }
+            entry.value_ptr.deinit();
+            allocator.free(entry.key_ptr.*);
+        }
+        res.functions.deinit();
+    }
     {
         for (res.errors.items) |item| {
             allocator.free(item);
@@ -63,15 +84,28 @@ pub fn deinitParseResult(res: *parse.ParseResult, allocator: std.mem.Allocator)
         res.config_data.deinit();
     }
 }
+
 pub fn mergeHashMaps(comptime V: type, dest: *std.StringHashMap(V), src: std.StringHashMap(V), allocator: std.mem.Allocator) !void {
     var it = src.iterator();
     while (it.next()) |entry| {
         try dest.put(try allocator.dupe(u8, entry.key_ptr.*), entry.value_ptr.*);
     }
 }
+
 pub fn mergeStringHashMaps(dest: *std.StringHashMap([]const u8), src: std.StringHashMap([]const u8), allocator: std.mem.Allocator) !void {
     var it = src.iterator();
     while (it.next()) |entry| {
         try dest.put(try allocator.dupe(u8, entry.key_ptr.*), try allocator.dupe(u8, entry.value_ptr.*));
     }
 }
+
+pub fn mergeFunctionMaps(dest: *std.StringHashMap(std.ArrayList([]const u8)), src: std.StringHashMap(std.ArrayList([]const u8)), allocator: std.mem.Allocator) !void {
+    var it = src.iterator();
+    while (it.next()) |entry| {
+        var new_list = std.ArrayList([]const u8).init(allocator);
+        for (entry.value_ptr.items) |item| {
+            try new_list.append(try allocator.dupe(u8, item));
+        }
+        try dest.put(try allocator.dupe(u8, entry.key_ptr.*), new_list);
+    }
+}