package main

import (
	"meow.tf/streamdeck/obs-replay/obsws"
)

type OBSWSClient struct {
	client *obsws.Client
	key    string
}

func NewOBSWSClient(key, host string, port int, password string) *OBSWSClient {
	obsc := &obsws.Client{Host: host, Port: port, Password: password}

	return &OBSWSClient{client: obsc, key: key}
}

func (c *OBSWSClient) Connect() error {
	err := c.client.Connect()

	if err != nil {
		return err
	}

	c.client.AddEventHandler("StreamStatus", streamStatusUpdate(c.key))
	c.client.AddEventHandler("ReplayStarted", func(obsws.Event) { loopContextState(c.key, 1) })
	c.client.AddEventHandler("ReplayStopped", func(obsws.Event) { loopContextState(c.key, 0) })

	return nil
}

func (c *OBSWSClient) Disconnect() error {
	return c.client.Disconnect()
}

func (c *OBSWSClient) Connected() bool {
	return c.client.Connected()
}

func (c *OBSWSClient) ToggleReplay() error {
	req := obsws.NewStartStopReplayBufferRequest()

	return req.Send(c.client)
}

func (c *OBSWSClient) SaveReplay() error {
	req := obsws.NewSaveReplayBufferRequest()

	return req.Send(c.client)
}

func streamStatusUpdate(key string) func(obsws.Event) {
	return func(e obsws.Event) {
		evt := e.(obsws.StreamStatusEvent)

		state := 0

		if evt.Replay {
			state = 1
		}

		loopContextState(key, state)
	}
}