package main

import (
	"fmt"
	"github.com/nareix/joy4/av"
	"github.com/nareix/joy4/av/avutil"
	//"github.com/nareix/joy4/av/transcode"
	//"github.com/nareix/joy4/cgo/ffmpeg"
	"github.com/nareix/joy4/format"
	"time"
)
func init() {
	format.RegisterAll()
}

func main() {
	//infile, _ := avutil.Open("http://117.169.120.142:8080/ysten-businessmobile/live/hdzhejiangstv/astv.m3u8")
	infile, _ := avutil.Open("rtmp://live.hkstv.hk.lxdns.com/live/hks")
	//infile, _ := avutil.Open("rtmp://localhost/app/publish")
	//infile, _ := avutil.Open("rtsp://184.72.239.149/vod/mp4://BigBuckBunny_175k.mov")
	//infile, _ := avutil.Open("/app/source_file/xiaoxin6.flv")
	streams, _ := infile.Streams()
	/*
		for _, stream := range streams {
			if stream.Type().IsAudio() {
				astream := stream.(av.AudioCodecData)
				fmt.Println(astream.Type(), astream.SampleRate(), astream.SampleFormat(), astream.ChannelLayout())
			} else if stream.Type().IsVideo() {
				vstream := stream.(av.VideoCodecData)
				fmt.Println(vstream.Type(), vstream.Width(), vstream.Height())
			}
		}
	*/
	outfile, _ := avutil.Create("out.mp4")

	fmt.Println("Start recording...")
	var err error
	if err = outfile.WriteHeader(streams); err != nil {
		return
	}

	var start bool = false
	var count int = 0
	var t0 time.Time
	for i := 0; i < 1500; i++ {
		var pkt av.Packet
		if pkt, err = infile.ReadPacket(); err != nil {
			break
		}

		//	fmt.Println("pkt", i, streams[pkt.Idx].Type(), "len", len(pkt.Data), "keyframe", pkt.IsKeyFrame)
		if streams[pkt.Idx].Type() == av.H264 && pkt.IsKeyFrame {
			start = true
			count++
		}

		if start {
			if count == 1 {
				t0 = time.Now()
			}
			outfile.WritePacket(pkt)
		}
	}

	t1 := time.Now()
	if err = outfile.WriteTrailer(); err != nil {
		return
	}

	outfile.Close()
	infile.Close()

	fmt.Printf("Stop recording, took %v to run.\n", t1.Sub(t0))
}