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

import pycurl

from module.plugins.internal.Addon import Addon
from module.plugins.internal.utils import encode


class UserAgentSwitcher(Addon):
    __name__    = "UserAgentSwitcher"
    __type__    = "hook"
    __version__ = "0.12"
    __status__  = "testing"

    __config__ = [("activated"     , "bool", "Activated"                                 , True                                                                      ),
                  ("connecttimeout", "int" , "Max timeout for link connection in seconds", 60                                                                        ),
                  ("maxredirs"     , "int" , "Maximum number of redirects to follow"     , 10                                                                        ),
                  ("useragent"     , "str" , "Custom user-agent string"                  , "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:38.0) Gecko/20100101 Firefox/38.0")]

    __description__ = """Custom user-agent"""
    __license__     = "GPLv3"
    __authors__     = [("Walter Purcaro", "vuolter@gmail.com")]


    def download_preparing(self, pyfile):
        connecttimeout = self.get_config('connecttimeout')
        maxredirs      = self.get_config('maxredirs')
        useragent      = self.get_config('useragent')

        if connecttimeout:
            pyfile.plugin.req.http.c.setopt(pycurl.CONNECTTIMEOUT, connecttimeout)

        if maxredirs:
            pyfile.plugin.req.http.c.setopt(pycurl.MAXREDIRS, maxredirs)

        if useragent:
            self.log_debug("Use custom user-agent string `%s`" % useragent)
            pyfile.plugin.req.http.c.setopt(pycurl.USERAGENT, encode(useragent))