78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"crypto/tls"
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/kixelated/invoker"
|
|
"github.com/kixelated/warp-demo/server/internal/warp"
|
|
)
|
|
|
|
func main() {
|
|
invoker.Panic = true
|
|
|
|
err := run(context.Background())
|
|
if err == nil {
|
|
return
|
|
}
|
|
|
|
log.Println(err)
|
|
|
|
var errPanic invoker.ErrPanic
|
|
|
|
// TODO use an interface
|
|
if errors.As(err, &errPanic) {
|
|
stack := string(errPanic.Stack())
|
|
|
|
scanner := bufio.NewScanner(strings.NewReader(stack))
|
|
for scanner.Scan() {
|
|
log.Println(scanner.Text())
|
|
}
|
|
}
|
|
|
|
os.Exit(1)
|
|
}
|
|
|
|
func run(ctx context.Context) (err error) {
|
|
addr := flag.String("addr", ":4443", "HTTPS server address")
|
|
cert := flag.String("tls-cert", "../cert/localhost.crt", "TLS certificate file path")
|
|
key := flag.String("tls-key", "../cert/localhost.key", "TLS certificate file path")
|
|
logDir := flag.String("log-dir", "", "logs will be written to the provided directory")
|
|
|
|
dash := flag.String("dash", "../media/playlist.mpd", "DASH playlist path")
|
|
|
|
flag.Parse()
|
|
|
|
media, err := warp.NewMedia(*dash)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open media: %w", err)
|
|
}
|
|
|
|
tlsCert, err := tls.LoadX509KeyPair(*cert, *key)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to load TLS certificate: %w", err)
|
|
}
|
|
|
|
config := warp.ServerConfig{
|
|
Addr: *addr,
|
|
Cert: &tlsCert,
|
|
LogDir: *logDir,
|
|
}
|
|
|
|
ws, err := warp.NewServer(config, media)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create warp server: %w", err)
|
|
}
|
|
|
|
log.Printf("listening on %s", *addr)
|
|
|
|
return invoker.Run(ctx, invoker.Interrupt, ws.Run)
|
|
}
|