kevm-pyk/utils: add byte_offset_to_lines

Author: ehildenb

kevm-pyk/src/kevm_pyk/utils.py

@@ -19,6 +19,26 @@ def write_cfg(_cfg: KCFG, _cfgpath: Path) -> None:
     _LOGGER.info(f'Updated CFG file: {_cfgpath}')
 
 
+def byte_offset_to_lines(lines: Iterable[str], byte_start: int, byte_width: int) -> Tuple[List[str], int, int]:
+    text_lines = []
+    line_start = 0
+    for l in lines:
+        if len(l) < byte_start:
+            byte_start -= len(l) + 1
+            line_start += 1
+        else:
+            break
+    line_end = line_start
+    for l in list(lines)[line_start:]:
+        if byte_start + byte_width < 0:
+            break
+        else:
+            text_lines.append(l)
+            byte_width -= len(l) + 1
+            line_end += 1
+    return (text_lines, line_start, line_end)
+
+
 def rpc_prove(
     kprove: KProve,
     cfgid: str,