#!/usr/bin/env filebot -script


def mesacc = login.split(':')
def mesadd = tryQuietly{ addshows.toBoolean() }
def mesupdate = tryQuietly { tick } ?: 'acquired'
def mesvalue = tryQuietly { value } ?: '1'

def mes = new MyEpisodesScraper(username:mesacc[0], password:mesacc[1])
def myshows = mes.getShowList()

// series name => series key (e.g. Doctor Who (2005) => doctorwho)
def collationKey = { s -> s == null ? '' : s.removeAll(/^(?i)(The|A)\b/).removeAll(/\(?\d{4}\)?$/).removeAll(/\W/).lower() }

args.getFiles{ it.isVideo() && parseEpisodeNumber(it) }.groupBy{ detectSeries(it).name }.each{ series, files ->
	if (series) {
		def show = myshows.find{ collationKey(it.name) == collationKey(series) }
		if (show == null && mesadd) {
			show = mes.getShows().find{ collationKey(it.name) == collationKey(series) }
			if (show == null) {
				println "[failure] '$series' not found"
				return
			}
			mes.addShow(show.id)
			println "[added] $show.name"
		}

		files.each{
			if (show != null) {
				def sxe = parseEpisodeNumber(it)
				mes.update(show.id, sxe.season, sxe.episode, mesupdate, mesvalue)
				println "[$mesupdate] $show.name $sxe [$it.name]"
			} else {
				println "[failure] '$series' has not been added [$it.name]"
			}
		}
	}
}



/****************************************************************************
 * MyEpisodes
 * 				http://www.myepisodes.com
 ****************************************************************************/
import org.jsoup.Jsoup
import org.jsoup.Connection.Method

class MyEpisodesScraper {
	def username
	def password

	def host = "https://www.myepisodes.com"

	def cache = Cache.getCache('myepisodes', CacheType.Monthly)
	def session = [:]

	def login() {
		def response = Jsoup.connect("${host}/login.php").data('username', username, 'password', password, 'action', 'Login', 'u', '').method(Method.POST).execute()
		session << response.cookies()
		return response.parse()
	}

	def get(url) {
		if (!session) {
			login()
		}

		def response = Jsoup.connect(url).cookies(session).method(Method.GET).execute()
		session << response.cookies()
		def html = response.parse()

		if (html.select('#frmLogin')) {
			session.clear()
			throw new Exception('Bad Login')
		}

		return html
	}

	def getShows() {
		def shows = cache.get('MyEpisodes.Shows')
		if (shows == null) {
			shows = ['other', 'A'..'Z'].flatten().findResults{ section ->
				get("${host}/shows.php?list=${section}").select('a').findResults{ a ->
					try {
						// e.g. http://www.myepisodes.com/epsbyshow/491/The%20A-Team
						return [id:a.absUrl('href').match(/\/(\d+)\//).toInteger(), name:a.text().trim()]
					} catch(e) {
						return null
					}
				}
			}.flatten().sort{ it.name }
			cache.put('MyEpisodes.Shows', shows)
		}
		return shows
	}

	def getShowList() {
		get("${host}/shows.php?type=manage").select('option').findResults{ option ->
			try {
				return [id:option.attr('value').toInteger(), name:option.text().trim()]
			} catch(e) {
				return null
			}
		}
	}

	def addShow(showid) {
		get("${host}/views.php?type=manageshow&mode=add&showid=${showid}")
	}

	def update(showid, season, episode, tick = 'acquired', value = '1') {
		get("${host}/myshows.php?action=Update&showid=${showid}&season=${season}&episode=${episode}&${tick}=${value}")
	}
}