package v8go

import (
	"context"
	"fmt"

	"golang.org/x/net/websocket"
)

type ClientConnection struct {
	Connection *websocket.Conn
	ctx        context.Context
	cancel     context.CancelFunc
	ProcessFun func(connection *ClientConnection, data []byte)
}

func NewClientConnection(parentCtx context.Context, conn *websocket.Conn, procss func(connection *ClientConnection, data []byte)) (*ClientConnection, context.Context) {
	ctx, cancel := context.WithCancel(parentCtx)
	c := &ClientConnection{
		Connection: conn,
		ctx:        ctx,
		cancel:     cancel,
		ProcessFun: procss,
	}
	go c.Run()
	return c, ctx
}

func (c *ClientConnection) Stop() {
	c.cancel()
}

func (c *ClientConnection) Run() {
	defer c.cancel()
	defer func() {
		fmt.Println("connection closed")
		_ = c.Connection.Close()
	}()
	go c.readLoop()
	select {
	case <-c.ctx.Done():
		return
	}

}

func (c *ClientConnection) readLoop() {
	defer c.cancel()

	for {
		select {
		case <-c.ctx.Done():
			return
		default:
			data2 := make([]byte, 4096)
			err := websocket.Message.Receive(c.Connection, &data2)
			if err != nil {
				return
			}
			c.ProcessFun(c, data2)
		}
	}
}