av/protocol/rtsp/cmd/record/main.go

/*
NAME
  record

DESCRIPTION
  record provides a program to connect to an RTSP server, request for an
  RTP stream and then save the RTP payload to file.

AUTHORS
  Saxon A. Nelson-Milton <saxon@ausocean.org>

LICENSE
  This is Copyright (C) 2019 the Australian Ocean Lab (AusOcean).

  It is free software: you can redistribute it and/or modify them
  under the terms of the GNU General Public License as published by the
  Free Software Foundation, either version 3 of the License, or (at your
  option) any later version.

  It is distributed in the hope that it will be useful, but WITHOUT
  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
  for more details.

  You should have received a copy of the GNU General Public License
  in gpl.txt.  If not, see http://www.gnu.org/licenses.
*/
package main

import (
	"flag"
	"fmt"
	"log"

	"bitbucket.org/ausocean/av/protocol/rtsp"
)

func main() {
	rtspServerPtr := flag.String("rtsp-server", "", "The RTSP server we would like to get video from.")
	clientPortPtr := flag.Uint("port", 6870, "The port on the client we would like to receive RTP on.")
	trackPtr := flag.String("track", "track1", "The track that we would like to receive media from.")
	flag.Parse()

	clt, err := rtsp.NewClient(*rtspServerPtr)
	if err != nil {
		panic(fmt.Sprintf("creating RTSP client failed with error: %v", err))
	}

	resp, err := clt.Options()
	if err != nil {
		log.Fatalln(err)
	}
	fmt.Println("Options:")
	fmt.Println(resp)

	resp, err = clt.Describe()
	if err != nil {
		log.Fatalln(err)
	}
	fmt.Println("Describe:")
	fmt.Println(resp)

	resp, err = clt.Setup(*trackPtr, fmt.Sprintf("RTP/AVP;unicast;client_port=%d-%d", *clientPortPtr, *clientPortPtr+1))
	if err != nil {
		log.Fatalln(err)
	}
	log.Println(resp)

	resp, err = clt.Play()
	if err != nil {
		log.Fatalln(err)
	}
	log.Println(resp)

	// TODO(saxon): use RTCP client here to maintain stream.
	for {
	}
}