73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
package tracker
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/rombintu/godpn/internal/models"
|
|
)
|
|
|
|
func (s *Server) registerHandler(c echo.Context) error {
|
|
var peer models.PeerManifest
|
|
if err := c.Bind(peer); err != nil {
|
|
return c.JSON(http.StatusBadRequest, map[string]string{"error": "invalid request"})
|
|
}
|
|
|
|
if peer.PubKey == "" {
|
|
return c.JSON(http.StatusBadRequest, map[string]string{"error": "pubkey is required"})
|
|
}
|
|
|
|
ip := c.RealIP()
|
|
if peer.IP != "" {
|
|
ip = peer.IP
|
|
}
|
|
|
|
s.mu.Lock()
|
|
s.tracker.peers[ip] = peer.PubKey
|
|
s.mu.Unlock()
|
|
|
|
return c.JSON(http.StatusOK, map[string]string{
|
|
"status": "registered",
|
|
"ip": ip,
|
|
})
|
|
}
|
|
|
|
// Обработчик списка peers (HTTP)
|
|
func (s *Server) listPeers(c echo.Context) error {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
return c.JSON(http.StatusOK, s.tracker.peers)
|
|
}
|
|
|
|
// WebSocket handler для реального времени
|
|
func (s *Server) websocketHandler(c echo.Context) error {
|
|
upgrader := websocket.Upgrader{
|
|
CheckOrigin: func(r *http.Request) bool { return true },
|
|
}
|
|
|
|
ws, err := upgrader.Upgrade(c.Response(), c.Request(), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer ws.Close()
|
|
|
|
// Отправляем текущий список peers при подключении
|
|
s.mu.RLock()
|
|
if err := ws.WriteJSON(s.tracker.peers); err != nil {
|
|
s.mu.RUnlock()
|
|
return err
|
|
}
|
|
s.mu.RUnlock()
|
|
|
|
// Ожидаем новые сообщения (можно добавить heartbeat)
|
|
for {
|
|
_, _, err := ws.ReadMessage()
|
|
if err != nil {
|
|
break
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|