499 lines
21 KiB
Python
499 lines
21 KiB
Python
import re
|
|
import requests
|
|
from dataclasses import dataclass
|
|
from requests import Session
|
|
from argparse import ArgumentParser
|
|
from sys import argv, stdout
|
|
from colorama import Fore, init
|
|
from telegram import Update
|
|
from telegram.ext import ApplicationBuilder, ContextTypes, CommandHandler
|
|
import logging
|
|
import os
|
|
from datetime import datetime
|
|
|
|
CUSTOM_INIT_PATH = '/custom-cont_init.d/'
|
|
CUSTOM_SERVICES_PATH = '/custom-services.d/'
|
|
STATUS_FALLBACK_LOCATION = '/custom-services.d/python/ARLStatus.txt'
|
|
EXTENDED_CONF_PATH = '/config/extended.conf'
|
|
NOT_FOUND_PATH = '/config/extended/logs/notfound'
|
|
FAILED_DOWNLOADS_PATH = '/config/extended/logs/downloaded/failed/deezer'
|
|
LOG_FILES_DIRECTORY = '/config/logs'
|
|
DEBUG_ROOT_PATH = './env'
|
|
|
|
# Web agent used to access Deezer
|
|
USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:83.0) Gecko/20100101 Firefox/110.0'
|
|
|
|
|
|
@dataclass
|
|
class Plan:
|
|
name: str
|
|
expires: str
|
|
active: bool
|
|
download: bool
|
|
lossless: bool
|
|
explicit: bool
|
|
|
|
|
|
@dataclass
|
|
class Account:
|
|
id: int
|
|
token: str
|
|
country: str
|
|
plan: Plan
|
|
|
|
|
|
class AuthError(Exception):
|
|
pass
|
|
|
|
|
|
class ParseError(Exception):
|
|
pass
|
|
|
|
|
|
class ServiceError(Exception):
|
|
pass
|
|
|
|
|
|
class DeezerPlatformProvider:
|
|
NAME = 'Deezer'
|
|
|
|
BASE_URL = 'http://www.deezer.com'
|
|
API_PATH = '/ajax/gw-light.php'
|
|
SESSION_DATA = {
|
|
'api_token': 'null',
|
|
'api_version': '1.0',
|
|
'input': '3',
|
|
'method': 'deezer.getUserData'
|
|
}
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.log = logging.getLogger('ARLChecker')
|
|
self.session = Session()
|
|
self.session.headers.update({'User-Agent': USER_AGENT})
|
|
|
|
def login(self, username, secret):
|
|
try:
|
|
res = self.session.post(
|
|
self.BASE_URL + self.API_PATH,
|
|
cookies={'arl': secret},
|
|
data=self.SESSION_DATA
|
|
)
|
|
res.raise_for_status()
|
|
except Exception as error:
|
|
self.log.error(Fore.RED + 'Could not connect! Service down, API changed, wrong credentials or code-related issue.' + Fore.RESET)
|
|
raise ConnectionError()
|
|
|
|
self.session.cookies.clear()
|
|
|
|
try:
|
|
res = res.json()
|
|
except Exception as error:
|
|
self.log.error(Fore.RED + "Could not parse JSON response from DEEZER!" + Fore.RESET)
|
|
raise ParseError()
|
|
|
|
if 'error' in res and res['error']:
|
|
self.log.error(Fore.RED + "Deezer returned the following error:{}".format(res["error"]) + Fore.RESET)
|
|
raise ServiceError()
|
|
|
|
res = res['results']
|
|
|
|
if res['USER']['USER_ID'] == 0:
|
|
raise AuthError()
|
|
|
|
return Account(username, secret, res['COUNTRY'], Plan(
|
|
res['OFFER_NAME'],
|
|
'Unknown',
|
|
True,
|
|
True,
|
|
res['USER']['OPTIONS']['web_sound_quality']['lossless'],
|
|
res['USER']['EXPLICIT_CONTENT_LEVEL']
|
|
))
|
|
|
|
|
|
class LidarrExtendedAPI:
|
|
# sets new token to extended.conf
|
|
def __init__(self):
|
|
self.root = ''
|
|
self.log = logging.getLogger('ARLChecker')
|
|
self.newARLToken = None
|
|
self.currentARLToken = None
|
|
self.arlLineText = None
|
|
self.arlLineIndex = None
|
|
self.fileText = None
|
|
self.enable_telegram_bot = False
|
|
self.telegram_bot_running = False
|
|
self.telegram_bot_token = None
|
|
self.telegram_user_chat_id = None
|
|
self.telegram_bot_enable_line_text = None
|
|
self.telegram_bot_enable_line_index = None
|
|
self.bot = None
|
|
self.enable_pushover_notify = False
|
|
self.pushover_user_key = None
|
|
self.pushover_app_api_key = None
|
|
self.enable_ntfy_notify = False
|
|
self.ntfy_sever_topic = None
|
|
self.ntfy_user_token = None
|
|
|
|
def parse_extended_conf(self):
|
|
self.currentARLToken = None
|
|
arl_token_match = None
|
|
deezer_active = False
|
|
re_search_pattern = r'"([^"]*)"'
|
|
try: # Try to open extended.conf and read all text into a var.
|
|
with open(self.root+EXTENDED_CONF_PATH, 'r', encoding='utf-8') as file:
|
|
self.fileText = file.readlines()
|
|
file.close()
|
|
except:
|
|
self.log.error(f"Could not find {self.root+EXTENDED_CONF_PATH}")
|
|
exit(1)
|
|
# Ensure Deezer is enabled and ARL token is populated
|
|
for line in self.fileText:
|
|
if 'dlClientSource="deezer"' in line or 'dlClientSource="both"' in line:
|
|
deezer_active = True
|
|
if 'arlToken=' in line:
|
|
self.arlLineText = line
|
|
self.arlLineIndex = self.fileText.index(self.arlLineText)
|
|
arl_token_match = re.search(re_search_pattern, line)
|
|
break
|
|
|
|
# ARL Token wrong flag error handling.
|
|
if arl_token_match is None:
|
|
self.log.error("ARL Token not found in extended.conf. Exiting.")
|
|
exit(1)
|
|
elif deezer_active is False:
|
|
self.log.error("Deezer not set as an active downloader in extended.conf. Exiting.")
|
|
file.close()
|
|
exit(1)
|
|
self.currentARLToken = arl_token_match[0]
|
|
self.log.info('ARL Found in extended.conf')
|
|
|
|
for line in self.fileText:
|
|
if 'telegramBotEnable=' in line:
|
|
self.telegram_bot_enable_line_text = line
|
|
self.telegram_bot_enable_line_index = self.fileText.index(self.telegram_bot_enable_line_text)
|
|
self.enable_telegram_bot = re.search(re_search_pattern, line)[0].replace('"', '').lower() in 'true'
|
|
if 'telegramBotToken=' in line:
|
|
self.telegram_bot_token = re.search(re_search_pattern, line)[0].replace('"', '')
|
|
if 'telegramUserChatID=' in line:
|
|
self.telegram_user_chat_id = re.search(re_search_pattern, line)[0].replace('"', '')
|
|
if 'pushoverEnable=' in line:
|
|
self.enable_pushover_notify = re.search(re_search_pattern, line)[0].replace('"', '').lower() in 'true'
|
|
if 'pushoverUserKey=' in line:
|
|
self.pushover_user_key = re.search(re_search_pattern, line)[0].replace('"', '')
|
|
if 'pushoverAppAPIKey=' in line:
|
|
self.pushover_app_api_key = re.search(re_search_pattern, line)[0].replace('"', '')
|
|
if 'ntfyEnable=' in line:
|
|
self.enable_ntfy_notify = re.search(re_search_pattern, line)[0].replace('"', '').lower() in 'true'
|
|
if 'ntfyServerTopic=' in line:
|
|
self.ntfy_sever_topic = re.search(re_search_pattern, line)[0].replace('"', '')
|
|
if 'ntfyUserToken=' in line:
|
|
self.ntfy_user_token = re.search(re_search_pattern, line)[0].replace('"', '')
|
|
|
|
if self.enable_telegram_bot:
|
|
self.log.info(Fore.GREEN+'Telegram bot is enabled.'+Fore.RESET)
|
|
if self.telegram_bot_token is None or self.telegram_user_chat_id is None:
|
|
self.log.error('Telegram bot token or user chat ID not set in extended.conf. Exiting')
|
|
exit(1)
|
|
else:
|
|
self.log.info('Telegram bot is disabled.')
|
|
|
|
# Report Notify/Bot Enable
|
|
if self.enable_pushover_notify:
|
|
self.log.info(Fore.GREEN+'Pushover notify is enabled.'+Fore.RESET)
|
|
else:
|
|
self.log.info('Pushover notify is disabled.')
|
|
if self.enable_ntfy_notify:
|
|
self.log.info(Fore.GREEN+'ntfy notify is enabled.'+Fore.RESET)
|
|
else:
|
|
self.log.info('ntfy notify is disabled.')
|
|
|
|
def check_token_wrapper(self): # adds Lidarr_extended specific logging and actions around check_token
|
|
self.log.info("Checking ARL Token from extended.conf")
|
|
if self.currentARLToken == '""':
|
|
self.log.info(Fore.YELLOW+"No ARL Token set in Extended.conf"+Fore.RESET)
|
|
self.report_status("NOT SET")
|
|
exit(0)
|
|
if self.currentARLToken is None:
|
|
self.log.error('Invalid ARL Token Entry (None Object)')
|
|
return False
|
|
validity_results = check_token(self.currentARLToken)
|
|
if validity_results is True:
|
|
self.report_status('VALID') # For text fallback method
|
|
else:
|
|
self.report_status('EXPIRED')
|
|
self.log.error(Fore.RED + 'Update the token in extended.conf' + Fore.RESET)
|
|
if self.telegram_bot_running: # Don't re-start the telegram bot if it's already running after bot invalid token entry
|
|
return False
|
|
if self.enable_pushover_notify:
|
|
pushover_notify(self.pushover_app_api_key, self.pushover_user_key, '---\U0001F6A8WARNING\U0001F6A8-----\nARL TOKEN EXPIRED\n Update arlToken in extended.conf"\n You can find a new ARL at:\nhttps://rentry.org/firehawk52#deezer-arls')
|
|
if self.enable_ntfy_notify:
|
|
ntfy_notify(self.ntfy_sever_topic, '---\U0001F6A8WARNING\U0001F6A8-----\nARL TOKEN EXPIRED\n Update arlToken in extended.conf\n You can find a new ARL at:\nhttps://rentry.org/firehawk52#deezer-arls', self.ntfy_user_token)
|
|
if self.enable_telegram_bot:
|
|
self.log.info(Fore.YELLOW + 'Starting Telegram bot...Check Telegram and follow instructions.' + Fore.RESET)
|
|
self.telegram_bot_running = True
|
|
self.start_telegram_bot()
|
|
exit(420)
|
|
|
|
def set_new_token(self): # Re-writes extended.conf with previously read-in text, replacing w/ new ARL
|
|
self.fileText[self.arlLineIndex] = self.arlLineText.replace(self.currentARLToken, self.newARLToken)
|
|
with open(self.root+EXTENDED_CONF_PATH, 'w', encoding='utf-8') as file:
|
|
file.writelines(self.fileText)
|
|
file.close()
|
|
self.log.info("New ARL token written to extended.conf")
|
|
self.parse_extended_conf()
|
|
|
|
# After new token is set, clean up notfound and failed downloads to bypass the default 30 day wait
|
|
def clear_not_found(self):
|
|
paths = [self.root + NOT_FOUND_PATH, self.root+FAILED_DOWNLOADS_PATH]
|
|
for path in paths:
|
|
for file in os.listdir(path):
|
|
file_to_delete = os.path.join(path, file)
|
|
os.remove(file_to_delete)
|
|
|
|
def report_status(self, status):
|
|
f = open(self.root+STATUS_FALLBACK_LOCATION, "w")
|
|
now = datetime.strftime(datetime.now(), "%b-%d-%Y at %H:%M:%S")
|
|
f.write(f"{now}: ARL Token is {status}.{' Please update arlToken in extended.conf' if status=='EXPIRED' else ''}")
|
|
f.close()
|
|
|
|
def start_telegram_bot(self):
|
|
try:
|
|
self.bot = TelegramBotControl(self, self.telegram_bot_token, self.telegram_user_chat_id)
|
|
except Exception as e:
|
|
if 'Chat not found' in str(e) or 'Chat_id' in str(e):
|
|
self.log.error(
|
|
Fore.RED + "Telegram Bot: Chat not found. Check your chat ID in extended.conf, or start a chat with your bot." + Fore.RESET)
|
|
elif 'The token' in str(e):
|
|
self.log.error(Fore.RED + "Telegram Bot: Check your Bot Token in extended.conf." + Fore.RESET)
|
|
else:
|
|
self.log.error('Telegram Bot: ' + str(e))
|
|
|
|
def disable_telegram_bot(self):
|
|
compiled = re.compile(re.escape('true'), re.IGNORECASE)
|
|
self.fileText[self.telegram_bot_enable_line_index] = compiled.sub('false', self.telegram_bot_enable_line_text)
|
|
with open(self.root+EXTENDED_CONF_PATH, 'w', encoding='utf-8') as file:
|
|
file.writelines(self.fileText)
|
|
file.close()
|
|
self.log.info("Telegram Bot Disabled.")
|
|
|
|
|
|
class TelegramBotControl:
|
|
def __init__(self, parent, telegram_bot_token, telegram_user_chat_id):
|
|
self.log = logging.getLogger('ARLChecker')
|
|
self.parent = parent
|
|
self.telegram_bot_token = telegram_bot_token
|
|
self.telegram_chat_id = telegram_user_chat_id
|
|
|
|
# Send initial notification
|
|
async def send_expired_token_notification(application):
|
|
await application.bot.sendMessage(chat_id=self.telegram_chat_id, text='---\U0001F6A8WARNING\U0001F6A8-----\nARL TOKEN EXPIRED\n Update Token by running "/set_token <TOKEN>"\n You can find a new ARL at:\nhttps://rentry.org/firehawk52#deezer-arls\n\n\n Other Commands:\n/cancel - Cancel this session\n/disable - Disable Telegram Bot', disable_web_page_preview=True)
|
|
self.log.info(Fore.YELLOW + "Telegram Bot Sent ARL Token Expiry Message " + Fore.RESET)
|
|
# TODO: Get Chat ID/ test on new bot
|
|
|
|
# start bot control
|
|
self.application = ApplicationBuilder().token(self.telegram_bot_token).post_init(send_expired_token_notification).build()
|
|
token_handler = CommandHandler('set_token', self.set_token)
|
|
cancel_handler = CommandHandler('cancel', self.cancel)
|
|
disable_handler = CommandHandler('disable', self.disable_bot)
|
|
self.application.add_handler(token_handler)
|
|
self.application.add_handler(cancel_handler)
|
|
self.application.add_handler(disable_handler)
|
|
self.application.run_polling(allowed_updates=Update.ALL_TYPES)
|
|
|
|
async def disable_bot(self, update, context: ContextTypes.DEFAULT_TYPE):
|
|
self.parent.disable_telegram_bot()
|
|
await update.message.reply_text('Disabled Telegram Bot. \U0001F614\nIf you would like to re-enable,\nset telegramBotEnable to true\nin extended.conf')
|
|
self.log.info(Fore.YELLOW + 'Telegram Bot: Send Disable Bot Message :(' + Fore.RESET)
|
|
self.application.stop_running()
|
|
|
|
async def cancel(self, update, context: ContextTypes.DEFAULT_TYPE):
|
|
await update.message.reply_text('Canceling...ARLToken is still expired.')
|
|
self.log.info(Fore.YELLOW + 'Telegram Bot: Canceling...ARLToken is still expired.' + Fore.RESET)
|
|
try:
|
|
self.application.stop_running()
|
|
except Exception:
|
|
pass
|
|
|
|
async def set_token(self, update, context: ContextTypes.DEFAULT_TYPE):
|
|
async def send_message(text, reply=False):
|
|
if reply is True:
|
|
await update.message.reply_text(text=text)
|
|
else:
|
|
await context.bot.send_message(chat_id=update.effective_chat.id, text=text)
|
|
self.log.info(Fore.YELLOW+"Telegram Bot: " + text + Fore.RESET)
|
|
try:
|
|
new_token = update.message.text.split('/set_token ')[1]
|
|
if new_token == '':
|
|
raise Exception
|
|
except:
|
|
await update.message.reply_text('Invalid Entry... Please try again.')
|
|
return
|
|
self.log.info(Fore.YELLOW+f"Telegram Bot:Token received: {new_token}" + Fore.RESET)
|
|
token_validity = check_token(new_token)
|
|
if token_validity:
|
|
await send_message("ARL valid, applying...")
|
|
self.parent.newARLToken = '"'+new_token+'"'
|
|
self.parent.set_new_token()
|
|
await send_message("Checking configuration...")
|
|
# reparse extended.conf
|
|
self.parent.parse_extended_conf()
|
|
token_validity = check_token(self.parent.currentARLToken)
|
|
if token_validity:
|
|
await send_message("ARL Token Updated! \U0001F44D", reply=True)
|
|
try:
|
|
self.application.stop_running()
|
|
except Exception:
|
|
pass
|
|
|
|
else: # If Token invalid
|
|
await send_message("Token expired or invalid. Try another token.", reply=True)
|
|
return
|
|
|
|
|
|
def pushover_notify(api_token, user_key, message): # Send Notification to Pushover
|
|
log = logging.getLogger('ARLChecker') # Get Logging
|
|
log.info(Fore.YELLOW + 'Attempting Pushover notification' + Fore.RESET)
|
|
response = requests.post("https://api.pushover.net/1/messages.json", data={
|
|
"token": api_token,
|
|
"user": user_key,
|
|
"message": message
|
|
})
|
|
if response.json()['status'] == 1:
|
|
log.info(Fore.GREEN+"Pushover notification sent successfully"+Fore.RESET)
|
|
else:
|
|
for message_error in response.json()['errors']:
|
|
log.error(Fore.RED+f"Pushover Response: {message_error}"+ Fore.RESET)
|
|
|
|
|
|
def ntfy_notify(server_plus_topic, message, token): # Send Notification to ntfy topic
|
|
log = logging.getLogger('ARLChecker') # Get Logging
|
|
log.info(Fore.YELLOW + 'Attempting ntfy notification' + Fore.RESET)
|
|
try:
|
|
requests.post(server_plus_topic,
|
|
data=message.encode(encoding='utf-8'),
|
|
headers={"Authorization": f"Bearer {token}"}
|
|
)
|
|
log.info(Fore.GREEN + 'ntfy notification sent successfully' + Fore.RESET)
|
|
except Exception as e:
|
|
if "Failed to resolve" in str(e):
|
|
log.error(Fore.RED + "ntfy ERROR: Check if server and user token correct in extended.conf"+Fore.RESET)
|
|
else:
|
|
log.error(Fore.RED + "NTFY ERROR: "+str(e)+Fore.RESET)
|
|
|
|
|
|
def check_token(token=None):
|
|
log = logging.getLogger('ARLChecker')
|
|
log.info(f"ARL Token to check: {token}")
|
|
log.info('Checking ARL Token Validity...')
|
|
try:
|
|
deezer_check = DeezerPlatformProvider()
|
|
account = deezer_check.login('', token.replace('"', ''))
|
|
if account.plan:
|
|
log.info(Fore.GREEN + f'Deezer Account Found.' + Fore.RESET)
|
|
log.info('-------------------------------')
|
|
log.info(f'Plan: {account.plan.name}')
|
|
log.info(f'Expiration: {account.plan.expires}')
|
|
log.info(f'Active: {Fore.GREEN+"Y" if account.plan.active else "N"}'+Fore.RESET)
|
|
log.info(f'Download: {Fore.GREEN+"Y" if account.plan.download else Fore.RED+"N"}'+Fore.RESET)
|
|
log.info(f'Lossless: {Fore.GREEN+"Y" if account.plan.lossless else Fore.RED+"N"}'+Fore.RESET)
|
|
log.info(f'Explicit: {Fore.GREEN+"Y" if account.plan.explicit else Fore.RED+"N"}'+Fore.RESET)
|
|
log.info('-------------------------------')
|
|
return True
|
|
except Exception as e:
|
|
if type(e) is AuthError:
|
|
log.error(Fore.RED + 'ARL Token Invalid/Expired.' + Fore.RESET)
|
|
return False
|
|
else:
|
|
log.error(e)
|
|
return
|
|
|
|
|
|
def parse_arguments():
|
|
parser = ArgumentParser(prog='Account Checker', description='Lidarr Extended Deezer ARL Token Tools')
|
|
parser.add_argument('-c', '--check', help='Check if currently set ARL Token is active/valid', required=False, default=False, action='store_true')
|
|
parser.add_argument('-n', '--new_token', help='Set new ARL Token', type=str, required=False, default=False)
|
|
parser.add_argument('-t', '--test_token', help='Test any token for validity', type=str, required=False, default=False)
|
|
parser.add_argument('-d', '--debug', help='For debug and development, sets root path to match testing env. See DEBUG_ROOT_PATH', required=False, default=False, action='store_true')
|
|
|
|
if not argv[1:]:
|
|
parser.print_help()
|
|
parser.exit()
|
|
|
|
return parser, parser.parse_args()
|
|
|
|
|
|
def get_version(root):
|
|
# Pull script version from bash script. will likely change this to a var passthrough
|
|
with open(root+CUSTOM_SERVICES_PATH+"ARLChecker", "r") as r:
|
|
for line in r:
|
|
if 'scriptVersion' in line:
|
|
return re.search(r'"([A-Za-z0-9_./\\-]*)"', line)[0].replace('"', '')
|
|
logging.error('Script Version not found! Exiting...')
|
|
exit(1)
|
|
|
|
|
|
def get_active_log(root):
|
|
# Get current log file
|
|
path = root + LOG_FILES_DIRECTORY
|
|
latest_file = max([os.path.join(path, f) for f in os.listdir(path) if 'ARLChecker' in f], key=os.path.getctime)
|
|
return latest_file
|
|
|
|
|
|
def init_logging(version, log_file_path):
|
|
# Logging Setup
|
|
logging.basicConfig(
|
|
format=f'%(asctime)s :: ARLChecker :: {version} :: %(levelname)s :: %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S',
|
|
level=logging.INFO,
|
|
handlers=[
|
|
logging.StreamHandler(stdout),
|
|
logging.FileHandler(log_file_path, mode="a", encoding='utf-8')
|
|
]
|
|
)
|
|
logger = logging.getLogger('ARLChecker')
|
|
|
|
# Initialize colorama
|
|
init(autoreset=True)
|
|
logger.info(Fore.GREEN + 'Logger initialized'+Fore.RESET)
|
|
|
|
return logger
|
|
|
|
|
|
def main():
|
|
root = ''
|
|
parser, args = parse_arguments()
|
|
if args.debug is True: # If debug flag set, works with IDE structure
|
|
root = DEBUG_ROOT_PATH
|
|
log = init_logging(get_version(root), get_active_log(root))
|
|
|
|
try:
|
|
if args.test_token:
|
|
log.info(Fore.CYAN+"CLI Token Tester"+Fore.RESET)
|
|
check_token(args.test_token)
|
|
exit(0)
|
|
arl_checker_instance = LidarrExtendedAPI()
|
|
arl_checker_instance.root = root
|
|
if args.check is True:
|
|
if arl_checker_instance.currentARLToken == '':
|
|
log.error("ARL Token not set. re-run with -n flag")
|
|
arl_checker_instance.parse_extended_conf()
|
|
arl_checker_instance.check_token_wrapper()
|
|
|
|
elif args.new_token:
|
|
if args.new_token == '':
|
|
log.error('Please pass new ARL token as an argument')
|
|
exit(96)
|
|
arl_checker_instance.newARLToken = '"'+args.new_token+'"'
|
|
arl_checker_instance.parse_extended_conf()
|
|
arl_checker_instance.set_new_token()
|
|
|
|
else:
|
|
parser.print_help()
|
|
except Exception as e:
|
|
logging.error(e, exc_info=True)
|
|
exit(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|