Add support for token.json file

- Before this change, DRIVE_TOKEN env variable was only way for authorization
- Now adding token.json file to repo will also work
- TOKEN_JSON_URL env variable can be used for authorization too
- Change DRIVE_TOKEN env variable from Required Config to Optional Config

bot/__init__.py

@@ -61,6 +61,22 @@ def get_config(name: str):
 except:
     pass
 
+try:
+    TOKEN_JSON_URL = get_config('TOKEN_JSON_URL')
+    if len(TOKEN_JSON_URL) == 0:
+        TOKEN_JSON_URL = None
+    else:
+        res = requests.get(TOKEN_JSON_URL)
+        if res.status_code == 200:
+            with open('token.json', 'wb+') as f:
+                f.write(res.content)
+                f.close()
+        else:
+            LOGGER.error(f"Failed to load token.json file [{res.status_code}]")
+            raise KeyError
+except KeyError:
+    pass
+
 try:
     BOT_TOKEN = get_config('BOT_TOKEN')
     OWNER_ID = int(get_config('OWNER_ID'))
@@ -75,13 +91,11 @@ def get_config(name: str):
     exit(1)
 
 try:
-    if len(get_config('DRIVE_TOKEN')) == 0 or str(get_config('DRIVE_TOKEN')).lower() == "empty":
-        LOGGER.error("DRIVE_TOKEN var is missing")
-        exit(1)
-    with open('token.json', 'wt') as f:
-        f.write(get_config('DRIVE_TOKEN').replace("\n",""))
+    if not os.path.exists('token.json'):
+        with open('token.json', 'wt') as f:
+            f.write(get_config('DRIVE_TOKEN').replace("\n", ""))
 except:
-    LOGGER.error("Failed to create token.json file")
+    LOGGER.error("token.json file is missing")
     exit(1)
 
 try:

config_sample.env

@@ -1,15 +1,16 @@
 # REQUIRED CONFIG
 BOT_TOKEN=
 OWNER_ID=
-DRIVE_TOKEN=
 DRIVE_FOLDER_ID=
 TELEGRAPH_ACCS=
 # OPTIONAL CONFIG
+DRIVE_TOKEN=
 AUTHORIZED_CHATS=
 DATABASE_URL=
 IS_TEAM_DRIVE=
 USE_SERVICE_ACCOUNTS=
 DRIVE_INDEX_URL=
+TOKEN_JSON_URL=
 ACCOUNTS_ZIP_URL=
 DRIVE_LIST_URL=
 APPDRIVE_EMAIL=