package socketio

import (
	"net"
	"net/http"

	"forge.cadoles.com/Pyxis/golang-socketio/transport"
	"github.com/gorilla/websocket"
)

type Transport struct {
	dial DialFunc
	ws   *transport.WebsocketTransport
}

// Connect implements transport.Transport.
func (t *Transport) Connect(url string) (conn transport.Connection, err error) {
	if t.dial == nil {
		return t.ws.Connect(url)
	} else {
		dialer := websocket.Dialer{
			NetDial: func(network, addr string) (net.Conn, error) {
				return t.dial(network, addr)
			},
		}

		socket, _, err := dialer.Dial(url, t.ws.RequestHeader)
		if err != nil {
			return nil, err
		}

		return transport.NewWebsocketConnection(socket, t.ws), nil
	}
}

// HandleConnection implements transport.Transport.
func (t *Transport) HandleConnection(w http.ResponseWriter, r *http.Request) (conn transport.Connection, err error) {
	return t.ws.HandleConnection(w, r)
}

// Serve implements transport.Transport.
func (t *Transport) Serve(w http.ResponseWriter, r *http.Request) {
	t.ws.Serve(w, r)
}

var _ transport.Transport = &Transport{}