--[[
	ZDF HBBTV Version 0.3
	Copyright (C) 2021 Jacek Jendrzej 'satbaby' 
	License: WTFPLv2
]]

function init()
	picfile = "/tmp/ZDFhbbtvEpg.jpg"
	dlPath = '/'
	lastmid = 1000
	json = require "json"
	fh = filehelpers.new()
	aktivelist = get_zdf_data('http://hbbtv.zdf.de/zdfm3/dyn/get.php')	
	n = neutrino()
	vPlay = video.new()
	nMisc = misc.new()
	last_menu = {}
	hid = 0
	dl = {}
	Epg = nil
	Title = nil
	Info1 = nil
	Info2 = nil
	UrlPic = nil
	videostream, audiostream = nil,nil
	have_ffmpeg = which("ffmpeg")
	zdfhbbtv_icon = script_path() .. '/zdfhbbtv_hint.png'
	if not fh:exist(zdfhbbtv_icon , "f") then
		zdfhbbtv_icon='streaming'
	end
end

function setmid(tab,mid)
	for k,v in pairs(tab) do 
		if type(v) == "table" then
			if v.type == 'specialcovers' or v.type == 'header' or v.type == 'infotext' then
				if v.type then
					v.type = nil
				end
				if v.refid then
					v.refid = nil
				end
				if v.title then
					v.title = nil
				end
				if v.subtype then
					v.subtype = nil
				end
				if v.logo then
					v.logo = nil
				end
				if v.optional then
					v.optional = nil
				end
				if v.xs then
					v.xs = nil
				end
				if v.addDocs then
					v.addDocs = nil
				end
				if v.img then
					v.img = nil
				end
				if v.variant then
					v.variant = nil
				end
			else
				if v.play then
					v.play = nil
				end
				if v.view then
					v.view = nil
				end
				if v.pause then
					v.pause = nil
				end
				if v.zdfView then
					v.zdfView = nil
				end
				if v.click then
					v.click = nil
				end
				if v.co then
					v.co = nil
				end
				if v.structureNodePath then
					v.structureNodePath = nil
				end
				if v.foottxt then
					v.foottxt = nil
				end
				if v.elems then
					if #v.elems == 0 then
						v.elems=nil
					end
				end
				v.myid = mid
				mid = mid + 1
				mid = setmid(v,mid)
			end
		end
	end
	return mid
end

function getmid(tab,mid)
	for k,v in pairs(tab) do 
		if type(v) == "table" then
			if v.myid == mid then
				return v
			end
			v = getmid(v,mid)
			if v then return v end
		end
	end
end

function getitkey(tab,key,str)
	if tab == nil then return str end
	for k,v in pairs(tab) do
			if k == key then
				if str == nil then
					str = v
				else
					str = str .. ' ' .. v
				end
			end
		if type(v) == "table" then
			str = getitkey(v,key,str)
		end
	end
	return str
end

function script_path()
	local path = (debug.getinfo(2, "S").source:sub(2))
	return path:match("(.*[/\\])")
end

function getMaxRes()
	local maxRes = 1280
	local Nconfig = configfile.new()
	if Nconfig then
		Nconfig:loadConfig("/var/tuxbox/config/neutrino.conf")
		maxRes = Nconfig:getInt32("livestreamResolution", 1280)
	end
	return maxRes
end

function getdata(Url,Postfields,outputfile,pass_headers,httpheaders)
	if Url == nil then return nil end
	if Curl == nil then
		Curl = curl.new()
	end

	if Url:sub(1, 2) == '//' then
		Url =  'https:' .. Url
	end

	local ret, data = Curl:download{ url=Url, A="Mozilla/5.0",maxRedirs=5,followRedir=false,postfields=Postfields,header=pass_headers,o=outputfile,httpheader=httpheaders }
	if ret == CURL.OK then
		if outputfile then
			return 1
		end
		return data
	else
		return nil
	end
end

