BlenderKit: split oauth into the original library and it's usage in blenderkit. (the lib is on Github as a new project)

1 files changed, 133 insertions, 0 deletions

diff --git a/blenderkit/bkit_oauth.py b/blenderkit/bkit_oauth.py
new file mode 100644
index 00000000..14274b04
--- /dev/null
+++ b/blenderkit/bkit_oauth.py
@@ -0,0 +1,133 @@
+# ##### 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.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)