#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @Author: KevinMidboe
# @Date:   2017-03-04 16:50:09
# @Last Modified by:   KevinMidboe
# @Last Modified time: 2017-03-04 22:15:21

import os, sqlite3, re
from fuzzywuzzy import process
from langdetect import detect
from pprint import pprint

showDir = '/Volumes/media/tv/'
dbPath = 'shows.db'
mediaExtensions = ['mkv', 'mp4', 'avi']
subExtensions = ['srt']

def XOR(list1, list2):
	return set(list1) ^ set(list2)

def getFuzzyName(query):
	return process.extractOne(query, getShowNames().keys())

def removeUploader(mediaItem):
	match = re.search('-[a-zA-Z\[\]\-]*.[a-z]{3}', mediaItem)
	if match:
		uploader, ext = match.group(0).split('.')
		return uploader

	return subtitles

def getLanguage(path, subtitles):
	f = open(path + subtitles, 'r', encoding= 'ISO-8859-15')
	language = detect(f.read())
	f.close()

	return language

# Finds the correct show name
def getShowNames():
	conn = sqlite3.connect(dbPath)
	c = conn.cursor()

	c.execute('SELECT show_names, date_added, date_modified FROM shows')

	returnList = {}
	for name, added, modified in c.fetchall():
		returnList[name] = [added, modified]

	conn.close()
	return returnList

def getNewFolderContents():
	showNames = getShowNames().keys()
	folderContents = filter( lambda f: not f.startswith('.'), os.listdir(showDir))

	return XOR(folderContents, showNames)

def checkForSingleEpisodes(folderItem):
	showName, hit = getFuzzyName(folderItem)
	episodeMatch = re.findall(re.sub(' ', '.', showName)+'\.S[0-9]{1,2}E[0-9]{1,2}\.', folderItem)
	
	if episodeMatch:
		return True

def getByIdentifier(folderItem, identifier):
	itemMatch = re.findall(identifier + '[0-9]{1,2}', folderItem)
	item = re.sub(identifier, '', itemMatch[0])
	return item 

def getItemChildren(folder):
	children = os.listdir(showDir + folder)

	media_items = []
	subtitles = []
	trash = []
	for childItem in children:
		if childItem[-3:] in mediaExtensions:
			media_items.append([childItem, removeUploader(childItem)])
		elif childItem[-3:] in subExtensions:
			subtitles.append([childItem, removeUploader(childItem), getLanguage(showDir + folder + '/', childItem)])
		else:
			trash.append(childItem)

	return media_items, subtitles, trash


def getEpisodeInfo(folderItem):
	showName, hit = getFuzzyName(folderItem)
	season = getByIdentifier(folderItem, 'S')
	episode = getByIdentifier(folderItem, 'E')
	media_items, subtitles, trash = getItemChildren(folderItem)
	
	episodeInfo = []
	episodeInfo = {'original': folderItem, 
		'full_path': showDir + folderItem,
		'name': showName, 
		'season': season, 
		'episode': episode,
		'media_items': media_items,
		'subtitles': subtitles,
		'trash': trash}


	return episodeInfo


def main():
	for item in getNewFolderContents():
		if checkForSingleEpisodes(item):
			pprint(getEpisodeInfo(item))


if __name__ == '__main__':
	main()