metadata {
    definition (name: "Denon HEOS Speaker", namespace: "dcm.heos", author: "dmeglio@gmail.com") {
		capability "AudioVolume"
		capability "MusicPlayer"
		capability "AudioNotification"
		capability "Refresh"
		capability "SpeechSynthesis"
		capability "Initialize"
		
		command "playTopResult", [[name:"Source*","type":"ENUM","description":"Source","constraints":["Pandora", "Rhapsody", "TuneIn", "Deezer", "Napster", "iHeartRadio", "Soundcloud", "Tidal", "Amazon Music"]],
		[name:"Type*","type":"ENUM","description":"Type","constraints":["Station", "Artist", "Album", "Track", "Playlist"]],
		[name:"Search*","type":"STRING",description:"Search"]]
		command "playPreset", [[name:"Preset*","type":"NUMBER",description:"Preset"]]
		command "playInput", [[name:"Input*","type":"ENUM","description":"Input","constraints":["aux_in_1", "aux_in_2", "aux_in_3", "aux_in_4", "aux_single", "aux1", "aux2", "aux3", "aux4", "aux5", "aux6", "aux7", "line_in_1", "line_in_2", "line_in_3", "line_in_4", "coax_in_1", "coax_in_2", "optical_in_1", "optical_in_2", "hdmi_in_1", "hdmi_in_2", "hdmi_in_3", "hdmi_in_4", "hdmi_arc_1", "cable_sat", "dvd", "bluray", "game", "mediaplayer", "cd", "tuner", "hdradio", "tvaudio", "phono", "usbdac", "analog"]]]
		
		attribute "artist", "STRING"
		attribute "album", "STRING"
		attribute "track", "STRING"
		attribute "albumArt", "STRING"
    }
}

preferences {
    input "debugOutput", "bool", title: "Enable debug logging?", defaultValue: true, displayDuringSetup: false, required: false
}

def installed() {
	log.debug "Installed with settings: ${settings}"
}

def updated() {
	log.debug "Updated with settings: ${settings}"
	unschedule()
	runIn(5, initialize)
}

def uninstalled() {
	if (getDataValue("master") == "true")
	{
		telnetClose()
		parent.notifyMasterRemoved(this)
	}
}

def initialize() {
	log.debug "initializing"
	telnetClose()
	if (getDataValue("master") == "true") 
	{
		connectToHeos()
		runIn(5,loginHeosAccount)
		runIn(5, getHeosSearchCriteria)
	}
}

def verifyAccount() {
	log.debug "verifying HEOS account"
	telnetClose()
	connectToHeos()
	state.verifyAccountOnly = true
	runIn(5,loginHeosAccount)
}

def getHeosSearchCriteria()
{
	state.searchCriteria = [:]
	state.browseCriteria = [:]
	sendHeosMessage("heos://browse/get_search_criteria?sid=2")
	sendHeosMessage("heos://browse/get_search_criteria?sid=3")
	sendHeosMessage("heos://browse/get_search_criteria?sid=5")
	sendHeosMessage("heos://browse/get_search_criteria?sid=6")
	sendHeosMessage("heos://browse/get_search_criteria?sid=7")
	sendHeosMessage("heos://browse/get_search_criteria?sid=9")
	sendHeosMessage("heos://browse/get_search_criteria?sid=10")
	sendHeosMessage("heos://browse/browse?sid=1")
	sendHeosMessage("heos://browse/browse?sid=13")
	 
}

def connectToHeos() {
	logDebug "Connecting to HEOS"
	telnetConnect([termChars:[13,10]], getDataValue("ip"), getDataValue("port").toInteger(), null, null)
}

