package main

import (
	"log"
	"net/http"
	"strings"
	"sync"
)

type Handler struct {
	managers map[string]*Manager
	mutex    sync.RWMutex
	close    chan string
}

func NewHandler() *Handler {
	h := &Handler{managers: make(map[string]*Manager), close: make(chan string)}
	go h.watchClose()
	return h
}

func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	url := r.URL.Path
	parts := make([]string, 0)

	for _, part := range strings.Split(url, "/") {
		if part != "" {
			parts = append(parts, part)
		}
	}

	if len(parts) < 3 {
		log.Println("Invalid URL", url)
		w.WriteHeader(http.StatusBadRequest)
		return
	}

	streamid := parts[0]
	path := "/" + strings.Join(parts[1:len(parts)-2], "/")
	chunk := parts[len(parts)-1]

	log.Println("Serving", path, streamid, chunk)

	if streamid == "" || chunk == "" || path == "" {
		w.WriteHeader(http.StatusBadRequest)
		return
	}

	manager := h.getManager(streamid)
	if manager == nil {
		manager = h.createManager(path, streamid)
	}
	manager.ServeHTTP(w, r, chunk)
}

func (h *Handler) getManager(streamid string) *Manager {
	h.mutex.RLock()
	defer h.mutex.RUnlock()
	return h.managers[streamid]
}

func (h *Handler) createManager(path string, streamid string) *Manager {
	h.mutex.Lock()
	defer h.mutex.Unlock()
	manager := NewManager(path, streamid, h.close)
	h.managers[streamid] = manager
	return manager
}

func (h *Handler) removeManager(streamid string) {
	h.mutex.Lock()
	defer h.mutex.Unlock()
	delete(h.managers, streamid)
}

func (h *Handler) watchClose() {
	for {
		id := <-h.close
		if id == "" {
			return
		}

		log.Println("Closing stream", id)
		h.removeManager(id)
	}
}

func (h *Handler) Close() {
	h.close <- ""
}

func main() {
	log.Println("Starting VOD server")

	h := NewHandler()

	http.Handle("/", h)
	http.ListenAndServe(":47788", nil)

	log.Println("Exiting VOD server")
	h.Close()
}