clipper/backend/main.go

61 lines
1.2 KiB
Go
Raw Normal View History

2021-09-04 22:29:12 +00:00
package main
import (
2021-09-05 05:24:51 +00:00
"bytes"
"encoding/binary"
2021-09-04 22:29:12 +00:00
"log"
2021-09-05 05:24:51 +00:00
"os/exec"
2021-09-04 22:29:12 +00:00
"github.com/kkdai/youtube/v2"
)
2021-09-05 05:24:51 +00:00
const (
ItagM4AAudio = 140
SizeOfInt16 = 2
)
// TODO: process as stream to avoid loading full buffer in memory
2021-09-04 22:29:12 +00:00
func main() {
var client youtube.Client
videoID := "s_oJYdRlrv0"
video, err := client.GetVideo(videoID)
if err != nil {
log.Fatal(err)
}
format := video.Formats.FindByItag(ItagM4AAudio)
stream, _, err := client.GetStream(video, format)
if err != nil {
log.Fatal(err)
}
2021-09-05 05:24:51 +00:00
// TODO: force 44.1khz
cmd := exec.Command("ffmpeg", "-i", "-", "-f", "s16le", "-acodec", "pcm_s16le", "-")
cmd.Stdin = stream
2021-09-04 22:29:12 +00:00
2021-09-05 05:24:51 +00:00
var out bytes.Buffer
cmd.Stdout = &out
2021-09-04 22:29:12 +00:00
2021-09-05 05:24:51 +00:00
var errOut bytes.Buffer
cmd.Stderr = &errOut
2021-09-04 22:29:12 +00:00
2021-09-05 05:24:51 +00:00
if err = cmd.Run(); err != nil {
log.Fatalf("err = %v, stdErr = %s", err, errOut.String())
}
log.Printf("byteLen = %d", out.Len())
// TODO: reflect.Slice to avoid copying?
data := make([]uint16, 0, out.Len()/SizeOfInt16)
for i := 0; i < out.Len(); i += SizeOfInt16 {
v := binary.LittleEndian.Uint16(out.Bytes()[i : i+SizeOfInt16])
data = append(data, v)
2021-09-04 22:29:12 +00:00
}
2021-09-05 05:24:51 +00:00
log.Printf("intLen = %d", len(data))
log.Printf("ints = %+v", data[0:8192])
2021-09-04 22:29:12 +00:00
}