#!/usr/bin/env python3
# ollie:parallel read
# ollie:tier cold
# description: Read a file with accurate line numbers. Prefer segments (start/end) over full-file reads.
# Args: <file_path> [start] [end] [--head=<n>]

import sys
import os
from itertools import islice

MAX_LINES = 500       # default window for whole-file reads (no start/end)
MAX_SEGMENT = 2000    # max window when start/end are given
MAX_LINE_LEN = 2000

# Parse --head=N from args
head = None
args = []
for a in sys.argv[1:]:
    if a.startswith('--head='):
        head = int(a.split('=', 1)[1])
    else:
        args.append(a)

if not args:
    print("usage: file_read <file_path> [start] [end] [--head=N]")
    sys.exit(1)

file_path = args[0]
start = int(args[1]) if len(args) > 1 else 1
end   = int(args[2]) if len(args) > 2 else None

if not os.path.isabs(file_path):
    print(f"STATUS=error")
    print(f"error: path must be absolute, got: {file_path}")
    sys.exit(1)

if not os.path.exists(file_path):
    print("STATUS=not_found")
    print("(new file)")
    sys.exit(0)

if not os.path.isfile(file_path):
    print(f"STATUS=error")
    print(f"error: not a regular file: {file_path}")
    sys.exit(1)

with open(file_path, 'rb') as f:
    raw_head = f.read(8192)
    if b'\x00' in raw_head:
        # Binary file -- try to detect image type
        IMAGE_SIGS = [
            b'\x89PNG\r\n\x1a\n', b'\xff\xd8\xff',
            b'GIF87a', b'GIF89a', b'RIFF',
        ]
        is_image = any(raw_head.startswith(s) for s in IMAGE_SIGS)
        if is_image:
            print("STATUS=ok BINARY=true IMAGE=true")
            print("(image file — use image_read for visual content)")
        else:
            print("STATUS=ok BINARY=true")
            print("(binary file)")
        sys.exit(0)

# Count total lines
with open(file_path, 'r', errors='replace') as f:
    total_lines = sum(1 for _ in f)

start = max(1, start)
if head:
    end = start + head - 1
elif end is None:
    end = start + MAX_LINES - 1
else:
    end = min(end, start + MAX_SEGMENT - 1)

with open(file_path, 'r', errors='replace') as f:
    for _ in islice(f, start - 1):
        pass
    segment_lines = list(islice(f, end - start + 1))

if not segment_lines:
    if start == 1:
        print(f"STATUS=ok LINES=0 TOTAL_LINES={total_lines} TRUNCATED=false EMPTY=true")
        print("(empty file)")
    else:
        print(f"STATUS=ok LINES=0 TOTAL_LINES={total_lines} TRUNCATED=false EMPTY=true")
        print("(no lines in range)")
    sys.exit(0)

out = []
for i, line in enumerate(segment_lines):
    lineno = start + i
    text = line.rstrip('\n')
    if len(text) > MAX_LINE_LEN:
        text = text[:MAX_LINE_LEN] + '...'
    out.append(f"{lineno:>6}\t{text}")

last_line_shown = start + len(out) - 1
truncated = last_line_shown < total_lines
print(f"STATUS=ok LINES={len(out)} TOTAL_LINES={total_lines} TRUNCATED={str(truncated).lower()}")
print('\n'.join(out))
