From 6563535a3d1ecaa07618de7620bdeb972daa1262 Mon Sep 17 00:00:00 2001
From: spaceface777 <spaceface7777@gmail.com>
Date: Thu, 26 Nov 2020 23:52:20 +0100
Subject: [PATCH] term.ui: simplify/fix the truecolor feature test (#6963)

---
 vlib/term/ui/termios_nix.c.v | 100 ++++++++---------------------------
 1 file changed, 22 insertions(+), 78 deletions(-)

diff --git a/vlib/term/ui/termios_nix.c.v b/vlib/term/ui/termios_nix.c.v
index 699c5f673d..cc12f21131 100644
--- a/vlib/term/ui/termios_nix.c.v
+++ b/vlib/term/ui/termios_nix.c.v
@@ -164,88 +164,30 @@ fn (mut ctx Context) termios_setup() ? {
 
 fn get_cursor_position() (int, int) {
 	print('\033[6n')
-	// ESC [ YYY `;` XXX `R`
-	mut reading_x, mut reading_y := false, false
-	mut x, mut y := 0, 0
-	for i := 0; i < 15 ; i++ {
-		ch := int(C.getchar())
-		b := byte(ch)
-		i++
-		// state management:
-		if b == `R` {
-			break
-		} else if b == `[` {
-			reading_y = true
-			reading_x = false
-			continue
-		} else if b == `;` {
-			reading_y = false
-			reading_x = true
-			continue
-		}
-		// converting string vals to ints:
-		if reading_x {
-			x = x * 10 + b - byte(`0`)
-		} else if reading_y {
-			y = y * 10 + b - byte(`0`)
-		}
-	}
-	return x, y
+	buf := malloc(25)
+	len := C.read(C.STDIN_FILENO, buf, 24)
+	unsafe { buf[len] = 0 }
+	s := tos(buf, len)
+	a := s[2..].split(';')
+	if a.len != 2 { return -1, -1 }
+	return a[0].int(), a[1].int()
 }
 
 fn supports_truecolor() bool {
+	// faster/simpler, but less reliable, check
+	if os.getenv('COLORTERM') in ['truecolor', '24bit'] {
+		return true
+	}
 	// set the bg color to some arbirtrary value (#010203), assumed not to be the default
 	print('\x1b[48:2:1:2:3m')
-
-	sx, sy := get_cursor_position()
-	// sequence to query the current cursor position
+	// andquery the current color
 	print('\x1bP\$qm\x1b\\')
-	color := get_current_bg_color()
-	ex, ey := get_cursor_position()
-	// if the terminal doesn't understand the "get current color",
-	// assume it doesn't support truecolor either
-	if !(sx == ex && sy == ey) {
-		println('>>> different pos: ($sx, $sy) -> ($ex, $ey)')
-		return false
-	}
-	// TODO: iTerm emits a different sequence, but it's compatible anyways, so
-	if color !in [0x010203, 0x01010203] {
-		println('>>> no match: $color')
-		return false
-	}
-	return true
-}
+	buf := malloc(25)
+	len := C.read(C.STDIN_FILENO, buf, 24)
+	unsafe { buf[len] = 0 }
+	s := tos(buf, len)
+	return '1:2:3' in s
 
-fn get_current_bg_color() int {
-	mut res := 0
-	mut colon_cnt := 0
-	mut cur_val := 0
-
-	for i := 0; i < 50 ; i++ {
-		ch := int(C.getchar())
-		b := byte(ch)
-
-		if b in [0, 255] {
-			return -1
-		} else if b == `m` {
-			if colon_cnt > 1 {
-				res = (res << 8) | cur_val
-				cur_val = 0
-			}
-			break
-		} else if b in [`:`, `;`] {
-			if colon_cnt > 1 {
-				res = (res << 8) | cur_val
-				cur_val = 0
-			}
-			colon_cnt++
-		} else if b.is_digit() {
-			if colon_cnt > 1 {
-				cur_val = cur_val * 10 + b - byte(`0`)
-			}
-		}
-	}
-	return res
 }
 
 fn termios_reset() {
@@ -278,9 +220,11 @@ fn (mut ctx Context) termios_loop() {
 		if !ctx.paused {
 			sw.restart()
 			if ctx.cfg.event_fn != voidptr(0) {
-				len := C.read(C.STDIN_FILENO, ctx.read_buf.data, ctx.read_buf.cap - ctx.read_buf.len)
-				if len > 0 {
-					ctx.resize_arr(len)
+				unsafe {
+					len := C.read(C.STDIN_FILENO, ctx.read_buf.data + ctx.read_buf.len, ctx.read_buf.cap - ctx.read_buf.len)
+					ctx.resize_arr(ctx.read_buf.len + len)
+				}
+				if ctx.read_buf.len > 0 {
 					ctx.parse_events()
 				}
 			}