'''
Compare Directories
Script was made because sometimes I have a directory with almost the same files, 
like if I was coping files and it failed somehow,
I wanted to know what files are missing and those existing which have
different file sizes.
'''
import os

def compare_directories(dir1, dir2):
    # Get a list of files in both directories
    files1 = {f: os.path.getsize(os.path.join(dir1, f)) for f in os.listdir(dir1) if os.path.isfile(os.path.join(dir1, f))}
    files2 = {f: os.path.getsize(os.path.join(dir2, f)) for f in os.listdir(dir2) if os.path.isfile(os.path.join(dir2, f))}

    # Find files present in both directories with differing sizes
    differing_files = [f for f in files1 if f in files2 and files1[f] != files2[f]]

    # Print the differing files
    if differing_files:
        print("Files with different sizes:")
        for f in differing_files:
            print(f"  {f}: {files1[f]} bytes (dir1) vs {files2[f]} bytes (dir2)")
    else:
        print("No differing files found.")

# Directories to compare
dir1 = "C:\Users\Heagan\Desktop\Meta\Games\Batman\com.camouflaj.manta"
dir2 = "This PC\Quest 3\Internal shared storage\Android\obb\com.camouflaj.manta"

compare_directories(dir1, dir2)

"""
pySmartDL Example
Its a download manager, script shows basic example for usage.
"""
from pySmartDL import SmartDL

url = input("URL: ")
dest = "." # or '~/Downloads/' on linux

obj = SmartDL(url, dest)
obj.start()
path = obj.get_dest()

"""
Youtube Downloader Example
Contains 2 example scripts on using youtube downloader.
One uses pytubefix, the other youtube_dl
"""

from pytubefix import YouTube
from pytubefix.cli import on_progress

url = "https://www.youtube.com/watch?v=KJwYBJMSbPI"

yt = YouTube(url, on_progress_callback=on_progress, use_po_token=True, allow_oauth_cache=True)
print(yt.title)
ys = yt.streams.get_highest_resolution()
ys.download()


from __future__ import unicode_literals
import youtube_dl

# https://www.youtube.com/watch?v=qaRkqu7PJpQ

ydl_opts = {
    'format': 'bestaudio/best',
    'postprocessors': [{
        'key': 'FFmpegExtractAudio',
        'preferredcodec': 'mp3',
        'preferredquality': '192',
    }],
}
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
    ydl.download(["https://www.youtube.com/watch?v=qaRkqu7PJpQ"])
    # ydl.download([input('url -> ')])

"""
Start an FTP server.
Made to share files on my local network easily.
"""
from pyftpdlib.authorizers import DummyAuthorizer
from pyftpdlib.handlers import FTPHandler
from pyftpdlib.servers import FTPServer

LOCAL_IP = '192.168.0.171'


def main():
    # Instantiate a dummy authorizer for managing 'virtual' users
    authorizer = DummyAuthorizer()

    # Define a new user having full r/w permissions and anonymous user
    authorizer.add_user('user', '12345', 'C:/Users/George/Downloads/Share', perm='elradfmw')
    authorizer.add_anonymous('C:/Users/George/Downloads/Share')

    # Instantiate FTP handler and apply authorizer
    handler = FTPHandler
    handler.authorizer = authorizer

    # Define a banner (string returned when client connects)
    handler.banner = "Welcome to my FTP server."

    # Instantiate FTP server and serve
    address = (LOCAL_IP, 21)  # listens on all interfaces on port 21
    server = FTPServer(address, handler)
    server.serve_forever()

if __name__ == '__main__':
    main()

"""
Video Compressor
Script using ffmpeg to compress all videos in a directory.
Made and run on my Google Drive to reduce videos file sizes.
"""

import os
import subprocess
import requests
from pathlib import Path
import os
import concurrent.futures
from tqdm import tqdm  # Install with: pip install tqdm
from threading import Lock
import shutil
from time import sleep
import json
import psutil


input_dir = 'D:\My Drive\Media'
output_dir = Path('C:\Processed')
video_extensions = [
    ".mp4",
    # ".mov"
]
lock = Lock()


def has_been_processed(input_file):
    command = [
        "ffprobe",
        "-v", "quiet",
        "-select_streams", "v:0",
        "-show_entries", "stream=codec_name,profile,level",
        "-of", "default=noprint_wrappers=1",
        input_file
    ]
    result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
    processed_with = result.stdout.strip()
    return processed_with == 'codec_name=hevc\nprofile=Main\nlevel=123'


def compress_video(input_file, outdir):
    command = [
        "ffmpeg",
        "-i", input_file,
        "-c:v", "hevc_nvenc",
        "-preset", "slow",
        "-rc", "vbr_hq",
        "-cq", "30",
        "-b:v", "0",
        "-rc-lookahead", "32",
        "-spatial_aq", "1",
        "-aq-strength", "8",
        "-c:a", "copy",
        "-metadata", "processed_with=hevc",
        outdir
    ]
    result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)


def convert(filepath):
    suf = Path(filepath).suffix.lower()
    if suf in video_extensions:
        try:
            full_output_dir = output_dir / '/'.join(filepath.parts[1:-1])
            full_output_dir.mkdir(parents=True, exist_ok=True)
            if not has_been_processed(filepath):
                shutil.move(filepath, full_output_dir)
                compress_video(full_output_dir / filepath.name, filepath)# if filepath.suffix == '.mp4' else str(filepath).replace(filepath.suffix, '.mp4'))

        except Exception as e:
            print(f'Error {e} with:', filepath)


def list_files_and_folders(directory):
    # Collect all file and folder paths
    items = []
    for root, dirs, files in os.walk(directory):
        items.extend([Path(os.path.join(root, f)) for f in files if Path(os.path.join(root, f)).suffix.lower() in video_extensions])
    total_items = len(items)
    if total_items == 0:
        print("No files or folders found.")
        return
    with tqdm(total=total_items, desc="Processing", unit="item") as pbar:
        for item in items:
            convert(item)
            pbar.update(1)


    # # Use tqdm to display a progress bar
    #     with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
    #         futures = [executor.submit(convert, item) for item in items]

    #         # Update the progress bar as each task completes
    #         for future in concurrent.futures.as_completed(futures):
    #             with lock:

list_files_and_folders(input_dir)

"""
Audio to Text
Converts an audio file clip and transcribes the text. 
"""

import speech_recognition as sr
from moviepy.audio.io.AudioFileClip import AudioFileClip

# Extract audio from video
video_path = input('Video file name: ')
audio_path = "example_audio.wav"
audio_clip = AudioFileClip(video_path)
audio_clip.write_audiofile(audio_path)

# Transcribe the audio
recognizer = sr.Recognizer()
with sr.AudioFile(audio_path) as source:
    audio_data = recognizer.record(source)
    text = recognizer.recognize_google(audio_data)
with open('transcribed_text.md', 'w') as f:
    f.write(text)