Skip to content
Snippets Groups Projects
bkit_oauth.py 4.84 KiB
Newer Older
  • Learn to ignore specific revisions
  • # ##### BEGIN GPL LICENSE BLOCK #####
    #
    #  This program is free software; you can redistribute it and/or
    #  modify it under the terms of the GNU General Public License
    #  as published by the Free Software Foundation; either version 2
    #  of the License, or (at your option) any later version.
    #
    #  This program is distributed in the hope that it will be useful,
    #  but WITHOUT ANY WARRANTY; without even the implied warranty of
    #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    #  GNU General Public License for more details.
    #
    #  You should have received a copy of the GNU General Public License
    #  along with this program; if not, write to the Free Software Foundation,
    #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
    #
    # ##### END GPL LICENSE BLOCK #####
    
    
    if "bpy" in locals():
        from importlib import reload
    
        tasks_queue = reload(tasks_queue)
        utils = reload(utils)
        paths = reload(paths)
        search = reload(search)
        categories = reload(categories)
        oauth = reload(oauth)
    
        from blenderkit import tasks_queue, utils, paths, search, categories, oauth, ui
    
    
    from bpy.props import (
        BoolProperty,
    )
    
    
    CLIENT_ID = "IdFRwa3SGA8eMpzhRVFMg5Ts8sPK93xBjif93x0F"
    
    PORTS = [62485, 65425, 55428, 49452]
    
    
    def login_thread(signup=False):
    
        r_url = paths.get_oauth_landing_url()
        url = paths.get_bkit_url()
        thread = threading.Thread(target=login, args=([signup, url, r_url]), daemon=True)
    
    def login(signup, url, r_url):
        authenticator = oauth.SimpleOAuthAuthenticator(server_url=url, client_id=CLIENT_ID, ports=PORTS)
        auth_token, refresh_token = authenticator.get_new_token(register=signup, redirect_url=r_url)
    
        tasks_queue.add_task((write_tokens, (auth_token, refresh_token)))
    
    
    
    def refresh_token_thread():
        preferences = bpy.context.preferences.addons['blenderkit'].preferences
        if len(preferences.api_key_refresh) > 0:
    
            url = paths.get_bkit_url()
            thread = threading.Thread(target=refresh_token, args=([preferences.api_key_refresh, url]), daemon=True)
    
    def refresh_token(api_key_refresh, url):
        authenticator = oauth.SimpleOAuthAuthenticator(server_url=url, client_id=CLIENT_ID, ports=PORTS)
    
        auth_token, refresh_token = authenticator.get_refreshed_token(api_key_refresh)
        if auth_token is not None and refresh_token is not None:
    
            tasks_queue.add_task((write_tokens, (auth_token, refresh_token)))
    
    
    
    def write_tokens(auth_token, refresh_token):
        utils.p('writing tokens')
        preferences = bpy.context.preferences.addons['blenderkit'].preferences
        preferences.api_key_refresh = refresh_token
        preferences.api_key = auth_token
        preferences.login_attempt = False
        props = utils.get_search_props()
    
        if props is not None:
            props.report = ''
    
        ui.add_report('BlenderKit Login success')
    
        search.get_profile()
        categories.fetch_categories_thread(auth_token)
    
    
    class RegisterLoginOnline(bpy.types.Operator):
        """Bring linked object hierarchy to scene and make it editable."""
    
        bl_idname = "wm.blenderkit_login"
        bl_label = "BlenderKit login or signup"
        bl_options = {'REGISTER', 'UNDO'}
    
    
        signup: BoolProperty(
            name="create a new account",
            description="True for register, otherwise login",
            default=False,
            options={'SKIP_SAVE'}
        )
    
    
        @classmethod
        def poll(cls, context):
            return True
    
        def execute(self, context):
            preferences = bpy.context.preferences.addons['blenderkit'].preferences
            preferences.login_attempt = True
    
            login_thread(self.signup)
    
            return {'FINISHED'}
    
    
    class Logout(bpy.types.Operator):
        """Bring linked object hierarchy to scene and make it editable."""
    
        bl_idname = "wm.blenderkit_logout"
        bl_label = "BlenderKit logout"
        bl_options = {'REGISTER', 'UNDO'}
    
        @classmethod
        def poll(cls, context):
            return True
    
        def execute(self, context):
            preferences = bpy.context.preferences.addons['blenderkit'].preferences
            preferences.login_attempt = False
            preferences.api_key_refresh = ''
            preferences.api_key = ''
            return {'FINISHED'}
    
    
    class CancelLoginOnline(bpy.types.Operator):
        """Cancel login attempt."""
    
        bl_idname = "wm.blenderkit_login_cancel"
        bl_label = "BlenderKit login cancel"
        bl_options = {'REGISTER', 'UNDO'}
    
        @classmethod
        def poll(cls, context):
            return True
    
        def execute(self, context):
            preferences = bpy.context.preferences.addons['blenderkit'].preferences
            preferences.login_attempt = False
            return {'FINISHED'}
    
    
    classess = (
        RegisterLoginOnline,
        CancelLoginOnline,
        Logout,
    )
    
    
    def register():
        for c in classess:
            bpy.utils.register_class(c)
    
    
    def unregister():
        for c in classess:
            bpy.utils.unregister_class(c)