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

from __future__ import with_statement

import base64
import binascii
import re

from Crypto.Cipher import AES

from module.plugins.Container import Container
from module.utils import fs_encode


class RSDF(Container):
    __name__    = "RSDF"
    __type__    = "container"
    __version__ = "0.26"

    __pattern__ = r'.+\.rsdf$'

    __description__ = """RSDF container decrypter plugin"""
    __license__     = "GPLv3"
    __authors__     = [("RaNaN", "RaNaN@pyload.org"),
                       ("spoob", "spoob@pyload.org")]


    KEY = "8C35192D964DC3182C6F84F3252239EB4A320D2500000000"
    IV  = "FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF"


    def decrypt(self, pyfile):
        Key       = binascii.unhexlify(self.KEY)
        IV_Cipher = AES.new(Key, AES.MODE_ECB)
        IV        = IV_Cipher.encrypt(binascii.unhexlify(self.IV))

        obj = AES.new(Key, AES.MODE_CFB, IV)

        try:
            file = fs_encode(pyfile.url.strip())
            with open(file, 'r') as rsdf:
                data = rsdf.read()

        except IOError, e:
            self.fail(e)

        if re.search(r"<title>404 - Not Found</title>", data):
            return

        for link in binascii.unhexlify(''.join(data.split())).splitlines():
            if not link:
                link         = obj.decrypt(base64.b64decode(link))
                decryptedUrl = link.replace('CCF: ', '')
                self.urls.append(decryptedUrl)