v/vlib/net/websocket/main.v

module main

import (
	ws
	eventbus
	time
	readline
	term
	benchmark
)

const (
	eb = eventbus.new()
)
#flag -I $PWD
#include "utf8.h"

fn C.utf8_validate_str() bool
fn main(){
	//println(sss)
	/* for sss in 0..10 {
		mut bm := benchmark.new_benchmark()
		for i in 0..10000 {
			for a, t in tests {
				ss := ws.utf8_validate(t.str, t.len)
				if !ss {
					panic("failed")
				}
				//println("${a}:${ss}")
			}
		}
		bm.stop()
		println( bm.total_message('remarks about the benchmark') )	
	} */
	mut websocket := ws.new("ws://localhost:9001/getCaseCount")
	//defer { }
	websocket.subscriber.subscribe("on_open", on_open)
	websocket.subscriber.subscribe("on_message", on_message)
	websocket.subscriber.subscribe("on_error", on_error)
	websocket.subscriber.subscribe("on_close", on_close)
	//go 
	websocket.connect()
	websocket.read()
	//time.usleep(2000000)
	//go websocket.listen()
	//term.erase_clear()
	/* text := read_line("[client]:")
	if text == "close" { 
		ws.close(1005, "done")
		time.usleep(1000000)
		exit(0)
	}
	ws.write(text, .text_frame) */
	/* time.usleep(1000000)
	ws.read() */
	//ws.close(1005, "done")
	/* 
	 */
	//websocket.close(1005, "done")
	//read_line("wait")
}

fn read_line(text string) string {
	mut r := readline.Readline{}
	mut output := r.read_line(text + " ") or { 
		panic(err)
	}
	output = output.replace("\n","")
	if output.len <= 0 {
		return ""
	}
	return output
}

fn on_open(params eventbus.Params){
	println("Websocket opened.")
}

fn on_message(params eventbus.Params){
	println("Message recieved. Sending it back.")
	typ := params.get_string("type")
	len := params.get_int("len")
	mut ws := params.get_caller(ws.Client{})
	if typ == "string" {
		message := params.get_string("payload")
		if ws.uri.ends_with("getCaseCount") {
			num := message.int()
			ws.close(1005, "done")
			start_tests(mut ws, num)
			return
		}
		//println("Message: " + message)
		ws.write(message.str, len, .text_frame)
	} else {
		println("Binary message.")
		message := params.get_raw("payload")
		ws.write(message, len, .binary_frame)
	}
}

fn start_tests(websocket mut ws.Client, num int) {
	for i := 1; i < num; i++ {
		println("Running test: " + i.str())
		websocket.uri = "ws://localhost:9001/runCase?case=${i.str()}&agent=vws/1.0a"
		if websocket.connect() >= 0 {
			websocket.listen()
		}
	}
	println("Done!")
	websocket.uri = "ws://localhost:9001/updateReports?agent=vws/1.0a"
	if websocket.connect() >= 0 {
		websocket.read()
		websocket.close(1000, "disconnecting...")
	}
	exit(0)
}

fn on_close(params eventbus.Params){
	println("Websocket closed.")
}

fn on_error(params eventbus.Params){
	println("we have an error.")
}