# -*- coding: utf-8 -*-

"""
Copyright(C) 2010  Christophe Benz, Romain Bignon

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, version 3 of the License.

This program 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
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

"""

from weboob.capabilities.video import ICapVideoProvider
from weboob.tools.application import ConsoleApplication

class Videoob(ConsoleApplication):
    APPNAME = 'videoob'
    CONFIG = {}

    def main(self, argv):
        self.weboob.load_modules(ICapVideoProvider, names=self.enabled_backends)
        return self.process_command(*argv[1:])

    @ConsoleApplication.command('Get video information')
    def command_info(self, _id):
        results = {}
        for backend in self.weboob.iter_backends():
            video = backend.get_video(_id)
            if video is None:
                continue
            result = []
            if video.title:
                result.append(('Video title', video.title))
            if video.duration:
                result.append(('Duration', '%d:%02d:%02d' % (
                    video.duration / 3600, (video.duration % 3600) / 60, video.duration % 60)))
            if video.url:
                result.append(('URL', video.url))
            if video.author:
                result.append(('Author', video.author))
            if video.date:
                result.append(('Date', video.date))
            if video.rating_max:
                result.append(('Rating', '%s / %s' % (video.rating, video.rating_max)))
            elif video.rating:
                result.append(('Rating', video.rating))
            results[backend.name] = result
        return results

    @ConsoleApplication.command('Search videos')
    def command_search(self, pattern=None):
        results = {}
        if pattern:
            results['HEADER'] = u'Search pattern: %s' % pattern
        else:
            results['HEADER'] = u'Last videos'
        for backend in self.weboob.iter_backends():
            result = []
            try:
                iterator = backend.iter_search_results(pattern)
            except NotImplementedError:
                continue
            else:
                for video in iterator:
                    result.append(('ID', video.id))
                    result.append(('Title', video.title))
            results[backend.name] = result
        return results

    @ConsoleApplication.command('Get video file URL from page URL')
    def command_file_url(self, url):
        for backend in self.weboob.iter_backends():
            video = backend.get_video(url)
            if video:
                print video.url
                break