From 78d7ed2c0a497675ba24a43c60d3e276c09e04a2 Mon Sep 17 00:00:00 2001
From: MrBounty <adrien.bouvais.pro@gmail.com>
Date: Sun, 13 Oct 2024 13:28:26 +0200
Subject: [PATCH] Fix environment variable

I was freeing the path just after getting it, changed that
---
 src/cli.zig | 13 +++++--------
 1 file changed, 5 insertions(+), 8 deletions(-)

diff --git a/src/cli.zig b/src/cli.zig
index 26b1a48..4778b6a 100644
--- a/src/cli.zig
+++ b/src/cli.zig
@@ -37,21 +37,19 @@ const State = enum {
 pub fn main() !void {
     var gpa = std.heap.GeneralPurposeAllocator(.{}){};
     const allocator = gpa.allocator();
-    defer {
-        switch (gpa.deinit()) {
-            .ok => std.log.debug("No memory leak baby !\n", .{}),
-            .leak => std.log.debug("We fucked it up bro...\n", .{}),
-        }
-    }
+    defer switch (gpa.deinit()) {
+        .ok => {},
+        .leak => std.log.debug("We fucked it up bro...\n", .{}),
+    };
 
     const path_env_variable = utils.getEnvVariables(allocator, "ZIPPONDB_PATH");
+    defer if (path_env_variable) |path| allocator.free(path);
     var file_engine: FileEngine = undefined;
     defer file_engine.deinit();
 
     if (path_env_variable) |path| {
         std.debug.print("Environment variable found: {s}\n", .{path});
         file_engine = FileEngine.init(allocator, path_env_variable.?);
-        defer allocator.free(path_env_variable.?);
     } else {
         file_engine = FileEngine.init(allocator, "");
         std.debug.print("No ZIPONDB_PATH envirionment variable found, please use the command:\n db use path/to/db \nor\n db new /path/to/dir\n", .{});
@@ -62,7 +60,6 @@ pub fn main() !void {
 
     var state: State = .expect_main_command;
 
-    // TODO: Use a State to prevent first_token and second_token
     while (true) {
         std.debug.print("> ", .{});
         const line = try std.io.getStdIn().reader().readUntilDelimiterOrEof(line_buf, '\n');