import machine, time, ili9341, network, socket, sys, select

# --- 1. HARDWARE ---
spi = machine.SPI(0, baudrate=40000000, sck=machine.Pin(18), mosi=machine.Pin(19))
# Standard Wokwi Landscape. If upside down, change r=3 to r=1.
display = ili9341.ILI9341(spi, cs=machine.Pin(17), dc=machine.Pin(16), rst=machine.Pin(15), w=320, h=240, r=3)

# --- 2. THE MANUAL CURSOR ENGINE ---
class Cursor:
    x = 0
    y = 0

def super_print(text):
    for char in text:
        if char == '\n':
            Cursor.x = 0
            Cursor.y += 12 
        elif char == '\r':
            Cursor.x = 0
        else:
            display.set_pos(Cursor.x, Cursor.y)
            display.print(char)
            Cursor.x += 8 
            
        if Cursor.x > 310:
            Cursor.x = 0
            Cursor.y += 12
            
        if Cursor.y > 220:
            display.fill_rectangle(0, 0, 320, 240, 0)
            Cursor.x = 0
            Cursor.y = 0

# --- 3. THE CHICKADEE ENGINE ---
def connect_wifi():
    display.fill_rectangle(0, 0, 320, 240, 0)
    display.set_color(0x07E0, 0x0000)
    super_print("CHICKADEE-OS v3.1\n")
    super_print("RECOVERING FROM UNICODE ERROR...\n")
    
    wlan = network.WLAN(network.STA_IF)
    wlan.active(True)
    wlan.connect("Wokwi-GUEST", "")
    while not wlan.isconnected():
        time.sleep(0.5)
        super_print(".")
    super_print("\nONLINE\n")
    return True

def strike_telehack():
    try:
        super_print("STRIKING TELEHACK...\n")
        addr = socket.getaddrinfo("tty.sdf.org", 23)[0][-1]
        s = socket.socket()
        s.settimeout(5)
        s.connect(addr)
        s.send(b"\r\n")
        s.setblocking(False)
        return s
    except:
        return None

# --- MAIN LOOP ---
if connect_wifi():
    connection = strike_telehack()
    if connection:
        poll = select.poll()
        poll.register(sys.stdin, select.POLLIN)
        while True:
            try:
                data = connection.recv(1024)
                if data:
                    # THE FIX: 'ignore' ensures weird bytes don't crash the script
                    clean_text = data.decode('ascii', 'ignore')
                    super_print(clean_text)
            except OSError: 
                pass
            except Exception as e:
                # Catch-all to keep the terminal from dying
                pass

            if poll.poll(0):
                line = sys.stdin.readline()
                if line:
                    super_print(f"\n> {line}")
                    connection.send(line.replace('\n', '\r\n').encode())
            time.sleep(0.01)