add movies section

1 files changed, 32 insertions, 17 deletions

diff --git a/main.lua b/main.lua
index 552cd14..f379dd0 100644
--- a/main.lua
+++ b/main.lua
@@ -31,8 +31,8 @@ osd_bg.z = -1
 osd_bg.data = '{\\pos(0,0)}' .. colours.bg ..
 	'{\\p1}m 0 0 l 7680 0 7680 720 0 720{\\p0}'
 
-local categories
-local streams
+local categories = {}
+local streams = {}
 local favourites
 local playing_stream_id
 
@@ -64,21 +64,42 @@ local function write_json_file(fn, data)
 	f:close()
 end
 
-local function load_data()
-	categories = read_json_file('categories.json')
-	for _, v in ipairs(categories) do
+local function load_section(prefix, name)
+	categories[#categories+1] = {
+		id=prefix .. ':0',
+		type='category',
+		name=name,
+		parent_id='root',
+	}
+
+	local tmp = read_json_file(prefix .. '_categories.json')
+	for _, v in ipairs(tmp) do
 		v.type = 'category'
-		v.id = v.category_id
+		v.id = prefix .. ':' .. v.category_id
 		v.name = v.category_name
-		v.parent_id = tostring(v.parent_id)
+		v.parent_id = prefix .. ':' .. v.parent_id
+		categories[#categories+1] = v
 	end
 
-	streams = read_json_file('streams.json')
-	for _, v in ipairs(streams) do
+	tmp = read_json_file(prefix .. '_streams.json')
+	for _, v in ipairs(tmp) do
 		v.type = 'stream'
 		v.id = v.stream_id
-		v.parent_id = v.category_id
+		v.parent_id = prefix .. ':' .. v.category_id
+		streams[#streams+1] = v
 	end
+end
+
+local function load_data()
+	categories[#categories+1] = {
+		id='favourites',
+		type='category',
+		name='Favourites',
+		parent_id='root',
+	}
+
+	load_section('live', 'Live TV')
+	load_section('movie', 'Movies')
 
 	favourites = read_json_file('favourites.json')
 	-- json loading/dumping breaks when the table is empty, so we need a
@@ -561,10 +582,4 @@ end)
 mp.add_forced_key_binding('TAB', 'toggle-menu', toggle_menu)
 bind_menu_keys()
 load_data()
-table.insert(categories, 1, {
-	id='favourites',
-	type='category',
-	name='Favourites',
-	parent_id='0',
-})
-push_category_menu('0')
+push_category_menu('root')