blenderkit/bkit_oauth.py

   1 # ##### BEGIN GPL LICENSE BLOCK #####
   2 #
   3 #  This program is free software; you can redistribute it and/or
   4 #  modify it under the terms of the GNU General Public License
   5 #  as published by the Free Software Foundation; either version 2
   6 #  of the License, or (at your option) any later version.
   7 #
   8 #  This program is distributed in the hope that it will be useful,
   9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 #  GNU General Public License for more details.
  12 #
  13 #  You should have received a copy of the GNU General Public License
  14 #  along with this program; if not, write to the Free Software Foundation,
  15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  16 #
  17 # ##### END GPL LICENSE BLOCK #####
  18
  19
  20 from blenderkit import tasks_queue, utils, paths, search, categories, oauth, ui, ui_panels
  21
  22 import bpy
  23
  24 import threading
  25 import requests
  26 import time
  27 import logging
  28 bk_logger = logging.getLogger('blenderkit')
  29
  30 from bpy.props import (
  31     BoolProperty,
  32 )
  33
  34 CLIENT_ID = "IdFRwa3SGA8eMpzhRVFMg5Ts8sPK93xBjif93x0F"
  35 PORTS = [62485, 65425, 55428, 49452]
  36
  37 active_authenticator = None
  38
  39
  40 def login_thread(signup=False):
  41     global active_authenticator
  42     r_url = paths.get_oauth_landing_url()
  43     url = paths.get_bkit_url()
  44     authenticator = oauth.SimpleOAuthAuthenticator(server_url=url, client_id=CLIENT_ID, ports=PORTS)
  45     # we store authenticator globally to be able to ping the server if connection fails.
  46     active_authenticator = authenticator
  47     thread = threading.Thread(target=login, args=([signup, url, r_url, authenticator]), daemon=True)
  48     thread.start()
  49
  50
  51 def login(signup, url, r_url, authenticator):
  52     auth_token, refresh_token, oauth_response = authenticator.get_new_token(register=signup, redirect_url=r_url)
  53     bk_logger.debug('tokens retrieved')
  54     tasks_queue.add_task((write_tokens, (auth_token, refresh_token, oauth_response)))
  55
  56
  57 def refresh_token_thread():
  58     preferences = bpy.context.preferences.addons['blenderkit'].preferences
  59     if len(preferences.api_key_refresh) > 0 and preferences.refresh_in_progress == False:
  60         preferences.refresh_in_progress = True
  61         url = paths.get_bkit_url()
  62         thread = threading.Thread(target=refresh_token, args=([preferences.api_key_refresh, url]), daemon=True)
  63         thread.start()
  64     else:
  65         ui.add_report('Already Refreshing token, will be ready soon. If this fails, please login again in Login panel.')
  66
  67
  68 def refresh_token(api_key_refresh, url):
  69     authenticator = oauth.SimpleOAuthAuthenticator(server_url=url, client_id=CLIENT_ID, ports=PORTS)
  70     auth_token, refresh_token, oauth_response = authenticator.get_refreshed_token(api_key_refresh)
  71     if auth_token is not None and refresh_token is not None:
  72         tasks_queue.add_task((write_tokens, (auth_token, refresh_token, oauth_response)))
  73     return auth_token, refresh_token, oauth_response
  74
  75
  76 def write_tokens(auth_token, refresh_token, oauth_response):
  77     bk_logger.debug('writing tokens')
  78     preferences = bpy.context.preferences.addons['blenderkit'].preferences
  79     preferences.api_key_refresh = refresh_token
  80     preferences.api_key = auth_token
  81     preferences.api_key_timeout = time.time() + oauth_response['expires_in']
  82     preferences.api_key_life = oauth_response['expires_in']
  83     preferences.login_attempt = False
  84     preferences.refresh_in_progress = False
  85     props = utils.get_search_props()
  86     if props is not None:
  87         props.report = ''
  88     ui.add_report('BlenderKit Re-Login success')
  89     search.get_profile()
  90     categories.fetch_categories_thread(auth_token, force = False)
  91
  92
  93 class RegisterLoginOnline(bpy.types.Operator):
  94     """Login online on BlenderKit webpage"""
  95
  96     bl_idname = "wm.blenderkit_login"
  97     bl_label = "BlenderKit login/signup"
  98     bl_options = {'REGISTER', 'UNDO'}
  99
 100     signup: BoolProperty(
 101         name="create a new account",
 102         description="True for register, otherwise login",
 103         default=False,
 104         options={'SKIP_SAVE'}
 105     )
 106
 107     message: bpy.props.StringProperty(
 108         name="Message",
 109         description="",
 110         default="You were logged out from BlenderKit.\n Clicking OK takes you to web login. ")
 111
 112     @classmethod
 113     def poll(cls, context):
 114         return True
 115
 116     def draw(self, context):
 117         layout = self.layout
 118         utils.label_multiline(layout, text=self.message, width = 300)
 119
 120     def execute(self, context):
 121         preferences = bpy.context.preferences.addons['blenderkit'].preferences
 122         preferences.login_attempt = True
 123         login_thread(self.signup)
 124         return {'FINISHED'}
 125
 126     def invoke(self, context, event):
 127         wm = bpy.context.window_manager
 128         preferences = bpy.context.preferences.addons['blenderkit'].preferences
 129         preferences.api_key_refresh = ''
 130         preferences.api_key = ''
 131         return wm.invoke_props_dialog(self)
 132
 133
 134 class Logout(bpy.types.Operator):
 135     """Logout from BlenderKit immediately"""
 136
 137     bl_idname = "wm.blenderkit_logout"
 138     bl_label = "BlenderKit logout"
 139     bl_options = {'REGISTER', 'UNDO'}
 140
 141     @classmethod
 142     def poll(cls, context):
 143         return True
 144
 145     def execute(self, context):
 146         preferences = bpy.context.preferences.addons['blenderkit'].preferences
 147         preferences.login_attempt = False
 148         preferences.api_key_refresh = ''
 149         preferences.api_key = ''
 150         if bpy.context.window_manager.get('bkit profile'):
 151             del (bpy.context.window_manager['bkit profile'])
 152         return {'FINISHED'}
 153
 154
 155 class CancelLoginOnline(bpy.types.Operator):
 156     """Cancel login attempt"""
 157
 158     bl_idname = "wm.blenderkit_login_cancel"
 159     bl_label = "BlenderKit login cancel"
 160     bl_options = {'REGISTER', 'UNDO'}
 161
 162     @classmethod
 163     def poll(cls, context):
 164         return True
 165
 166     def execute(self, context):
 167         global active_authenticator
 168         preferences = bpy.context.preferences.addons['blenderkit'].preferences
 169         preferences.login_attempt = False
 170         try:
 171             if active_authenticator is not None:
 172                 requests.get(active_authenticator.redirect_uri)
 173                 active_authenticator = None
 174         except Exception as e:
 175             print('stopped login attempt')
 176             print(e)
 177         return {'FINISHED'}
 178
 179
 180 classes = (
 181     RegisterLoginOnline,
 182     CancelLoginOnline,
 183     Logout,
 184 )
 185
 186
 187 def register():
 188     for c in classes:
 189         bpy.utils.register_class(c)
 190
 191
 192 def unregister():
 193     for c in classes:
 194         bpy.utils.unregister_class(c)