Switch to c++
This commit is contained in:
329
__old__/editor.rb
Normal file
329
__old__/editor.rb
Normal file
@@ -0,0 +1,329 @@
|
||||
class Editor
|
||||
attr_accessor :text, :cursor, :selection_start, :filename
|
||||
|
||||
def initialize(filename, x, y, width, height)
|
||||
contents = File.exist?(filename) ? File.read(filename) : "dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd"
|
||||
@text = contents.grapheme_clusters
|
||||
@cursor = @text.size
|
||||
@selection_start = 0
|
||||
@scroll = 0
|
||||
@x = x
|
||||
@y = y
|
||||
@width = width
|
||||
@height = height
|
||||
@filename = filename
|
||||
@folds = []
|
||||
@highlights = Array.new(@text.size)
|
||||
@highlight_generation = 0
|
||||
@tree = nil
|
||||
@num_width = (@text.count("\n") + 1).to_s.size + 3
|
||||
|
||||
@parser = C.ts_parser_new
|
||||
@lang_ts = C.tree_sitter_ruby
|
||||
C.ts_parser_set_language @parser, @lang_ts
|
||||
@cached_tree = nil
|
||||
C.load_query "/home/syed/main/crib/grammar/ruby.scm"
|
||||
|
||||
highlight!
|
||||
|
||||
rebuild_folded_lines!
|
||||
end
|
||||
|
||||
def save!
|
||||
File.write @filename, @text.join
|
||||
end
|
||||
|
||||
def highlight!
|
||||
gen = @highlight_generation
|
||||
src = @text.join
|
||||
text_bytes = []
|
||||
byte_offset = 0
|
||||
@text.each do |cluster|
|
||||
text_bytes << byte_offset
|
||||
byte_offset += cluster.bytesize
|
||||
end
|
||||
text_bytes << byte_offset
|
||||
line_map = []
|
||||
line = 0
|
||||
@text.each_with_index do |ch, i|
|
||||
line_map[i] = line
|
||||
line += 1 if ch == "\n"
|
||||
end
|
||||
# TODO: add ts_tree_edit calls in erase/insert
|
||||
count_ptr = FFI::MemoryPointer.new :uint32
|
||||
@cached_tree = C.ts_parser_parse_string @parser, nil, src, src.bytesize
|
||||
root = C.ts_tree_root_node @cached_tree
|
||||
ptr = C.ts_collect_tokens root, count_ptr, @lang_ts, src
|
||||
count = count_ptr.read_uint32
|
||||
ints = ptr.read_array_of_uint32 count
|
||||
C.free_tokens ptr
|
||||
count_ptr.free
|
||||
C.ts_tree_delete @cached_tree
|
||||
return if gen != @highlight_generation
|
||||
@highlight_generation = 0
|
||||
@highlights = Array.new(@text.size)
|
||||
done_folds = []
|
||||
ints.each_slice(3).map do |start_byte, end_byte, sym_i|
|
||||
start_cluster = byte_to_cluster start_byte, text_bytes
|
||||
end_cluster = byte_to_cluster end_byte - 1, text_bytes
|
||||
next if start_cluster.nil? || end_cluster.nil?
|
||||
sym = TS_SYMBOL_MAP[sym_i]
|
||||
if sym == "@fold"
|
||||
start_row = line_map[start_cluster]
|
||||
end_row = line_map[end_cluster]
|
||||
next if start_row == end_row
|
||||
hash_value = Zlib.crc32 @text[start_cluster...end_cluster].join
|
||||
done_folds << hash_value
|
||||
if (f = @folds.find { |f| f.crc32 == hash_value })
|
||||
f.start = start_row
|
||||
f.end = end_row
|
||||
else
|
||||
@folds << Fold.new(start_row, end_row, false, hash_value)
|
||||
end
|
||||
next
|
||||
end
|
||||
hl = TS_RUBY[sym]
|
||||
next unless hl
|
||||
(start_cluster..end_cluster).each do |i|
|
||||
@highlights[i] = hl if @highlights[i].nil? || @highlights[i].priority < hl.priority
|
||||
end
|
||||
end
|
||||
@folds.select! { |f| done_folds.include? f.crc32 }
|
||||
rebuild_folded_lines!
|
||||
end
|
||||
|
||||
def byte_to_cluster(byte, byte_starts)
|
||||
left = 0
|
||||
right = byte_starts.size - 2
|
||||
while left <= right
|
||||
mid = (left + right) / 2
|
||||
if byte_starts[mid] <= byte && byte < byte_starts[mid + 1]
|
||||
return mid
|
||||
elsif byte < byte_starts[mid]
|
||||
right = mid - 1
|
||||
else
|
||||
left = mid + 1
|
||||
end
|
||||
end
|
||||
nil
|
||||
end
|
||||
|
||||
def erase(x0, x1)
|
||||
return if x0 < 0 || x1 < x0
|
||||
lines_deleted = @text[x0...x1].count "\n"
|
||||
@text[x0...x1] = []
|
||||
@highlights[x0...x1] = []
|
||||
@highlight_generation += 1
|
||||
@folds.map! do |f|
|
||||
if f.start >= @text[...x0].count("\n")
|
||||
f.start -= lines_deleted
|
||||
f.end -= lines_deleted
|
||||
end
|
||||
f
|
||||
end
|
||||
rebuild_folded_lines!
|
||||
@num_width = (@text.count("\n") + 1).to_s.size + 3
|
||||
end
|
||||
|
||||
def insert(x, str)
|
||||
@text.insert x, *str.grapheme_clusters
|
||||
@highlights.insert x, *Array.new(str.grapheme_clusters.size)
|
||||
lines_added = str.count "\n"
|
||||
@folds.map! do |f|
|
||||
if f.start >= @text[...x].count("\n")
|
||||
f.start += lines_added
|
||||
f.end += lines_added
|
||||
end
|
||||
f
|
||||
end
|
||||
rebuild_folded_lines!
|
||||
@highlight_generation += 1
|
||||
@num_width = (@text.count("\n") + 1).to_s.size + 3
|
||||
end
|
||||
|
||||
def move_up!
|
||||
precursor_text = @text[...@cursor]
|
||||
last_newline = precursor_text.rindex "\n"
|
||||
return unless last_newline
|
||||
@preferred_col ||= @cursor - last_newline
|
||||
prev_start = precursor_text[...last_newline].rindex("\n") || -1
|
||||
@cursor = [
|
||||
prev_start + @preferred_col,
|
||||
@text[prev_start + 1..].index("\n")&.+(prev_start + 1) || @text.size
|
||||
].min
|
||||
end
|
||||
|
||||
def move_down!
|
||||
next_newline = @text[@cursor..].index("\n")&.+(@cursor)
|
||||
return unless next_newline
|
||||
@preferred_col ||= (@cursor - (@text[...@cursor].rindex("\n") || -1)).abs
|
||||
@cursor = [
|
||||
next_newline + @preferred_col,
|
||||
@text[next_newline + 1..].index("\n")&.+(next_newline + 1) || @text.size
|
||||
].min
|
||||
end
|
||||
|
||||
def move_left!
|
||||
@cursor = [@cursor - 1, 0].max
|
||||
@preferred_col = nil
|
||||
end
|
||||
|
||||
def move_right!
|
||||
@cursor = [@cursor + 1, @text.size].min
|
||||
@preferred_col = nil
|
||||
end
|
||||
|
||||
def cursor_valid?
|
||||
!@folded_lines.include? @text[0...@cursor].count("\n")
|
||||
end
|
||||
|
||||
def handle_event(event)
|
||||
return if event.nil?
|
||||
case KEY_TYPE[event[:key_type]]
|
||||
when :char
|
||||
ch = event[:c].chr
|
||||
ch = "\n" if ch == "\r"
|
||||
if ch == "'"
|
||||
cr = @text[0...@cursor].count("\n")
|
||||
@folds.find { |f| f.active = !f.active if f.start <= cr && f.end >= cr }
|
||||
rebuild_folded_lines!
|
||||
elsif ch == ctrl_key('s').chr
|
||||
save!
|
||||
elsif ch =~ /[[:print:]]|\n/
|
||||
insert @cursor, ch
|
||||
@cursor += 1
|
||||
elsif ch == "\b" || ch == "\x7f"
|
||||
erase @cursor - 1, @cursor
|
||||
@cursor = [@cursor - 1, 0].max
|
||||
end
|
||||
@preferred_col = nil
|
||||
when :special
|
||||
return unless MODIFIER[event[:special_modifier]] == :none
|
||||
case SPECIAL_KEY[event[:special_key]]
|
||||
when :up
|
||||
move_up!
|
||||
move_up! until cursor_valid?
|
||||
when :down
|
||||
move_down!
|
||||
move_down! until cursor_valid?
|
||||
when :left
|
||||
move_left!
|
||||
move_left! until cursor_valid?
|
||||
when :right
|
||||
move_right!
|
||||
move_right! until cursor_valid?
|
||||
when :delete
|
||||
erase @cursor, @cursor + 1
|
||||
@cursor = [@cursor, @text.size].min
|
||||
@preferred_col = nil
|
||||
end
|
||||
end
|
||||
move_right! until cursor_valid?
|
||||
adjust_scroll!
|
||||
end
|
||||
|
||||
def adjust_scroll!
|
||||
cr = cursor_row_visual
|
||||
start = @scroll
|
||||
last = @scroll + @height - 1
|
||||
if cr < start
|
||||
@scroll = cr
|
||||
elsif cr > last
|
||||
@scroll = cr - @height + 1
|
||||
end
|
||||
end
|
||||
|
||||
def cursor_row_visual
|
||||
row = col = 0
|
||||
nc = 0
|
||||
@text.each_with_index do |ch, i|
|
||||
return row if i == @cursor
|
||||
if @folded_lines.include? nc
|
||||
nc += 1 if ch == "\n"
|
||||
next
|
||||
end
|
||||
if ch == "\n"
|
||||
row += 1
|
||||
nc += 1
|
||||
col = 0
|
||||
next
|
||||
end
|
||||
w = C.display_width(ch)
|
||||
if col + w > @width - @num_width
|
||||
row += 1
|
||||
col = 0
|
||||
end
|
||||
col += w
|
||||
end
|
||||
row
|
||||
end
|
||||
|
||||
def rebuild_folded_lines!
|
||||
@folded_lines = Set.new
|
||||
@folds.each do |f|
|
||||
(f.start + 1..f.end).each { |line_num| @folded_lines.add(line_num) } if f.active
|
||||
end
|
||||
end
|
||||
|
||||
def render
|
||||
(0...@height).each { |h| (0...@width).each { |w|
|
||||
C.update @y + h, @x + w, ' ', 0x000000, 0x000000, 0
|
||||
} }
|
||||
row = col = nc = 0
|
||||
cursor_row = cursor_col = nil
|
||||
@text.each_with_index do |ch, i|
|
||||
if @folded_lines.include? nc
|
||||
nc += 1 if ch == "\n"
|
||||
next
|
||||
end
|
||||
if i == @cursor
|
||||
cursor_row = row
|
||||
cursor_col = col
|
||||
end
|
||||
if col == 0
|
||||
color = 0x666666
|
||||
color = 0xFFFFFF if cursor_row == row
|
||||
num = (nc + 1).to_s.rjust @num_width - 1, '👪👩💻👨🏿🚀🏳️🌈🔥🛠️🌍🥶✅❌💡⏰🚀'#ddasasdasda
|
||||
num.each_char.with_index do |c, j|
|
||||
C.update @y + row - @scroll, @x + j, c, color, 0x000000, 0
|
||||
end
|
||||
if (fold = @folds.find { |f| nc == f.start })
|
||||
icon = fold.active ? "" : "" # "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : "" "" : ""
|
||||
C.update @y + row - @scroll, @x, icon, 0xFF0000, 0x000000, 0
|
||||
end
|
||||
end
|
||||
if ch == "\n"
|
||||
row += 1
|
||||
nc += 1
|
||||
col = 0
|
||||
next
|
||||
end
|
||||
w = C.display_width(ch)
|
||||
if col + w > @width - @num_width
|
||||
row += 1
|
||||
col = 0
|
||||
end
|
||||
col += w
|
||||
next if row < @scroll
|
||||
break if row - @scroll >= @height
|
||||
fg = 0xFFFFFF
|
||||
bg = 0x000000
|
||||
fl = 0
|
||||
if (h = @highlights[i])
|
||||
fg = h.color_fg
|
||||
bg = h.color_bg
|
||||
fl = h.flags
|
||||
end
|
||||
C.update @y + row - @scroll, @x + @num_width + col - w, ch, fg, bg, fl
|
||||
end
|
||||
if @text[-1] == "\n" || @text.size == 0
|
||||
num = (nc + 1).to_s.rjust @num_width - 1, ' '
|
||||
color = 0x666666
|
||||
color = 0xFFFFFF if (cursor_row || row) == row
|
||||
num.each_char.with_index do |c, j|
|
||||
C.update @y + row - @scroll, @x + j, c, color, 0x000000, 0
|
||||
end
|
||||
end
|
||||
C.set_cursor @y - @scroll + (cursor_row || row), @x + @num_width + (cursor_col || col), 1
|
||||
end
|
||||
end
|
||||
Reference in New Issue
Block a user