initial commit - from neko open source repository..

internal/websocket/admin.go

@@ -0,0 +1,298 @@
+package websocket
+
+import (
+	"strings"
+
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+)
+
+func (h *MessageHandler) adminLock(id string, session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	if h.locked {
+		h.logger.Debug().Msg("server already locked...")
+		return nil
+	}
+
+	h.locked = true
+
+	if err := h.sessions.Broadcast(
+		message.Admin{
+			Event: event.ADMIN_LOCK,
+			ID:    id,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_LOCK)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) adminUnlock(id string, session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	if !h.locked {
+		h.logger.Debug().Msg("server not locked...")
+		return nil
+	}
+
+	h.locked = false
+
+	if err := h.sessions.Broadcast(
+		message.Admin{
+			Event: event.ADMIN_UNLOCK,
+			ID:    id,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_UNLOCK)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) adminControl(id string, session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	host, ok := h.sessions.GetHost()
+
+	h.sessions.SetHost(id)
+
+	if ok {
+		if err := h.sessions.Broadcast(
+			message.AdminTarget{
+				Event:  event.ADMIN_CONTROL,
+				ID:     id,
+				Target: host.ID(),
+			}, nil); err != nil {
+			h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_CONTROL)
+			return err
+		}
+	} else {
+		if err := h.sessions.Broadcast(
+			message.Admin{
+				Event: event.ADMIN_CONTROL,
+				ID:    id,
+			}, nil); err != nil {
+			h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_CONTROL)
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) adminRelease(id string, session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	host, ok := h.sessions.GetHost()
+
+	h.sessions.ClearHost()
+
+	if ok {
+		if err := h.sessions.Broadcast(
+			message.AdminTarget{
+				Event:  event.ADMIN_RELEASE,
+				ID:     id,
+				Target: host.ID(),
+			}, nil); err != nil {
+			h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_RELEASE)
+			return err
+		}
+	} else {
+		if err := h.sessions.Broadcast(
+			message.Admin{
+				Event: event.ADMIN_RELEASE,
+				ID:    id,
+			}, nil); err != nil {
+			h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_RELEASE)
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) adminGive(id string, session types.Session, payload *message.Admin) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	if !h.sessions.Has(payload.ID) {
+		h.logger.Debug().Str("id", payload.ID).Msg("user does not exist")
+		return nil
+	}
+
+	// set host
+	h.sessions.SetHost(payload.ID)
+
+	// let everyone know
+	if err := h.sessions.Broadcast(
+		message.AdminTarget{
+			Event:  event.CONTROL_GIVE,
+			ID:     id,
+			Target: payload.ID,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.CONTROL_LOCKED)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) adminMute(id string, session types.Session, payload *message.Admin) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	target, ok := h.sessions.Get(payload.ID)
+	if !ok {
+		h.logger.Debug().Str("id", payload.ID).Msg("can't find session id")
+		return nil
+	}
+
+	if target.Admin() {
+		h.logger.Debug().Msg("target is an admin, baling")
+		return nil
+	}
+
+	target.SetMuted(true)
+
+	if err := h.sessions.Broadcast(
+		message.AdminTarget{
+			Event:  event.ADMIN_MUTE,
+			Target: target.ID(),
+			ID:     id,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_UNMUTE)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) adminUnmute(id string, session types.Session, payload *message.Admin) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	target, ok := h.sessions.Get(payload.ID)
+	if !ok {
+		h.logger.Debug().Str("id", payload.ID).Msg("can't find target session")
+		return nil
+	}
+
+	target.SetMuted(false)
+
+	if err := h.sessions.Broadcast(
+		message.AdminTarget{
+			Event:  event.ADMIN_UNMUTE,
+			Target: target.ID(),
+			ID:     id,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_UNMUTE)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) adminKick(id string, session types.Session, payload *message.Admin) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	target, ok := h.sessions.Get(payload.ID)
+	if !ok {
+		h.logger.Debug().Str("id", payload.ID).Msg("can't find session id")
+		return nil
+	}
+
+	if target.Admin() {
+		h.logger.Debug().Msg("target is an admin, baling")
+		return nil
+	}
+
+	if err := target.Kick("kicked"); err != nil {
+		return err
+	}
+
+	if err := h.sessions.Broadcast(
+		message.AdminTarget{
+			Event:  event.ADMIN_KICK,
+			Target: target.ID(),
+			ID:     id,
+		}, []string{payload.ID}); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_KICK)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) adminBan(id string, session types.Session, payload *message.Admin) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	target, ok := h.sessions.Get(payload.ID)
+	if !ok {
+		h.logger.Debug().Str("id", payload.ID).Msg("can't find session id")
+		return nil
+	}
+
+	if target.Admin() {
+		h.logger.Debug().Msg("target is an admin, baling")
+		return nil
+	}
+
+	remote := target.Address()
+	if remote == "" {
+		h.logger.Debug().Msg("no remote address, baling")
+		return nil
+	}
+
+	address := strings.SplitN(remote, ":", -1)
+	if len(address[0]) < 1 {
+		h.logger.Debug().Str("address", remote).Msg("no remote address, baling")
+		return nil
+	}
+
+	h.logger.Debug().Str("address", remote).Msg("adding address to banned")
+
+	h.banned[address[0]] = true
+
+	if err := target.Kick("banned"); err != nil {
+		return err
+	}
+
+	if err := h.sessions.Broadcast(
+		message.AdminTarget{
+			Event:  event.ADMIN_BAN,
+			Target: target.ID(),
+			ID:     id,
+		}, []string{payload.ID}); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.ADMIN_BAN)
+		return err
+	}
+
+	return nil
+}

