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

# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.

"""Extractors for https://gofile.io/"""

from .common import Extractor, Message
from .. import exception
from ..cache import memcache
import hashlib
import time


class GofileFolderExtractor(Extractor):
    category = "gofile"
    subcategory = "folder"
    root = "https://gofile.io"
    directory_fmt = ("{category}", "{name} ({code})")
    archive_fmt = "{id}"
    pattern = r"(?:https?://)?(?:www\.)?gofile\.io/d/([^/?#]+)"
    example = "https://gofile.io/d/ID"

    def items(self):
        recursive = self.config("recursive")
        password = self.config("password")

        token = self.config("api-token")
        if not token:
            token = self._create_account()
        self.cookies.set("accountToken", token, domain=".gofile.io")
        self.api_token = token

        folder = self._get_content(self.groups[0], password)
        yield Message.Directory, "", folder

        try:
            contents = folder.pop("children")
        except KeyError:
            raise exception.AuthorizationError("Password required")

        num = 0
        for content in contents.values():
            content["folder"] = folder

            if content["type"] == "file":
                num += 1
                content["num"] = num
                content["filename"], _, content["extension"] = \
                    content["name"].rpartition(".")
                yield Message.Url, content["link"], content

            elif content["type"] == "folder":
                if recursive:
                    url = "https://gofile.io/d/" + content["id"]
                    content["_extractor"] = GofileFolderExtractor
                    yield Message.Queue, url, content
                else:
                    self.log.debug("Skipping subfolder '%s'", content["id"])
            else:
                self.log.debug("'%s' is of unknown type (%s)",
                               content.get("name"), content["type"])

    def request_api(self, endpoint, params=None, headers=None, method="GET"):
        if headers is None:
            headers = {}
        headers["Referer"] = self.root + "/"
        headers["Origin"] = self.root

        response = self.request_json(
            "https://api.gofile.io" + endpoint,
            method=method, params=params, headers=headers)

        if response["status"] != "ok":
            if response["status"] == "error-notFound":
                raise exception.NotFoundError("content")
            if response["status"] == "error-passwordRequired":
                raise exception.AuthorizationError("Password required")
            raise exception.AbortExtraction(
                f"{endpoint} failed (Status: {response['status']})")

        return response["data"]

    @memcache()
    def _create_account(self):
        self.log.debug("Creating temporary account")
        return self.request_api("/accounts", method="POST")["token"]

    def _generate_website_token(self, lang="en-US"):
        # https://gofile.io/dist/js/wt.obf.js
        data = (f"{self.session.headers['User-Agent']}::"
                f"{lang}::"
                f"{self.api_token}::"
                f"{int(time.time() / 14400)}::"
                f"f4s58gs6")
        return hashlib.sha256(data.encode()).hexdigest()

    def _get_content(self, content_id, password=None):
        params = {
            "contentFilter": ""	,
            "page"         : "1",
            "pageSize"     : "1000",
            "sortField"    : "name",
            "sortDirection": "1",
            "password"     : None if password is None else
                             hashlib.sha256(password.encode()).hexdigest(),
        }
        headers = {
            "Authorization"  : "Bearer " + self.api_token,
            "X-Website-Token": self._generate_website_token("en-US"),
            "X-BL"           : "en-US",
        }
        return self.request_api("/contents/" + content_id, params, headers)
