compiler: color only the relevant part of the source code in error messages

vlib/v/util/errors.v

@@ -105,44 +105,36 @@ pub fn source_context(kind, source string, column int, pos token.Position) []str
 	tab_spaces := '    '
 	for iline := bline; iline <= aline; iline++ {
 		sline := source_lines[iline]
-		mut cline := sline.replace('\t', tab_spaces)
+		start_column := imin(column, sline.len)
-		if iline == pos.line_nr {
+		end_column := imin(column + pos.len, sline.len)
-			cline = color(kind, cline)
+		cline := if iline == pos.line_nr {
+			sline[..start_column] + color(kind, sline[start_column..end_column]) + sline[end_column..]
+		} else {
+			sline
 		}
-		clines << '${iline+1:5d} | ' + cline
+		clines << '${iline+1:5d} | ' + cline.replace('\t', tab_spaces)
 		//
 		if iline == pos.line_nr {
 			// The pointerline should have the same spaces/tabs as the offending
 			// line, so that it prints the ^ character exactly on the *same spot*
 			// where it is needed. That is the reason we can not just
 			// use strings.repeat(` `, col) to form it.
-			mut pointerline := []string{}
+			mut pointerline := ''
-			for i, bchar in sline {
+			for bchar in sline[..start_column] {
-				if i < column {
+				x := if bchar.is_space() {
-					mut x := bchar
+					bchar
-					if x == `\t` {
-						pointerline << tab_spaces
-					} else {
-						x = if x.is_space() {
-							bchar
-						} else {
-							` `
-						}
-						pointerline << x.str()
-					}
-					continue
-				}
-				if pos.len > 1 {
-					max_len := sline.len - pointerline.len // rest of the line
-					len := imin(max_len, pos.len)
-					underline := '~'.repeat(len)
-					pointerline << bold(color(kind, underline))
 				} else {
-					pointerline << bold(color(kind, '^'))
+					` `
 				}
-				break
+				pointerline += x.str()
 			}
-			clines << '      | ' + pointerline.join('')
+			underline := if pos.len > 1 {
+				'~'.repeat(end_column - start_column)
+			} else {
+				'^'
+			}
+			pointerline += bold(color(kind, underline))
+			clines << '      | ' + pointerline.replace('\t', tab_spaces)
 		}
 	}
 	return clines