internal/websocket/broadcast.go

@@ -0,0 +1,56 @@
+package websocket
+
+import (
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+)
+
+func (h *MessageHandler) boradcastCreate(session types.Session, payload *message.BroadcastCreate) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	h.broadcast.Create(payload.URL)
+
+	if err := h.boradcastStatus(session); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) boradcastDestroy(session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	h.broadcast.Destroy()
+
+	if err := h.boradcastStatus(session); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) boradcastStatus(session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	if err := session.Send(
+		message.BroadcastStatus{
+			Event:    event.BORADCAST_STATUS,
+			IsActive: h.broadcast.IsActive(),
+			URL:      h.broadcast.GetUrl(),
+		}); err != nil {
+		h.logger.Warn().Err(err).Msgf("sending event %s has failed", event.BORADCAST_STATUS)
+		return err
+	}
+
+	return nil
+}

internal/websocket/chat.go

@@ -0,0 +1,41 @@
+package websocket
+
+import (
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+)
+
+func (h *MessageHandler) chat(id string, session types.Session, payload *message.ChatReceive) error {
+	if session.Muted() {
+		return nil
+	}
+
+	if err := h.sessions.Broadcast(
+		message.ChatSend{
+			Event:   event.CHAT_MESSAGE,
+			Content: payload.Content,
+			ID:      id,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.CONTROL_RELEASE)
+		return err
+	}
+	return nil
+}
+
+func (h *MessageHandler) chatEmote(id string, session types.Session, payload *message.EmoteReceive) error {
+	if session.Muted() {
+		return nil
+	}
+
+	if err := h.sessions.Broadcast(
+		message.EmoteSend{
+			Event: event.CHAT_EMOTE,
+			Emote: payload.Emote,
+			ID:    id,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.CONTROL_RELEASE)
+		return err
+	}
+	return nil
+}

internal/websocket/control.go

