module main

import os
import rand
import time
import veb
import veb.sse

struct App {
	veb.StaticHandler
}

struct Context {
	veb.Context
}

fn main() {
	mut app := &App{}
	app.serve_static('/favicon.ico', 'favicon.ico')!
	app.mount_static_folder_at(os.resource_abs_path('.'), '/')!
	veb.run[App, Context](mut app, 8081)
}

// XTODO template broken (@)
pub fn (mut app App) index() veb.Result {
	title := 'SSE Example'
	return $veb.html()
}

fn (mut app App) sse() veb.Result {
	ctx.takeover_conn()
	spawn handle_sse_conn(mut ctx)
	return veb.no_result()
}

fn handle_sse_conn(mut ctx Context) {
	mut session := sse.start_connection(mut ctx.Context)
	session.send_message(data: 'ok') or { return }
	for {
		data := '{"time": "${time.now().str()}", "random_id": "${rand.ulid()}"}'
		session.send_message(event: 'ping', data: data) or { break }
		println('> sent event: ${data}')
		time.sleep(1 * time.second)
	}
	session.close()
}