def parse(message) {
	def json = parseJson(message)
	logDebug "Telnet Message: ${json}"
	
	if (json.heos.result == "fail" && json.heos.command == "system/sign_in" && state.verifyAccountOnly) {
		updateDataValue("accountVerified", "false")
		telnetClose()
	}
	else if (json.heos.result == "success" || json.heos.result == null || (json.heos.result == "fail" && json.heos.command == "player/clear_queue")) {
		if (json.heos.command == "player/get_players") {
			parent.distributeMessage(json.heos.command, json.payload)
			registerHeosChangeEvents()
		}
		else if (json.heos.command == "system/sign_in") {
			if (!json.heos.message.contains("command under process")) {
				if (!state.verifyAccountOnly) {
					if (json.heos.message.startsWith("signed_in")) {
						queryHeosPlayers()
					}
				}
				else {
					if (json.heos.message.startsWith("signed_in"))
						updateDataValue("accountVerified", "true")
					else 
						updateDataValue("accountVerified", "false")
					telnetClose()
				}
			}
		}  
		else if (json.heos.command == "event/player_now_playing_progress") {
			def payload = parseHeosQueryString(json.heos.message)
			if (payload.cur_pos.toInteger() <= 2000)
				sendHeosMessage("heos://player/get_now_playing_media?pid=${payload.pid}")
		}
		else if (json.heos.command == "player/get_now_playing_media") {
			json.payload << parseHeosQueryString(json.heos.message)
			parent.distributeMessage(json.heos.command, json.payload)
		}
		else if (json.heos.command == "player/get_mute") {
			json.payload = parseHeosQueryString(json.heos.message)
			parent.distributeMessage(json.heos.command, json.payload)
		}
		else if (json.heos.command == "player/get_volume") {
			json.payload = parseHeosQueryString(json.heos.message)
			parent.distributeMessage(json.heos.command, json.payload)
		}
		else if (json.heos.command == "player/get_play_state") {
			json.payload = parseHeosQueryString(json.heos.message)
			parent.distributeMessage(json.heos.command, json.payload)
		}
		else if (json.heos.command.startsWith("event/")) {
			if (json.heos.message != null)
				parent.distributeMessage(json.heos.command,parseHeosQueryString(json.heos.message))
		}
		else if (json.heos.command == "browse/get_search_criteria") {
			def sid = parseHeosQueryString(json.heos.message).sid.toInteger()
			state.searchCriteria[sid] = json.payload
		}
		else if (json.heos.command == "player/clear_queue") {
		log.debug json.heos.message
			if (!json.heos.message.startsWith("command under process")) {
				def query = parseHeosQueryString(json.heos.message)
				log.debug query
				sendHeosMessage("heos://browse/play_stream?pid=${query.pid}&url=${query.url}")
			}
		}
		else if (json.heos.command == "browse/browse") {
			if (!json.heos.message.startsWith("command under process")) {
				def msgParams = parseHeosQueryString(json.heos.message)
				def sid = msgParams.sid.toInteger()
				if (!msgParams.cid)
					state.browseCriteria[sid] = json.payload
				else {
					def cid = msgParams.cid
					def pid = msgParams.pid
					for (container in json.payload) {
						
						
						if (container.playable == "yes") {
							def containerName = java.net.URLDecoder.decode(container?.name?.toLowerCase(), "UTF-8")
							log.debug "${containerName} --- ${state.searchString}"
							if (containerName.contains(state.searchString))
							{
								state.searchString = null
								sendHeosMessage("heos://browse/play_stream?pid=${pid}&sid=${sid}&cid=${cid}&mid=${container.mid}&name=${container.name}")
								break
							}
						}
					}
				}
			}
		}
		else if (json.heos.command == "browse/search") {
			if (!json.heos.message.startsWith("command under process")) {
				for (container in json.payload) {
					if (container.playable == "yes") {
						sendHeosMessage("heos://browse/play_stream?pid=${pid}&sid=${sid}&cid=${cid}&mid=${container.mid}&name=${container.name}")
						break
					}
				}
			}
		}
	}
	else {
	}
}

def telnetStatus(message) {
	log.error "Status: ${message}"
	// Reconnect if there is a telnet error
	initialize()
}

def setMaster(isMaster) {
	updateDataValue("master", isMaster.toString())
}

def setDeviceDetails(ip, port) {
	updateDataValue("ip", ip)
	updateDataValue("port", port.toString())
}

def setPlayerId(pid) {
	updateDataValue("pid", pid.toString())
}

def sendHeosMessage(String msg) {
	if (getDataValue("master") == "true") {
		logDebug "Sending HEOS command: ${msg}"
		sendHubCommand(new hubitat.device.HubAction(msg, hubitat.device.Protocol.TELNET))
	}
	else
		parent.sendHeosMessage(msg)
}

def loginHeosAccount()
{
	def username = parent.getUsername()
	def password = parent.getPassword().replaceAll("%","%25").replaceAll("&","%26").replaceAll("=","%3D")
	sendHeosMessage("heos://system/sign_in?un=${username}&pw=${password}")
}

def queryHeosPlayers() {
	sendHeosMessage("heos://player/get_players")
}

def registerHeosChangeEvents() {
	sendHeosMessage("heos://system/register_for_change_events?enable=on")
}

def parseHeosQueryString(queryString) {
	def queryKvp = queryString.split("&")
	def result = [:]
	for (kvp in queryKvp) {
		def splitVals = kvp.split("=")
		if (splitVals.size() == 2)
			result."${splitVals[0]}" = splitVals[1]
	}
	return result
}

def logDebug(msg) {
    if (settings?.debugOutput) {
		log.debug msg
	}
}


def mute() {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/set_mute?pid=${pid}&state=on")
}

def unmute() {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/set_mute?pid=${pid}&state=off")
}

