Added Allocator to Editor.prompt & Editor.promptEx

1 files changed, 8 insertions, 7 deletions

diff --git a/src/Editor.zig b/src/Editor.zig
index 469fc81..d721600 100644
--- a/src/Editor.zig
+++ b/src/Editor.zig
@@ -188,7 +188,7 @@ pub fn open(self: *Editor, name: []const u8) !void {
 }
 
 pub fn openFile(self: *Editor) !void {
-    const fname_opt = try self.prompt("File name");
+    const fname_opt = try self.prompt(self.allocator, "File name");
     if (fname_opt) |fname| {
         defer self.allocator.free(fname);
         return self.open(fname);
@@ -200,19 +200,20 @@ pub fn processKeypress(self: *Editor) !void {
     try self.current_state(self, self.buffer, key);
 }
 
-pub fn prompt(self: *Editor, prompt_str: []const u8) !?[]u8 {
-    return self.promptEx(void, error{}, prompt_str, null, {});
+pub fn prompt(self: *Editor, allocator: Allocator, prompt_str: []const u8) !?[]u8 {
+    return self.promptEx(void, error{}, allocator, prompt_str, null, {});
 }
 
 pub fn promptEx(
     self: *Editor,
     comptime CallbackData: type,
     comptime CallbackError: type,
+    allocator: Allocator,
     prompt_str: []const u8,
     callback: ?PromptCallback(CallbackData, CallbackError),
     cb_data: CallbackData,
 ) !?[]u8 {
-    var buf = ArrayList(u8).init(self.allocator);
+    var buf = ArrayList(u8).init(allocator);
     defer buf.deinit();
 
     while (true) {
@@ -262,7 +263,7 @@ pub fn promptYN(self: *Editor, prompt_str: []const u8) !bool {
     const full_prompt = try std.fmt.allocPrint(self.allocator, "{s} (Y/N)", .{prompt_str});
     defer self.allocator.free(full_prompt);
 
-    var response = try self.prompt(full_prompt);
+    var response = try self.prompt(self.allocator, full_prompt);
     defer if (response) |str| self.allocator.free(str);
     // TODO: This can be improved
     while (response != null
@@ -275,7 +276,7 @@ pub fn promptYN(self: *Editor, prompt_str: []const u8) !bool {
         )
     ) {
         if (response) |str| self.allocator.free(str);
-        response = try self.prompt(full_prompt);
+        response = try self.prompt(self.allocator, full_prompt);
     }
 
     return response != null and (response.?[0] == 'y' or response.?[0] == 'Y');
@@ -361,7 +362,7 @@ pub fn setStatusMessage(self: *Editor, comptime fmt: []const u8, args: anytype)
 
 pub fn switchBuffer(self: *Editor) !void {
     // TODO: completion
-    const bufname_opt = try self.prompt("Switch to buffer");
+    const bufname_opt = try self.prompt(self.allocator, "Switch to buffer");
     if (bufname_opt) |bufname| {
         defer self.allocator.free(bufname);