Skip to content
Snippets Groups Projects
bkit_oauth.py 4.14 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 #####
    
    import bpy
    
    import threading
    import blenderkit
    from blenderkit import tasks_queue, utils, paths, search, categories, oauth
    
    CLIENT_ID = "IdFRwa3SGA8eMpzhRVFMg5Ts8sPK93xBjif93x0F"
    PORTS = [62485, 1234]
    
    def login_thread():
        thread = threading.Thread(target=login, args=([]), daemon=True)
        thread.start()
    
    
    def login():
        authenticator = oauth.SimpleOAuthAuthenticator(server_url=paths.get_bkit_url(), client_id=CLIENT_ID, ports=PORTS)
        auth_token, refresh_token = authenticator.get_new_token()
        utils.p('tokens retrieved')
        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:
            thread = threading.Thread(target=refresh_token, args=([preferences.api_key_refresh]), daemon=True)
            thread.start()
    
    
    def refresh_token(api_key_refresh):
        authenticator = oauth.SimpleOAuthAuthenticator(server_url=paths.get_bkit_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((blenderkit.bkit_oauth.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()
        props.report = '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'}
    
        @classmethod
        def poll(cls, context):
            return True
    
        def execute(self, context):
            preferences = bpy.context.preferences.addons['blenderkit'].preferences
            preferences.login_attempt = True
            login_thread()
            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)