cam2ip/handlers/mjpeg.go

package handlers

import (
	"fmt"
	"log"
	"mime/multipart"
	"net/http"
	"net/textproto"
	"time"

	"github.com/gen2brain/cam2ip/image"
)

// MJPEG handler.
type MJPEG struct {
	reader  ImageReader
	delay   int
	quality int
}

// NewMJPEG returns new MJPEG handler.
func NewMJPEG(reader ImageReader, delay, quality int) *MJPEG {
	return &MJPEG{reader, delay, quality}
}

// ServeHTTP handles requests on incoming connections.
func (m *MJPEG) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	if r.Method != "GET" && r.Method != "HEAD" {
		http.Error(w, "405 Method Not Allowed", http.StatusMethodNotAllowed)

		return
	}

	mimeWriter := multipart.NewWriter(w)
	_ = mimeWriter.SetBoundary("--boundary")

	w.Header().Add("Connection", "close")
	w.Header().Add("Cache-Control", "no-store, no-cache")
	w.Header().Add("Content-Type", fmt.Sprintf("multipart/x-mixed-replace;boundary=%s", mimeWriter.Boundary()))

	done := r.Context().Done()

loop:
	for {
		select {
		case <-done:
			break loop

		default:
			partHeader := make(textproto.MIMEHeader)
			partHeader.Add("Content-Type", "image/jpeg")

			partWriter, err := mimeWriter.CreatePart(partHeader)
			if err != nil {
				log.Printf("mjpeg: createPart: %v", err)
				continue
			}

			img, err := m.reader.Read()
			if err != nil {
				log.Printf("mjpeg: read: %v", err)
				continue
			}

			err = image.NewEncoder(partWriter, m.quality).Encode(img)
			if err != nil {
				log.Printf("mjpeg: encode: %v", err)
				continue
			}

			if m.delay > 0 {
				time.Sleep(time.Duration(m.delay) * time.Millisecond)
			}
		}
	}

	_ = mimeWriter.Close()
}