__init__.py

"""
Copyright 2022 Sketchfab

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    https://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""

import os
import urllib
import requests
import threading
import time
from collections import OrderedDict
import subprocess
import tempfile
import json
import shutil
from uuid import UUID

import bpy
import bpy.utils.previews
from bpy.props import (StringProperty,
                       EnumProperty,
                       BoolProperty,
                       IntProperty,
                       PointerProperty)

bl_info = {
    'name': 'Sketchfab Plugin',
    'description': 'Browse and download free Sketchfab downloadable models',
    'author': 'Sketchfab',
    'license': 'APACHE2',
    'deps': '',
    'version': (1, 6, 1),
    "blender": (2, 80, 0),
    'location': 'View3D > Tools > Sketchfab',
    'warning': '',
    'wiki_url': 'https://github.com/sketchfab/blender-plugin/releases',
    'tracker_url': 'https://github.com/sketchfab/blender-plugin/issues',
    'link': 'https://github.com/sketchfab/blender-plugin',
    'support': 'COMMUNITY',
    'category': 'Import-Export'
}
bl_info['blender'] = getattr(bpy.app, "version")


PLUGIN_VERSION = str(bl_info['version']).strip('() ').replace(',', '.')
preview_collection = {}
thumbnailsProgress = set([])
ongoingSearches = set([])
is_plugin_enabled = False


class Config:

    ADDON_NAME = 'io_sketchfab'
    GITHUB_REPOSITORY_URL = 'https://github.com/sketchfab/blender-plugin'
    GITHUB_REPOSITORY_API_URL = 'https://api.github.com/repos/sketchfab/blender-plugin'
    SKETCHFAB_REPORT_URL = 'https://help.sketchfab.com/hc/en-us/requests/new?type=exporters&subject=Blender+Plugin'

    SKETCHFAB_URL = 'https://sketchfab.com'
    CLIENTID = 'hGC7unF4BHyEB0s7Orz5E1mBd3LluEG0ILBiZvF9'
    SKETCHFAB_OAUTH = SKETCHFAB_URL + '/oauth2/token/'
    SKETCHFAB_API = 'https://api.sketchfab.com'
    SKETCHFAB_SEARCH = SKETCHFAB_API + '/v3/search'
    SKETCHFAB_MODEL = SKETCHFAB_API + '/v3/models'
    SKETCHFAB_ORGS = SKETCHFAB_API + '/v3/orgs'
    SKETCHFAB_# = 'https://sketchfab.com/#'

    # BASE_SEARCH = SKETCHFAB_SEARCH + '?type=models&downloadable=true'
    BASE_SEARCH = SKETCHFAB_SEARCH + '?type=models'
    DEFAULT_FLAGS = '&staffpicked=true&sort_by=-staffpickedAt'
    DEFAULT_SEARCH = SKETCHFAB_SEARCH + \
                     '?type=models' + \
                     '&restricted=1' + \
                     '&downloadable=true' + \
                     DEFAULT_FLAGS

    SKETCHFAB_ME = '{}/v3/me'.format(SKETCHFAB_API)
    # BASE_SEARCH_OWN_MODELS = SKETCHFAB_ME + '/search?type=models&downloadable=true'
    BASE_SEARCH_OWN_MODELS = SKETCHFAB_ME + '/search?type=models'
    PURCHASED_MODELS = SKETCHFAB_ME + "/models/purchases?"

    SKETCHFAB_PLUGIN_VERSION = '{}/releases'.format(GITHUB_REPOSITORY_API_URL)

    # Those will be set during plugin initialization, or upon setting a new cache directory
    SKETCHFAB_TEMP_DIR = ""
    SKETCHFAB_THUMB_DIR = ""
    SKETCHFAB_MODEL_DIR = ""

    SKETCHFAB_CATEGORIES = (('ALL', 'All categories', 'All categories'),
                            ('animals-pets', 'Animals & Pets', 'Animals and Pets'),
                            ('architecture', 'Architecture', 'Architecture'),
                            ('art-abstract', 'Art & Abstract', 'Art & Abstract'),
                            ('cars-vehicles', 'Cars & vehicles', 'Cars & vehicles'),
                            ('characters-creatures', 'Characters & Creatures', 'Characters & Creatures'),
                            ('cultural-heritage-history', 'Cultural Heritage & History', 'Cultural Heritage & History'),
                            ('electronics-gadgets', 'Electronics & Gadgets', 'Electronics & Gadgets'),
                            ('fashion-style', 'Fashion & Style', 'Fashion & Style'),
                            ('food-drink', 'Food & Drink', 'Food & Drink'),
                            ('furniture-home', 'Furniture & Home', 'Furniture & Home'),
                            ('music', 'Music', 'Music'),
                            ('nature-plants', 'Nature & Plants', 'Nature & Plants'),
                            ('news-politics', 'News & Politics', 'News & Politics'),
                            ('people', 'People', 'People'),
                            ('places-travel', 'Places & Travel', 'Places & Travel'),
                            ('science-technology', 'Science & Technology', 'Science & Technology'),
                            ('sports-fitness', 'Sports & Fitness', 'Sports & Fitness'),
                            ('weapons-military', 'Weapons & Military', 'Weapons & Military'))

    SKETCHFAB_FACECOUNT = (('ANY', "All", ""),
                           ('10', "Up to 10", ""),
                           ('100', "Up to 100", ""),
                           ('1K', "Up to 1k", ""),
                           ('10K', "Up to 10k", ""),
                           ('50K', "10k to 50k", ""),
                           ('100K', "50k to 100k", ""),
                           ('250K', "100k to 250k", ""),
                           ('250KP', "250k +", ""))

    SKETCHFAB_SORT_BY = (('RELEVANCE', "Relevance", ""),
                         ('LIKES', "Likes", ""),
                         ('VIEWS', "Views", ""),
                         ('RECENT', "Recent", ""))

    SKETCHFAB_SEARCH_DOMAIN = (('DEFAULT', "All site", "", 0),
                               ('OWN', "Own Models (PRO)", "", 1),
                               ('STORE', "Store purchases", "", 2))

    MAX_THUMBNAIL_HEIGHT = 256

    SKETCHFAB_UPLOAD_LIMITS = {
        "basic" : 100 * 1024 * 1024,
        "pro": 200 * 1024 * 1024,
        "prem": 500 * 1024 * 1024,
        "ent": 500 * 1024 * 1024
    }

class Utils:
    def humanify_size(size):
        suffix = 'B'
        readable = size

        # Megabyte
        if size > 1048576:
            suffix = 'MB'
            readable = size / 1048576.0
        # Kilobyte
        elif size > 1024:
            suffix = 'KB'
            readable = size / 1024.0

        readable = round(readable, 2)
        return '{}{}'.format(readable, suffix)

    def humanify_number(number):
        suffix = ''
        readable = number

        if number > 1000000:
            suffix = 'M'
            readable = number / 1000000.0

        elif number > 1000:
            suffix = 'K'
            readable = number / 1000.0

        readable = round(readable, 2)
        return '{}{}'.format(readable, suffix)

    def build_download_url(uid, use_org_profile=False, active_org=None):
        if use_org_profile:
            return '{}/{}/models/{}/download'.format(Config.SKETCHFAB_ORGS, active_org["uid"], uid)
        else:
            return '{}/{}/download'.format(Config.SKETCHFAB_MODEL, uid)

    def thumbnail_file_exists(uid):
        return os.path.exists(os.path.join(Config.SKETCHFAB_THUMB_DIR, '{}.jpeg'.format(uid)))

    def clean_thumbnail_directory():
        if not os.path.exists(Config.SKETCHFAB_THUMB_DIR):
            return

        from os import listdir
        for file in listdir(Config.SKETCHFAB_THUMB_DIR):
            os.remove(os.path.join(Config.SKETCHFAB_THUMB_DIR, file))

    def clean_downloaded_model_dir(uid):
        shutil.rmtree(os.path.join(Config.SKETCHFAB_MODEL_DIR, uid))

    def get_thumbnail_url(thumbnails_json):
        min_height  = 1e6
        min_thumbnail = None
        best_height = 0
        best_thumbnail = None
        for image in thumbnails_json['images']:
            h = image['height']
            if h <= Config.MAX_THUMBNAIL_HEIGHT and h > best_height:
                best_height = h
                best_thumbnail = image['url']
            elif h < min_height:
                min_height = h
                min_thumbnail = image['url']
        # Ensure we have a thumbnail if available thumbnails are all above MAX_THUMBNAIL_HEIGHT
        if best_thumbnail is None and min_thumbnail is not None:
            return min_thumbnail
        return best_thumbnail

    def setup_plugin():
        if not os.path.exists(Config.SKETCHFAB_THUMB_DIR):
            os.makedirs(Config.SKETCHFAB_THUMB_DIR)

    def get_uid_from_thumbnail_url(thumbnail_url):
        return thumbnail_url.split('/')[4]

    def get_uid_from_model_url(model_url, use_org_profile=False):
        try:
            return model_url.split('/')[7] if use_org_profile else model_url.split('/')[5]
        except:
            ShowMessage("ERROR", "Url parsing error", "Error getting uid from url: {}".format(model_url))
            return None

    def get_uid_from_download_url(model_url):
        return model_url.split('/')[6]

    def clean_node_hierarchy(objects, root_name):
        """
        Removes the useless nodes in a hierarchy
        TODO: Keep the transform (might impact Yup/Zup)
        """
        # Find the parent object
        root = None
        for object in objects:
            if object.parent is None:
                root = object
        if root is None:
            return None

        # Go down its hierarchy until one child has multiple children, or a single mesh
        # Keep the name while deleting objects in the hierarchy
        diverges = False
        while diverges==False:
            children = root.children
            if children is not None:

                if len(children)>1:
                    diverges = True
                    root.name = root_name

                if len(children)==1:
                    if children[0].type != "EMPTY":
                        diverges = True
                        root.name = root_name
                        if children[0].type == "MESH": # should always be the case
                            matrixcopy = children[0].matrix_world.copy()
                            children[0].parent = None
                            children[0].matrix_world = matrixcopy
                            bpy.data.objects.remove(root)
                            children[0].name = root_name
                            root = children[0]

                    elif children[0].type == "EMPTY":
                        diverges = False
                        matrixcopy = children[0].matrix_world.copy()
                        children[0].parent = None
                        children[0].matrix_world = matrixcopy
                        bpy.data.objects.remove(root)
                        root = children[0]
            else:
                break

        # Select the root Empty node
        root.select_set(True)

    def is_valid_uuid(uuid_to_test, version=4):
        try:
            uuid_obj = UUID(hex=uuid_to_test, version=version)
            return True
        except ValueError:
            return False

class Cache:
    SKETCHFAB_CACHE_FILE = os.path.join(
        bpy.utils.user_resource("SCRIPTS", path="sketchfab_cache", create=True),
        ".cache"
    ) # Use a user path to avoid permission-related errors

    def read():
        if not os.path.exists(Cache.SKETCHFAB_CACHE_FILE):
            return {}

        with open(Cache.SKETCHFAB_CACHE_FILE, 'rb') as f:
            data = f.read().decode('utf-8')
            return json.loads(data)

    def get_key(key):
        cache_data = Cache.read()
        if key in cache_data:
            return cache_data[key]

    def save_key(key, value):
        cache_data = Cache.read()
        cache_data[key] = value
        with open(Cache.SKETCHFAB_CACHE_FILE, 'wb+') as f:
            f.write(json.dumps(cache_data).encode('utf-8'))

    def delete_key(key):
        cache_data = Cache.read()
        if key in cache_data:
            del cache_data[key]

        with open(Cache.SKETCHFAB_CACHE_FILE, 'wb+') as f:
            f.write(json.dumps(cache_data).encode('utf-8'))


# helpers
def get_sketchfab_login_props():
    return bpy.context.window_manager.sketchfab_api


def get_sketchfab_props():
    return bpy.context.window_manager.sketchfab_browser


def get_sketchfab_props_proxy():
    return bpy.context.window_manager.sketchfab_browser_proxy

def get_sketchfab_model(uid):
    skfb = get_sketchfab_props()
    if "current" in skfb.search_results and uid in skfb.search_results["current"]:
        return skfb.search_results['current'][uid]
    else:
        return None

def run_default_search():
    searchthr = GetRequestThread(Config.DEFAULT_SEARCH, parse_results)
    searchthr.start()


def get_plugin_enabled():
    global is_plugin_enabled
    return is_plugin_enabled


def refresh_search(self, context):
    pprops = get_sketchfab_props_proxy()
    if pprops.is_refreshing:
        return

    props = get_sketchfab_props()

    if pprops.search_domain != props.search_domain:
        props.search_domain = pprops.search_domain
    if pprops.sort_by != props.sort_by:
        props.sort_by = pprops.sort_by

    if 'current' in props.search_results:
        del props.search_results['current']

    props.query = pprops.query
    props.animated = pprops.animated
    props.pbr = pprops.pbr
    props.staffpick = pprops.staffpick
    props.restricted = pprops.restricted
    props.downloadable = pprops.downloadable
    props.categories = pprops.categories
    props.face_count = pprops.face_count
    bpy.ops.wm.sketchfab_search('EXEC_DEFAULT')


def set_login_status(status_type, status):
    login_props = get_sketchfab_login_props()
    login_props.status = status
    login_props.status_type = status_type


def set_import_status(status):
    props = get_sketchfab_props()
    props.import_status = status


class SketchfabApi:
    def __init__(self):
        self.access_token = ''
        self.api_token = ''
        self.headers = {}
        self.username = ''
        self.display_name = ''
        self.plan_type = ''
        self.next_results_url = None
        self.prev_results_url = None
        self.user_orgs = []
        self.user_has_orgs = False
        self.active_org = None
        self.use_org_profile = False

    def build_headers(self):
        if self.access_token:
            self.headers = {'Authorization': 'Bearer ' + self.access_token}
        elif self.api_token:
            self.headers = {'Authorization': 'Token ' + self.api_token}
        else:
            print("Empty authorization header")
            self.headers = {}

    def login(self, email, password, api_token):
        bpy.ops.wm.login_modal('INVOKE_DEFAULT')

    def is_user_logged(self):
        if (self.access_token or self.api_token) and self.headers:
            return True

        return False

    def is_user_pro(self):
        return len(self.plan_type) and self.plan_type not in ['basic', 'plus']

    def logout(self):
        self.access_token = ''
        self.api_token = ''
        self.headers = {}
        Cache.delete_key('username')
        Cache.delete_key('access_token')
        Cache.delete_key('api_token')
        Cache.delete_key('key')

        props = get_sketchfab_props()
        #props.search_domain = "DEFAULT"
        if 'current' in props.search_results:
            del props.search_results['current']
        pprops = get_sketchfab_props_proxy()
        #pprops.search_domain = "DEFAULT"

        self.user_orgs = []
        self.user_has_orgs = False
        self.active_org = None
        self.use_org_profile = False
        props.use_org_profile = False
        pprops.use_org_profile = False

        bpy.ops.wm.sketchfab_search('EXEC_DEFAULT')

    def request_user_info(self):
        requests.get(Config.SKETCHFAB_ME, headers=self.headers, hooks={'response': self.parse_user_info})

    def get_user_info(self):
        if self.display_name and self.plan_type:
            return '{} ({})'.format(self.display_name, self.plan_type)
        else:
            return ('', '')

    def parse_user_info(self, r, *args, **kargs):
        if r.status_code == 200:
            user_data = r.json()
            self.username = user_data['username']
            self.display_name = user_data['displayName']
            self.plan_type = user_data['account']
            requests.get(Config.SKETCHFAB_ME + "/orgs", headers=self.headers, hooks={'response': self.on_user_orgs_check})
        else:
            print('\nInvalid access or API token\nYou can get your API token here:\nhttps://sketchfab.com/settings/password\n')
            set_login_status('ERROR', 'Failed to authenticate')
            ShowMessage("ERROR", "Failed to authenticate", "Invalid access or API token")
            self.access_token = ''
            self.api_token = ''
            self.headers = {}

    def request_user_orgs(self):
        if not self.active_org:
            requests.get(Config.SKETCHFAB_ME + "/orgs", headers=self.headers, hooks={'response': self.parse_orgs_info})
            pass

    def on_user_orgs_check(self, r, *args, **kargs):
        self.user_has_orgs = bool((r.status_code == 200) and len(r.json().get("results", [])))

    def parse_orgs_info(self, r, *args, **kargs):
        """
        Get and store information about user's orgs, and its orgs projects
        """
        if r.status_code == 200:
            orgs_data = r.json()

            # Get a list of the user's orgs
            for org in orgs_data["results"]:
                self.user_orgs.append({
                    "uid":         org["uid"],
                    "displayName": org["displayName"],
                    "username":    org["username"],
                    "url":         org["publicProfileUrl"],
                    "projects":    [],
                })
            self.user_orgs.sort(key = lambda x : x["displayName"])

            # Iterate on the orgs to get lists of their projects
            for org in self.user_orgs:

                # Create the callback inline to keep a reference to the org uid
                def parse_projects_info(r, *args, **kargs):
                    """
                    Get and store information about an org projects
                    """
                    if r.status_code == 200:
                        projects_data = r.json()
                        projects = projects_data["results"]
                        # Add the projects to the orgs dict object
                        for proj in projects:
                            org_uid = proj["org"]["uid"]
                            org = next((x for x in self.user_orgs if x["uid"] == org_uid))
                            org["projects"].append({
                                "uid":         proj["uid"],
                                "name":        proj["name"],
                                "slug":        proj["slug"],
                                "modelCount":  proj["modelCount"],
                                "memberCount": proj["memberCount"],
                            })
                        org["projects"].sort(key = lambda x : x["name"])

                        # Iterate on all projects (not just the 24 first)
                        if projects_data["next"] is not None:
                            requests.get(
                                projects_data["next"],
                                headers=self.headers,
                                hooks={'response': parse_projects_info}
                            )

                    else:
                        print('Can not get projects info')

                requests.get("%s/%s/projects" % (Config.SKETCHFAB_ORGS, org["uid"]),
                    headers=self.headers,
                    hooks={'response': parse_projects_info})

            # Set the first org as active
            if len(self.user_orgs):
                self.active_org = self.user_orgs[0]
                self.user_has_orgs = True

            # Iterate on all orgs (not just the 24 first)
            if orgs_data["next"] is not None:
                requests.get(orgs_data["next"], headers=self.headers, hooks={'response': self.parse_orgs_info})

    def request_thumbnail(self, thumbnails_json, model_uid):
        # Avoid requesting twice the same data
        if model_uid not in thumbnailsProgress:
            thumbnailsProgress.add(model_uid)
            url = Utils.get_thumbnail_url(thumbnails_json)
            thread = ThumbnailCollector(url)
            thread.start()

    def request_model_info(self, uid, callback=None):
        callback = self.handle_model_info if callback is None else callback
        url = Config.SKETCHFAB_MODEL + '/' + uid
        if self.use_org_profile and self.active_org.get("uid"):
            url = Config.SKETCHFAB_ORGS + "/" + self.active_org["uid"] + "/models/" + uid

        model_infothr = GetRequestThread(url, callback, self.headers)
        model_infothr.start()

    def handle_model_info(self, r, *args, **kwargs):
        skfb = get_sketchfab_props()
        uid = Utils.get_uid_from_model_url(r.url, self.use_org_profile)

        # Dirty fix to avoid processing obsolete result data
        if 'current' not in skfb.search_results or uid is None or uid not in skfb.search_results['current']:
            return

        model = skfb.search_results['current'][uid]
        json_data = r.json()
        model.license = json_data.get('license', {})
        if model.license is not None:
            model.license = model.license.get('fullName', 'Personal (you own this model)')
        anim_count = int(json_data.get('animationCount', 0))
        model.animated = 'Yes ({} animation(s))'.format(anim_count) if anim_count > 0 else 'No'
        skfb.search_results['current'][uid] = model

    def search(self, query, search_cb):
        skfb = get_sketchfab_props()
        url = Config.BASE_SEARCH
        if skfb.search_domain == "OWN":
            url = Config.BASE_SEARCH_OWN_MODELS
        elif skfb.search_domain == "STORE":
            url = Config.PURCHASED_MODELS
        elif skfb.search_domain == "ACTIVE_ORG":
            url = Config.SKETCHFAB_ORGS + "/%s/models?isArchivesReady=true" % self.active_org["uid"]
        elif len(skfb.search_domain) == 32:
            url = Config.SKETCHFAB_ORGS + "/%s/models?isArchivesReady=true&projects=%s" % (self.active_org["uid"], skfb.search_domain)

        search_query = '{}{}'.format(url, query)
        print(f"Search query a:\n{search_query}")
        if search_query not in ongoingSearches:
            ongoingSearches.add(search_query)
            searchthr = GetRequestThread(search_query, search_cb, self.headers)
            searchthr.start()

    def search_cursor(self, url, search_cb):
        requests.get(url, headers=self.headers, hooks={'response': search_cb})

    def write_model_info(self, title, author, authorUrl, license, uid):
        try:
            downloadHistory = bpy.context.preferences.addons[__name__.split('.')[0]].preferences.downloadHistory
            if downloadHistory != "":
                downloadHistory = os.path.abspath(downloadHistory)
                createFile = False
                if not os.path.exists(downloadHistory):
                    createFile = True
                with open(downloadHistory, 'a+') as f:
                    if createFile:
                        f.write("Model name, Author name, Author url, License, Model link,\n")
                    f.write("{}, {}, https://sketchfab.com/{}, {}, https://sketchfab.com/models/{},\n".format(
                        title.replace(",", " "),
                        author.replace(",", " "),
                        authorUrl.replace(",", " "),
                        license.replace(",", " "),
                        uid
                    ))
        except:
            print("Error encountered while saving data to history file")

    def parse_model_info_request(self, r, *args, **kargs):
        try:
            if r.status_code == 200:
                result = r.json()
                title = result['name']
                author = result['user']['displayName']
                username = result['user']['username']
                license = result["license"]["label"]
                uid = result['uid']
                self.write_model_info(title, author, username, license, uid)
            else:
                print("Error encountered while getting model info ({})\n{}\n{}".format(r.status_code, r.url, str(r.json())))
        except:
            print("Error encountered while parsing model info request: {}".format(r.url))

    def download_model(self, uid):
        skfb_model = get_sketchfab_model(uid)
        if skfb_model is not None: # The model comes from the search results
            if skfb_model.download_url and (time.time() - skfb_model.time_url_requested < skfb_model.url_expires):
                self.get_archive(skfb_model.download_url)
            else:
                skfb_model.download_url = None
                skfb_model.url_expires = None
                skfb_model.time_url_requested = None
                self.write_model_info(skfb_model.title, skfb_model.author, skfb_model.username, skfb_model.license, uid)
                original_URL = Utils.build_download_url(uid, self.use_org_profile, self.active_org)
                print(f"Downloading URL:\n{original_URL}")
                requests.get(original_URL, headers=self.headers, hooks={'response': self.handle_download})
        else: # Model comes from a direct link
            skfb = get_sketchfab_props()
            download_url = ""

            # If the model is in an org, find if the user has access to it
            if "/orgs/" in skfb.manualImportPath:
                try:
                    orgName = skfb.manualImportPath.split("/orgs/")[1].split("/")[0]
                    if skfb.skfb_api.user_has_orgs and not skfb.skfb_api.active_org:
                        skfb.skfb_api.request_user_orgs()

                    user_orgs = skfb.skfb_api.user_orgs
                    orgUid = ""
                    for org in user_orgs:
                        if org["username"] == orgName:
                            orgUid = org["uid"]
                            break
                    if orgUid:
                        download_url = '{}/{}/models/{}/download'.format(Config.SKETCHFAB_ORGS, orgUid, uid)
                    else:
                        ShowMessage("ERROR", "User not in Organization", "User does not appear to belong to org %s" % (orgName))
                        return
                except:
                    ShowMessage("ERROR", "Invalid url", "Cannot parse org name from url %s" % skfb.manualImportPath)
                    return
            # Otherwise, request a direct download and get model info
            else:
                download_url = Utils.build_download_url(uid)
                requests.get('{}/{}'.format(Config.SKETCHFAB_MODEL, uid), headers=skfb.skfb_api.headers, hooks={'response': self.parse_model_info_request})

            print(f"Downloading URL:\n{download_url}")
            requests.get(download_url, headers=self.headers, hooks={'response': self.handle_download})

    def handle_download(self, r, *args, **kwargs):
        if r.status_code != 200 or 'gltf' not in r.json():
            ShowMessage("ERROR", "This model is not downloadable", "Make sure your account has enough rights to download the model")
            return

        skfb = get_sketchfab_props()
        uid = Utils.get_uid_from_model_url(r.url, self.use_org_profile)
        if uid is None:
            return

        gltf = r.json()['gltf']
        skfb_model = get_sketchfab_model(uid)

        # If the model name is not known at this step, we could try to do an additional API call to get it
        # This can happen when the user chose to import a model from its url
        # However this adds an additional call and a bit of complexity for org models (need additional parsing),
        # so for a simple hotfix models imported this way will be called "Sketchfab model"
        self.get_archive(gltf['url'], skfb_model.title if skfb_model else "Sketchfab Model")

    def get_archive(self, url, title):
        if url is None:
            print('Url is None')
            return

        r = requests.get(url, stream=True)
        uid = Utils.get_uid_from_download_url(url)
        temp_dir = os.path.join(Config.SKETCHFAB_MODEL_DIR, uid)
        if not os.path.exists(temp_dir):
            os.makedirs(temp_dir)

        archive_path = os.path.join(temp_dir, '{}.zip'.format(uid))
        if not os.path.exists(archive_path):
            wm = bpy.context.window_manager
            wm.progress_begin(0, 100)
            set_log("Downloading model..")
            with open(archive_path, "wb") as f:
                total_length = r.headers.get('content-length')
                if total_length is None:  # no content length header
                    f.write(r.content)
                else:
                    dl = 0
                    total_length = int(total_length)
                    for data in r.iter_content(chunk_size=4096):
                        dl += len(data)
                        f.write(data)
                        done = int(100 * dl / total_length)
                        wm.progress_update(done)
                        set_log("Downloading model..{}%".format(done))

            wm.progress_end()
        else:
            print('Model already downloaded')

        gltf_path, gltf_zip = unzip_archive(archive_path)
        if gltf_path:
            try:
                import_model(gltf_path, uid, title)
            except Exception as e:
                import traceback
                print(traceback.format_exc())
        else:
            ShowMessage("ERROR", "Download error", "Failed to download model (url might be invalid)")
            model = get_sketchfab_model(uid)
            set_import_status("Import model ({})".format(model.download_size if model.download_size else 'fetching data'))
        return

class SketchfabLoginProps(bpy.types.PropertyGroup):
    def update_tr(self, context):
        self.status = ''
        if self.email != self.last_username or self.password != self.last_password:
            self.last_username = self.email
            self.last_password = self.password
            if not self.password:
                set_login_status('ERROR', 'Password is empty')
            bpy.ops.wm.sketchfab_login('EXEC_DEFAULT')


    email : StringProperty(
        name="email",
        description="User email",
        default=""
    )

    api_token : StringProperty(
        name="API Token",
        description="User API Token",
        default=""
    )

    use_mail : BoolProperty(
            name="Use mail / password",
            description="Use mail/password login or API Token",
            default=True,
    )

    password : StringProperty(
        name="password",
        description="User password",
        subtype='PASSWORD',
        default="",
        update=update_tr
    )

    access_token : StringProperty(
            name="access_token",
            description="oauth access token",
            subtype='PASSWORD',
            default=""
            )

    status : StringProperty(name='', default='')
    status_type : EnumProperty(
            name="Login status type",
            items=(('ERROR', "Error", ""),
                       ('INFO', "Information", ""),
                       ('FILE_REFRESH', "Progress", "")),
            description="Determines which icon to use",
            default='FILE_REFRESH'
            )

    last_username : StringProperty(default="default")
    last_password : StringProperty(default="default")

    skfb_api = SketchfabApi()

def get_user_orgs(self, context):
    api  = get_sketchfab_props().skfb_api
    if not api.user_has_orgs:
        api.request_user_orgs()
    return [(org["uid"], org["displayName"], "") for org in api.user_orgs]

def get_org_projects(self, context):
    api  = get_sketchfab_props().skfb_api
    return [(proj["uid"], proj["name"], proj["name"]) for proj in api.active_org["projects"]]

def get_available_search_domains(self, context):
    api = get_sketchfab_props().skfb_api

    search_domains = [domain for domain in Config.SKETCHFAB_SEARCH_DOMAIN]

    if api.user_has_orgs and api.use_org_profile:
        search_domains = [
            ("ACTIVE_ORG", "Active Organization", api.active_org["displayName"], 0)
        ]
        for p in get_org_projects(self, context):
            search_domains.append(p)

    return tuple(search_domains)

def refresh_orgs(self, context):

    pprops = get_sketchfab_props_proxy()
    if pprops.is_refreshing:
        return

    props = get_sketchfab_props()
    api   = props.skfb_api

    api.use_org_profile = pprops.use_org_profile

    if api.user_has_orgs and not api.active_org :
        bpy.context.window.cursor_set("WAIT")
        api.request_user_orgs()
        bpy.context.window.cursor_set("DEFAULT")

    orgs = [org for org in api.user_orgs if org["uid"] == pprops.active_org]
    api.active_org = orgs[0] if len(orgs) else None

    if pprops.use_org_profile != props.use_org_profile:
        props.use_org_profile = pprops.use_org_profile
    if pprops.active_org != props.active_org:
        props.active_org = pprops.active_org

    if props.use_org_profile:
        props.search_domain = "ACTIVE_ORG"
        pprops.search_domain = "ACTIVE_ORG"
    else:
        props.search_domain = "DEFAULT"
        pprops.search_domain = "DEFAULT"

    refresh_search(self, context)

def get_sorting_options(self, context):
    api = get_sketchfab_props().skfb_api
    if api.user_has_orgs and api.use_org_profile:
        return (
            ('RELEVANCE', "Relevance", ""),
            ('RECENT', "Recent", "")
        )
    else:
        return Config.SKETCHFAB_SORT_BY

class SketchfabBrowserPropsProxy(bpy.types.PropertyGroup):
    # Search
    query : StringProperty(
            name="",
            update=refresh_search,
            description="Query to search",
            default="",
            options={'SKIP_SAVE'}
            )

    pbr : BoolProperty(
            name="PBR",
            description="Search for PBR model only",
            default=False,
            update=refresh_search,
            )

    categories : EnumProperty(
            name="Categories",
            items=Config.SKETCHFAB_CATEGORIES,
            description="Show only models of category",
            default='ALL',
            update=refresh_search
            )
    face_count : EnumProperty(
            name="Face Count",
            items=Config.SKETCHFAB_FACECOUNT,
            description="Determines which meshes are exported",
            default='ANY',
            update=refresh_search
            )

    sort_by : EnumProperty(
            name="Sort by",
            items=get_sorting_options,
            description="Sort ",
            update=refresh_search,
            )

    animated : BoolProperty(
            name="Animated",
            description="Show only models with animation",
            default=False,
            update=refresh_search
            )

    staffpick : BoolProperty(
            name="Staffpick",
            description="Show only staffpick models",
            default=False,
            update=refresh_search
            )

    downloadable : BoolProperty(
            name="Downloadable",
            description="Show only downloadable models",
            default=True,
            update=refresh_search
            )

    restricted : BoolProperty(
            name="Restricted",
            description="Show restricted models",
            default=True,
            update=refresh_search
            )

    search_domain : EnumProperty(
            name="",
            items=get_available_search_domains,
            description="Search domain ",
            update=refresh_search,
            default=None
            )

    use_org_profile : BoolProperty(
        name="Use organisation profile",
        description="Download/Upload as a member of an organization.\nSearch queries and uploads will be performed to\nthe organisation and project selected below",
        default=False,
        update=refresh_orgs
    )

    active_org : EnumProperty(
        name="Org",
        items=get_user_orgs,
        description="Active org",
        update=refresh_orgs
    )

    is_refreshing : BoolProperty(
        name="Refresh",
        description="Refresh",
        default=False,
    )
    expanded_filters : bpy.props.BoolProperty(default=False)

class SketchfabBrowserProps(bpy.types.PropertyGroup):
    # Search
    query : StringProperty(
            name="Search",
            description="Query to search",
            default=""
            )

    pbr : BoolProperty(
            name="PBR",
            description="Search for PBR model only",
            default=False
            )

    categories : EnumProperty(
            name="Categories",
            items=Config.SKETCHFAB_CATEGORIES,
            description="Show only models of category",
            default='ALL',
             )

    face_count : EnumProperty(
            name="Face Count",
            items=Config.SKETCHFAB_FACECOUNT,
            description="Determines which meshes are exported",
            default='ANY',
            )

    sort_by : EnumProperty(
            name="Sort by",
            items=get_sorting_options,
            description="Sort ",
            )

    animated : BoolProperty(
            name="Animated",
            description="Show only models with animation",
            default=False,
            )

    staffpick : BoolProperty(
            name="Staffpick",
            description="Show only staffpick models",
            default=False,
            )

    downloadable : BoolProperty(
            name="Downloadable",
            description="Show only downloadable models",
            default=False,
            )

    restricted : BoolProperty(
            name="Restricted",
            description="Show restricted models",
            default=False,
            )

    search_domain : EnumProperty(
            name="Search domain",
            items=get_available_search_domains,
            description="Search domain ",
            )

    use_org_profile : BoolProperty(
        name="Use organisation profile",
        description="Import/Export as a member of an organization\nLOL",
        default=False,
    )

    active_org : EnumProperty(
        name="Org",
        items=get_user_orgs,
        description="Active org",
    )

    status : StringProperty(name='status', default='idle')

    use_preview : BoolProperty(
        name="Use Preview",
        description="Show results using preview widget instead of regular buttons with thumbnails as icons",
        default=True
        )

    search_results = {}
    current_key : StringProperty(name='current', default='current')
    has_searched_next : BoolProperty(name='next', default=False)
    has_searched_prev : BoolProperty(name='prev', default=False)

    skfb_api = SketchfabLoginProps.skfb_api
    custom_icons = bpy.utils.previews.new()
    has_loaded_thumbnails : BoolProperty(default=False)

    is_latest_version : IntProperty(default=-1)

    import_status : StringProperty(name='import', default='')

    manualImportBoolean : BoolProperty(
            name="Import from url",
            description="Import a downloadable model from a url",
            default=False,
            )
    manualImportPath : StringProperty(
            name="Url",
            description="Paste full model url:\n* https://sketchfab.com/models/mymodel-XXXX\n* https://sketchfab.com/orgs/XXXX/3d-models/mymodel-YYYY",
            default="",
            maxlen=1024,
            options={'TEXTEDIT_UPDATE'})


def list_current_results(self, context):
    skfb = get_sketchfab_props()

    # No results:
    if 'current' not in skfb.search_results:
        return preview_collection['default']

    if skfb.has_loaded_thumbnails and 'thumbnails' in preview_collection:
        return preview_collection['thumbnails']

    res = []
    missing_thumbnail = False
    if 'current' in skfb.search_results and len(skfb.search_results['current']):
        skfb_results = skfb.search_results['current']
        for i, result in enumerate(skfb_results):
            if result in skfb_results:
                model = skfb_results[result]
                if model.uid in skfb.custom_icons:
                    res.append((model.uid, model.title, "", skfb.custom_icons[model.uid].icon_id, i))
                else:
                    res.append((model.uid, model.title, "", preview_collection['skfb']['0'].icon_id, i))
                    missing_thumbnail = True
            else:
                print('Result issue')

    # Default element to avoid having an empty preview collection
    if not res:
        res.append(('NORESULTS', 'empty', "", preview_collection['skfb']['0'].icon_id, 0))

    preview_collection['thumbnails'] = tuple(res)
    skfb.has_loaded_thumbnails = not missing_thumbnail
    return preview_collection['thumbnails']


def draw_model_info(layout, model, context):
    ui_model_props = layout.box().column(align=True)

    row = ui_model_props.row()
    row.label(text="{}".format(model.title), icon='OBJECT_DATA')
    row.operator("wm.sketchfab_view", text="", icon='LINKED').model_uid = model.uid

    ui_model_props.label(text='{}'.format(model.author), icon='ARMATURE_DATA')

    if model.license:
        ui_model_props.label(text='{}'.format(model.license), icon='TEXT')
    else:
        ui_model_props.label(text='Fetching..')

    if model.vertex_count and model.face_count:
        ui_model_stats = ui_model_props.row()
        ui_model_stats.label(text='Verts: {}  |  Faces: {}'.format(Utils.humanify_number(model.vertex_count), Utils.humanify_number(model.face_count)), icon='MESH_DATA')

    if(model.animated):
        ui_model_props.label(text='Animated: ' + model.animated, icon='ANIM_DATA')

    layout.separator()

def draw_import_button(layout, model, context):

    import_ops = layout.row()
    skfb = get_sketchfab_props()

    import_ops.enabled = skfb.skfb_api.is_user_logged() and bpy.context.mode == 'OBJECT' and Utils.is_valid_uuid(model.uid)
    if not skfb.skfb_api.is_user_logged():
        downloadlabel = 'Log in to download models'
    elif bpy.context.mode != 'OBJECT':
        downloadlabel = "Import is available only in object mode"
    else:
        downloadlabel = "Import model"
        if model.download_size:
            downloadlabel += " ({})".format(model.download_size)
    if skfb.import_status:
        downloadlabel = skfb.import_status

    download_icon = 'IMPORT' if import_ops.enabled else 'INFO'
    import_ops.scale_y = 2.0
    import_ops.operator("wm.sketchfab_download", icon=download_icon, text=downloadlabel, translate=False, emboss=True).model_uid = model.uid

def set_log(log):
    get_sketchfab_props().status = log

def unzip_archive(archive_path):
    if os.path.exists(archive_path):
        set_import_status('Unzipping model')
        import zipfile
        try:
            zip_ref = zipfile.ZipFile(archive_path, 'r')
            extract_dir = os.path.dirname(archive_path)
            zip_ref.extractall(extract_dir)
            zip_ref.close()
        except zipfile.BadZipFile:
            print('Error when dezipping file')
            os.remove(archive_path)
            print('Invaild zip. Try again')
            set_import_status('')
            return None, None

        gltf_file = os.path.join(extract_dir, 'scene.gltf')
        return gltf_file, archive_path

    else:
        print('ERROR: archive doesn\'t exist')


def run_async(func):
    from threading import Thread
    from functools import wraps

    @wraps(func)
    def async_func(*args, **kwargs):
        func_hl = Thread(target=func, args=args, kwargs=kwargs)
        func_hl.start()
        return func_hl

    return async_func


def import_model(gltf_path, uid, title):
    bpy.ops.wm.import_modal('INVOKE_DEFAULT', gltf_path=gltf_path, uid=uid, title=title)


def build_search_request(query, pbr, animated, staffpick, downloadable, restricted, face_count, category, sort_by):
    final_query = '&q={}'.format(query) if query else ''

    if animated:
        final_query = final_query + '&animated=true'

    if staffpick:
        final_query = final_query + '&staffpicked=true'

    if restricted:
        final_query = final_query + '&restricted=1'

    if downloadable:
        final_query = final_query + '&downloadable=true'

    if sort_by == 'LIKES':
        final_query = final_query + '&sort_by=-likeCount'
    elif sort_by == 'RECENT':
        final_query = final_query + '&sort_by=-publishedAt'
    elif sort_by == 'VIEWS':
        final_query = final_query + '&sort_by=-viewCount'

    if face_count == '10':
        final_query = final_query + '&max_face_count=10'
    elif face_count == '100':
        final_query = final_query + '&max_face_count=100'
    elif face_count == '1K':
        final_query = final_query + '&max_face_count=1000'
    elif face_count == '10K':
        final_query = final_query + '&max_face_count=10000'
    elif face_count == '50K':
        final_query = final_query + '&min_face_count=10000&max_face_count=50000'
    elif face_count == '100K':
        final_query = final_query + '&min_face_count=50000&max_face_count=100000'
    elif face_count == '250K':
        final_query = final_query + "&min_face_count=100000&max_face_count=250000"
    elif face_count == '250KP':
        final_query = final_query + "&min_face_count=250000"

    if category != 'ALL':
        final_query = final_query + '&categories={}'.format(category)

    if pbr:
        final_query = final_query + '&pbr_type=metalness'

    return final_query


def parse_results(r, *args, **kwargs):

    ongoingSearches.discard(r.url)

    skfb = get_sketchfab_props()
    json_data = r.json()

    if 'current' in skfb.search_results:
        skfb.search_results['current'].clear()
        del skfb.search_results['current']

    skfb.search_results['current'] = OrderedDict()

    for result in list(json_data.get('results', [])):

        # Dirty fix to avoid parsing obsolete data
        if 'current' not in skfb.search_results:
            return

        uid = result['uid']
        skfb.search_results['current'][result['uid']] = SketchfabModel(result)

        if not os.path.exists(os.path.join(Config.SKETCHFAB_THUMB_DIR, uid) + '.jpeg'):
            skfb.skfb_api.request_thumbnail(result['thumbnails'], uid)
        elif uid not in skfb.custom_icons:
            skfb.custom_icons.load(uid, os.path.join(Config.SKETCHFAB_THUMB_DIR, "{}.jpeg".format(uid)), 'IMAGE')

        # Make a request to get the download_size for org and own models
        """
        model = skfb.search_results['current'][result['uid']]
        if model.download_size is None:
            api = skfb.skfb_api
            def set_download_size(r, *args, **kwargs):
                json_data = r.json()
                print(json_data)
                if 'gltf' in json_data and 'size' in json_data['gltf']:
                    model.download_size = Utils.humanify_size(json_data['gltf']['size'])
            requests.get(Utils.build_download_url(uid, api.use_org_profile, api.active_org), headers=api.headers, hooks={'response': set_download_size})
        """

    if json_data['next']:
        skfb.skfb_api.next_results_url = json_data['next']
    else:
        skfb.skfb_api.next_results_url = None

    if json_data['previous']:
        skfb.skfb_api.prev_results_url = json_data['previous']
    else:
        skfb.skfb_api.prev_results_url = None


class ThumbnailCollector(threading.Thread):
    def __init__(self, url):
        self.url = url
        threading.Thread.__init__(self)

    def set_url(self, url):
        self.url = url

    def run(self):
        if not self.url:
            return
        requests.get(self.url, stream=True, hooks={'response': self.handle_thumbnail})

    def handle_thumbnail(self, r, *args, **kwargs):
        uid = r.url.split('/')[4]
        if not os.path.exists(Config.SKETCHFAB_THUMB_DIR):
            try:
                os.makedirs(Config.SKETCHFAB_THUMB_DIR)
            except:
                pass
        thumbnail_path = os.path.join(Config.SKETCHFAB_THUMB_DIR, uid) + '.jpeg'

        with open(thumbnail_path, "wb") as f:
            total_length = r.headers.get('content-length')

            if total_length is None and r.content:
                f.write(r.content)
            else:
                dl = 0
                total_length = int(total_length)
                for data in r.iter_content(chunk_size=4096):
                    dl += len(data)
                    f.write(data)

        thumbnailsProgress.discard(uid)

        props = get_sketchfab_props()
        if uid not in props.custom_icons:
            props.custom_icons.load(uid, os.path.join(Config.SKETCHFAB_THUMB_DIR, "{}.jpeg".format(uid)), 'IMAGE')


class LoginModal(bpy.types.Operator):
    """Login into your account"""
    bl_idname = "wm.login_modal"
    bl_label = ""
    bl_options = {'INTERNAL'}

    is_logging : BoolProperty(default=False)
    error : BoolProperty(default=False)
    error_message : StringProperty(default='')

    def execute(self, context):
        return {'FINISHED'}

    def handle_mail_login(self, r, *args, **kwargs):
        browser_props = get_sketchfab_props()
        if r.status_code == 200 and 'access_token' in r.json():
            browser_props.skfb_api.access_token = r.json()['access_token']
            login_props = get_sketchfab_login_props()
            Cache.save_key('username', login_props.email)
            Cache.save_key('access_token', browser_props.skfb_api.access_token)

            browser_props.skfb_api.build_headers()
            set_login_status('INFO', '')
            browser_props.skfb_api.request_user_info()

        else:
            if 'error_description' in r.json():
                set_login_status('ERROR', 'Failed to authenticate: bad login/password')
            else:
                set_login_status('ERROR', 'Failed to authenticate: bad login/password')
                print('Cannot login.\n {}'.format(r.json()))

        self.is_logging = False

    def handle_token_login(self, api_token):
        browser_props = get_sketchfab_props()
        browser_props.skfb_api.api_token = api_token
        login_props = get_sketchfab_login_props()
        Cache.save_key('api_token', login_props.api_token)

        browser_props.skfb_api.build_headers()
        set_login_status('INFO', '')
        browser_props.skfb_api.request_user_info()
        self.is_logging = False

    def modal(self, context, event):
        if self.error:
            self.error = False
            set_login_status('ERROR', '{}'.format(self.error_message))
            return {"FINISHED"}

        if self.is_logging:
            set_login_status('FILE_REFRESH', 'Loging in to your Sketchfab account...')
            return {'RUNNING_MODAL'}
        else:
            return {'FINISHED'}

    def invoke(self, context, event):
        self.is_logging = True
        try:
            context.window_manager.modal_handler_add(self)
            login_props = get_sketchfab_login_props()
            if(login_props.use_mail):
                data = {
                    'grant_type': 'password',
                    'client_id': Config.CLIENTID,
                    'username': login_props.email,
                    'password': login_props.password,
                }
                requests.post(Config.SKETCHFAB_OAUTH, data=data, hooks={'response': self.handle_mail_login})
            else:
                self.handle_token_login(login_props.api_token)
        except Exception as e:
            self.error = True
            self.error_message = str(e)

        return {'RUNNING_MODAL'}


class ImportModalOperator(bpy.types.Operator):
    """Imports the selected model into Blender"""
    bl_idname = "wm.import_modal"
    bl_label = "Import glTF model into Sketchfab"
    bl_options = {'INTERNAL'}

    gltf_path : StringProperty()
    uid : StringProperty()
    title: StringProperty()

    def execute(self, context):
        print('IMPORT')
        return {'FINISHED'}

    def modal(self, context, event):
        if bpy.context.scene.render.engine not in ["CYCLES", "BLENDER_EEVEE"]:
            bpy.context.scene.render.engine = "BLENDER_EEVEE"
        try:
            old_objects = [o.name for o in bpy.data.objects] # Get the current objects inorder to find the new node hierarchy
            bpy.ops.import_scene.gltf(filepath=self.gltf_path)
            set_import_status('')
            Utils.clean_downloaded_model_dir(self.uid)
            Utils.clean_node_hierarchy([o for o in bpy.data.objects if o.name not in old_objects], self.title)
            return {'FINISHED'}
        except Exception:
            import traceback
            print(traceback.format_exc())
            set_import_status('')
            return {'FINISHED'}

    def invoke(self, context, event):
        context.window_manager.modal_handler_add(self)
        set_import_status('Importing...')
        return {'RUNNING_MODAL'}


class GetRequestThread(threading.Thread):
    def __init__(self, url, callback, headers={}):
        self.url = url
        self.callback = callback
        self.headers = headers
        threading.Thread.__init__(self)

    def run(self):
        requests.get(self.url, headers=self.headers, hooks={'response': self.callback})


class View3DPanel:
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'TOOLS' if bpy.app.version < (2, 80, 0) else 'UI'
    bl_category = 'Sketchfab'
    bl_context = 'objectmode'

class SketchfabPanel(View3DPanel, bpy.types.Panel):
    bl_options = {'DEFAULT_CLOSED'}
    bl_idname = "VIEW3D_PT_sketchfab_about"
    bl_label = "About"

    @classmethod
    def poll(cls, context):
        return (context.scene is not None)

    def draw(self, context):

        skfb = get_sketchfab_props()

        if skfb.is_latest_version == 1:
            self.bl_label = "Sketchfab plugin v{} (up-to-date)".format(PLUGIN_VERSION)
        elif skfb.is_latest_version == 0:
            self.bl_label = "Sketchfab plugin v{} (outdated)".format(PLUGIN_VERSION)
            self.layout.operator('wm.skfb_new_version', text='New version available', icon='ERROR')
        elif skfb.is_latest_version == -2:
            self.bl_label = "Sketchfab plugin v{}".format(PLUGIN_VERSION)

        # External links
        #doc_ui = self.layout.row()
        self.layout.operator('wm.skfb_help', text='Documentation', icon='QUESTION')
        self.layout.operator('wm.skfb_report_issue', text='Report an issue', icon='ERROR')
        self.layout.label(text="Download folder:")
        self.layout.label(text="  " + Config.SKETCHFAB_TEMP_DIR)

class LoginPanel(View3DPanel, bpy.types.Panel):
    bl_idname = "VIEW3D_PT_sketchfab_login"
    bl_label = "Activation / Log in"
    #bl_parent_id = "VIEW3D_PT_sketchfab"

    is_logged = BoolProperty()

    def draw(self, context):
        global is_plugin_enabled
        if not is_plugin_enabled:
            self.layout.operator('wm.skfb_enable', text='Activate add-on', icon="LINKED").enable = True
        else:
            # LOGIN
            skfb_login = get_sketchfab_login_props()
            layout = self.layout.box().column(align=True)
            layout.enabled = get_plugin_enabled()
            if skfb_login.skfb_api.is_user_logged():
                login_row = layout.row()
                login_row.label(text='Logged in as {}'.format(skfb_login.skfb_api.get_user_info()))
                login_row.operator('wm.sketchfab_login', text='Logout', icon='DISCLOSURE_TRI_RIGHT').authenticate = False
                if skfb_login.status:
                    layout.prop(skfb_login, 'status', icon=skfb_login.status_type)
            else:
                layout.label(text="Login to your Sketchfab account", icon='INFO')
                layout.prop(skfb_login, "use_mail")
                if skfb_login.use_mail:
                    layout.prop(skfb_login, "email")
                    layout.prop(skfb_login, "password")
                else:
                    layout.prop(skfb_login, "api_token")
                ops_row = layout.row()
                ops_row.operator('wm.sketchfab_#', text='Create an account', icon='PLUS')
                login_icon = "LINKED" if bpy.app.version < (2,80,0) else "USER"
                ops_row.operator('wm.sketchfab_login', text='Log in', icon=login_icon).authenticate = True
                if skfb_login.status:
                    layout.prop(skfb_login, 'status', icon=skfb_login.status_type)

class TeamsPanel(View3DPanel, bpy.types.Panel):
    bl_idname = "VIEW3D_PT_sketchfab_teams"
    bl_label = "Sketchfab for Teams"
    bl_options = {'DEFAULT_CLOSED'}

    def draw(self, context):

        skfb = get_sketchfab_props()
        api  = skfb.skfb_api

        self.layout.enabled = get_plugin_enabled() and api.is_user_logged()

        if not api.user_has_orgs:
            self.layout.label(text="You are not part of an organization", icon='INFO')
            self.layout.operator("wm.url_open", text='Learn about Sketchfab for Teams').url = "https://sketchfab.com/features/teams"
        else:
            props = get_sketchfab_props_proxy()

            use_org_profile_row = self.layout.row()
            use_org_profile_row.prop(props, "use_org_profile")

            org_row = self.layout.row()
            org_row.prop(props, "active_org")
            org_row.enabled = skfb.skfb_api.use_org_profile

            pprops = get_sketchfab_props()

class Model:
    def __init__(self, _uid):
        self.uid = _uid
        self.download_size = 0

class SketchfabBrowse(View3DPanel, bpy.types.Panel):
    bl_idname = "VIEW3D_PT_sketchfab_browse"
    bl_label = "Import"

    uid   = ''
    label = "Search results"

    def draw_search(self, layout, context):
        prop = get_sketchfab_props()
        props = get_sketchfab_props_proxy()
        skfb_api = prop.skfb_api

        # Add an option to import from url or uid
        col = layout.box().column(align=True)
        row = col.row()
        row.prop(prop, "manualImportBoolean")

        if prop.manualImportBoolean:
            row = col.row()
            row.prop(prop, "manualImportPath")

        else:
            col = layout.box().column(align=True)
            ro = col.row()
            ro.label(text="Search")
            domain_col = ro.column()
            domain_col.scale_x = 1.5
            domain_col.enabled = skfb_api.is_user_logged()
            domain_col.prop(props, "search_domain")

            ro = col.row()
            ro.scale_y = 1.25
            ro.prop(props, "query")
            ro.operator("wm.sketchfab_search", text="", icon='VIEWZOOM')

            # User selected own models but is not pro
            if props.search_domain == "OWN" and skfb_api.is_user_logged() and not skfb_api.is_user_pro():
                col.label(text='A PRO account is required', icon='QUESTION')
                col.label(text='to access your personal library')

            # Display a collapsible box for filters
            col = layout.box().column(align=True)
            col.enabled = (props.search_domain != "STORE")
            row = col.row()
            row.prop(props, "expanded_filters", icon="TRIA_DOWN" if props.expanded_filters else "TRIA_RIGHT", icon_only=True, emboss=False)
            row.label(text="Search filters")
            if props.expanded_filters:
                if props.search_domain in ["DEFAULT", "OWN"]:
                    col.separator()
                    col.prop(props, "categories")
                    col.prop(props, "sort_by")
                    col.prop(props, "face_count")
                    row = col.row()
                    row.prop(props, "pbr")
                    row.prop(props, "staffpick")
                    row.prop(props, "downloadable")
                    row.prop(props, "restricted")
                    row.prop(props, "animated")
                else:
                    col.separator()
                    col.prop(props, "sort_by")
                    col.prop(props, "face_count")

        pprops = get_sketchfab_props()

    def draw_results(self, layout, context):

        props = get_sketchfab_props()

        col = layout.box().column(align=True)

        if not props.manualImportBoolean:

            #results = layout.column(align=True)
            col.label(text=self.label)

            model = None

            result_pages_ops = col.row()
            if props.skfb_api.prev_results_url:
                result_pages_ops.operator("wm.sketchfab_search_prev", text="Previous page", icon='FRAME_PREV')

            if props.skfb_api.next_results_url:
                result_pages_ops.operator("wm.sketchfab_search_next", text="Next page", icon='FRAME_NEXT')

            #result_label = 'Click below to see more results'
            #col.label(text=result_label, icon='INFO')
            try:
                col.template_icon_view(bpy.context.window_manager, 'result_previews', show_labels=True, scale=8)
            except Exception:
                print('ResultsPanel: Failed to display results')
                pass

            if 'current' not in props.search_results or not len(props.search_results['current']):
                self.label = 'No results'
                return
            else:
                self.label = "Search results"

            if "current" in props.search_results:

                if bpy.context.window_manager.result_previews not in props.search_results['current']:
                    return

                model = props.search_results['current'][bpy.context.window_manager.result_previews]

                if not model:
                    return

                if self.uid != model.uid:
                    self.uid = model.uid

                    if not model.info_requested:
                        props.skfb_api.request_model_info(model.uid)
                        model.info_requested = True

                draw_model_info(col, model, context)
                draw_import_button(col, model, context)
        else:
            uid = ""
            if "sketchfab.com" in props.manualImportPath:
                uid = props.manualImportPath[-32:]
            m = Model(uid)
            draw_import_button(col, m, context)

    def draw(self, context):
        self.layout.enabled = get_plugin_enabled()
        self.draw_search(self.layout, context)
        self.draw_results(self.layout, context)

    def invoke(self, context, event):
        wm = context.window_manager
        return wm.invoke_props_dialog(self, width=900, height=850)

class SketchfabExportPanel(View3DPanel, bpy.types.Panel):
    #bl_idname = "wm.sketchfab_export" if bpy.app.version == (2, 79, 0) else "VIEW3D_PT_sketchfab_export"
    bl_options = {'DEFAULT_CLOSED'}
    bl_label = "Export"
    bl_idname = "VIEW3D_PT_sketchfab_export"

    def draw(self, context):

        api = get_sketchfab_props().skfb_api
        self.layout.enabled = get_plugin_enabled() and api.is_user_logged()

        wm = context.window_manager
        props = wm.sketchfab_export

        layout = self.layout

        # Selection only
        layout.prop(props, "selection")

        # Model properties
        col = layout.box().column(align=True)
        if not props.reuploadBoolean:
            col.prop(props, "title")
            col.prop(props, "description")
            col.prop(props, "tags")
            col.prop(props, "draft")
            col.prop(props, "private")
            if props.private:
                col.prop(props, "password")
        col.prop(props, "reuploadBoolean")
        if props.reuploadBoolean:
            col.prop(props, "reuploadPath")

        # Project selection if member of an org
        if api.active_org and api.use_org_profile:
            row = layout.row()
            row.prop(props, "active_project")

        # Upload button
        row = layout.row()
        row.scale_y = 2.0
        upload_label = "Reupload" if props.reuploadBoolean else "Upload"
        upload_icon  = "EXPORT"
        upload_enabled = api.is_user_logged() and bpy.context.mode == 'OBJECT'
        if not upload_enabled:
            if not api.is_user_logged():
                upload_label = "Log in to upload models"
            elif bpy.context.mode != 'OBJECT':
                upload_label = "Export is only available in object mode"
        if sf_state.uploading:
            upload_label = "Uploading %s" % sf_state.size_label
            upload_icon  = "SORTTIME"
        row.operator("wm.sketchfab_export", icon=upload_icon, text=upload_label)

        model_url = sf_state.model_url
        if model_url:
            layout.operator("wm.url_open", text="View Online Model", icon='URL').url = model_url

class SketchfabLogger(bpy.types.Operator):
    """Log in / out your Sketchab.com account"""
    bl_idname = 'wm.sketchfab_login'
    bl_label = 'Sketchfab Login'
    bl_options = {'INTERNAL'}

    authenticate : BoolProperty(default=True)

    def execute(self, context):
        set_login_status('FILE_REFRESH', 'Login to your Sketchfab account...')
        wm = context.window_manager
        if self.authenticate:
            wm.sketchfab_browser.skfb_api.login(wm.sketchfab_api.email, wm.sketchfab_api.password, wm.sketchfab_api.api_token)
        else:
            wm.sketchfab_browser.skfb_api.logout()
            wm.sketchfab_api.password = ''
            wm.sketchfab_api.last_password = "default"
            set_login_status('FILE_REFRESH', '')
        return {'FINISHED'}


class SketchfabModel:
    def __init__(self, json_data):
        self.title = str(json_data['name'])
        self.author = json_data['user']['displayName']
        self.username = json_data['user']['username']
        self.uid = json_data['uid']
        self.vertex_count = json_data['vertexCount']
        self.face_count = json_data['faceCount']

        if 'archives' in json_data and  'gltf' in json_data['archives']:
            if 'size' in json_data['archives']['gltf'] and json_data['archives']['gltf']['size']:
                self.download_size = Utils.humanify_size(json_data['archives']['gltf']['size'])
        else:
            self.download_size = None

        self.thumbnail_url = os.path.join(Config.SKETCHFAB_THUMB_DIR, '{}.jpeg'.format(self.uid))

        # Model info request
        self.info_requested = False
        self.license = None
        self.animated = False

        # Download url data
        self.download_url = None
        self.time_url_requested = None
        self.url_expires = None

def ShowMessage(icon = "INFO", title = "Info", message = "Information"):
    def draw(self, context):
        self.layout.label(text=message)
    print("\n{}: {}".format(icon, message))
    bpy.context.window_manager.popup_menu(draw, title = title, icon = icon)


class SketchfabDownloadModel(bpy.types.Operator):
    """Import the selected model"""
    bl_idname = "wm.sketchfab_download"
    bl_label = "Downloading"
    bl_options = {'INTERNAL'}

    model_uid : bpy.props.StringProperty(name="uid")

    def execute(self, context):
        skfb_api = context.window_manager.sketchfab_browser.skfb_api
        skfb_api.download_model(self.model_uid)
        return {'FINISHED'}


class ViewOnSketchfab(bpy.types.Operator):
    """Upload your model to Sketchfab"""
    bl_idname = "wm.sketchfab_view"
    bl_label = "View the model on Sketchfab"
    bl_options = {'INTERNAL'}

    model_uid : bpy.props.StringProperty(name="uid")

    def execute(self, context):
        import webbrowser
        webbrowser.open('{}/models/{}'.format(Config.SKETCHFAB_URL, self.model_uid))
        return {'FINISHED'}


def clear_search():
    skfb = get_sketchfab_props()
    skfb.has_loaded_thumbnails = False
    skfb.search_results.clear()
    skfb.custom_icons.clear()
    bpy.data.window_managers['WinMan']['result_previews'] = 0


class SketchfabSearch(bpy.types.Operator):
    """Send a search query to Sketchfab
    Searches on the selected domain (all site, own models for PRO+ users, organization...)
    and takes into accounts various search filters"""
    bl_idname = "wm.sketchfab_search"
    bl_label = "Search Sketchfab"
    bl_options = {'INTERNAL'}

    def execute(self, context):
        # prepare request for search
        clear_search()
        skfb = get_sketchfab_props()
        skfb.skfb_api.prev_results_url = None
        skfb.skfb_api.next_results_url = None
        final_query = build_search_request(skfb.query, skfb.pbr, skfb.animated, skfb.staffpick, skfb.downloadable, skfb.restricted, skfb.face_count, skfb.categories, skfb.sort_by)
        skfb.skfb_api.search(final_query, parse_results)
        return {'FINISHED'}


class SketchfabSearchNextResults(bpy.types.Operator):
    """Loads the next batch of 24 models from the search results"""
    bl_idname = "wm.sketchfab_search_next"
    bl_label = "Search Sketchfab"
    bl_options = {'INTERNAL'}

    def execute(self, context):
        # prepare request for search
        clear_search()
        skfb_api = get_sketchfab_props().skfb_api
        skfb_api.search_cursor(skfb_api.next_results_url, parse_results)
        return {'FINISHED'}


class SketchfabSearchPreviousResults(bpy.types.Operator):
    """Loads the previous batch of 24 models from the search results"""
    bl_idname = "wm.sketchfab_search_prev"
    bl_label = "Search Sketchfab"
    bl_options = {'INTERNAL'}

    def execute(self, context):
        # prepare request for search
        clear_search()
        skfb_api = get_sketchfab_props().skfb_api
        skfb_api.search_cursor(skfb_api.prev_results_url, parse_results)
        return {'FINISHED'}

class SketchfabCreateAccount(bpy.types.Operator):
    """Create an account on sketchfab.com"""
    bl_idname = "wm.sketchfab_#"
    bl_label = "Sketchfab"
    bl_options = {'INTERNAL'}

    def execute(self, context):
        import webbrowser
        webbrowser.open(Config.SKETCHFAB_#)
        return {'FINISHED'}


class SketchfabNewVersion(bpy.types.Operator):
    """Opens addon latest available release on github"""
    bl_idname = "wm.skfb_new_version"
    bl_label = "Sketchfab"
    bl_options = {'INTERNAL'}

    def execute(self, context):
        import webbrowser
        webbrowser.open('{}/releases/latest'.format(Config.GITHUB_REPOSITORY_URL))
        return {'FINISHED'}


class SketchfabReportIssue(bpy.types.Operator):
    """Open an issue on github tracker"""
    bl_idname = "wm.skfb_report_issue"
    bl_label = "Sketchfab"
    bl_options = {'INTERNAL'}

    def execute(self, context):
        import webbrowser
        webbrowser.open(Config.SKETCHFAB_REPORT_URL)
        return {'FINISHED'}


class SketchfabHelp(bpy.types.Operator):
    """Opens the addon README on github"""
    bl_idname = "wm.skfb_help"
    bl_label = "Sketchfab"
    bl_options = {'INTERNAL'}

    def execute(self, context):
        import webbrowser
        webbrowser.open('{}/releases/latest'.format(Config.GITHUB_REPOSITORY_URL))
        return {'FINISHED'}


def activate_plugin():
    props = get_sketchfab_props()
    login = get_sketchfab_login_props()

    # Fill login/access_token
    cache_data = Cache.read()
    if 'username' in cache_data:
        login.email = cache_data['username']

    if 'access_token' in cache_data:
        props.skfb_api.access_token = cache_data['access_token']
        props.skfb_api.build_headers()
        props.skfb_api.request_user_info()
        props.skfb_api.use_mail = True
    elif 'api_token' in cache_data:
        props.skfb_api.api_token = cache_data['api_token']
        props.skfb_api.build_headers()
        props.skfb_api.request_user_info()
        props.skfb_api.use_mail = False

    global is_plugin_enabled
    is_plugin_enabled = True

    try:
        requests.get(Config.SKETCHFAB_PLUGIN_VERSION, hooks={'response': check_plugin_version})
    except Exception as e:
        print('Error when checking for version: {}'.format(e))

    run_default_search()


class SketchfabEnable(bpy.types.Operator):
    """Activate the addon (checks login, cache folders...)"""
    bl_idname = "wm.skfb_enable"
    bl_label = "Sketchfab"
    bl_options = {'INTERNAL'}

    enable : BoolProperty(default=True)
    def execute(self, context):
        if self.enable:
            activate_plugin()

        return {'FINISHED'}


class SketchfabExportProps(bpy.types.PropertyGroup):
    description : StringProperty(
            name="Description",
            description="Description of the model (optional)",
            default="",
            maxlen=1024)
    filepath : StringProperty(
            name="Filepath",
            description="internal use",
            default="",
            )
    selection : BoolProperty(
            name="Selection only",
            description="Determines which meshes are exported",
            default=False,
            )
    private : BoolProperty(
            name="Private",
            description="Upload as private (requires a pro account)",
            default=False,
            )
    draft : BoolProperty(
            name="Draft",
            description="Do not publish the model",
            default=True,
            )
    password : StringProperty(
            name="Password",
            description="Password-protect your model (requires a pro account)",
            default="",
            )
    tags : StringProperty(
            name="Tags",
            description="List of tags (42 max), separated by spaces (optional)",
            default="",
            )
    title : StringProperty(
            name="Title",
            description="Title of the model (determined automatically if left empty)",
            default="",
            maxlen=48
            )
    reuploadBoolean : BoolProperty(
            name="Reupload",
            description="Reupload the model over an existing one",
            default=False,
            )
    reuploadPath : StringProperty(
            name="Url",
            description="Paste full model url to reupload to",
            default="",
            maxlen=1024)
    active_project : EnumProperty(
        name="Project",
        items=get_org_projects,
        description="Active project",
        update=refresh_orgs
    )


class _SketchfabState:
    """Singleton to store state"""
    __slots__ = (
        "uploading",
        "size_label",
        "model_url",
        "report_message",
        "report_type",
        )

    def __init__(self):
        self.uploading = False
        self.size_label = ""
        self.model_url = ""
        self.report_message = ""
        self.report_type = ''

sf_state = _SketchfabState()
del _SketchfabState

# remove file copy
def terminate(filepath):
    print(filepath)
    os.remove(filepath)
    os.rmdir(os.path.dirname(filepath))

def upload_report(report_message, report_type):
    sf_state.report_message = report_message
    sf_state.report_type = report_type

# upload the blend-file to sketchfab
def upload(filepath, filename):

    props = get_sketchfab_props()
    api   = props.skfb_api

    wm = bpy.context.window_manager
    props = wm.sketchfab_export

    title = props.title
    if not title:
        title = os.path.splitext(os.path.basename(bpy.data.filepath))[0]

    # Limit the number of tags to 42
    props.tags = " ".join(props.tags.split(" ")[:42])

    _data = {
        "name": title,
        "description": props.description,
        "tags": props.tags,
        "private": props.private,
        "isPublished": not props.draft,
        "password": props.password,
        "source": "blender-exporter",
    }

    _files = {
        "modelFile": open(filepath, 'rb'),
    }

    _headers = api.headers

    uploadUrl = ""
    modelUid  = ""
    requestFunction = requests.post

    # Are we reuploading ?
    if props.reuploadBoolean:

        requestFunction = requests.put

        if "sketchfab.com/" not in props.reuploadPath:
            return upload_report("reupload url is malformed %s" % props.reuploadPath, 'ERROR')

        # Get the model uid
        try:
            modelUid = props.reuploadPath[-32:]
            if not Utils.is_valid_uuid(modelUid):
                return upload_report("reupload url does not end with a valid uid (32 characters string): %s" % props.reuploadPath, 'ERROR')
        except:
            return upload_report("reupload url is malformed %s" % props.reuploadPath, 'ERROR')

        # If the model is in an org, find if the user has access to it
        if "/orgs/" in props.reuploadPath:
            if True:#try:
                orgName = props.reuploadPath.split("/orgs/")[1].split("/")[0]
                user_orgs = api.user_orgs
                orgUid = ""
                for org in user_orgs:
                    if org["username"] == orgName:
                        orgUid = org["uid"]
                        break
                if orgUid:
                    uploadUrl = '{}/{}/models/{}'.format(Config.SKETCHFAB_ORGS, orgUid, modelUid)
                else:
                    return upload_report("User does not appear to belong to org %s" % (orgName), 'ERROR')
            else:#aexcept:
                return upload_report("Cannot parse the org name from the url %s" % props.reuploadPath, 'ERROR')
        # Otherwise, request a direct reupload
        else:
            uploadUrl = '{}/{}'.format(Config.SKETCHFAB_MODEL, modelUid)

        _data = {
            "uid" : modelUid,
            "source": "blender-exporter"
        }

    else:

        # Org or not
        if api.user_has_orgs and api.use_org_profile:
            uploadUrl = "%s/%s/models" % (Config.SKETCHFAB_ORGS, api.active_org["uid"])
            _data["orgProject"] = props.active_project
        else:
            uploadUrl = Config.SKETCHFAB_MODEL

    # Upload and parse the result
    try:
        print("Uploading to %s" % uploadUrl)
        r = requestFunction(
            uploadUrl,
            data    = _data,
            files   = _files,
            headers = _headers
        )
    except requests.exceptions.RequestException as e:
        return upload_report("Upload failed. Error: %s" % str(e), 'WARNING')

    if r.status_code not in [requests.codes.ok, requests.codes.created, requests.codes.no_content]:
        return upload_report("Upload failed. Error code: %s\nMessage:\n%s" % (str(r.status_code), str(r)), 'WARNING')
    else:
        try:
            result = r.json()
            sf_state.model_url = Config.SKETCHFAB_URL + "/models/" + result["uid"]
        except:
            sf_state.model_url = Config.SKETCHFAB_URL + "/models/" + modelUid
        return upload_report("Upload complete. Available on your sketchfab.com dashboard.", 'INFO')


class ExportSketchfab(bpy.types.Operator):
    """Upload your model to Sketchfab"""
    bl_idname = "wm.sketchfab_export"
    bl_label = "Upload"

    _timer = None
    _thread = None

    def modal(self, context, event):
        if event.type == 'TIMER':
            if not self._thread.is_alive():
                wm = context.window_manager
                props = wm.sketchfab_export

                terminate(props.filepath)

                if context.area:
                    context.area.tag_redraw()

                # forward message from upload thread
                if not sf_state.report_type:
                    sf_state.report_type = 'ERROR'
                self.report({sf_state.report_type}, sf_state.report_message)

                wm.event_timer_remove(self._timer)
                self._thread.join()
                sf_state.uploading = False
                return {'FINISHED'}

        return {'PASS_THROUGH'}

    def execute(self, context):

        if sf_state.uploading:
            self.report({'WARNING'}, "Please wait till current upload is finished")
            return {'CANCELLED'}

        wm = context.window_manager
        props = wm.sketchfab_export
        sf_state.model_url = ""

        # Prepare to save the file
        binary_path = bpy.app.binary_path
        script_path = os.path.dirname(os.path.realpath(__file__))
        basename, ext = os.path.splitext(bpy.data.filepath)
        if not basename:
            basename = os.path.join(basename, "temp")
        if not ext:
            ext = ".blend"
        tempdir = tempfile.mkdtemp()
        filepath = os.path.join(tempdir, "export-sketchfab" + ext)

        SKETCHFAB_EXPORT_DATA_FILE = os.path.join(tempdir, "export-sketchfab.json")

        try:
            # save a copy of actual scene but don't interfere with the users models
            bpy.ops.wm.save_as_mainfile(filepath=filepath, compress=True, copy=True)

            with open(SKETCHFAB_EXPORT_DATA_FILE, 'w') as s:
                json.dump({
                        "selection": props.selection,
                        }, s)

            subprocess.check_call([
                    binary_path,
                    "--background",
                    "-noaudio",
                    filepath,
                    "--python", os.path.join(script_path, "pack_for_export.py"),
                    "--", tempdir
                    ])

            os.remove(filepath)

            # read subprocess call results
            with open(SKETCHFAB_EXPORT_DATA_FILE, 'r') as s:
                r = json.load(s)
                size = r["size"]
                props.filepath = r["filepath"]
                filename = r["filename"]

            os.remove(SKETCHFAB_EXPORT_DATA_FILE)

        except Exception as e:
            self.report({'WARNING'}, "Error occured while preparing your file: %s" % str(e))
            return {'FINISHED'}

        # Check the generated file size against the user plans, to know if the upload will succeed
        upload_limit = Config.SKETCHFAB_UPLOAD_LIMITS[get_sketchfab_props().skfb_api.plan_type]
        if get_sketchfab_props().skfb_api.use_org_profile:
            upload_limit = Config.SKETCHFAB_UPLOAD_LIMITS["ent"]
        if size > upload_limit:
            human_size_limit    = Utils.humanify_size(upload_limit)
            human_exported_size = Utils.humanify_size(size)
            self.report({'ERROR'}, "Upload size is above your plan upload limit: %s > %s" % (human_exported_size, human_size_limit))
            return {'FINISHED'}

        sf_state.uploading = True
        sf_state.size_label = Utils.humanify_size(size)
        self._thread = threading.Thread(
                target=upload,
                args=(props.filepath, filename),
                )
        self._thread.start()

        wm.modal_handler_add(self)
        self._timer = wm.event_timer_add(1.0, window=context.window)

        return {'RUNNING_MODAL'}

    def cancel(self, context):
        wm = context.window_manager
        wm.event_timer_remove(self._timer)
        self._thread.join()

def get_temporary_path():

    # Get the preferences cache directory
    cachePath = bpy.context.preferences.addons[__name__.split('.')[0]].preferences.cachePath

    # The cachePath was set in the preferences
    if cachePath:
        return cachePath
    else:
        # Rely on Blender temporary directory
        if bpy.app.version == (2, 79, 0):
            if bpy.context.user_preferences.filepaths.temporary_directory:
                return bpy.context.user_preferences.filepaths.temporary_directory
            else:
                return tempfile.mkdtemp()
        else:
            if bpy.context.preferences.filepaths.temporary_directory:
                return bpy.context.preferences.filepaths.temporary_directory
            else:
                return tempfile.mkdtemp()

def updateCacheDirectory(self, context):

    # Get the cache path from the preferences, or a default temporary
    path = os.path.abspath(get_temporary_path())

    # Delete the old directory
    # Won't delete anything upon plugin intialization, only when switching path in preferences
    if Config.SKETCHFAB_TEMP_DIR and os.path.exists(Config.SKETCHFAB_TEMP_DIR) and os.path.isdir(Config.SKETCHFAB_TEMP_DIR):
        shutil.rmtree(Config.SKETCHFAB_TEMP_DIR)

    # Create the paths and directories for temporary directories
    Config.SKETCHFAB_TEMP_DIR = os.path.join(path, "sketchfab_downloads")
    Config.SKETCHFAB_THUMB_DIR = os.path.join(Config.SKETCHFAB_TEMP_DIR, 'thumbnails')
    Config.SKETCHFAB_MODEL_DIR = os.path.join(Config.SKETCHFAB_TEMP_DIR, 'imports')
    if not os.path.exists(Config.SKETCHFAB_TEMP_DIR): os.makedirs(Config.SKETCHFAB_TEMP_DIR)
    if not os.path.exists(Config.SKETCHFAB_THUMB_DIR): os.makedirs(Config.SKETCHFAB_THUMB_DIR)
    if not os.path.exists(Config.SKETCHFAB_MODEL_DIR): os.makedirs(Config.SKETCHFAB_MODEL_DIR)

class SketchfabAddonPreferences(bpy.types.AddonPreferences):
    bl_idname = __name__
    cachePath: StringProperty(
        name="Cache folder",
        description=(
            "Temporary directory for downloads from sketchfab.com\n"
            "Set by the OS by default, make sure to have write access\n"
            "to this directory if you set it manually"
        ),
        subtype='DIR_PATH',
        update=updateCacheDirectory
    )
    downloadHistory : StringProperty(
        name="Download history file",
        description=(
            ".csv file containing your downloads from sketchfab.com\n"
            "If valid, the name, license and url of every model you\n"
            "download through the plugin will be saved in this file"
        ),
        subtype='FILE_PATH'
    )
    def draw(self, context):
        layout = self.layout
        layout.prop(self, "cachePath", text="Download directory")
        layout.prop(self, "downloadHistory", text="Download history (.csv)")

classes = (
    SketchfabAddonPreferences,

    # Properties
    SketchfabBrowserProps,
    SketchfabLoginProps,
    SketchfabBrowserPropsProxy,
    SketchfabExportProps,

    # Panels
    LoginPanel,
    TeamsPanel,
    SketchfabBrowse,
    SketchfabExportPanel,
    SketchfabPanel,

    # Operators
    SketchfabEnable,
    SketchfabCreateAccount,
    LoginModal,
    SketchfabNewVersion,
    SketchfabHelp,
    SketchfabReportIssue,
    SketchfabSearch,
    SketchfabSearchPreviousResults,
    SketchfabSearchNextResults,
    ImportModalOperator,
    ViewOnSketchfab,
    SketchfabDownloadModel,
    SketchfabLogger,
    ExportSketchfab,
    )

def check_plugin_version(request, *args, **kwargs):
    response = request.json()
    skfb = get_sketchfab_props()
    if response and len(response):
        latest_release_version = response[0]['tag_name'].replace('.', '')
        current_version = str(bl_info['version']).replace(',', '').replace('(', '').replace(')', '').replace(' ', '')

        if latest_release_version == current_version:
            print('You are using the latest version({})'.format(response[0]['tag_name']))
            skfb.is_latest_version = 1
        else:
            print('A new version is available: {}'.format(response[0]['tag_name']))
            skfb.is_latest_version = 0
    else:
        print('Failed to retrieve plugin version')
        skfb.is_latest_version = -2

def register():
    sketchfab_icon = bpy.utils.previews.new()
    icons_dir      = os.path.dirname(__file__)
    sketchfab_icon.load("skfb", os.path.join(icons_dir, "logo.png"), 'IMAGE')
    sketchfab_icon.load("0",    os.path.join(icons_dir, "placeholder.png"), 'IMAGE')

    res = []
    res.append(('NORESULTS', 'empty', "", sketchfab_icon['0'].icon_id, 0))
    preview_collection['default'] = tuple(res)
    preview_collection['skfb'] = sketchfab_icon
    bpy.types.WindowManager.result_previews = EnumProperty(items=list_current_results)

    for cls in classes:
        bpy.utils.register_class(cls)

    bpy.types.WindowManager.sketchfab_browser = PointerProperty(
                type=SketchfabBrowserProps)

    bpy.types.WindowManager.sketchfab_browser_proxy = PointerProperty(
                type=SketchfabBrowserPropsProxy)

    bpy.types.WindowManager.sketchfab_api = PointerProperty(
                type=SketchfabLoginProps,
                )

    bpy.types.WindowManager.sketchfab_export = PointerProperty(
                type=SketchfabExportProps,
                )

    # If a cache path was set in preferences, use it
    updateCacheDirectory(None, context=bpy.context)

def unregister():
    for cls in classes:
        bpy.utils.unregister_class(cls)

    del bpy.types.WindowManager.sketchfab_api
    del bpy.types.WindowManager.sketchfab_browser
    del bpy.types.WindowManager.sketchfab_browser_proxy
    del bpy.types.WindowManager.sketchfab_export

    bpy.utils.previews.remove(preview_collection['skfb'])
    del bpy.types.WindowManager.result_previews
    Utils.clean_thumbnail_directory()


if __name__ == "__main__":
    register()