Compare commits

...

6 commits

5 changed files with 81 additions and 20 deletions

View file

@ -17,6 +17,8 @@ pub const Cursor = @import("Cursor.zig");
pub const View = @import("View.zig");
pub const Selection = @import("Selection.zig");
pub const FindMode = enum { exact, case_folded };
pub const Metrics = struct {
ctx: *const anyopaque,
egc_length: egc_length_func,
@ -943,10 +945,10 @@ const Node = union(enum) {
}
}
pub const FindMode = enum { exact, case_folded };
pub const FindAllCallback = fn (data: *anyopaque, begin_row: usize, begin_col: usize, end_row: usize, end_col: usize) error{Stop}!void;
pub fn find_all_ranges(self: *const Node, pattern: []const u8, data: *anyopaque, callback: *const FindAllCallback, mode: FindMode, allocator: Allocator) error{ OutOfMemory, Stop }!void {
const Ctx = struct {
allocator: std.mem.Allocator,
pattern: []const u8,
data: *anyopaque,
callback: *const FindAllCallback,
@ -985,7 +987,11 @@ const Node = union(enum) {
input = input[input_consume_size..];
},
.case_folded => {
@panic("unimplemented");
const input_consume_size = @min(ctx.buf.len - ctx.rest.len, input.len);
var writer = std.Io.Writer.fixed(ctx.buf[ctx.rest.len..]);
unicode.case_folded_write(&writer, input[0..input_consume_size]) catch return error.WriteFailed;
ctx.rest = ctx.buf[0 .. ctx.rest.len + writer.end];
input = input[input_consume_size..];
},
}
@ -1030,6 +1036,7 @@ const Node = union(enum) {
}
};
var ctx: Ctx = .{
.allocator = allocator,
.pattern = pattern,
.data = data,
.callback = callback,

View file

@ -113,9 +113,7 @@ pub const TransformError = error{
WriteFailed,
};
fn utf8_transform(comptime field: uucode.FieldEnum, allocator: std.mem.Allocator, text: []const u8) TransformError![]u8 {
var result: std.Io.Writer.Allocating = .init(allocator);
defer result.deinit();
fn utf8_write_transform(comptime field: uucode.FieldEnum, writer: *std.Io.Writer, text: []const u8) TransformError!void {
const view: std.unicode.Utf8View = try .init(text);
var it = view.iterator();
while (it.nextCodepoint()) |cp| {
@ -126,12 +124,18 @@ fn utf8_transform(comptime field: uucode.FieldEnum, allocator: std.mem.Allocator
};
var utf8_buf: [6]u8 = undefined;
const size = try std.unicode.utf8Encode(cp_, &utf8_buf);
try result.writer.writeAll(utf8_buf[0..size]);
try writer.writeAll(utf8_buf[0..size]);
}
}
fn utf8_transform(comptime field: uucode.FieldEnum, allocator: std.mem.Allocator, text: []const u8) TransformError![]u8 {
var result: std.Io.Writer.Allocating = .init(allocator);
defer result.deinit();
try utf8_write_transform(field, &result.writer, text);
return result.toOwnedSlice();
}
fn utf8_predicate(comptime field: uucode.FieldEnum, text: []const u8) TransformError!bool {
fn utf8_predicate(comptime field: uucode.FieldEnum, text: []const u8) error{InvalidUtf8}!bool {
const view: std.unicode.Utf8View = try .init(text);
var it = view.iterator();
while (it.nextCodepoint()) |cp| {
@ -162,6 +166,10 @@ pub fn case_fold(allocator: std.mem.Allocator, text: []const u8) TransformError!
return utf8_transform(.case_folding_simple, allocator, text);
}
pub fn case_folded_write(writer: *std.Io.Writer, text: []const u8) TransformError!void {
return utf8_write_transform(.case_folding_simple, writer, text);
}
pub fn switch_case(allocator: std.mem.Allocator, text: []const u8) TransformError![]u8 {
return if (try utf8_predicate(.is_lowercase, text))
to_upper(allocator, text)
@ -169,5 +177,9 @@ pub fn switch_case(allocator: std.mem.Allocator, text: []const u8) TransformErro
to_lower(allocator, text);
}
pub fn is_lowercase(text: []const u8) error{InvalidUtf8}!bool {
return try utf8_predicate(.is_lowercase, text);
}
const std = @import("std");
const uucode = @import("vaxis").uucode;

View file

@ -558,6 +558,7 @@
["ctrl+space", "mini_mode_cancel"],
["ctrl+enter", "mini_mode_insert_bytes", "\n"],
["ctrl+backspace", "mini_mode_reset"],
["alt+c", "toggle_find_mode"],
["alt+v", "system_paste"],
["alt+n", "goto_next_match"],
["alt+p", "goto_prev_match"],

View file

@ -5218,12 +5218,16 @@ pub const Editor = struct {
pub fn find_query(self: *Self, ctx: Context) Result {
var query: []const u8 = undefined;
var match_type: Match.Type = undefined;
var find_mode: Buffer.FindMode = .exact;
if (ctx.args.match(.{tp.extract(&query)}) catch false) {
self.match_type = .find;
try self.find_in_buffer(query, .none);
try self.find_in_buffer(query, .none, find_mode);
self.clamp();
} else if (ctx.args.match(.{ tp.extract(&query), tp.extract(&match_type) }) catch false) {
try self.find_in_buffer(query, match_type);
try self.find_in_buffer(query, match_type, find_mode);
self.clamp();
} else if (ctx.args.match(.{ tp.extract(&query), tp.extract(&match_type), tp.extract(&find_mode) }) catch false) {
try self.find_in_buffer(query, match_type, find_mode);
self.clamp();
} else return error.InvalidFindQueryArgument;
}
@ -5233,7 +5237,7 @@ pub const Editor = struct {
_ = ctx;
const query: []const u8 = try self.copy_word_at_cursor(self.allocator);
defer self.allocator.free(query);
try self.find_in_buffer(query, .find);
try self.find_in_buffer(query, .find, .exact);
}
pub const find_word_at_cursor_meta: Meta = .{ .description = "Search for the word under the cursor" };
@ -5274,13 +5278,13 @@ pub const Editor = struct {
} else self.last_find_query = self.allocator.dupe(u8, query) catch return;
}
pub fn find_in_buffer(self: *Self, query: []const u8, match_type: Match.Type) !void {
pub fn find_in_buffer(self: *Self, query: []const u8, match_type: Match.Type, find_mode: Buffer.FindMode) !void {
self.set_last_find_query(query, match_type);
self.match_type = match_type;
return self.find_in_buffer_sync(query);
return self.find_in_buffer_sync(query, find_mode);
}
fn find_in_buffer_sync(self: *Self, query: []const u8) !void {
fn find_in_buffer_sync(self: *Self, query: []const u8, mode: Buffer.FindMode) !void {
const Ctx = struct {
matches: usize = 0,
self: *Self,
@ -5296,7 +5300,7 @@ pub const Editor = struct {
defer self.add_match_done();
var ctx: Ctx = .{ .self = self };
self.init_matches_update();
try root.find_all_ranges(query, &ctx, Ctx.cb, .exact, self.allocator);
try root.find_all_ranges(query, &ctx, Ctx.cb, mode, self.allocator);
}
fn find_in_buffer_async(self: *Self, query: []const u8) !void {
@ -5511,7 +5515,7 @@ pub const Editor = struct {
if (self.matches.items.len == 0) {
if (self.last_find_query) |last| {
self.find_operation = .goto_next_match;
try self.find_in_buffer(last, self.last_find_query_match_type);
try self.find_in_buffer(last, self.last_find_query_match_type, .exact);
}
}
try self.move_cursor_next_match(ctx);
@ -5540,7 +5544,7 @@ pub const Editor = struct {
if (self.matches.items.len == 0) {
if (self.last_find_query) |last| {
self.find_operation = .goto_prev_match;
try self.find_in_buffer(last, self.last_find_query_match_type);
try self.find_in_buffer(last, self.last_find_query_match_type, .exact);
}
}
try self.move_cursor_prev_match(ctx);

View file

@ -1,9 +1,11 @@
const tp = @import("thespian");
const cbor = @import("cbor");
const input = @import("input");
const keybind = @import("keybind");
const command = @import("command");
const EventHandler = @import("EventHandler");
const Buffer = @import("Buffer");
const tui = @import("../../tui.zig");
const ed = @import("../../editor.zig");
@ -14,11 +16,17 @@ const ArrayList = @import("std").ArrayList;
const Self = @This();
const name = "󱎸 find";
const name_auto = name;
const name_exact = name ++ "";
const name_case_folded = name ++ "";
const Commands = command.Collection(cmds);
const Mode = enum { auto, exact, case_folded };
allocator: Allocator,
input_: ArrayList(u8),
find_mode: Mode = .auto,
last_input: ArrayList(u8),
start_view: ed.View,
start_cursor: ed.Cursor,
@ -26,7 +34,7 @@ editor: *ed.Editor,
history_pos: ?usize = null,
commands: Commands = undefined,
pub fn create(allocator: Allocator, _: command.Context) !struct { tui.Mode, tui.MiniMode } {
pub fn create(allocator: Allocator, ctx: command.Context) !struct { tui.Mode, tui.MiniMode } {
const editor = tui.get_active_editor() orelse return error.NotFound;
const self = try allocator.create(Self);
errdefer allocator.destroy(self);
@ -39,7 +47,11 @@ pub fn create(allocator: Allocator, _: command.Context) !struct { tui.Mode, tui.
.editor = editor,
};
try self.commands.init(self);
if (editor.get_primary().selection) |sel| ret: {
_ = ctx.args.match(.{cbor.extract(&self.find_mode)}) catch {};
var query: []const u8 = undefined;
if (ctx.args.match(.{ cbor.extract(&self.find_mode), cbor.extract(&query) }) catch false) {
try self.input_.appendSlice(self.allocator, query);
} else if (editor.get_primary().selection) |sel| ret: {
const text = editor.get_selection(sel, self.allocator) catch break :ret;
defer self.allocator.free(text);
try self.input_.appendSlice(self.allocator, text);
@ -48,7 +60,11 @@ pub fn create(allocator: Allocator, _: command.Context) !struct { tui.Mode, tui.
.insert_command = "mini_mode_insert_bytes",
});
mode.event_handler = EventHandler.to_owned(self);
return .{ mode, .{ .name = name } };
return .{ mode, .{ .name = switch (self.find_mode) {
.auto => name_auto,
.exact => name_exact,
.case_folded => name_case_folded,
} } };
}
pub fn deinit(self: *Self) void {
@ -91,13 +107,21 @@ fn flush_input(self: *Self) !void {
const primary = self.editor.get_primary();
primary.selection = null;
primary.cursor = self.start_cursor;
try self.editor.find_in_buffer(self.input_.items, .find);
try self.editor.find_in_buffer(self.input_.items, .find, switch (self.find_mode) {
.auto => self.auto_detect_mode(),
.exact => .exact,
.case_folded => .case_folded,
});
} else {
self.editor.get_primary().selection = null;
self.editor.init_matches_update();
}
}
fn auto_detect_mode(self: *Self) Buffer.FindMode {
return if (Buffer.unicode.is_lowercase(self.input_.items) catch return .exact) .case_folded else .exact;
}
fn cmd(self: *Self, name_: []const u8, ctx: command.Context) tp.result {
self.flush_input() catch {};
return command.executeName(name_, ctx);
@ -153,6 +177,19 @@ const cmds = struct {
const Meta = command.Metadata;
const Result = command.Result;
pub fn toggle_find_mode(self: *Self, _: Ctx) Result {
const new_find_mode: Buffer.FindMode = switch (self.find_mode) {
.exact => .case_folded,
.auto, .case_folded => .exact,
};
const allocator = self.allocator;
const query = try allocator.dupe(u8, self.input_.items);
defer allocator.free(query);
self.cancel();
command.executeName("find", command.fmt(.{ new_find_mode, query })) catch {};
}
pub const toggle_find_mode_meta: Meta = .{ .description = "Toggle find mode" };
pub fn mini_mode_reset(self: *Self, _: Ctx) Result {
self.input_.clearRetainingCapacity();
self.update_mini_mode_text();