Initial commit

streamer.go

@@ -0,0 +1,59 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"os/exec"
+
+	"gopkg.in/djherbis/buffer.v1"
+	"gopkg.in/djherbis/nio.v2"
+)
+
+func stream(url string, w http.ResponseWriter) error {
+	if url == "" {
+		err := errors.New("empty url")
+
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return err
+	}
+
+	w.Header().Set("Content-Type", config.ContentType)
+	w.Header().Set("Accept-Ranges", "none")
+	w.WriteHeader(http.StatusOK)
+
+	return encode(url, w)
+}
+
+func encode(url string, w io.Writer) error {
+	cmd := exec.Command("ffmpeg",
+		"-i", url,
+		"-vn",
+		"-c:a", config.Codec,
+		"-b:a", fmt.Sprintf("%dk", config.Bitrate),
+		"-f", config.Format,
+		"-ac", "2",
+		"pipe:1")
+	out, err := cmd.StdoutPipe()
+	if err != nil {
+		return err
+	}
+
+	err = cmd.Start()
+	if err != nil {
+		return err
+	}
+
+	go func() {
+		cmd.Wait()
+	}()
+
+	buf := buffer.New(int64(config.BufferSize) * 1024 * 1024)
+	_, err = nio.Copy(w, out, buf)
+	if err != nil {
+		cmd.Process.Kill()
+	}
+
+	return nil
+}