mdisk.py

import requests
import json
import os
import subprocess
import threading
import shutil
import zipfile

# setting
currentFile = __file__
realPath = os.path.realpath(currentFile)
dirPath = os.path.dirname(realPath)
dirName = os.path.basename(dirPath)

# is Windows ?
if os.name == 'nt': iswin = "1"
else: iswin =  "0"

# binary setting
if iswin == "0":
    ytdlp = dirPath + "/binaries/yt-dlp"
    aria2c = dirPath + "/binaries/aria2c"
    ffmpeg = dirPath + "/ffmpeg/ffmpeg"
    ffprobe = dirPath + "/ffmpeg/ffprobe"
    ffmpeg_Lzip = dirPath + "/ffmpeg/ffmpeg_L.zip"
    ffprobe_Lzip = dirPath + "/ffmpeg/ffprobe_L.zip"

    if not os.path.exists(ffmpeg):
        with zipfile.ZipFile(ffmpeg_Lzip,"r") as zip_ref:
            zip_ref.extractall("ffmpeg")
        os.remove(ffmpeg_Lzip)
    if not os.path.exists(ffprobe):
        with zipfile.ZipFile(ffprobe_Lzip,"r") as zip_ref:
            zip_ref.extractall("ffmpeg")
        os.remove(ffprobe_Lzip)

    os.system(f"chmod 777 {ytdlp} {aria2c} {ffmpeg} {ffprobe} ffmpeg/qt-faststart")
    
else:

    ytdlp = dirPath + "/binaries/yt-dlp.exe"
    aria2c = dirPath + "/binaries/aria2c.exe"
    ffmpeg = dirPath + "/ffmpeg/ffmpeg.exe"
    ffprobe = dirPath + "/ffmpeg/ffprobe.exe"
    ffmpeg_Wzip = dirPath + "/ffmpeg/ffmpeg.zip"
    ffprobe_Wzip = dirPath + "/ffmpeg/ffprobe.zip"

    if not os.path.exists(ffmpeg):
        with zipfile.ZipFile(ffmpeg_Wzip,"r") as zip_ref:
            zip_ref.extractall("ffmpeg")
        os.remove(ffmpeg_Wzip)
    if not os.path.exists(ffprobe):
        with zipfile.ZipFile(ffprobe_Wzip,"r") as zip_ref:
            zip_ref.extractall("ffmpeg")
        os.remove(ffprobe_Wzip)
    


# header for request
header = {
        'Accept': '*/*',
        'Accept-Language': 'en-US,en;q=0.5',
        'Accept-Encoding': 'gzip, deflate, br',
        'Referer': 'https://mdisk.me/',
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/93.0.4577.82 Safari/537.36'
    	 }

# actual function
def mdow(allinfo,message):

    #setting
    os.mkdir(str(message.id))
    input_video = dirPath + f'/{message.id}/vid.mp4'
    input_audio = dirPath + f'/{message.id}' 
        
    result = subprocess.run([ytdlp, '--no-warning', '-k', '--user-agent','Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/93.0.4577.82 Safari/537.36', '--allow-unplayable-formats', '-F', allinfo["source"]], capture_output=True, text=True)
    with open(f"{message.id}.txt","w") as temp:
        temp.write(result.stdout)
    if iswin == "0": os.system(f"sed -i 1,6d {message.id}.txt")

    # getting ids
    with open(f"{message.id}.txt", 'r') as file1:
        Lines = file1.readlines()
    
    os.remove(f"{message.id}.txt")
    audids = []
    audname = []
    i = 1
    vid_format = str(0)

    for line in Lines:
        line = line.strip()
        if "audio" in line:
            audids.append(line.split(" ")[0])
            if "[" in line and "]" in line:
                audname.append(line.split("[")[1].split("]")[0])
                i = i + 1
            else:
                audname.append(f"Track - {i}")
                i = i + 1
        if "video" in line:
            vid_format = line.split(" ")[0]
       
    # threding audio download   
    audi = threading.Thread(target=lambda:downaud(input_audio,audids,allinfo["source"]),daemon=True)
    audi.start()    

    # video download
    subprocess.run([ytdlp, '--no-warning', '-k', '-f', vid_format, allinfo["source"], '-o', input_video, '--user-agent', 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.212 Safari/537.36',
                   '--allow-unplayable-formats', '--external-downloader', aria2c, '--external-downloader-args', '-x 16 -s 16 -k 1M'])
    
    # check if video downloaded
    if not os.path.exists(input_video):
        print("Video Not Downloaded")
        return allinfo["source"],-1,None
    print("Video Downloaded")
    
    # renaming
    output = allinfo['filename']
    filename = output[:1000]
    output = output.replace(".mkv", "").replace(".mp4", "")
    output = "".join( x for x in output if (x.isalnum() or x in "_ "))
    output = output[:200]

    # check if normal video
    if len(audids) == 0:
        foutput = f"{output}.mkv"
        os.rename(input_video,foutput)
        shutil.rmtree(str(message.id))
        return foutput,0,filename

    # merge
    audi.join()
    cmd = f'{ffmpeg} -y -i "{input_video}" '

    leng = 0
    for ele in audids:
        out_audio = input_audio + f'/aud-{ele}.m4a'
        cmd = cmd + f'-i "{out_audio}" '
        leng = leng + 1
    
    cmd = cmd + "-map 0 "
    i = 1
    while(i<=leng):
        cmd = cmd + f"-map {i} "
        i = i + 1

    i = 1
    for ele in audname:
        cmd = cmd + f'-metadata:s:a:{i} language="{ele}" '

    return cmd,output,filename


def merge(message,cmd,output,filename):

    tcmd = cmd    
    cmd = cmd + f'-c copy "{output}.mkv"'
    print(cmd)
    subprocess.call(cmd, shell=True)                        

    # cleaning
    if os.path.exists(output + '.mkv'):
        print('Cleaning Leftovers...')
        shutil.rmtree(str(message.id))
        foutput = f"{output}.mkv"
        return foutput,1,filename

    else:
        print("Trying with Changes")
        ffoutput = f" {output}.mkv"
        cmd = f'{tcmd} -c copy "{ffoutput}"'
        subprocess.call(cmd, shell=True)
        
        if os.path.exists(output+'.mkv'):
            print('Cleaning Leftovers...')
            
            shutil.rmtree(str(message.id))
            return ffoutput,1,filename


# multi-threding audio download      
def downaud(input_audio,audids,resp):
    threadlist = []
    for i in range(len(audids)):
        threadlist.append(threading.Thread(target=lambda:downaudio(input_audio,audids[i],resp),daemon=True))
        threadlist[i].start()
    
    for ele in threadlist:
        ele.join()
    
    print("Audio Downloaded")
    

# actual audio download
def downaudio(input_audio,ele,resp):             
        out_audio = input_audio + f'/aud-{ele}.m4a'
        subprocess.run([ytdlp, '--no-warning', '-k', '-f', ele, resp, '-o', out_audio, '--user-agent', 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.212 Safari/537.36',
                   '--allow-unplayable-formats', '--external-downloader', aria2c, '--external-downloader-args', '-x 16 -s 16 -k 1M'])


# getting info
def getinfo(link):
    URL = f'https://diskuploader.entertainvideo.com/v1/file/cdnurl?param={link.split("/")[-1]}'
    try: return requests.get(url=URL, headers=header).json()
    except: return {}