Fixed memory leaks.

1 files changed, 18 insertions, 14 deletions

diff --git a/ttt.jai b/ttt.jai
index 0715544..929a736 100644
--- a/ttt.jai
+++ b/ttt.jai
@@ -590,12 +590,14 @@ load_database :: (db: *Database, path: string) -> success: bool {
 // Returns success.
 export_to_csv :: (db: Database, path: string) -> success: bool {
     assert(xx path, ASSERT_NOT_EMPTY, "path");
+
+    auto_release_temp();
     
     builder: String_Builder;
     defer reset(*builder);
     
     CSV_HEADER :: string.[ "task", "sunday", "monday", "tuesday", "wednesday", "thursday", "friday", "saturday" ];
-    print_to_builder(*builder, "%\n", join(..CSV_HEADER, separator = ","));
+    print_to_builder(*builder, "%\n", join(..CSV_HEADER, separator = ",",, temporary_allocator));
     
     buffer: [Task.name.count] u8;
     name: string = xx buffer;
@@ -701,15 +703,16 @@ import_from_csv :: (db: *Database, path: string) -> bool {
     }
     
     { // Parse CSV lines.
-        auto_release_temp(); // TODO Needs to be tested.
         line := csv;
         while (true) {
+            auto_release_temp();
+            
             line, success := consume_next_line(*csv);
     //         line, success := next_line(*csv);
             if success == false break;
             
             task: Task;
-            csv_values := split(line, ",",, temporary_allocator); // TODO Temporary memory... if file is too big this may break.
+            csv_values := split(line, ",",, temporary_allocator);
             
             // Import task name.
             name_length := min(task.name.count, csv_values[0].count);
@@ -891,6 +894,8 @@ update_layout :: () {
 
 draw_tui :: (db: *Database, layout: *Layout) {
 
+    // TODO During dirty_flag revamp...we may also start using the String_Builder. 
+
     adjust_first_day_of_week := int.[
         (0 + FIRST_DAY_OF_WEEK) % NUM_WEEK_DAYS,
         (1 + FIRST_DAY_OF_WEEK) % NUM_WEEK_DAYS,
@@ -990,7 +995,6 @@ draw_tui :: (db: *Database, layout: *Layout) {
     // Display up to rows allowed by the layout, or less if reached end of database.
     idx_stop := idx_start + (ifx layout_tasks_rows > db.tasks.count - idx_start then db.tasks.count - idx_start else layout_tasks_rows);
     for task_idx: idx_start..idx_stop-1 {
-        auto_release_temp(); // TODO Temporary memory being trashed?!
         task := *db.tasks[task_idx];
         y += 1;
         x = 1;
@@ -1138,7 +1142,7 @@ read_input_int :: (y: int, message: string) -> value: int, success: bool {
     input_pos_x := x + message.count + 2;
     input_width := size_x - input_pos_x;
     
-    str := read_input_string(input_pos_x, y, input_width);
+    str := read_input_string(input_pos_x, y, input_width,, temporary_allocator);
     
     value, success := parse_int(*str);
     return value, success;
@@ -1170,7 +1174,6 @@ main :: () {
     defer free_memory();
     
     {   // Initialize app directory.
-        auto_release_temp();
         home_dir, success_dir := get_home_directory(); // Returns system owned memory.
         if success_dir == false {
             home_dir = ".";
@@ -1195,7 +1198,6 @@ main :: () {
     }
     
     { // Initialize database and archive files if needed.
-        auto_release_temp();
         if (file_exists(db_file_path) == false) {
             if (store_database(database, db_file_path) == false) {
                 print_error("Failed to initialize database.");
@@ -1333,7 +1335,8 @@ main :: () {
         }
         
         if is_exit_requested {
-            exit(0);
+            // exit(0); // TODO fucking exit does not report memory leaks.
+            return;
         }
     }
     
@@ -1353,7 +1356,9 @@ main :: () {
     TUI.flush_input();
     TUI.set_next_key(TUI.Keys.Resize);
     while (true) {
-
+        
+        reset_temporary_storage();
+        
         TUI.set_style(style_default);
         
         if (is_terminal_too_small) {
@@ -1366,7 +1371,6 @@ main :: () {
             draw_error_window();
         }
         
-        reset_temporary_storage();
         key := TUI.get_key(INPUT_TIMEOUT_MS);
         if key == #char "q" || key == #char "Q" break;
         update_times(*database);
@@ -1456,7 +1460,7 @@ main :: () {
 
                 // Change task name.
                 TUI.using_style(action_style);
-                input := read_input_string(2, selected_task_row, Task.name.count, size_x - 2 - Task.name.count);
+                input := read_input_string(2, selected_task_row, Task.name.count, size_x - 2 - Task.name.count,, temporary_allocator);
                 if is_empty_string(input) == false {
                     replace_chars(input, "\t\x0B\x0C\r", #char " "); // Replace weird spaces with space.
                     memset(selected_task.name.data, 0, Task.name.count);
@@ -1492,9 +1496,9 @@ main :: () {
             case #char "6"; #through;
             case #char "7";
                 if (selected_task == null) continue;
-                
+
                 // Prepare position to input time operation.
-                selected_day := cast(int)(key - #char "1"); // TODO DAM this cast...
+                selected_day := cast(int)(key - #char "1");
                 input_width := layout.columns[L_DAYS_IDX + selected_day].width;
                 input_pos_x := 2 + layout.columns[L_TITLE_IDX].width;
                 for 0..selected_day-1 {
@@ -1504,7 +1508,7 @@ main :: () {
 
                 // Get input string.
                 TUI.using_style(action_style);
-                input := read_input_string(input_pos_x, selected_task_row, input_width); // TODO Temp stringzes.
+                input := read_input_string(input_pos_x, selected_task_row, input_width,, temporary_allocator);
                 
                 // Abort if input if empty.
                 if is_empty_string(input) continue;