package ring

import (
	"encoding/json"
	"errors"
	"fmt"
	"net/url"
	"strconv"

	"github.com/AlexxIT/go2rtc/pkg/core"
	"github.com/AlexxIT/go2rtc/pkg/webrtc"
	"github.com/google/uuid"
	pion "github.com/pion/webrtc/v4"
)

type Client struct {
	api       *RingApi
	wsClient  *WSClient
	prod      core.Producer
	cameraID  int
	dialogID  string
	connected core.Waiter
	closed    bool
}

func Dial(rawURL string) (*Client, error) {
	u, err := url.Parse(rawURL)
	if err != nil {
		return nil, err
	}

	query := u.Query()
	encodedToken := query.Get("refresh_token")
	cameraID := query.Get("camera_id")
	deviceID := query.Get("device_id")
	_, isSnapshot := query["snapshot"]

	if encodedToken == "" || deviceID == "" || cameraID == "" {
		return nil, errors.New("ring: wrong query")
	}

	client := &Client{
		dialogID: uuid.NewString(),
	}

	client.cameraID, err = strconv.Atoi(cameraID)
	if err != nil {
		return nil, fmt.Errorf("ring: invalid camera_id: %w", err)
	}

	refreshToken, err := url.QueryUnescape(encodedToken)
	if err != nil {
		return nil, fmt.Errorf("ring: invalid refresh token encoding: %w", err)
	}

	client.api, err = NewRestClient(RefreshTokenAuth{RefreshToken: refreshToken}, nil)
	if err != nil {
		return nil, err
	}

	// Snapshot Flow
	if isSnapshot {
		client.prod = NewSnapshotProducer(client.api, client.cameraID)
		return client, nil
	}

	client.wsClient, err = StartWebsocket(client.cameraID, client.api)
	if err != nil {
		client.Stop()
		return nil, err
	}

	// Create Peer Connection
	conf := pion.Configuration{
		ICEServers: []pion.ICEServer{
			{URLs: []string{
				"stun:stun.kinesisvideo.us-east-1.amazonaws.com:443",
				"stun:stun.kinesisvideo.us-east-2.amazonaws.com:443",
				"stun:stun.kinesisvideo.us-west-2.amazonaws.com:443",
				"stun:stun.l.google.com:19302",
				"stun:stun1.l.google.com:19302",
				"stun:stun2.l.google.com:19302",
				"stun:stun3.l.google.com:19302",
				"stun:stun4.l.google.com:19302",
			}},
		},
		ICETransportPolicy: pion.ICETransportPolicyAll,
		BundlePolicy:       pion.BundlePolicyBalanced,
	}

	api, err := webrtc.NewAPI()
	if err != nil {
		client.Stop()
		return nil, err
	}

	pc, err := api.NewPeerConnection(conf)
	if err != nil {
		client.Stop()
		return nil, err
	}

	// protect from sending ICE candidate before Offer
	var sendOffer core.Waiter

	// protect from blocking on errors
	defer sendOffer.Done(nil)

	prod := webrtc.NewConn(pc)
	prod.FormatName = "ring/webrtc"
	prod.Mode = core.ModeActiveProducer
	prod.Protocol = "ws"
	prod.URL = rawURL

	client.wsClient.onMessage = func(msg WSMessage) {
		client.onWSMessage(msg)
	}

	client.wsClient.onError = func(err error) {
		// fmt.Printf("ring: error: %s\n", err.Error())
		client.Stop()
		client.connected.Done(err)
	}

	client.wsClient.onClose = func() {
		// fmt.Println("ring: disconnect")
		client.Stop()
		client.connected.Done(errors.New("ring: disconnect"))
	}

	prod.Listen(func(msg any) {
		switch msg := msg.(type) {
		case *pion.ICECandidate:
			_ = sendOffer.Wait()

			iceCandidate := msg.ToJSON()

			// skip empty ICE candidates
			if iceCandidate.Candidate == "" {
				return
			}

			icePayload := map[string]any{
				"ice":        iceCandidate.Candidate,
				"mlineindex": iceCandidate.SDPMLineIndex,
			}

			if err = client.wsClient.sendSessionMessage("ice", icePayload); err != nil {
				client.connected.Done(err)
				return
			}

		case pion.PeerConnectionState:
			switch msg {
			case pion.PeerConnectionStateNew:
				break
			case pion.PeerConnectionStateConnecting:
				break
			case pion.PeerConnectionStateConnected:
				client.connected.Done(nil)
			default:
				client.Stop()
				client.connected.Done(errors.New("ring: " + msg.String()))
			}
		}
	})

	client.prod = prod

	// Setup media configuration
	medias := []*core.Media{
		{
			Kind:      core.KindAudio,
			Direction: core.DirectionSendRecv,
			Codecs: []*core.Codec{
				{
					Name:      "opus",
					ClockRate: 48000,
					Channels:  2,
				},
			},
		},
		{
			Kind:      core.KindVideo,
			Direction: core.DirectionRecvonly,
			Codecs: []*core.Codec{
				{
					Name:      "H264",
					ClockRate: 90000,
				},
			},
		},
	}

	// Create offer
	offer, err := prod.CreateOffer(medias)
	if err != nil {
		client.Stop()
		return nil, err
	}

	// Send offer
	offerPayload := map[string]any{
		"stream_options": map[string]bool{
			"audio_enabled": true,
			"video_enabled": true,
		},
		"sdp": offer,
	}

	if err = client.wsClient.sendSessionMessage("live_view", offerPayload); err != nil {
		client.Stop()
		return nil, err
	}

	sendOffer.Done(nil)

	if err = client.connected.Wait(); err != nil {
		return nil, err
	}

	return client, nil
}

