#!/usr/bin/env python3

import sys
import os
import subprocess
import re
from subprocess import check_output, STDOUT, CalledProcessError, Popen, PIPE
import pexpect
import time
import datetime

def path_rm_end_slash(string):
    string = re.sub("\/$", "", string)
    return(string)

if len(sys.argv) >= 3 and len(sys.argv) <= 4:
    if len(sys.argv) == 3:
        sourcedir = sys.argv[1]
        outdir = sys.argv[2]
        arg = "-n"
    else:
        sourcedir = sys.argv[2]
        outdir = sys.argv[3]
        arg = sys.argv[1]

    if not os.path.isdir(sourcedir):
        print("No such file or directory: '"+str(sourcedir)+"'")
        sys.exit()
else:
    print("Simple tool for convert mp4, mkv or webm file to mp3")
    print("Usage:\nconvert.py [-n|-y] [SOURCE] [DESTINATION]")
    print("  -n ffmpeg: Do not overwrite output files, and exit immediately if a specified output file already exists. (Default option)")
    print("  -y ffmpeg: Overwrite output files without asking.")
    sys.exit()

def dirSize(path,ext1,ext2):
  list_dir = []
  list_dir = os.listdir(path)
  count = 0
  for file in list_dir:
    if file.endswith(ext1):
      count += 1
    if file.endswith(ext2):
      count += 1
  return count

if not os.path.exists(outdir):
    os.makedirs(outdir)

total = dirSize(sourcedir, ".mp4", ".mkv")

count = 0
for file in sorted(os.listdir(sourcedir)):
  try:
    name = file[:file.rfind(".")]
    ext = [".mp4", ".mkv", ".webm"]
    if file.endswith(tuple(ext)):
      exist = False
      if os.path.isfile(outdir+"/"+name+".mp3"):
        if os.path.getsize(path_rm_end_slash(sourcedir)+"/"+file) == os.path.getsize(outdir+"/"+name+".mp3"):
          print("File "+path_rm_end_slash(sourcedir)+"/"+name+".mp3 already exist. Use -y for overwrite.")
      else:
        print("Converting : "+file)
      count += 1
      #cmd = ["ffmpeg", "-n", "-i", sourcedir+"/"+name+".mp4", "-c:a", "libmp3lame", outdir+"/"+name+".mp3"]²
      cmd = 'ffmpeg '+arg+' -i "'+path_rm_end_slash(sourcedir)+'/'+file+'" -c:a libmp3lame "'+outdir+'/'+name+'.mp3"'

      thread = pexpect.spawn(cmd)
      cpl = thread.compile_pattern_list([pexpect.EOF,"frame= *\d+",'(.+)'])

      while True:
        i = thread.expect_list(cpl, timeout=None)
        if i == 0: # EOF
          break
        elif i == 1:
          time_number = thread.match.group(0)
          print(time_number)
          thread.close
        elif i == 2:
          unknown_line = thread.match.group(0)
          if re.match(r'.*Duration.*', str(unknown_line)):
            count = 0
            for element in unknown_line.split():
              count += 1
              if re.match(r"^b'Duration:'$", str(element)):
                break
            total = str(unknown_line.split()[count]).split("'")[1].split(',')[0]
            totalRange = time.strptime(total.split('.')[0],'%H:%M:%S')
            totalRangeS = datetime.timedelta(hours=totalRange.tm_hour,minutes=totalRange.tm_min,seconds=totalRange.tm_sec).total_seconds()
          if re.match(r'.*time=.*', str(unknown_line)):
            timeRange = time.strptime(str(unknown_line).partition("time=")[2].split('.')[0],'%H:%M:%S')
            timeRangeS = datetime.timedelta(hours=timeRange.tm_hour,minutes=timeRange.tm_min,seconds=timeRange.tm_sec).total_seconds()
            print(str(round(100*timeRangeS/totalRangeS, 1))+"% / 100% "+str(exist), end='\r')

          thread.close

  except KeyboardInterrupt:
    # Read the comment here => https://stackoverflow.com/a/1112357 !
    os.remove(outdir+"/"+name+".mp3")
    sys.exit()