package main

	import (
		"log"
		"os"

		"github.com/bluenviron/gortsplib/v3"
		"github.com/bluenviron/gortsplib/v3/pkg/formats"
		"github.com/bluenviron/gortsplib/v3/pkg/url"
		"github.com/pion/rtp"
		"github.com/youpy/go-wav"
	)

	// This example shows how to
	// 1. connect to a RTSP server
	// 2. check if there's a G711 media
	// 3. get G711 frames of that media

	func main() {
		c := gortsplib.Client{}

		// parse URL
		u, err := url.Parse("rtsp://xxx:554")
		if err != nil {
			panic(err)
		}

		// connect to the server
		err = c.Start(u.Scheme, u.Host)
		if err != nil {
			panic(err)
		}
		defer c.Close()

		// find published medias
		medias, baseURL, _, err := c.Describe(u)
		if err != nil {
			panic(err)
		}

		// find the G711 media and format
		var forma *formats.G711
		medi := medias.FindFormat(&forma)
		if medi == nil {
			panic("media not found")
		}

		// create decoder
		rtpDec := forma.CreateDecoder()

		// setup a single media
		_, err = c.Setup(medi, baseURL, 0, 0)
		if err != nil {
			panic(err)
		}

		// Create PCM file
		f, err := os.Create("out.pcm")
		if err != nil {
			panic(err)
		}
		defer f.Close()

		// Create a WAV file
		f2, err := os.Create("out.wav")
		if err != nil {
			panic(err)
		}
		defer f2.Close()

		// Use go-wav to encde the PCM file to WAV
                     //
		wavOut := wav.NewWriter(f2, 0, 1, uint32(rtpDec.SampleRate), 16)

		// called when a RTP packet arrives
		c.OnPacketRTP(medi, forma, func(pkt *rtp.Packet) {
			// extract G711 frames from RTP packets
			op, _, err := rtpDec.Decode(pkt)
			if err != nil {
				log.Printf("ERR: %v", err)
				return
			}

			// write to wav
			wavOut.Write(op)

			// print
			log.Printf("received G711 frame of size %d\n", len(op))
		})

		// start playing
		_, err = c.Play(nil)
		if err != nil {
			panic(err)
		}

		// wait until a fatal error
		panic(c.Wait())
	}