/
web.go
114 lines (101 loc) · 2.23 KB
/
web.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package services
import (
"fmt"
"net"
"net/http"
"github.com/pkg/errors"
log "github.com/sirupsen/logrus"
"github.com/urfave/cli"
)
const (
webHostFlag = "host"
webPortFlag = "port"
webPlayerFlag = "player"
)
func RegisterWebFlags(c *cli.App) {
c.Flags = append(c.Flags, cli.StringFlag{
Name: webHostFlag + ", H",
Usage: "host",
EnvVar: "HOST",
})
c.Flags = append(c.Flags, cli.IntFlag{
Name: webPortFlag + ", P",
Usage: "port",
Value: 8080,
EnvVar: "PORT",
})
c.Flags = append(c.Flags, cli.BoolFlag{
Name: webPlayerFlag,
Usage: "player",
EnvVar: "PLAYER",
})
}
type Web struct {
h *HLSParser
host string
port int
player bool
output string
handler http.Handler
ln net.Listener
}
func getParam(headerName string, getName string, r *http.Request) string {
param := r.Header.Get(headerName)
if param != "" {
return param
}
return r.URL.Query().Get(getName)
}
func NewWeb(c *cli.Context, h *HLSParser) *Web {
we := &Web{
host: c.String(webHostFlag),
port: c.Int(webPortFlag),
player: c.Bool(webPlayerFlag),
output: c.String(OutputFlag),
h: h,
}
we.buildHandler()
return we
}
func (s *Web) Handle(h func(h http.Handler) http.Handler) {
s.handler = h(s.handler)
}
type Handleable interface {
Handle(h func(h http.Handler) http.Handler)
}
func (s *Web) buildHandler() {
mux := http.NewServeMux()
if s.player {
mux.Handle("/player/", http.StripPrefix("/player/", http.FileServer(http.Dir("./player"))))
}
mux.HandleFunc("/touch", func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
})
fileH := http.FileServer(http.Dir(s.output))
enrichH := enrichPlaylistHandler(fileH)
corsH := allowCORSHandler(enrichH)
mux.Handle("/", corsH)
s.handler = mux
}
func (s *Web) Serve() error {
addr := fmt.Sprintf("%s:%d", s.host, s.port)
ln, err := net.Listen("tcp", addr)
if err != nil {
return errors.Wrap(err, "Failed to bind address")
}
s.ln = ln
log.Infof("Serving Web at %v", addr)
if s.player {
log.Info(fmt.Sprintf("Player available at http://%v/player/", addr))
}
return http.Serve(ln, s.handler)
}
func (s *Web) Close() {
log.Info("Closing Web")
defer func() {
log.Info("Web closed")
}()
if s.ln != nil {
s.ln.Close()
}
}