#!/usr/bin/env python3

# This file is a part of Lector, a Qt based ebook reader
# Copyright (C) 2017-18 BasioMeusPuga

# 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, either version 3 of the License, or
# (at your option) any later version.

# 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, see <http://www.gnu.org/licenses/>.

# TODO
# Account for files with passwords

import os
import time
import zipfile
from lector.rarfile import rarfile


class ParseCOMIC:
    def __init__(self, filename, *args):
        self.filename = filename
        self.book = None
        self.image_list = None
        self.book_extension = os.path.splitext(self.filename)

    def read_book(self):
        try:
            if self.book_extension[1] == '.cbz':
                self.book = zipfile.ZipFile(
                    self.filename, mode='r', allowZip64=True)
                self.image_list = [i.filename for i in self.book.infolist() if not i.is_dir()]

            elif self.book_extension[1] == '.cbr':
                self.book = rarfile.RarFile(self.filename)
                self.image_list = [i.filename for i in self.book.infolist() if not i.isdir()]

            self.image_list.sort()
        except:  # Specifying no exception here is warranted
            print('Cannot parse ' + self.filename)
            return

    def get_title(self):
        title = os.path.basename(self.book_extension[0]).strip(' ')
        return title

    def get_author(self):
        return None

    def get_year(self):
        creation_time = time.ctime(os.path.getctime(self.filename))
        creation_year = creation_time.split()[-1]
        return creation_year

    def get_cover_image(self):
        # The first image in the archive may not be the cover
        # It is implied, however, that the first image in order
        # will be the cover
        return self.book.read(self.image_list[0])

    def get_isbn(self):
        return None

    def get_tags(self):
        return None

    def get_contents(self):
        file_settings = {'images_only': True}
        contents = [(f'Page {count + 1}', i) for count, i in enumerate(self.image_list)]

        return contents, file_settings