aboutsummaryrefslogtreecommitdiff
path: root/internal/pvfm/recording/recording_demo.go
blob: 3264841455d793b58dd0ea96317b8cd2984dc077 (plain)
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
// +build ignore

package main

import (
	"flag"
	"log"
	"os"
	"os/signal"

	"github.com/PonyvilleFM/aura/recording"
)

var (
	url   = flag.String("url", "", "url to record")
	fname = flag.String("fname", "", "filename to record to")
	debug = flag.Bool("debug", false, "debug mode")

	askedToDie bool
)

func main() {
	flag.Parse()

	r, err := recording.New(*url, *fname)
	if err != nil {
		log.Printf("%s -> %s: %v", *url, *fname, err)
		log.Fatal(err)
	}

	r.Debug = *debug

	go func() {
		log.Printf("Starting download of stream %s to %s", *url, *fname)
		err := r.Start()
		if err != nil {
			log.Fatal(err)
		}
	}()

	go func() {
		c := make(chan os.Signal, 1)
		signal.Notify(c, os.Interrupt)

		for _ = range c {
			if askedToDie {
				os.Exit(2)
			}

			log.Println("Stopping recording... (^C again to kill now)")
			r.Cancel()

			askedToDie = true
		}
	}()

	<-r.Done()
	log.Printf("stream %s recorded to %s", *url, *fname)
}