def setVolume(volume) {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/set_volume?pid=${pid}&level=${volume}")
}

def setLevel(volumelevel) {
	setVolume(volumelevel)
}

def volumeDown() {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/volume_down?pid=${pid}&step=5")
}

def volumeUp() {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/volume_up?pid=${pid}&step=5")
}

def nextTrack() {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/play_next?pid=${pid}")
}

def previousTrack() {
	def pid = getDataValue("pid")
	sendHeosMessage(" heos://player/play_previous?pid=${pid}")
}

def play() {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/set_play_state?pid=${pid}&state=play")
}

def pause() {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/set_play_state?pid=${pid}&state=pause")
}

def stop() {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/set_play_state?pid=${pid}&state=stop")
}

def playTrack(uri) {
	clearQueue(uri)
}

def restoreTrack(uri) {
	playTrack(uri)
}

def resumeTrack(uri) {
	playTrack(uri)
}

def setTrack(uri) {
	playTrack(uri)
}

def playText(text) {
	def ttsTrack = textToSpeech(text)
	playTrack(ttsTrack.uri)
}

def playText(text, volumelevel) {
	setVolume(volumelevel)
	playText(text)
}

def playTextAndRestore(text, volumelevel) {
	playText(text, volumelevel)
}

def playTextAndResume(text, volumelevel) {
	playText(text, volumelevel)
}

def playTrack(trackuri, volumelevel) {
	setVolume(volumelevel)
	playTrack(trackuri)
}

def playTrackAndRestore(trackuri, volumelevel) {
	playTrack(trackuri, volumelevel)
}

def playTrackAndResume(trackuri, volumelevel) {
	playTrack(trackuri, volumelevel)
}

def speak(text) {
	playText(text)
}

def refresh() {
	def pid = getDataValue("pid")

	sendHeosMessage("heos://player/get_play_state?pid=${pid}")
	sendHeosMessage("heos://player/get_now_playing_media?pid=${pid}")
	sendHeosMessage("heos://player/get_volume?pid=${pid}")
	sendHeosMessage("heos://player/get_mute?pid=${pid}")
}

def internalPlayTopResult(pid, source, type, search) {
	if (getDataValue("master") == "true") {
		def sources = 
		["Pandora":1, "Rhapsody":2, "TuneIn":3, "Deezer":5, "Napster":6, "iHeartRadio":7, "Soundcloud":9, "Tidal":10, "Amazon Music":13]
		
		def sid = sources[source]
		
		if (source == "Amazon Music") {
			def cid = getCidBySourceAndType(sid, type)
			if (cid != null) {
				state.searchString = search.toLowerCase()
				sendHeosMessage("heos://browse/browse?sid=${sid}&cid=${cid}&pid=${pid}")
			}
			else
				log.error "${type} search not supported by ${source}"
		}
		else if (source == "Pandora") {
			def cid = getCidBySourceAndType(sid, type)
			if (cid != null) {
				state.searchString = search.toLowerCase()
				sendHeosMessage("heos://browse/browse?sid=${sid}&cid=${cid}&pid=${pid}")
			}
			else
				log.error "${type} search not supported by ${source}"
		}
		else {
			def scid = getScidBySourceAndType(sid, type)
			if (scid != null) {
				state.searchString = null
				sendHeosMessage("heos://browse/search?sid=${sid}&search=${search}&scid=${scid}&pid=${pid}")
			}
			else
				log.error "${type} search not supported by ${source}"
		}
	}
	else {
		parent.playTopResult(pid, source, type, search)
	}
}

def playTopResult(source, type, search) {
	def pid = getDataValue("pid")
	
	internalPlayTopResult(pid, source, type, search)
	
}

def playPreset(preset) {
	def pid = getDataValue("pid")
	
	sendHeosMessage("heos://browse/play_preset?pid=${pid}&preset=${preset}")
}

def playInput(input) {
	def pid = getDataValue("pid")
	
	sendHeosMessage("heos://browse/play_input?pid=${pid}&input=inputs/${input}")
}

def getScidBySourceAndType(sid, type) {
	for (criteria in state.searchCriteria."$sid")
	{
		if (criteria.name == type)
			return criteria.scid
	}
	return null
}

def getCidBySourceAndType(sid, type) {

	if (sid == 13) {
		if (type == "Station")
			type = "Prime Stations"
		else if (type == "Playlist")
			type = "Playlists"
	}
	else if (sid == 1) {
		if (type == "Station" || type == "Playlist")
			type = "A-Z"
	}

	for (container in state.browseCriteria."$sid")
	{
		if (container.name == type)
			return container.cid
	}
	return null
}

def clearQueue(url) {
	def pid = getDataValue("pid")
	sendHeosMessage("heos://player/clear_queue?pid=${pid}&url=${url}")
}