@@ -0,0 +1,163 @@
+package websocket
+
+import (
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+)
+
+func (h *MessageHandler) controlRelease(id string, session types.Session) error {
+
+	// check if session is host
+	if !h.sessions.IsHost(id) {
+		h.logger.Debug().Str("id", id).Msg("is not the host")
+		return nil
+	}
+
+	// release host
+	h.logger.Debug().Str("id", id).Msgf("host called %s", event.CONTROL_RELEASE)
+	h.sessions.ClearHost()
+
+	// tell everyone
+	if err := h.sessions.Broadcast(
+		message.Control{
+			Event: event.CONTROL_RELEASE,
+			ID:    id,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.CONTROL_RELEASE)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) controlRequest(id string, session types.Session) error {
+	// check for host
+	if !h.sessions.HasHost() {
+		// set host
+		h.sessions.SetHost(id)
+
+		// let everyone know
+		if err := h.sessions.Broadcast(
+			message.Control{
+				Event: event.CONTROL_LOCKED,
+				ID:    id,
+			}, nil); err != nil {
+			h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.CONTROL_LOCKED)
+			return err
+		}
+
+		return nil
+	}
+
+	// get host
+	host, ok := h.sessions.GetHost()
+	if ok {
+
+		// tell session there is a host
+		if err := session.Send(message.Control{
+			Event: event.CONTROL_REQUEST,
+			ID:    host.ID(),
+		}); err != nil {
+			h.logger.Warn().Err(err).Str("id", id).Msgf("sending event %s has failed", event.CONTROL_REQUEST)
+			return err
+		}
+
+		// tell host session wants to be host
+		if err := host.Send(message.Control{
+			Event: event.CONTROL_REQUESTING,
+			ID:    id,
+		}); err != nil {
+			h.logger.Warn().Err(err).Str("id", host.ID()).Msgf("sending event %s has failed", event.CONTROL_REQUESTING)
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) controlGive(id string, session types.Session, payload *message.Control) error {
+	// check if session is host
+	if !h.sessions.IsHost(id) {
+		h.logger.Debug().Str("id", id).Msg("is not the host")
+		return nil
+	}
+
+	if !h.sessions.Has(payload.ID) {
+		h.logger.Debug().Str("id", payload.ID).Msg("user does not exist")
+		return nil
+	}
+
+	// set host
+	h.sessions.SetHost(payload.ID)
+
+	// let everyone know
+	if err := h.sessions.Broadcast(
+		message.ControlTarget{
+			Event:  event.CONTROL_GIVE,
+			ID:     id,
+			Target: payload.ID,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.CONTROL_LOCKED)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) controlClipboard(id string, session types.Session, payload *message.Clipboard) error {
+	// check if session is host
+	if !h.sessions.IsHost(id) {
+		h.logger.Debug().Str("id", id).Msg("is not the host")
+		return nil
+	}
+
+	h.remote.WriteClipboard(payload.Text)
+	return nil
+}
+
+func (h *MessageHandler) controlKeyboard(id string, session types.Session, payload *message.Keyboard) error {
+	// check if session is host
+	if !h.sessions.IsHost(id) {
+		h.logger.Debug().Str("id", id).Msg("is not the host")
+		return nil
+	}
+
+	// change layout
+	if payload.Layout != nil {
+		h.remote.SetKeyboardLayout(*payload.Layout)
+	}
+
+	// set num lock
+	var NumLock = 0
+	if payload.NumLock == nil {
+		NumLock = -1
+	} else if *payload.NumLock {
+		NumLock = 1
+	}
+
+	// set caps lock
+	var CapsLock = 0
+	if payload.CapsLock == nil {
+		CapsLock = -1
+	} else if *payload.CapsLock {
+		CapsLock = 1
+	}
+
+	// set scroll lock
+	var ScrollLock = 0
+	if payload.ScrollLock == nil {
+		ScrollLock = -1
+	} else if *payload.ScrollLock {
+		ScrollLock = 1
+	}
+
+	h.logger.Debug().
+		Int("NumLock", NumLock).
+		Int("CapsLock", CapsLock).
+		Int("ScrollLock", ScrollLock).
+		Msg("setting keyboard modifiers")
+
+	h.remote.SetKeyboardModifiers(NumLock, CapsLock, ScrollLock)
+	return nil
+}

internal/websocket/handler.go

@@ -0,0 +1,179 @@
+package websocket
+
+import (
+	"encoding/json"
+
+	"github.com/pkg/errors"
+	"github.com/rs/zerolog"
+
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+	"n.eko.moe/neko/internal/utils"
+)
+
+type MessageHandler struct {
+	logger    zerolog.Logger
+	sessions  types.SessionManager
+	webrtc    types.WebRTCManager
+	remote    types.RemoteManager
+	broadcast types.BroadcastManager
+	banned    map[string]bool
+	locked    bool
+}
+
+func (h *MessageHandler) Connected(id string, socket *WebSocket) (bool, string, error) {
+	address := socket.Address()
+	if address == "" {
+		h.logger.Debug().Msg("no remote address")
+	} else {
+		ok, banned := h.banned[address]
+		if ok && banned {
+			h.logger.Debug().Str("address", address).Msg("banned")
+			return false, "banned", nil
+		}
+	}
+
+	if h.locked {
+		session, ok := h.sessions.Get(id)
+		if !ok || !session.Admin() {
+			h.logger.Debug().Msg("server locked")
+			return false, "locked", nil
+		}
+	}
+
+	return true, "", nil
+}
+
+func (h *MessageHandler) Disconnected(id string) error {
+	if h.locked && len(h.sessions.Admins()) == 0 {
+		h.locked = false
+	}
+
+	return h.sessions.Destroy(id)
+}
+
+func (h *MessageHandler) Message(id string, raw []byte) error {
+	header := message.Message{}
+	if err := json.Unmarshal(raw, &header); err != nil {
+		return err
+	}
+
+	session, ok := h.sessions.Get(id)
+	if !ok {
+		errors.Errorf("unknown session id %s", id)
+	}
+
+	switch header.Event {
+	// Signal Events
+	case event.SIGNAL_ANSWER:
+		payload := &message.SignalAnswer{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.signalAnswer(id, session, payload)
+			}), "%s failed", header.Event)
+
+	// Control Events
+	case event.CONTROL_RELEASE:
+		return errors.Wrapf(h.controlRelease(id, session), "%s failed", header.Event)
+	case event.CONTROL_REQUEST:
+		return errors.Wrapf(h.controlRequest(id, session), "%s failed", header.Event)
+	case event.CONTROL_GIVE:
+		payload := &message.Control{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.controlGive(id, session, payload)
+			}), "%s failed", header.Event)
+	case event.CONTROL_CLIPBOARD:
+		payload := &message.Clipboard{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.controlClipboard(id, session, payload)
+			}), "%s failed", header.Event)
+	case event.CONTROL_KEYBOARD:
+		payload := &message.Keyboard{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.controlKeyboard(id, session, payload)
+			}), "%s failed", header.Event)
+
+
+	// Chat Events
+	case event.CHAT_MESSAGE:
+		payload := &message.ChatReceive{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.chat(id, session, payload)
+			}), "%s failed", header.Event)
+	case event.CHAT_EMOTE:
+		payload := &message.EmoteReceive{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.chatEmote(id, session, payload)
+			}), "%s failed", header.Event)
+
+	// Screen Events
+	case event.SCREEN_RESOLUTION:
+		return errors.Wrapf(h.screenResolution(id, session), "%s failed", header.Event)
+	case event.SCREEN_CONFIGURATIONS:
+		return errors.Wrapf(h.screenConfigurations(id, session), "%s failed", header.Event)
+	case event.SCREEN_SET:
+		payload := &message.ScreenResolution{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.screenSet(id, session, payload)
+			}), "%s failed", header.Event)
+
+	// Boradcast Events
+	case event.BORADCAST_CREATE:
+		payload := &message.BroadcastCreate{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.boradcastCreate(session, payload)
+			}), "%s failed", header.Event)
+	case event.BORADCAST_DESTROY:
+		return errors.Wrapf(h.boradcastDestroy(session), "%s failed", header.Event)
+
+	// Admin Events
+	case event.ADMIN_LOCK:
+		return errors.Wrapf(h.adminLock(id, session), "%s failed", header.Event)
+	case event.ADMIN_UNLOCK:
+		return errors.Wrapf(h.adminUnlock(id, session), "%s failed", header.Event)
+	case event.ADMIN_CONTROL:
+		return errors.Wrapf(h.adminControl(id, session), "%s failed", header.Event)
+	case event.ADMIN_RELEASE:
+		return errors.Wrapf(h.adminRelease(id, session), "%s failed", header.Event)
+	case event.ADMIN_GIVE:
+		payload := &message.Admin{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.adminGive(id, session, payload)
+			}), "%s failed", header.Event)
+	case event.ADMIN_BAN:
+		payload := &message.Admin{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.adminBan(id, session, payload)
+			}), "%s failed", header.Event)
+	case event.ADMIN_KICK:
+		payload := &message.Admin{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.adminKick(id, session, payload)
+			}), "%s failed", header.Event)
+	case event.ADMIN_MUTE:
+		payload := &message.Admin{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.adminMute(id, session, payload)
+			}), "%s failed", header.Event)
+	case event.ADMIN_UNMUTE:
+		payload := &message.Admin{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.adminUnmute(id, session, payload)
+			}), "%s failed", header.Event)
+	default:
+		return errors.Errorf("unknown message event %s", header.Event)
+	}
+}

