#!/usr/bin/env python

# -*- coding: utf8 -*-

u"""

Arte-DL

-------

Downloads a video from the french-german TV channel ARTE.

Dependencies:

- ``rtmpdump`` to download the video stream.

.. :Authors:

       Aurélien Bompard <aurelien@bompard.org> <http://aurelien.bompard.org>

.. :License:

       GNU GPL v3 or later

"""

import sys

import os

import re

import urllib2

from xml.etree import ElementTree as etree

import subprocess

from optparse import OptionParser

MAX_TRIES = 10

def get_video_id(url):

    match = re.match("http://videos\.arte\.tv/(\w\w)/videos/([^/]+-\d+)\.html", url)

    if not match:

        print >>sys.stderr, "Can't find the video ID in the URL"

        if options.debug:

            print url

        sys.exit(1)

    return match.group(1), match.group(2)

def get_video_info(lang, basename):

    url = "http://videos.arte.tv/%s/do_delegate/videos/%s,view,asPlayerXml.xml" % (lang, basename)

    videoref = etree.parse(urllib2.urlopen(url))

    video_info_tag = [ t for t in videoref.findall("videos/video")

                               if t.get("lang") == lang ]

    if not video_info_tag:

        print >>sys.stderr, "Can't find the video info URL in the videoref info XML"

        if options.debug:

            etree.dump(videoref)

        sys.exit(1)

    video_info_url = video_info_tag[0].get("ref")

    video_info = etree.parse(urllib2.urlopen(video_info_url))

    return video_info

def get_video_url(video_info, quality):

    url_tag = [ t for t in video_info.findall("urls/url") if t.get("quality") == quality.lower() ]

    if not url_tag:

        print >>sys.stderr, "Can't find the video URL in the video info XML"

        if options.debug:

            etree.dump(video_info)

        sys.exit(1)

    return url_tag[0].text.strip()

def get_video_name(video_info):

    return video_info.findtext("name").strip()

def download_video(url, filename):

    command = ["rtmpdump", "-r", url, "-o", filename]

    if os.path.exists(filename):

        command.insert(1, "--resume")

    for arg in command:

        if arg.count(" ") or arg.count("&"):

            print '"%s"' % arg,

        else:

            print arg,

    print

    if options.debug:

        return

    retcode = subprocess.call(command)

    return retcode

def main():

    global options

    usage = "%prog [options] URL"

    parser = OptionParser(usage)

    parser.add_option("-q", "--quality", dest="quality",

                      default="HD", metavar="SD/HD",

                      help="quality (SD or HD, defaults to %default)")

    parser.add_option("-o", "--output", dest="output", metavar="FILENAME",

                      help="filename for the downloaded video")

    parser.add_option("-d", "--debug", action="store_true",

                      dest="debug", default=False,

                      help="only display which command would be run")

    options, args = parser.parse_args()

    if not len(args) > 0:

        parser.error("Missing URL")

    url = args[0]

    lang, basename = get_video_id(url)

    video_info = get_video_info(lang, basename)

    video_url = get_video_url(video_info, options.quality)

    video_name = get_video_name(video_info)

    if options.debug:

        print "URL: ", video_url

    if options.output:

        filename = options.output

        if not filename.endswith(".flv"):

            filename += ".flv"

    else:

        filename = "Arte - %s.flv" % video_name

        print u"Downloading video %s in file %s." \

                % (video_name, filename)

        print "Choose a new filename, or leave empty to accept the default filename."

        newfilename = raw_input("--> ")

        if newfilename:

            filename = newfilename

    retcode = download_video(video_url, filename)

    # rtmpdump returns 2 if the download is incomplete

    current_try = 1

    while retcode == 2:

        print "Trying again..."

        retcode = download_video(video_url, filename)

        current_try += 1

        if current_try > MAX_TRIES:

            print "Too many tries, aborting."

            break

if __name__ == "__main__":

    main()