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

from module.plugins.internal.MultiHoster import MultiHoster
from module.plugins.internal.misc import json


class SmoozedCom(MultiHoster):
    __name__    = "SmoozedCom"
    __type__    = "hoster"
    __version__ = "0.13"
    __status__  = "testing"

    __pattern__ = r'^unmatchable$'  #: Since we want to allow the user to specify the list of hoster to use we let MultiHoster.activate
    __config__  = [("activated"   , "bool", "Activated"                                        , True ),
                   ("use_premium" , "bool", "Use premium account if available"                 , True ),
                   ("fallback"    , "bool", "Fallback to free download if premium fails"       , False),
                   ("chk_filesize", "bool", "Check file size"                                  , True ),
                   ("max_wait"    , "int" , "Reconnect if waiting time is greater than minutes", 10   ),
                   ("revertfailed", "bool", "Revert to standard download if fails"             , True )]

    __description__ = """Smoozed.com hoster plugin"""
    __license__     = "GPLv3"
    __authors__     = [(None, None)]


    FILE_ERRORS = [("Error", r'{"state":"error"}'),
                   ("Retry", r'{"state":"retry"}')]


    def handle_free(self, pyfile):
        #: In some cases hostsers do not supply us with a filename at download, so we
        #: Are going to set a fall back filename (e.g. for freakshare or xfileshare)
        pyfile.name = pyfile.name.split('/').pop()  #: Remove everthing before last slash

        #: Correction for automatic assigned filename: Removing html at end if needed
        suffix_to_remove = ["html", "htm", "php", "php3", "asp", "shtm", "shtml", "cfml", "cfm"]
        temp             = pyfile.name.split('.')

        if temp.pop() in suffix_to_remove:
            pyfile.name = ".".join(temp)

        #: Check the link
        get_data = {'session_key': self.account.get_data('session'),
                    'url'        : pyfile.url}

        html = self.load("http://www2.smoozed.com/api/check", get=get_data)
        data = json.loads(html)

        if data['state'] != "ok":
            self.fail(data['message'])

        if data['data'].get('state', 'ok') != "ok":
            if data['data'] == "Offline":
                self.offline()
            else:
                self.fail(data['data']['message'])

        pyfile.name = data['data']['name']
        pyfile.size = int(data['data']['size'])

        #: Start the download
        header = self.load("http://www2.smoozed.com/api/download", get=get_data, just_header=True)

        if not "location" in header:
            self.fail(_("Unable to initialize download"))
        else:
            self.link = header.get('location')[-1] if isinstance(header.get('location'), list) else header.get('location')
