stream-server/http/server.go

package http

import (
	"net/http"
	"strings"
	"os"
	"html/template"
)

func NewServer(port string) (error) {
	server := http.NewServeMux()
	server_setup(server)
	if err := http.ListenAndServe(":" + port, server); err != nil {
		return err
	}
	return nil
}

func server_setup(server *http.ServeMux) {
	server.HandleFunc("/live/", serve_main_page)
	server.HandleFunc("/list/", serve_live_playlist)
	server.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("static/"))))
	server.HandleFunc("/vid/", serve_vid_segment)
}

func serve_vid_segment(w http.ResponseWriter, r *http.Request) {
	base_dir, _ := os.UserHomeDir()
	tmp := strings.Split(strings.TrimPrefix(r.URL.Path, "/vid/"), "/")
	stream_key := tmp[0]
	segment_file := tmp[1]
	http.ServeFile(w, r, base_dir + "/live/" + stream_key + "/" + segment_file)
}

func serve_live_playlist(w http.ResponseWriter, r *http.Request) {
	stream_id := strings.TrimPrefix(r.URL.Path, "/list/")
	base_dir, _ := os.UserHomeDir()
	stream_playlist_path := base_dir + "/live/" + stream_id + "/stream.m3u8"
	if _, err := os.Stat(stream_playlist_path); err != nil {
		http.NotFound(w, r)
	} else {
		http.ServeFile(w, r, stream_playlist_path)
	}
}

func serve_main_page(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Access-Control-Allow-Origin", "*")
	page := template.Must(template.ParseFiles("static/index.html"))
	stream_user := strings.TrimPrefix(r.URL.Path, "/live/")
	page.Execute(w, stream_user)
}