Initial commit

handlers/socket.go

@@ -0,0 +1,53 @@
+package handlers
+
+import (
+	"bytes"
+	"encoding/base64"
+	"log"
+	"time"
+
+	"golang.org/x/net/websocket"
+
+	"github.com/gen2brain/cam2ip/camera"
+)
+
+// Socket handler.
+type Socket struct {
+	camera *camera.Camera
+	delay  int
+}
+
+// NewSocket returns new socket handler.
+func NewSocket(camera *camera.Camera, delay int) websocket.Handler {
+	s := &Socket{camera, delay}
+	return websocket.Handler(s.write)
+}
+
+// write writes images to socket
+func (s *Socket) write(ws *websocket.Conn) {
+	for {
+		img, err := s.camera.Read()
+		if err != nil {
+			log.Printf("socket: read: %v", err)
+			continue
+		}
+
+		w := new(bytes.Buffer)
+		enc := camera.NewEncoder(w)
+
+		err = enc.Encode(img)
+		if err != nil {
+			log.Printf("socket: encode: %v", err)
+			continue
+		}
+
+		b64 := base64.StdEncoding.EncodeToString(w.Bytes())
+
+		_, err = ws.Write([]byte(b64))
+		if err != nil {
+			break
+		}
+
+		time.Sleep(time.Duration(s.delay) * time.Millisecond)
+	}
+}