source: ipk/source.sh4/network_xupnpd/_path_/etc/xupnpd/plugins/xupnpd_titan.lua @ 27875

-- Plugin for converting channels lists from coolstream receivers
-- Author focus.cst@gmail.com
-- License GPL v2
-- Copyright (C) 2013 CoolStream International Ltd

-- flag to test as plain script, without xupnpd - cfg not defined in this case
local cst_test =  false

if not cfg then
cfg={}
cfg.tmp_path='/tmp/'
cfg.feeds_path='/tmp/'
cfg.debug=1
cst_test = true
end

function cst_debug(level, msg)
        if cfg.debug>level then
                print(msg)
        end
end

function cst_get_bouquets(file)
        local btable={}
        repeat
                local string=file:read()
                if string then
                        cst_debug(1, "########## bouquet="..string)
                        local num = string.match(string, "%d+");
                        if num then
                                local len = string.len(num);
                                local name = string.sub(string, len+1);
                                btable[num] = name
                                cst_debug(1, "num="..num.." name="..btable[num]);
                        end
                        --break; -- one bouquet
                end
        until not string
        return btable
end

function cst_get_channels(file)
        local ctable={}
        repeat
                local string=file:read()
                idx = 1;
                if string then
                        cst_debug(1, "########## channel="..string)
                        local num = string.match(string, "%d+");
                        if num then
                                local len = string.len(num);
                                local rest = string.sub(string, len+1);
                                local id = string.match(rest, "%x+ ");
                                len = string.len(id);
                                local name = string.sub(rest, len+2);
                                cst_debug(1, "num="..num.." id="..id.." name="..name)
                                if id and name then
                                        table.insert(ctable, {id, name});
                                        idx = idx + 1;
                                end
                        end
                end
        until not string        
        return ctable
end

-- all bouquets
-- local burl = "getbouquets"
-- only favorites
local burl = "getbouquets?fav=true"

-- without epg
-- local curl = "getbouquet?bouquet="
-- with epg
local curl = "getbouquet?epg=true&bouquet="

function cst_updatefeed(feed,friendly_name)
        local rc=false
        local feedspath = cfg.feeds_path
        if not friendly_name then
                friendly_name = feed
        end
        local wget = "wget -q -O- "
        local cst_url = 'http://'..feed..'/control/'

        cst_debug(0, wget..cst_url..burl)
        local bouquetsfile = io.popen(wget..cst_url..burl)
        local bouquets = cst_get_bouquets(bouquetsfile)
        bouquetsfile:close()

        if not bouquets then
                return rc
        end
        local bindex
        local bouquett = {}
        for bindex,bouquett in pairs(bouquets) do
                local cindex
                local channelt = {}
                cst_debug(0,wget.."\""..cst_url..curl..bindex.."\"")
                local xmlbouquetfile = io.popen(wget.."\""..cst_url..curl..bindex.."\"")
                local bouquet = cst_get_channels(xmlbouquetfile)
                xmlbouquetfile:close()
                if bouquet then
                        local m3ufilename = cfg.tmp_path.."cst_"..friendly_name.."_bouquet_"..bindex..".m3u"
                        cst_debug(0, m3ufilename)
                        local m3ufile = io.open(m3ufilename,"w")
                        m3ufile:write("#EXTM3U name=\""..bouquett.." ("..friendly_name..")\" plugin=coolstream type=ts\n")
                        for cindex,channelt in pairs(bouquet) do
                                local id = channelt[1];
                                local name = channelt[2];
                                m3ufile:write("#EXTINF:0,"..name.."\n")
                                -- m3ufile:write(cst_url.."zapto?"..id.."\n")
                                m3ufile:write("http://"..feed..":31339/id="..id.."\n")
                        end
                        m3ufile:close()
                        os.execute(string.format('mv %s %s',m3ufilename,feedspath))
                        rc=true
                end
        end
        return rc
end

function cst_read_url(url)
        local wget = "wget -q -O- "
        local turl = wget..url
        cst_debug(0, turl)
        local file = io.popen(turl)
        local string = file:read()
        file:close()
        return string
end

function cst_zapto(urlbase,id)
        local zap = urlbase.."/control/zapto?"..id;
        cst_read_url(zap)
end

function cst_sendurl(cst_url,range)
        local i,j,baseurl = string.find(cst_url,"(.+):.+")
        cst_debug(0, "cst_sendurl: url="..cst_url.." baseurl="..baseurl)

        i,j,id = string.find(cst_url,".*id=(.+)")
        local surl = baseurl.."/control/standby"
        local standby = cst_read_url(surl)

        if standby then
                cst_debug(0, "standby="..standby)

                -- wakeup from standby
                if string.find(standby,"on") then
                        cst_read_url(surl.."?off")
                end
        end
        -- zap to channel
        cst_zapto(baseurl,id)

        if not cst_test then
                plugin_sendurl(cst_url,cst_url,range)
        end
end

if cst_test then
cst_updatefeed("172.16.1.20","tank")
-- cst_updatefeed("172.16.1.10","tank")
-- cst_sendurl("http://172.16.1.20:31339/id=c1f000010070277a", 0)
end

if not cst_test then
plugins['coolstream']={}
plugins.coolstream.name="CoolStream"
plugins.coolstream.desc="IP address (example: <i>192.168.0.1</i>)"
plugins.coolstream.updatefeed=cst_updatefeed
plugins.coolstream.sendurl=cst_sendurl
end