function godirectkey(d)
	if d  == nil then return d end
	local  _dkey = ""
	if d == 1 then
		_dkey = RC.red
	elseif d == 2 then
		_dkey = RC.green
	elseif d == 3 then
		_dkey = RC.yellow
	elseif d == 4 then
		_dkey = RC.blue
	elseif d < 14 then
		_dkey = RC[""..d - 4 ..""]
	elseif d == 14 then
		_dkey = RC["0"]
	else
		-- rest
		_dkey = ""
	end
	return _dkey
end

function hideMenu(menu)
	if menu then menu:hide() end
end


function sleep(a)
	local sec = tonumber(os.clock() + a)
	while (os.clock() < sec) do
	end
end

function rescalePic(picW,picH,maxW,maxH)
	if picW and picW > 0 and picH and picH > 0 then
		local aspect = picW / picH
		if not maxH then
			maxH = getMaxScreenHeight()
		end
		if not maxW then
			maxW = getMaxScreenWidth()
		end
		if picW / maxW > picH / maxH then
			picW = maxW
			picH = maxW/aspect
		else
			picH = maxH
			picW = maxH * aspect
		end
		picH = math.floor(picH)
		picW = math.floor(picW)
	end
	return picW,picH
end

function getMaxScreenWidth()
	local max_w = SCREEN.END_X - SCREEN.OFF_X
	return max_w
end

function getMaxScreenHeight()
	local max_h = SCREEN.END_Y - SCREEN.OFF_Y
	return max_h
end

function epgInfo(xres, yres, aspectRatio, framerate)
	local dltxt = ''
	local dl_possible = dl_check(videostream)
	local dl = {}
	if dl_possible then
		dl = gen_dl(videostream, audiostream, Title, Epg)
		dltxt = 'Download Video'
	end
	local withPic = false
	if not fh:exist(picfile , "f") then
		local ok = getdata(UrlPic, nil, picfile)
		if ok then
			withPic = true
		end
	else
		withPic = true
	end
	
	local diff,x,y,w,h  = 0,0,0,0,0
	local space = 10
	local wow = cwindow.new{x=x, y=y, dx=w, dy=h, title=Title, btnRed=dltxt }
	local tf = wow:headerHeight() + wow:footerHeight()
	w,h = n:scale2Res(600), n:scale2Res(300) + tf
	local tw = n:getRenderWidth(FONT.MENU_TITLE,Title) + (wow:headerHeight() * 2)
	if tw > w then
		w = tw
		if w > 1200 then w = 1200 end
	end
	if withPic then
		local maxW ,maxH = n:scale2Res(440), n:scale2Res(368)
		local picW, picH = n:GetSize(picfile)
		maxW,maxH = rescalePic(picW,picH,maxW,maxH)
		diff,h = maxW,maxH
		cpicture.new{parent=wow, x=space, y=space, dx=maxW, dy=maxH, image=picfile}
		h = maxH + tf + (space*2)
		local wP =  (maxW * 2) + (space * 3)
		if w < wP then
			w = wP
		end
	end

	ct = ctext.new{parent=wow, x=diff + (space*2), y=space, dx=w-diff, dy=h-tf, text=Epg, mode="ALIGN_TOP | ALIGN_SCROLL"}
	if withPic == false then
		local ctLines = ct:getLines() + 1
		local th = ctLines * n:FontHeight(FONT.MENU) + tf + (2*space)
		if th < 720 then
			h = th
		end
	end
	wow:setDimensionsAll(x , y, w, h)
	wow:setCenterPos{3}
	wow:paint()

	local msg, data = nil,nil
	local stop = false
	repeat
		msg, data = n:GetInput(500)
		if ct and (msg == RC.up or msg == RC.page_up) then
			ct:scroll{dir="up"}
		elseif ct and (msg == RC.down or msg == RC.page_down) then
			ct:scroll{dir="down"}
		elseif dl_possible and msg == RC.red then
			stop = true
		end
	until msg == RC.ok or msg == RC.home or msg == RC.info or stop
	wow:hide()

	if dl_possible and msg == RC.red  then
		local h = hintbox.new{caption="Download gestartet   ", text=Title}
		h:paint()
		dl_stream(dl)
		sleep(3)
		h:hide()
	end
