mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
lint fix.
This commit is contained in:
parent
1ff743c598
commit
9c014336ff
@ -47,7 +47,7 @@ func init() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if _, err := os.Stat(logs); os.IsNotExist(err) {
|
if _, err := os.Stat(logs); os.IsNotExist(err) {
|
||||||
os.Mkdir(logs, os.ModePerm)
|
_ = os.Mkdir(logs, os.ModePerm)
|
||||||
}
|
}
|
||||||
|
|
||||||
latest := filepath.Join(logs, "neko-latest.log")
|
latest := filepath.Join(logs, "neko-latest.log")
|
||||||
|
@ -30,6 +30,11 @@ func New(remote *config.Remote, config *config.Broadcast) *BroadcastManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (manager *BroadcastManager) Shutdown() error {
|
||||||
|
manager.Destroy()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func (manager *BroadcastManager) Start() error {
|
func (manager *BroadcastManager) Start() error {
|
||||||
if !manager.enabled || manager.IsActive() {
|
if !manager.enabled || manager.IsActive() {
|
||||||
return nil
|
return nil
|
||||||
|
@ -63,10 +63,8 @@ func errResponse(input interface{}) *ErrResponse {
|
|||||||
case *error:
|
case *error:
|
||||||
e := err.(error)
|
e := err.(error)
|
||||||
res.Details = e.Error()
|
res.Details = e.Error()
|
||||||
break
|
|
||||||
default:
|
default:
|
||||||
res.Details = fmt.Sprintf("%+v", err)
|
res.Details = fmt.Sprintf("%+v", err)
|
||||||
break
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -3,7 +3,6 @@ package http
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
@ -32,21 +31,22 @@ func New(conf *config.Server, webSocketHandler types.WebSocketHandler) *Server {
|
|||||||
router.Use(middleware.Logger) // Log API request calls
|
router.Use(middleware.Logger) // Log API request calls
|
||||||
|
|
||||||
router.Get("/ws", func(w http.ResponseWriter, r *http.Request) {
|
router.Get("/ws", func(w http.ResponseWriter, r *http.Request) {
|
||||||
webSocketHandler.Upgrade(w, r)
|
err := webSocketHandler.Upgrade(w, r)
|
||||||
|
if err != nil {
|
||||||
|
logger.Warn().Err(err).Msg("failed to upgrade websocket conection")
|
||||||
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
router.Get("/stats", func(w http.ResponseWriter, r *http.Request) {
|
router.Get("/stats", func(w http.ResponseWriter, r *http.Request) {
|
||||||
password := r.URL.Query().Get("pwd")
|
password := r.URL.Query().Get("pwd")
|
||||||
isAdmin, err := webSocketHandler.IsAdmin(password)
|
isAdmin, err := webSocketHandler.IsAdmin(password)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
w.WriteHeader(http.StatusForbidden)
|
http.Error(w, err.Error(), http.StatusForbidden)
|
||||||
fmt.Fprint(w, err)
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if !isAdmin {
|
if !isAdmin {
|
||||||
w.WriteHeader(http.StatusUnauthorized)
|
http.Error(w, "bad authorization", http.StatusUnauthorized)
|
||||||
fmt.Fprint(w, "bad authorization")
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -59,7 +59,7 @@ func New(conf *config.Server, webSocketHandler types.WebSocketHandler) *Server {
|
|||||||
})
|
})
|
||||||
|
|
||||||
router.Get("/health", func(w http.ResponseWriter, r *http.Request) {
|
router.Get("/health", func(w http.ResponseWriter, r *http.Request) {
|
||||||
w.Write([]byte("true"))
|
_, _ = w.Write([]byte("true"))
|
||||||
})
|
})
|
||||||
|
|
||||||
fs := http.FileServer(http.Dir(conf.Static))
|
fs := http.FileServer(http.Dir(conf.Static))
|
||||||
@ -67,8 +67,7 @@ func New(conf *config.Server, webSocketHandler types.WebSocketHandler) *Server {
|
|||||||
if _, err := os.Stat(conf.Static + r.URL.Path); !os.IsNotExist(err) {
|
if _, err := os.Stat(conf.Static + r.URL.Path); !os.IsNotExist(err) {
|
||||||
fs.ServeHTTP(w, r)
|
fs.ServeHTTP(w, r)
|
||||||
} else {
|
} else {
|
||||||
w.WriteHeader(http.StatusNotFound)
|
http.NotFound(w, r)
|
||||||
fmt.Fprint(w, "404 page not found")
|
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
@ -250,7 +250,7 @@ func (manager *RemoteManager) GetScreenSize() *types.ScreenSize {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (manager *RemoteManager) SetKeyboardLayout(layout string) {
|
func (manager *RemoteManager) SetKeyboardLayout(layout string) {
|
||||||
exec.Command("setxkbmap", layout).Run()
|
_ = exec.Command("setxkbmap", layout).Run()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (manager *RemoteManager) SetKeyboardModifiers(NumLock int, CapsLock int, ScrollLock int) {
|
func (manager *RemoteManager) SetKeyboardModifiers(NumLock int, CapsLock int, ScrollLock int) {
|
||||||
|
@ -141,7 +141,7 @@ func (manager *SessionManager) Members() []*types.Member {
|
|||||||
return members
|
return members
|
||||||
}
|
}
|
||||||
|
|
||||||
func (manager *SessionManager) Destroy(id string) error {
|
func (manager *SessionManager) Destroy(id string) {
|
||||||
manager.mu.Lock()
|
manager.mu.Lock()
|
||||||
session, ok := manager.members[id]
|
session, ok := manager.members[id]
|
||||||
if ok {
|
if ok {
|
||||||
@ -154,11 +154,11 @@ func (manager *SessionManager) Destroy(id string) error {
|
|||||||
manager.mu.Unlock()
|
manager.mu.Unlock()
|
||||||
|
|
||||||
manager.emmiter.Emit("destroyed", id, session)
|
manager.emmiter.Emit("destroyed", id, session)
|
||||||
return err
|
manager.logger.Err(err).Str("session_id", id).Msg("destorying session")
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
manager.mu.Unlock()
|
manager.mu.Unlock()
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (manager *SessionManager) Clear() error {
|
func (manager *SessionManager) Clear() error {
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
package types
|
package types
|
||||||
|
|
||||||
type BroadcastManager interface {
|
type BroadcastManager interface {
|
||||||
|
Shutdown() error
|
||||||
Start() error
|
Start() error
|
||||||
Stop()
|
Stop()
|
||||||
IsActive() bool
|
IsActive() bool
|
||||||
|
@ -38,7 +38,7 @@ type SessionManager interface {
|
|||||||
Get(id string) (Session, bool)
|
Get(id string) (Session, bool)
|
||||||
Members() []*Member
|
Members() []*Member
|
||||||
Admins() []*Member
|
Admins() []*Member
|
||||||
Destroy(id string) error
|
Destroy(id string)
|
||||||
Clear() error
|
Clear() error
|
||||||
Broadcast(v interface{}, exclude interface{}) error
|
Broadcast(v interface{}, exclude interface{}) error
|
||||||
OnHost(listener func(id string))
|
OnHost(listener func(id string))
|
||||||
|
@ -15,7 +15,7 @@ type WebSocket interface {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type WebSocketHandler interface {
|
type WebSocketHandler interface {
|
||||||
Start() error
|
Start()
|
||||||
Shutdown() error
|
Shutdown() error
|
||||||
Upgrade(w http.ResponseWriter, r *http.Request) error
|
Upgrade(w http.ResponseWriter, r *http.Request) error
|
||||||
Stats() Stats
|
Stats() Stats
|
||||||
|
@ -93,7 +93,7 @@ func (manager *WebRTCManager) CreatePeer(id string, session types.Session) (stri
|
|||||||
settings.SetLite(true)
|
settings.SetLite(true)
|
||||||
}
|
}
|
||||||
|
|
||||||
settings.SetEphemeralUDPPortRange(manager.config.EphemeralMin, manager.config.EphemeralMax)
|
_ = settings.SetEphemeralUDPPortRange(manager.config.EphemeralMin, manager.config.EphemeralMax)
|
||||||
settings.SetNAT1To1IPs(manager.config.NAT1To1IPs, webrtc.ICECandidateTypeHost)
|
settings.SetNAT1To1IPs(manager.config.NAT1To1IPs, webrtc.ICECandidateTypeHost)
|
||||||
settings.SetICETimeouts(6*time.Second, 6*time.Second, 3*time.Second)
|
settings.SetICETimeouts(6*time.Second, 6*time.Second, 3*time.Second)
|
||||||
settings.SetSRTPReplayProtectionWindow(512)
|
settings.SetSRTPReplayProtectionWindow(512)
|
||||||
@ -101,8 +101,8 @@ func (manager *WebRTCManager) CreatePeer(id string, session types.Session) (stri
|
|||||||
// Create MediaEngine based off sdp
|
// Create MediaEngine based off sdp
|
||||||
engine := webrtc.MediaEngine{}
|
engine := webrtc.MediaEngine{}
|
||||||
|
|
||||||
engine.RegisterCodec(manager.audioCodec, webrtc.RTPCodecTypeAudio)
|
_ = engine.RegisterCodec(manager.audioCodec, webrtc.RTPCodecTypeAudio)
|
||||||
engine.RegisterCodec(manager.videoCodec, webrtc.RTPCodecTypeVideo)
|
_ = engine.RegisterCodec(manager.videoCodec, webrtc.RTPCodecTypeVideo)
|
||||||
|
|
||||||
i := &interceptor.Registry{}
|
i := &interceptor.Registry{}
|
||||||
if err := webrtc.RegisterDefaultInterceptors(&engine, i); err != nil {
|
if err := webrtc.RegisterDefaultInterceptors(&engine, i); err != nil {
|
||||||
|
@ -66,7 +66,10 @@ func (h *MessageHandler) adminControl(id string, session types.Session) error {
|
|||||||
|
|
||||||
host, ok := h.sessions.GetHost()
|
host, ok := h.sessions.GetHost()
|
||||||
|
|
||||||
h.sessions.SetHost(id)
|
err := h.sessions.SetHost(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
if ok {
|
if ok {
|
||||||
if err := h.sessions.Broadcast(
|
if err := h.sessions.Broadcast(
|
||||||
@ -138,7 +141,10 @@ func (h *MessageHandler) adminGive(id string, session types.Session, payload *me
|
|||||||
}
|
}
|
||||||
|
|
||||||
// set host
|
// set host
|
||||||
h.sessions.SetHost(payload.ID)
|
err := h.sessions.SetHost(payload.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
// let everyone know
|
// let everyone know
|
||||||
if err := h.sessions.Broadcast(
|
if err := h.sessions.Broadcast(
|
||||||
|
@ -35,7 +35,10 @@ func (h *MessageHandler) controlRequest(id string, session types.Session) error
|
|||||||
// check for host
|
// check for host
|
||||||
if !h.sessions.HasHost() {
|
if !h.sessions.HasHost() {
|
||||||
// set host
|
// set host
|
||||||
h.sessions.SetHost(id)
|
err := h.sessions.SetHost(id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
// let everyone know
|
// let everyone know
|
||||||
if err := h.sessions.Broadcast(
|
if err := h.sessions.Broadcast(
|
||||||
@ -89,7 +92,10 @@ func (h *MessageHandler) controlGive(id string, session types.Session, payload *
|
|||||||
}
|
}
|
||||||
|
|
||||||
// set host
|
// set host
|
||||||
h.sessions.SetHost(payload.ID)
|
err := h.sessions.SetHost(payload.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
// let everyone know
|
// let everyone know
|
||||||
if err := h.sessions.Broadcast(
|
if err := h.sessions.Broadcast(
|
||||||
|
@ -42,8 +42,8 @@ func (h *MessageHandler) Connected(admin bool, socket *WebSocket) (bool, string,
|
|||||||
return true, "", nil
|
return true, "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *MessageHandler) Disconnected(id string) error {
|
func (h *MessageHandler) Disconnected(id string) {
|
||||||
return h.sessions.Destroy(id)
|
h.sessions.Destroy(id)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *MessageHandler) Message(id string, raw []byte) error {
|
func (h *MessageHandler) Message(id string, raw []byte) error {
|
||||||
|
@ -60,7 +60,7 @@ type WebSocketHandler struct {
|
|||||||
conns uint32
|
conns uint32
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ws *WebSocketHandler) Start() error {
|
func (ws *WebSocketHandler) Start() {
|
||||||
ws.sessions.OnCreated(func(id string, session types.Session) {
|
ws.sessions.OnCreated(func(id string, session types.Session) {
|
||||||
if err := ws.handler.SessionCreated(id, session); err != nil {
|
if err := ws.handler.SessionCreated(id, session); err != nil {
|
||||||
ws.logger.Warn().Str("id", id).Err(err).Msg("session created with and error")
|
ws.logger.Warn().Str("id", id).Err(err).Msg("session created with and error")
|
||||||
@ -99,25 +99,33 @@ func (ws *WebSocketHandler) Start() error {
|
|||||||
case <-ws.shutdown:
|
case <-ws.shutdown:
|
||||||
return
|
return
|
||||||
default:
|
default:
|
||||||
if ws.sessions.HasHost() {
|
time.Sleep(100 * time.Millisecond)
|
||||||
text := ws.remote.ReadClipboard()
|
|
||||||
if text != current {
|
if !ws.sessions.HasHost() {
|
||||||
session, ok := ws.sessions.GetHost()
|
continue
|
||||||
if ok {
|
}
|
||||||
session.Send(message.Clipboard{
|
|
||||||
Event: event.CONTROL_CLIPBOARD,
|
text := ws.remote.ReadClipboard()
|
||||||
Text: text,
|
if text == current {
|
||||||
})
|
continue
|
||||||
}
|
}
|
||||||
current = text
|
|
||||||
|
session, ok := ws.sessions.GetHost()
|
||||||
|
if ok {
|
||||||
|
err := session.Send(message.Clipboard{
|
||||||
|
Event: event.CONTROL_CLIPBOARD,
|
||||||
|
Text: text,
|
||||||
|
})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
ws.logger.Err(err).Msg("unable to synchronize clipboard")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
time.Sleep(100 * time.Millisecond)
|
|
||||||
|
current = text
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ws *WebSocketHandler) Shutdown() error {
|
func (ws *WebSocketHandler) Shutdown() error {
|
||||||
|
@ -129,12 +129,12 @@ func WriteClipboard(data string) {
|
|||||||
|
|
||||||
func ResetKeys() {
|
func ResetKeys() {
|
||||||
for code := range debounce_button {
|
for code := range debounce_button {
|
||||||
ButtonUp(code)
|
_ = ButtonUp(code)
|
||||||
|
|
||||||
delete(debounce_button, code)
|
delete(debounce_button, code)
|
||||||
}
|
}
|
||||||
for code := range debounce_key {
|
for code := range debounce_key {
|
||||||
KeyUp(code)
|
_ = KeyUp(code)
|
||||||
|
|
||||||
delete(debounce_key, code)
|
delete(debounce_key, code)
|
||||||
}
|
}
|
||||||
@ -146,7 +146,7 @@ func CheckKeys(duration time.Duration) {
|
|||||||
if t.Sub(start) < duration {
|
if t.Sub(start) < duration {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
ButtonUp(code)
|
_ = ButtonUp(code)
|
||||||
|
|
||||||
delete(debounce_button, code)
|
delete(debounce_button, code)
|
||||||
}
|
}
|
||||||
@ -154,7 +154,7 @@ func CheckKeys(duration time.Duration) {
|
|||||||
if t.Sub(start) < duration {
|
if t.Sub(start) < duration {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
KeyUp(code)
|
_ = KeyUp(code)
|
||||||
|
|
||||||
delete(debounce_key, code)
|
delete(debounce_key, code)
|
||||||
}
|
}
|
||||||
|
@ -136,6 +136,7 @@ func (neko *Neko) Start() {
|
|||||||
server := http.New(neko.Server, webSocketHandler)
|
server := http.New(neko.Server, webSocketHandler)
|
||||||
server.Start()
|
server.Start()
|
||||||
|
|
||||||
|
neko.broadcastManager = broadcastManager
|
||||||
neko.sessionManager = sessionManager
|
neko.sessionManager = sessionManager
|
||||||
neko.remoteManager = remoteManager
|
neko.remoteManager = remoteManager
|
||||||
neko.webRTCManager = webRTCManager
|
neko.webRTCManager = webRTCManager
|
||||||
@ -144,6 +145,12 @@ func (neko *Neko) Start() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (neko *Neko) Shutdown() {
|
func (neko *Neko) Shutdown() {
|
||||||
|
if err := neko.broadcastManager.Shutdown(); err != nil {
|
||||||
|
neko.logger.Err(err).Msg("broadcast manager shutdown with an error")
|
||||||
|
} else {
|
||||||
|
neko.logger.Debug().Msg("broadcast manager shutdown")
|
||||||
|
}
|
||||||
|
|
||||||
if err := neko.remoteManager.Shutdown(); err != nil {
|
if err := neko.remoteManager.Shutdown(); err != nil {
|
||||||
neko.logger.Err(err).Msg("remote manager shutdown with an error")
|
neko.logger.Err(err).Msg("remote manager shutdown with an error")
|
||||||
} else {
|
} else {
|
||||||
|
Loading…
Reference in New Issue
Block a user