From 5a0ff96324621e852944089148d0e9edbbc44d45 Mon Sep 17 00:00:00 2001
From: Sebastian Morr <sebastian@morr.cc>
Date: Mon, 28 Sep 2020 18:40:41 +0200
Subject: [PATCH] Add a chapter select button

---
 main.gd   | 43 +++++++++++++++++++++++++++++++++++++------
 main.tscn | 23 ++++++++++++++++++-----
 2 files changed, 55 insertions(+), 11 deletions(-)

diff --git a/main.gd b/main.gd
index 1e93f0a..60c1c1a 100644
--- a/main.gd
+++ b/main.gd
@@ -2,8 +2,7 @@ extends Control
 
 var dragged = null
 
-#var level_set = "top-down"
-var level_set = "bottom-up"
+var chapter = "bottom-up"
 var current_level = 0
 
 onready var terminal = $HBoxContainer/RightSide/Terminal
@@ -12,6 +11,7 @@ onready var output = terminal.output
 onready var goal_repository = $HBoxContainer/Repositories/GoalRepository
 onready var active_repository = $HBoxContainer/Repositories/ActiveRepository
 onready var level_select = $HBoxContainer/RightSide/TopStuff/Menu/LevelSelect
+onready var chapter_select = $HBoxContainer/RightSide/TopStuff/Menu/ChapterSelect
 onready var next_level_button = $HBoxContainer/RightSide/TopStuff/Menu/NextLevelButton
 onready var level_name = $HBoxContainer/RightSide/TopStuff/LevelPanel/LevelName
 onready var level_description = $HBoxContainer/RightSide/TopStuff/LevelPanel/Text/LevelDescription
@@ -20,17 +20,37 @@ onready var level_congrats = $HBoxContainer/RightSide/TopStuff/LevelPanel/Text/L
 func _ready():
 	# Initialize level select.
 	var options = level_select.get_popup()
-	repopulate_levels()
 	options.connect("id_pressed", self, "load_level")
 	
+	# Initialize chapter select.
+	options = chapter_select.get_popup()
+	options.connect("id_pressed", self, "load_chapter")
+	
 	# Load first level.
 	load_level(0)
 	input.grab_focus()
 	
+func list_chapters():
+	var chapters = []
+	var dir = Directory.new()
+	dir.open("res://levels/")
+	dir.list_dir_begin()
+
+	while true:
+		var file = dir.get_next()
+		if file == "":
+			break
+		elif not file.begins_with("."):
+			chapters.append(file)
+
+	dir.list_dir_end()
+	chapters.sort()
+	return chapters
+	
 func list_levels():
 	var levels = []
 	var dir = Directory.new()
-	dir.open("res://levels/%s" % level_set)
+	dir.open("res://levels/%s" % chapter)
 	dir.list_dir_begin()
 
 	while true:
@@ -45,7 +65,7 @@ func list_levels():
 	
 	var final_level_sequence = []
 	
-	var level_sequence = Array(game.read_file("res://levels/%s/sequence" % level_set, "").split("\n"))
+	var level_sequence = Array(game.read_file("res://levels/%s/sequence" % chapter, "").split("\n"))
 	
 	for level in level_sequence:
 		if level == "":
@@ -59,6 +79,11 @@ func list_levels():
 	
 	return final_level_sequence
 
+func load_chapter(id):
+	var chapters = list_chapters()
+	chapter = chapters[id]
+	chapter_select.text = chapter
+
 func load_level(id):
 	AudioServer.set_bus_mute(AudioServer.get_bus_index("Master"), true)
 	
@@ -70,7 +95,7 @@ func load_level(id):
 	var levels = list_levels()
 	
 	var level = levels[id]
-	var level_prefix = "res://levels/%s/" % level_set
+	var level_prefix = "res://levels/%s/" % chapter
 	
 	var goal_repository_path = "/tmp/goal/"
 	var active_repository_path = "/tmp/active/"
@@ -163,3 +188,9 @@ func repopulate_levels():
 	options.clear()
 	for level in list_levels():
 		options.add_item(level)
+
+func repopulate_chapters():
+	var options = chapter_select.get_popup()
+	options.clear()
+	for chapter in list_chapters():
+		options.add_item(chapter)
diff --git a/main.tscn b/main.tscn
index cac3218..38a4755 100644
--- a/main.tscn
+++ b/main.tscn
@@ -97,18 +97,30 @@ size_flags_vertical = 3
 margin_right = 631.0
 margin_bottom = 35.0
 
+[node name="ChapterSelect" type="MenuButton" parent="HBoxContainer/RightSide/TopStuff/Menu"]
+margin_right = 154.0
+margin_bottom = 35.0
+text = "Select chapter..."
+flat = false
+expand_icon = true
+__meta__ = {
+"_edit_use_anchors_": false
+}
+
 [node name="LevelSelect" type="MenuButton" parent="HBoxContainer/RightSide/TopStuff/Menu"]
-margin_right = 130.0
+margin_left = 159.0
+margin_right = 289.0
 margin_bottom = 35.0
 text = "Select level..."
 flat = false
+expand_icon = true
 __meta__ = {
 "_edit_use_anchors_": false
 }
 
 [node name="ReloadButton" type="Button" parent="HBoxContainer/RightSide/TopStuff/Menu"]
-margin_left = 135.0
-margin_right = 214.0
+margin_left = 294.0
+margin_right = 373.0
 margin_bottom = 35.0
 focus_mode = 0
 enabled_focus_mode = 0
@@ -118,8 +130,8 @@ __meta__ = {
 }
 
 [node name="NextLevelButton" type="Button" parent="HBoxContainer/RightSide/TopStuff/Menu"]
-margin_left = 219.0
-margin_right = 329.0
+margin_left = 378.0
+margin_right = 488.0
 margin_bottom = 35.0
 focus_mode = 0
 custom_styles/hover = SubResource( 1 )
@@ -199,6 +211,7 @@ caret_blink = true
 __meta__ = {
 "_edit_use_anchors_": false
 }
+[connection signal="button_down" from="HBoxContainer/RightSide/TopStuff/Menu/ChapterSelect" to="." method="repopulate_chapters"]
 [connection signal="button_down" from="HBoxContainer/RightSide/TopStuff/Menu/LevelSelect" to="." method="repopulate_levels"]
 [connection signal="pressed" from="HBoxContainer/RightSide/TopStuff/Menu/ReloadButton" to="." method="reload_level"]
 [connection signal="pressed" from="HBoxContainer/RightSide/TopStuff/Menu/NextLevelButton" to="." method="load_next_level"]