end

function toUcode(s)
	s=s:gsub("'","&apos;")
	s=s:gsub("<","&lt;")
	s=s:gsub(">","&gt;")
	s=s:gsub('"',"&quot;")
	s=s:gsub("\x0a","&#x0a;")
	s=s:gsub("\x0d","&#x0d;")
	s=s:gsub("&","&amp;")
	return s
end

function writeXML(ch, title, info1, info2, filename)
	ch = ch or ""
	title = title or ""
	info1 = info1 or ""
	info2 = info2 or ""
local xml='<?xml version="1.0" encoding="UTF-8"?>\
\
<neutrino commandversion="1">\
	<record command="record">\
		<channelname>' .. ch .. '</channelname>\
		<epgtitle>' .. toUcode(title) .. '</epgtitle>\
		<id>0</id>\
		<info1>' .. toUcode(info1) .. '</info1>\
		<info2>' .. info2 .. '</info2>\
		<epgid>0</epgid>\
		<mode>1</mode>\
		<videopid>0</videopid>\
		<videotype>1</videotype>\
		<audiopids>\
			<audio pid="1" audiotype="0" selected="0" name=""/>\
		</audiopids>\
		<vtxtpid>0</vtxtpid>\
		<genremajor>0</genremajor>\
		<genreminor>0</genreminor>\
		<seriename></seriename>\
		<length>0</length>\
		<productioncountry></productioncountry>\
		<productiondate>0</productiondate>\
		<rating>0</rating>\
		<quality>0</quality>\
		<parentallockage>0</parentallockage>\
		<dateoflastplay>0</dateoflastplay>\
		<bookmark>\
			<bookmarkstart>0</bookmarkstart>\
			<bookmarkend>0</bookmarkend>\
			<bookmarklast>0</bookmarklast>\
			<bookmarkuser bookmarkuserpos="0" bookmarkusertype="0" bookmarkusername=""/>\
		</bookmark>\
	</record>\
</neutrino>\n'

	local file = io.open(filename,'w')
	file:write(xml)
	file:close()
end

function dl_stream(dl)
	local Format = nil
	if dl and dl.streamUrl then
		if dl.streamUrl:sub(-4) == ".mp4" then
			Format = 'mp4'
		elseif dl.streamUrl:find("m3u8") then
			Format = 'ts'
		end
		local dlname = nil
		if dl.info1 then
			dlname = dl.ch .. "_" .. dl.name .. "_" .. dl.info1
			dlname = dlname:gsub("[%p%s/]", "_")
		end
		if dlname and Format then
			local dls  = "/tmp/.zdfhbbtv_dl.sh"
			local filenamexml = "/tmp/.zdfhbbtv_dl_xml"
			writeXML(dl.ch, dl.name, dl.info1, dl.info2, filenamexml)
			dlname = dlPath .. "/" .. dlname
			local script=io.open(dls,"w")
			script:write('echo "download start" ;\n')
			if Format == 'mp4' then
				script:write('wget -q --continue ' .. dl.streamUrl .. ' -O ' .. dlname .. '.mp4 ;\n')
			elseif Format == 'ts' or Format == 'mkv' then
				if dl.streamUrl2 then
					script:write("ffmpeg -y -nostdin -loglevel 30 -i '" .. dl.streamUrl .. "' -i '" .. dl.streamUrl2  .. "' -c copy  " .. dlname   .. "." .. Format .. "\n")
				else
					script:write("ffmpeg -y -nostdin -loglevel 30 -i '" .. dl.streamUrl .. "' -c copy  " .. dlname   .. "." .. Format .. "\n")
				end
			end
			script:write('if [ $? -eq 0 ]; then \n')
			script:write('wget -q http://127.0.0.1/control/message?popup="Video ' .. dl.name .. ' wurde heruntergeladen." -O /dev/null ; \n')
			script:write('mv ' .. filenamexml .. ' ' .. dlname .. '.xml ; \n')
			script:write('else \n')
			script:write('wget -q http://127.0.0.1/control/message?popup="Download ' .. dl.name .. ' FEHLGESCHLAGEN" -O /dev/null ; \n')
			script:write('rm ' .. filenamexml .. ' ; \n')
			script:write('fi \n')
			script:write('rm ' .. dls .. '; \n')
			script:close()
			os.execute('sh  ' .. dls .. ' &')
			return true
		end
	end
	return false