internal/websocket/screen.go

@@ -0,0 +1,66 @@
+package websocket
+
+import (
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+)
+
+func (h *MessageHandler) screenSet(id string, session types.Session, payload *message.ScreenResolution) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	if err := h.remote.ChangeResolution(payload.Width, payload.Height, payload.Rate); err != nil {
+		h.logger.Warn().Err(err).Msgf("unable to change screen size")
+		return err
+	}
+
+	if err := h.sessions.Broadcast(
+		message.ScreenResolution{
+			Event:  event.SCREEN_RESOLUTION,
+			ID:     id,
+			Width:  payload.Width,
+			Height: payload.Height,
+			Rate:   payload.Rate,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.SCREEN_RESOLUTION)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) screenResolution(id string, session types.Session) error {
+	if size := h.remote.GetScreenSize(); size != nil {
+		if err := session.Send(message.ScreenResolution{
+			Event:  event.SCREEN_RESOLUTION,
+			Width:  size.Width,
+			Height: size.Height,
+			Rate:   int(size.Rate),
+		}); err != nil {
+			h.logger.Warn().Err(err).Msgf("sending event %s has failed", event.SCREEN_RESOLUTION)
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) screenConfigurations(id string, session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	if err := session.Send(message.ScreenConfigurations{
+		Event:          event.SCREEN_CONFIGURATIONS,
+		Configurations: h.remote.ScreenConfigurations(),
+	}); err != nil {
+		h.logger.Warn().Err(err).Msgf("sending event %s has failed", event.SCREEN_CONFIGURATIONS)
+		return err
+	}
+
+	return nil
+}

internal/websocket/session.go

@@ -0,0 +1,93 @@
+package websocket
+
+import (
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+)
+
+func (h *MessageHandler) SessionCreated(id string, session types.Session) error {
+	// send sdp and id over to client
+	if err := h.signalProvide(id, session); err != nil {
+		return err
+	}
+
+	if session.Admin() {
+		// send screen configurations if admin
+		if err := h.screenConfigurations(id, session); err != nil {
+			return err
+		}
+
+		// send broadcast status if admin
+		if err := h.boradcastStatus(session); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) SessionConnected(id string, session types.Session) error {
+	// send list of members to session
+	if err := session.Send(message.MembersList{
+		Event:    event.MEMBER_LIST,
+		Memebers: h.sessions.Members(),
+	}); err != nil {
+		h.logger.Warn().Str("id", id).Err(err).Msgf("sending event %s has failed", event.MEMBER_LIST)
+		return err
+	}
+
+	// send screen current resolution
+	if err := h.screenResolution(id, session); err != nil {
+		return err
+	}
+
+	// tell session there is a host
+	host, ok := h.sessions.GetHost()
+	if ok {
+		if err := session.Send(message.Control{
+			Event: event.CONTROL_LOCKED,
+			ID:    host.ID(),
+		}); err != nil {
+			h.logger.Warn().Str("id", id).Err(err).Msgf("sending event %s has failed", event.CONTROL_LOCKED)
+			return err
+		}
+	}
+
+	// let everyone know there is a new session
+	if err := h.sessions.Broadcast(
+		message.Member{
+			Event:  event.MEMBER_CONNECTED,
+			Member: session.Member(),
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.CONTROL_RELEASE)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) SessionDestroyed(id string) error {
+	// clear host if exists
+	if h.sessions.IsHost(id) {
+		h.sessions.ClearHost()
+		if err := h.sessions.Broadcast(message.Control{
+			Event: event.CONTROL_RELEASE,
+			ID:    id,
+		}, nil); err != nil {
+			h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.CONTROL_RELEASE)
+		}
+	}
+
+	// let everyone know session disconnected
+	if err := h.sessions.Broadcast(
+		message.MemberDisconnected{
+			Event: event.MEMBER_DISCONNECTED,
+			ID:    id,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.MEMBER_DISCONNECTED)
+		return err
+	}
+
+	return nil
+}

internal/websocket/signal.go

@@ -0,0 +1,38 @@
+package websocket
+
+import (
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+)
+
+func (h *MessageHandler) signalProvide(id string, session types.Session) error {
+	sdp, lite, ice, err := h.webrtc.CreatePeer(id, session)
+	if err != nil {
+		return err
+	}
+
+	if err := session.Send(message.SignalProvide{
+		Event: event.SIGNAL_PROVIDE,
+		ID:    id,
+		SDP:   sdp,
+		Lite:  lite,
+		ICE:   ice,
+	}); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) signalAnswer(id string, session types.Session, payload *message.SignalAnswer) error {
+	if err := session.SetName(payload.DisplayName); err != nil {
+		return err
+	}
+
+	if err := session.SignalAnswer(payload.SDP); err != nil {
+		return err
+	}
+
+	return nil
+}

internal/websocket/socket.go

@@ -0,0 +1,55 @@
+package websocket
+
+import (
+	"encoding/json"
+	"strings"
+	"sync"
+
+	"github.com/gorilla/websocket"
+)
+
+type WebSocket struct {
+	id         string
+	address    string
+	ws         *WebSocketHandler
+	connection *websocket.Conn
+	mu         sync.Mutex
+}
+
+func (socket *WebSocket) Address() string {
+	//remote := socket.connection.RemoteAddr()
+	address := strings.SplitN(socket.address, ":", -1)
+	if len(address[0]) < 1 {
+		return socket.address
+	}
+	return address[0]
+}
+
+func (socket *WebSocket) Send(v interface{}) error {
+	socket.mu.Lock()
+	defer socket.mu.Unlock()
+	if socket.connection == nil {
+		return nil
+	}
+
+	raw, err := json.Marshal(v)
+	if err != nil {
+		return err
+	}
+
+	socket.ws.logger.Debug().
+		Str("session", socket.id).
+		Str("address", socket.connection.RemoteAddr().String()).
+		Str("raw", string(raw)).
+		Msg("sending message to client")
+
+	return socket.connection.WriteMessage(websocket.TextMessage, raw)
+}
+
+func (socket *WebSocket) Destroy() error {
+	if socket.connection == nil {
+		return nil
+	}
+
+	return socket.connection.Close()
+}

internal/websocket/websocket.go

@@ -0,0 +1,268 @@
+package websocket
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gorilla/websocket"
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/config"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+	"n.eko.moe/neko/internal/utils"
+)
+
+func New(sessions types.SessionManager, remote types.RemoteManager, broadcast types.BroadcastManager, webrtc types.WebRTCManager, conf *config.WebSocket) *WebSocketHandler {
+	logger := log.With().Str("module", "websocket").Logger()
+
+	return &WebSocketHandler{
+		logger:    logger,
+		conf:      conf,
+		sessions:  sessions,
+		remote:    remote,
+		upgrader:  websocket.Upgrader{
+			CheckOrigin: func(r *http.Request) bool {
+				return true
+			},
+		},
+		handler: &MessageHandler{
+			logger:    logger.With().Str("subsystem", "handler").Logger(),
+			remote:    remote,
+			broadcast: broadcast,
+			sessions:  sessions,
+			webrtc:    webrtc,
+			banned:    make(map[string]bool),
+			locked:    false,
+		},
+	}
+}
+
+// Send pings to peer with this period. Must be less than pongWait.
+const pingPeriod = 60 * time.Second
+
+type WebSocketHandler struct {
+	logger    zerolog.Logger
+	upgrader  websocket.Upgrader
+	sessions  types.SessionManager
+	remote    types.RemoteManager
+	conf      *config.WebSocket
+	handler   *MessageHandler
+	shutdown  chan bool
+}
+
+func (ws *WebSocketHandler) Start() error {
+	ws.sessions.OnCreated(func(id string, session types.Session) {
+		if err := ws.handler.SessionCreated(id, session); err != nil {
+			ws.logger.Warn().Str("id", id).Err(err).Msg("session created with and error")
+		} else {
+			ws.logger.Debug().Str("id", id).Msg("session created")
+		}
+	})
+
+	ws.sessions.OnConnected(func(id string, session types.Session) {
+		if err := ws.handler.SessionConnected(id, session); err != nil {
+			ws.logger.Warn().Str("id", id).Err(err).Msg("session connected with and error")
+		} else {
+			ws.logger.Debug().Str("id", id).Msg("session connected")
+		}
+	})
+
+	ws.sessions.OnDestroy(func(id string, session types.Session) {
+		if err := ws.handler.SessionDestroyed(id); err != nil {
+			ws.logger.Warn().Str("id", id).Err(err).Msg("session destroyed with and error")
+		} else {
+			ws.logger.Debug().Str("id", id).Msg("session destroyed")
+		}
+	})
+
+	go func() {
+		defer func() {
+			ws.logger.Info().Msg("shutdown")
+		}()
+
+		current := ws.remote.ReadClipboard()
+
+		for {
+			select {
+			case <-ws.shutdown:
+				return
+			default:
+				if ws.sessions.HasHost() {
+					text := ws.remote.ReadClipboard()
+					if text != current {
+						session, ok := ws.sessions.GetHost()
+						if ok {
+							session.Send(message.Clipboard{
+								Event: event.CONTROL_CLIPBOARD,
+								Text:  text,
+							})
+						}
+						current = text
+					}
+				}
+				time.Sleep(100 * time.Millisecond)
+			}
+		}
+	}()
+
+	return nil
+}
+
+func (ws *WebSocketHandler) Shutdown() error {
+	ws.shutdown <- true
+	return nil
+}
+
+func (ws *WebSocketHandler) Upgrade(w http.ResponseWriter, r *http.Request) error {
+	ws.logger.Debug().Msg("attempting to upgrade connection")
+
+	connection, err := ws.upgrader.Upgrade(w, r, nil)
+	if err != nil {
+		ws.logger.Error().Err(err).Msg("failed to upgrade connection")
+		return err
+	}
+
+	id, ip, admin, err := ws.authenticate(r)
+	if err != nil {
+		ws.logger.Warn().Err(err).Msg("authentication failed")
+
+		if err = connection.WriteJSON(message.Disconnect{
+			Event:   event.SYSTEM_DISCONNECT,
+			Message: "invalid_password",
+		}); err != nil {
+			ws.logger.Error().Err(err).Msg("failed to send disconnect")
+		}
+
+		if err = connection.Close(); err != nil {
+			return err
+		}
+		return nil
+	}
+
+	socket := &WebSocket{
+		id:         id,
+		ws:         ws,
+		address:    ip,
+		connection: connection,
+	}
+
+	ok, reason, err := ws.handler.Connected(id, socket)
+	if err != nil {
+		ws.logger.Error().Err(err).Msg("connection failed")
+		return err
+	}
+
+	if !ok {
+		if err = connection.WriteJSON(message.Disconnect{
+			Event:   event.SYSTEM_DISCONNECT,
+			Message: reason,
+		}); err != nil {
+			ws.logger.Error().Err(err).Msg("failed to send disconnect")
+		}
+
+		if err = connection.Close(); err != nil {
+			return err
+		}
+
+		return nil
+	}
+
+	ws.sessions.New(id, admin, socket)
+
+	ws.logger.
+		Debug().
+		Str("session", id).
+		Str("address", connection.RemoteAddr().String()).
+		Msg("new connection created")
+
+	defer func() {
+		ws.logger.
+			Debug().
+			Str("session", id).
+			Str("address", connection.RemoteAddr().String()).
+			Msg("session ended")
+	}()
+
+	ws.handle(connection, id)
+	return nil
+}
+
+func (ws *WebSocketHandler) authenticate(r *http.Request) (string, string, bool, error) {
+	ip := r.RemoteAddr
+
+	if ws.conf.Proxy {
+		ip = utils.ReadUserIP(r)
+	}
+
+	id, err := utils.NewUID(32)
+	if err != nil {
+		return "", ip, false, err
+	}
+
+	passwords, ok := r.URL.Query()["password"]
+	if !ok || len(passwords[0]) < 1 {
+		return "", ip, false, fmt.Errorf("no password provided")
+	}
+
+	if passwords[0] == ws.conf.AdminPassword {
+		return id, ip, true, nil
+	}
+
+	if passwords[0] == ws.conf.Password {
+		return id, ip, false, nil
+	}
+
+	return "", ip, false, fmt.Errorf("invalid password: %s", passwords[0])
+}
+
+func (ws *WebSocketHandler) handle(connection *websocket.Conn, id string) {
+	bytes := make(chan []byte)
+	cancel := make(chan struct{})
+	ticker := time.NewTicker(pingPeriod)
+
+	go func() {
+		defer func() {
+			ticker.Stop()
+			ws.logger.Debug().Str("address", connection.RemoteAddr().String()).Msg("handle socket ending")
+			ws.handler.Disconnected(id)
+		}()
+
+		for {
+			_, raw, err := connection.ReadMessage()
+			if err != nil {
+				if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
+					ws.logger.Warn().Err(err).Msg("read message error")
+				} else {
+					ws.logger.Debug().Err(err).Msg("read message error")
+				}
+				close(cancel)
+				break
+			}
+			bytes <- raw
+		}
+	}()
+
+	for {
+		select {
+		case raw := <-bytes:
+			ws.logger.Debug().
+				Str("session", id).
+				Str("address", connection.RemoteAddr().String()).
+				Str("raw", string(raw)).
+				Msg("received message from client")
+			if err := ws.handler.Message(id, raw); err != nil {
+				ws.logger.Error().Err(err).Msg("message handler has failed")
+			}
+		case <-cancel:
+			return
+		case _ = <-ticker.C:
+			if err := connection.WriteMessage(websocket.PingMessage, nil); err != nil {
+				return
+			}
+		}
+	}
+}