/* Author: Ian Collier <imc@comlab.ox.ac.uk>
   Licence: GPL <http://www.gnu.org/copyleft/gpl.html>
   Support: None whatsoever - please don't write asking for help

   This program is a hack and may or may not work.  Use at your own risk.
*/

#include <stdio.h>
#include <string.h>
#include "librtsp/rtsp_session.h"

int main (int argc, char **argv) {
   rtsp_session_t *rtsp;
   char *mrl=0;
   char *file=0;
   FILE *fp;
   static char buffer[8192];
   int r;
   int w;
   int v=1;
   unsigned long t=0;

   for (r=1; r<argc; r++) {
      if (argv[r][0]=='-') switch(argv[r][1]) {
         case 'q': v--; break;
         case 'v': v++; break;
         case 'o':
            if (file) {
               fputs("Extra parameter or redundant -o flag\n",stderr);
               exit(1);
            }
            if (r+1<argc) file=argv[++r];
            else if (mrl) {
               file=strrchr(mrl,'/');
               if (file) file++;
               else file=mrl;
            }
            break;
         default: fprintf(stderr, "Unrecognised flag `%s'\n", argv[r]);
            exit(1);
      }
      else if (!mrl) mrl=argv[r];
      else if (!file) file=argv[r];
      else {
         fputs("Too many parameters\n",stderr);
         exit(1);
      }
   }
   if (!mrl || strncasecmp (mrl, "rtsp://", 6)) {
      fputs("Please supply a MRL starting with rtsp://\n",stderr);
      exit(1);
   }
   if (file) {
      fp=fopen(file,"w");
      if (!fp) {
         perror(file);
         fputs("Error opening output file\n",stderr);
         exit(2);
      }
   }
   else {
      file="stdout";
      fp=stdout;
   }

   if (v>1) fprintf(stderr, "About to try dumping %s to %s\n", mrl, file);

   rtsp = rtsp_session_start(mrl);
   if (!rtsp) {
      fputs("RTSP session did not start for some reason\n",stderr);
      exit(3);
   }
   if (v>1) fputs("RTSP session successfully started\n", stderr);

   if (v) fprintf(stderr, "%9lu bytes read\r", 0L);
   while ((r=rtsp_session_read(rtsp, buffer, sizeof buffer))>0) {
      t+=r;
      if (r<sizeof buffer && v>1)
         fprintf(stderr, " short read of %d bytes (total so far %lu)\n", r, t);
      w=fwrite(buffer, 1, r, fp);
      if (w<r) {
         perror(file);
         fprintf(stderr, "short write (%d of %d bytes written)\n", w, r);
         exit(2);
      }
      if (v) fprintf(stderr, "%9lu bytes read\r", t);
   }
   if (v) fputc('\n',stderr);
   if (fclose(fp)) {
      perror(file);
      fputs("(error while closing file - may not be complete)\n", stderr);
      exit(2);
   }

   rtsp_session_end (rtsp);
   if (v>1) fputs("Finished successfully\n",stderr);
   exit(0);
}