end

function dl_check(streamUrl)
	local check = false
	local Nconfig = configfile.new()
	if Nconfig then
		Nconfig:loadConfig("/var/tuxbox/config/neutrino.conf")
		dlPath = Nconfig:getString("network_nfs_recordingdir", '/tmp')
	end

	local dl_not_possible = dlPath == '/tmp' or dlPath == '/'
	if dl_not_possible then return check end
	if fh:exist('/tmp/.zdfhbbtv_dl.sh', 'f') then return check end
	if streamUrl:sub(-4) == ".mp4" then
		check = true
	elseif have_ffmpeg and streamUrl:find('m3u8') then
		check = true
	end
	return check
end

function gen_dl(streamUrl,streamUrl2,title,info1)
	local dl = {}
	dl.name = title
	dl.streamUrl = streamUrl
	dl.streamUrl2 = streamUrl2
	dl.info1 = ''
	dl.ch = 'ZDF Hbbtv'
	dl.date = ''
	if info1 then
		dl.info2 = toUcode(info1)
	end

	return dl
end

function which(bin_name)
	local path = os.getenv("PATH") or "/bin"
	for v in path:gmatch("([^:]+):?") do
		local file = v .. "/" .. bin_name
		if fh:exist(file , "f") then
			return true
		end
	end
	return false
end

