.{math, log, string, random, buffer, memory} := @use("../../stn/src/lib.hb");
.{Color, Surface, new_surface, put_surface, sync, put_rect, put_filled_rect, text, put_text} := @use("../../render/src/lib.hb");
.{Channel, Window, WindowProps, WindowData, MessageHeader, BUFFER_SERVER, BUFFER_CLIENT, message, permissions, recv_header, recv_message, send_message, send_header, await_message} := @use("./lib.hb")

WindowServer := struct {
	window_count: uint,
	channel: Channel,
	// ! replace this with a collection when we get an allocator
	windows: [?Window; 10],
	font: text.Font,
}

// ! in the future this should be safely handled
server := @as(WindowServer, idk)

psf := @embed("../../../assets/consolefonts/tamsyn/10x20r.psf")

start := fn(): void {
	font := text.font_from_psf2(@bitcast(&psf), false)
	if font == null {
		log.error("server: failed to load asset\0")
		return
	}
	server = .(0, .{client: buffer.create(BUFFER_CLIENT), server: buffer.create(BUFFER_SERVER)}, .(null, null, null, null, null, null, null, null, null, null), @as(text.Font, font))
	log.info("server: started server\0")
}

incoming := fn(): bool {
	msg := recv_header(server.channel.server)
	if msg == null {
		return true
	}
	if msg.kind == message.syn {
		log.info("server: recv syn\0")
		channel := Channel.(buffer.create_nameless(), buffer.create_nameless())
		send_message(Channel, message.ack, channel, server.channel.client)
		props := await_message(WindowProps, channel.server)
		if props.header.kind != message.props {
			return true
		}
		log.info("server: recv props\0")
		// ! do inspection of requested props here
		data := WindowData.(props.body, channel, permissions.default)
		send_message(WindowData, message.ack, data, channel.client)
		surface := new_window_decorations(data.props.dimensions)
		put_filled_rect(surface, .(0, 0), data.props.dimensions + .(DECO_WIDTH, DECO_HEIGHT_TOP + DECO_HEIGHT_BOTTOM), DECO_COLOUR)
		put_filled_rect(surface, .(0, 0), .(data.props.dimensions.x + DECO_WIDTH, DECO_HEIGHT_TOP), DECO_COLOUR)
		put_text(surface, server.font, .(2, 1), .(0, 0, 0, 255), data.props.title)
		server.windows[server.window_count] = .(data, surface)
		server.window_count += 1
	}
	return true
}

$DECO_WIDTH := 2
$DECO_HEIGHT_TOP := 20
$DECO_HEIGHT_BOTTOM := 1
$DECO_COLOUR := Color.(100, 200, 255, 255)

new_window_decorations := fn(dimensions: math.Vec2(uint)): Surface {
	return new_surface(dimensions.x + DECO_WIDTH, dimensions.y + DECO_HEIGHT_TOP + DECO_HEIGHT_BOTTOM)
}

// ! compositor code. this currently disallows tearing.

collect_frames := fn(): void {
	window := server.windows[0]
	if window == null {
		return
	}
	header := recv_header(window.data.channel.server)
	if header == null | @unwrap(header).kind != message.frame_ready {
		return
	}
	send_header(message.ack, window.data.channel.client)
	ptr := await_message(^Color, window.data.channel.server)
	// ! weird compiler error
	// if ptr.header.kind != message.ack {
	// return
	// }
	put_surface(window.surface, Surface.(ptr.body, window.data.props.dimensions.x, window.data.props.dimensions.y, window.data.props.dimensions.x * window.data.props.dimensions.y), .(DECO_WIDTH / 2, DECO_HEIGHT_TOP), false)
	send_header(message.ack, window.data.channel.client)
}

render_clients := fn(screen: Surface): void {
	window := server.windows[0]
	if window == null {
		return
	}
	put_surface(screen, window.surface, window.data.props.position, false)
	sync(screen)
}