server.v4.runback.go

package socketio

import (
	"errors"
	"fmt"

	siop "github.com/njones/socketio/protocol"
	siot "github.com/njones/socketio/transport"
)

func doConnectPacketV4(v4 *ServerV4) func(SocketID, siot.Socket, *Request) error {
	return func(socketID SocketID, socket siot.Socket, req *Request) (err error) {
		unlock := v4.prev.prev.prev.r()
		tr := v4.tr()
		unlock()

		tr.Join(socket.Namespace, socketID, socketID.Room(socketIDPrefix))

		stopBuffer := tr.(rawTransport).Transport(socketID).StartBuffer()
		defer stopBuffer()

		v4.setPrefix()
		v4.setSocketID(socketID)
		v4.setNsp(socket.Namespace)

		var h = handshakeV4{}
		switch val := socket.Data.(type) {
		case map[string]interface{}:
			h.Auth = func() map[string]interface{} {
				return val
			}
		}

		if fn, ok := v4.onConnect[socket.Namespace]; ok {
			return fn(&SocketV4{inSocketV4: v4.inSocketV4.clone(), req: req, han: h})
		}

		return ErrNamespaceNotFound.F(socket.Namespace)
	}
}

func runV4(v4 *ServerV4) func(SocketID, *Request) error {
	return func(socketID SocketID, req *Request) error {
		unlock := v4.prev.prev.prev.r()
		tr := v4.tr()
		unlock()

		for socket := range tr.Receive(socketID) {
			if err := doV4(v4, socketID, socket, req); err != nil {
				return err
			}
		}
		return nil
	}
}

func doV4(v4 *ServerV4, socketID SocketID, socket siot.Socket, req *Request) error {
	v1 := v4.prev.prev.prev

	switch socket.Type {
	case siop.ConnectPacket.Byte():
		unlock := v1.r()
		tr := v4.tr()
		unlock()

		if err := v1.doConnectPacket(socketID, socket, req); err != nil {
			if errors.Is(err, ErrNamespaceNotFound) {
				tr.Send(socketID, serviceError(fmt.Errorf("%valid namespace", "Inv")), siop.WithNamespace(socket.Namespace), siop.WithType(byte(siop.ConnectErrorPacket)))
				return nil
			}
			tr.Send(socketID, serviceError(err), siop.WithType(byte(siop.ConnectErrorPacket)))
			return nil
		}

		connectResponse := map[string]interface{}{"sid": socketID.String()}
		tr.Send(socketID, connectResponse, siop.WithType(siop.ConnectPacket.Byte()), siop.WithNamespace(socket.Namespace))
		tr.(rawTransport).Transport(socketID).SendBuffer()
		return nil
	}
	return doV3(v4.prev, socketID, socket, req)
}