function getVideoUrlM3U8(m3u8_url)
	if m3u8_url == nil then return nil end
	local res = 0
	local videoUrl = nil
	local audioUrl = nil
	local data = getdata(m3u8_url)
	if data then
		local host = m3u8_url:match('([%a]+[:]?//[_%w%-%.]+)/')
		local lastpos = (m3u8_url:reverse()):find("/")
		local hosttmp = m3u8_url:sub(1,#m3u8_url-lastpos)
		if hosttmp then
			host = hosttmp .."/"
		end
		local revision = 0
		if APIVERSION ~= nil and (APIVERSION.MAJOR > 1 or ( APIVERSION.MAJOR == 1 and APIVERSION.MINOR > 82 )) then
			revision = nMisc:GetRevision()
		end

		local audio_url = nil
		if revision == 1 then -- separate audio for hd51 and co
			local Nconfig	= configfile.new()
			local lang1,lang2,lang3 = nil,nil,nil
			Nconfig:loadConfig("/var/tuxbox/config/neutrino.conf")
			lang1 = Nconfig:getString("pref_lang_0", "#")
			lang2 = Nconfig:getString("pref_lang_1", "#")
			lang3 = Nconfig:getString("pref_lang_2", "#")
			if lang1 == "#" then lang1 = nil else lang1 = lang1:lower() lang1 = lang1:sub(1,3) end
			if lang2 == "#" then lang2 = nil else lang2 = lang2:lower() lang2 = lang2:sub(1,3) end
			if lang3 == "#" then lang3 = nil else lang3 = lang3:lower() lang3 = lang3:sub(1,3) end
			if lang1 == nil then
				lang1 = Nconfig:getString("language", "english")
				if lang1 == nil then
					lang1 = "eng"
				else
					lang1 = lang1:lower() lang1 = lang1:sub(1,3)
				end
			end

			local l1,l2,l3,l4,l = nil,nil,nil,nil,nil
			for adata in data:gmatch('TYPE%=AUDIO.GROUP%-ID=".-",(.-)\n') do
				local lname = adata:match('NAME="(.-)"')
				local lang = adata:match('LANGUAGE="(.-)"')
				local aurl = adata:match('URI="(.-)"')
				if aurl then
					local low_lang = lang:lower()
					if l1 == nil and lname and lang1 and low_lang == lang1 then
						l1 = aurl
					elseif l2 == nil and lname and lang2 and low_lang == lang2 then
						l2 = aurl
					elseif l3 == nil and lname and lang3 and low_lang == lang3 then
						l3 = aurl
					elseif l4 == nil and lname and low_lang == "deu" then
						l4 = aurl
					elseif l == nil then
						l = aurl
					end
				end
			end
			audio_url = l1 or l2 or l3 or l4 or l
		end
		local maxRes = getMaxRes()
		local allres = {}
		local j = 1
		local minRes = 0
		for band, res1, res2, url in data:gmatch('BANDWIDTH=(%d+).-RESOLUTION=(%d+)x(%d+).-\n(.-)\n') do
			local nr = tonumber(res1)
			if nr <= maxRes then
				minRes = nr
			end
			allres[j] = nr
			j=j+1
		end
		if minRes == 0 and j>1 then maxRes = math.min(unpack(allres)) end

		for band, res1, res2, url in data:gmatch('BANDWIDTH=(%d+).-RESOLUTION=(%d+)x(%d+).-\n(.-)\n') do
			if url and res1 then
				local nr = tonumber(res1)
				if (nr <= maxRes and nr > res) then
					res=nr
					if host and url:sub(1,4) ~= "http" then
						url = host .. url
					end
					if audio_url and host and audio_url:sub(1,4) ~= "http" then
						audio_url = host .. audio_url
					end
					videoUrl  = url
					audioUrl  = audio_url
				end
			end
		end
	else
		return m3u8_url, nil
	end
	if videoUrl then videoUrl = videoUrl:gsub("\x0d","") end
	if audioUrl then audioUrl = audioUrl:gsub("\x0d","") end
	return videoUrl, audioUrl
end

function getZDFstream(tab)
	local url = 'https://hbbtv.zdf.de/zdfm3/dyn/get.php?id=' .. tab.link.id
	local jdata = getdata(url)
	if jdata then
		local jnTab = json:decode(jdata)
		if jnTab and jnTab.streams then
			local streams = jnTab.streams[1]
			if streams then
				local maxRes = getMaxRes()
				tab.audiostream = nil
				local mp4 = streams.h264_aac_mp4_http_na_na
				local m3u8 = streams.h264_aac_ts_http_m3u8_http
				local mpd = streams.h264_aac_mp4_http_mpd_http
				if maxRes > 1281 and mp4 and mp4.main and mp4.main.deu and mp4.main.deu.q3 then
					tab.stream = mp4.main.deu.q3
				elseif maxRes < 1281 and mp4 and mp4.main and mp4.main.deu and mp4.main.deu.q1 then
					tab.stream = mp4.main.deu.q1
				elseif m3u8 and m3u8.main and m3u8.main.deu and m3u8.main.deu.q3 then
					tab.stream , tab.audiostream = getVideoUrlM3U8(m3u8.main.deu.q3)
				elseif mpd and mpd.main and mpd.main.deu then
					tab.stream = mpd.main.deu
				end
			end
			Epg,Title,Info1,Info2,UrlPic = nil,nil,nil,nil,nil
			videostream, audiostream = nil,nil
			os.remove(picfile)
			if jnTab.text then Epg = jnTab.text end
			if jnTab.title then Title = jnTab.title end
			if jnTab.cpix and jnTab.cpix.nielsen and jnTab.cpix.nielsen.program then
				Info1 = jnTab.cpix.nielsen.program
				if jnTab.cpix.nielsen.nol_c5 then
					Info2 = jnTab.cpix.nielsen.nol_c5:match(',(.*)')
				end
			end
			local str = nil
			if jnTab.displayAvailability then
				str = getitkey(jnTab.displayAvailability.lineOne,'title',str)
				str = getitkey(jnTab.displayAvailability.lineTwo,'title',str)
			end
			if str then
				if Info2 then
					Info2 = Info2 .. ': ' .. str
				else
					Info2 = str
				end
			end
			if jnTab.img then UrlPic = jnTab.img end
			tab.Epg = Epg
			tab.Title = Title
			tab.Info1 = Info1
			tab.Info2 = Info2
		end
	end
end

function play_video(tab)
	if tab.stream then
		hideMenu(last_menu[hid])
		if tab.Epg and tab.Title then
			Epg = tab.Epg
			Title = tab.Title
			videostream, audiostream = tab.stream,tab.audiostream
			vPlay:setInfoFunc("epgInfo")
		end
		vPlay:PlayFile(tab.Title, tab.stream, tab.Info1 or "",tab.Info2 or 'ZDF hbbtv',tab.audiostream or "")
	end
end

function get_zdf_data(link,data)
	if data == nil then
		data = getdata(link)
	end
	if data then
		local jnTab = json:decode(data)
		if jnTab then
			lastmid = setmid(jnTab,lastmid)
		end
		return jnTab
	end
end

function selPlay(id)
	hideMenu(last_menu[hid])
	id = tonumber(id)
	local h = hintbox.new{caption="Please Wait ...", text="I'm Thinking."}
	if h then
		h:paint()
	end
	local vTab = getmid(aktivelist,id)
	if vTab then
		if vTab.stream == nil then
			getZDFstream(vTab)
		end
	end
	if h then
		h:hide()
	end
	if vTab.stream  then
		play_video(vTab)
	end
end
function selList(id)
	hideMenu(last_menu[hid])
	id = tonumber(id)
	local myTab = getmid(aktivelist,id)
	if myTab.elems == nil then
		local newTab = get_zdf_data('https://hbbtv.zdf.de/zdfm3/dyn/get.php?id=' .. myTab.link.id)
		myTab.elems = {}
		myTab.elems=newTab.elems
	end
	main_menu(myTab)
end

function main_menu(liste)
	hid = hid + 1
	local tname = liste.title or liste.titletxt or liste.myid or liste.id
	if #tname == 0 then tname = 'NoName' end
	local menu  = menu.new{name = tname, icon=zdfhbbtv_icon}
	last_menu[hid] = menu
	menu:addItem{type='back'}
	menu:addItem{type='separatorline'}
	local d =  0
	for i, v in ipairs(liste.elems) do
		if v.myid and v.hasVideo  then
			if d == 0 then menu:addItem{type="subhead", name='Videos'} end
			d=d+1
			local mact = 'selPlay'
			local hico = 'hint_movie'
			local mname =  v.titletxt or v.title or v.myid or '## error ##'
			if #mname == 0 then mname = 'NoName' end
			local vhint = v.headtxt .. ' ' .. v.infoline.text
			if v.text then
				 vhint = vhint .. ' - ' .. v.text
			end
			menu:addItem{type="forwarder" , name=mname, action=mact,hint=vhint,hint_icon=hico ,id=v.myid ,directkey=godirectkey(d)}
		end
	end
	local one = true
	for i, v in ipairs(liste.elems) do
		if v.myid and (v.hasVideo==nil or v.hasVideo==false) and v.type ~= 'specialcovers' and v.type ~= 'header' then
			if one then 	menu:addItem{type='subhead', name='Untermenü'} one = false end
			d=d+1
			local mact = 'selList'
			local hico = 'hint_next'
			local mname =  v.titletxt or v.title or v.myid or '## error ##'
			if #mname == 0 then mname = 'NoName' end
			local vhint = v.headtxt
			if v.text then
				if not vhint or #vhint == 0 then
					vhint = v.text
				else
					vhint = vhint .. ' ' .. v.text
				end
			end
			if v.infoline and v.infoline.text then
				vhint = vhint .. ' - ' .. v.infoline.text
			end
			menu:addItem{type="forwarder" , name=mname, action=mact,hint=vhint ,hint_icon=hico ,id=v.myid ,directkey=godirectkey(d)}
		end
	end
	menu:exec()
	hid = hid - 1
end

function main()
	local h = hintbox.new{caption="Please Wait ...", text="I'm Thinking."}
	h:paint()
	init()
	h:hide()
	main_menu(aktivelist)
	os.remove(picfile)
	collectgarbage()
end

main()