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

from __future__ import with_statement

import os
import re

from module.plugins.internal.Addon import Addon, threaded
from module.plugins.internal.utils import fs_join


class MergeFiles(Addon):
    __name__    = "MergeFiles"
    __type__    = "hook"
    __version__ = "0.18"
    __status__  = "testing"

    __config__ = [("activated", "bool", "Activated", True)]

    __description__ = """Merges parts splitted with hjsplit"""
    __license__     = "GPLv3"
    __authors__     = [("and9000", "me@has-no-mail.com")]


    BUFFER_SIZE = 4096


    @threaded
    def package_finished(self, pack):
        files = {}
        fid_dict = {}
        for fid, data in pack.getChildren().items():
            if re.search("\.\d{3}$", data['name']):
                if data['name'][:-4] not in files:
                    files[data['name'][:-4]] = []
                files[data['name'][:-4]].append(data['name'])
                files[data['name'][:-4]].sort()
                fid_dict[data['name']] = fid

        dl_folder = self.pyload.config.get("general", "download_folder")

        if self.pyload.config.get("general", "folder_per_package"):
            dl_folder = fs_join(dl_folder, pack.folder)

        for name, file_list in files.items():
            self.log_info(_("Starting merging of"), name)

            with open(fs_join(dl_folder, name), "wb") as final_file:
                for splitted_file in file_list:
                    self.log_debug("Merging part", splitted_file)

                    pyfile = self.pyload.files.getFile(fid_dict[splitted_file])

                    pyfile.setStatus("processing")

                    try:
                        with open(fs_join(dl_folder, splitted_file), "rb") as s_file:
                            size_written = 0
                            s_file_size = int(os.path.getsize(os.path.join(dl_folder, splitted_file)))
                            while True:
                                f_buffer = s_file.read(self.BUFFER_SIZE)
                                if f_buffer:
                                    final_file.write(f_buffer)
                                    size_written += self.BUFFER_SIZE
                                    pyfile.setProgress((size_written * 100) / s_file_size)
                                else:
                                    break
                        self.log_debug("Finished merging part", splitted_file)

                    except Exception, e:
                        self.log_error(e, trace=True)

                    finally:
                        pyfile.setProgress(100)
                        pyfile.setStatus("finished")
                        pyfile.release()

            self.log_info(_("Finished merging of"), name)