functions.py

import subprocess
from variables import *
from spotify import get_track_image
import requests
from log import *
import telebot
import shutil

def download(track_link):
    try:
        # download track
        print("start downloading: " + track_link)
        normal_download_command = ['../spotdl', "--bitrate", "320k", track_link] # nomal download
        warp_download_command = ['proxychains', '-f', '../proxychains.conf', '../spotdl', "--bitrate", "320k", track_link] # download with warp
        if warp_mode:
            command = warp_download_command # download with proxychains and warp from port 40000
        else:
            command = normal_download_command # normal download
        # download in a subprocess with a set timeout
        subprocess.run(command, cwd=directory, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, timeout=300)
        print("end downloading: " + track_link)

        # download cover image
        image_url = get_track_image(track_link)
        print("start downloading cover image: " + image_url)
        subprocess.run(f"wget -O cover.jpg -o /dev/null \"{image_url}\"", shell=True, cwd=directory, timeout=300)
        print("end downloading cover image: " + image_url)

    except Exception as e:
        # more info: apparently they give "SongError: Track no longer exists"
        # I can handle them more properly later without passing them to this function in the first place
        # e.g. by using spotify api
        log(bot_name + " error in download function (not critical and important). track: " + track_link)
        return

def file_list(directory):
    file_list = []
    for file in os.listdir(directory):
        if file.endswith(".mp3"):
            file_list.append(file)
    return file_list

def clear_files(folder_path):
    directory = folder_path
    # note: there is a .gitkeep file in output folder

    for filename in os.listdir(directory):
        if filename != ".gitkeep":
            file_path = os.path.join(directory, filename)
            try:
                if os.path.isfile(file_path):
                    os.remove(file_path)
                    print(f"Deleted {filename}")
            except Exception as e:
                print(f"Error deleting {filename}: {e}")

def check_membership(channel, user):
    # Send a request to the Telegram Bot API
    response = requests.get(f'https://api.telegram.org/bot{bot_api}/getChatMember', params={'chat_id': channel, 'user_id': user})
    
    # Parse the response
    data = response.json()
    if data['ok']:
        member_status = data['result']['status']
        if member_status == 'member' or member_status == 'creator' or member_status == 'administrator':
            print('The user is a member of the channel.')
            return True
        else:
            print('The user is not a member of the channel.')
            return False
    else:
        print('Failed to retrieve chat member information.')
        return False

def try_to_delete_message(chat_id, message_id):
    # initialize bot
    bot = telebot.TeleBot(bot_api)

    try:
        bot.delete_message(chat_id, message_id)
    except:
        pass # ignore errors if user has already deleted the message

# experimental - to see if has effect on spotdl rate limits
def delete_spotdl_cache():
    # Path to the directory
    directory = spotdl_cache_path

    # Check if the directory exists and remove it
    if os.path.exists(directory):
        shutil.rmtree(directory)
        print(f'{directory} has been removed.')
    else:
        print(f'{directory} does not exist.')

def setup_spotdl_executable():
    # Remove the existing spotdl file if it exists
    if os.path.exists("spotdl"):
        os.remove("spotdl")
        print("Old spotdl file removed.")
    
    # Define the URL for the latest version of spotdl
    url = spotdl_executable_link
    
    # Download spotdl using wget and name the file 'spotdl'
    download_command = ["wget", "-O", "spotdl", url]
    
    # Run the wget command to download spotdl
    subprocess.run(download_command, check=True)
    
    # Give the downloaded file executable permissions
    chmod_command = ["chmod", "+x", "spotdl"]
    
    # Run the chmod command to make the file executable
    subprocess.run(chmod_command, check=True)