func (c *Client) onWSMessage(msg WSMessage) {
	rawMsg, _ := json.Marshal(msg)

	// fmt.Printf("ring: onWSMessage: %s\n", string(rawMsg))

	// check if "doorbot_id" is present
	if _, ok := msg.Body["doorbot_id"]; !ok {
		return
	}

	// check if the message is from the correct doorbot
	doorbotID := msg.Body["doorbot_id"].(float64)
	if int(doorbotID) != c.cameraID {
		return
	}

	if msg.Method == "session_created" || msg.Method == "session_started" {
		if _, ok := msg.Body["session_id"]; ok && c.wsClient.sessionID == "" {
			c.wsClient.sessionID = msg.Body["session_id"].(string)
		}
	}

	// check if the message is from the correct session
	if _, ok := msg.Body["session_id"]; ok {
		if msg.Body["session_id"].(string) != c.wsClient.sessionID {
			return
		}
	}

	switch msg.Method {
	case "sdp":
		if prod, ok := c.prod.(*webrtc.Conn); ok {
			// Get answer
			var msg AnswerMessage
			if err := json.Unmarshal(rawMsg, &msg); err != nil {
				c.Stop()
				c.connected.Done(err)
				return
			}

			if err := prod.SetAnswer(msg.Body.SDP); err != nil {
				c.Stop()
				c.connected.Done(err)
				return
			}

			if err := c.wsClient.activateSession(); err != nil {
				c.Stop()
				c.connected.Done(err)
				return
			}

			prod.SDP = msg.Body.SDP
		}

	case "ice":
		if prod, ok := c.prod.(*webrtc.Conn); ok {
			var msg IceCandidateMessage
			if err := json.Unmarshal(rawMsg, &msg); err != nil {
				break
			}

			// Skip empty candidates
			if msg.Body.Ice == "" {
				break
			}

			if err := prod.AddCandidate(msg.Body.Ice); err != nil {
				c.Stop()
				c.connected.Done(err)
				return
			}
		}

	case "close":
		c.Stop()
		c.connected.Done(errors.New("ring: close"))

	case "pong":
		// Ignore
	}
}

func (c *Client) GetMedias() []*core.Media {
	return c.prod.GetMedias()
}

func (c *Client) GetTrack(media *core.Media, codec *core.Codec) (*core.Receiver, error) {
	return c.prod.GetTrack(media, codec)
}

func (c *Client) AddTrack(media *core.Media, codec *core.Codec, track *core.Receiver) error {
	if webrtcProd, ok := c.prod.(*webrtc.Conn); ok {
		if media.Kind == core.KindAudio {
			// Enable speaker
			speakerPayload := map[string]any{
				"stealth_mode": false,
			}
			_ = c.wsClient.sendSessionMessage("camera_options", speakerPayload)
		}
		return webrtcProd.AddTrack(media, codec, track)
	}

	return fmt.Errorf("add track not supported for snapshot")
}

func (c *Client) Start() error {
	return c.prod.Start()
}

func (c *Client) Stop() error {
	if c.closed {
		return nil
	}

	c.closed = true

	if c.prod != nil {
		_ = c.prod.Stop()
	}

	if c.wsClient != nil {
		_ = c.wsClient.Close()
	}

	return nil
}

func (c *Client) MarshalJSON() ([]byte, error) {
	return json.Marshal(c.prod)
}
