1 files changed, 181 insertions, 34 deletions

@@ -1,46 +1,193 @@
+require 'set'
 require 'yaml'
 
 configpath = ARGV[0]
 outputpath = ARGV[1]
 
+panel_to_id = {}
+door_groups = {}
+
+panel_output = []
+door_ids_by_item_id = {}
+painting_ids_by_item_id = {}
+panel_ids_by_location_id = {}
+mentioned_doors = Set[]
+mentioned_paintings = Set[]
+painting_output = {}
+
 config = YAML.load_file(configpath)
-output = config.map do |panel|
-  ret = panel
-  if ret["color"].kind_of? String
-    ret["color"] = [ret["color"]]
-  end
-  ret
-end.map do |panel|
-  ret = {}
-  ret["id"] = "\"#{panel["id"]}\""
-  ret["color"] = "[\"" + panel["color"].join("\",\"") + "\"]"
-  ret["tag"] = "\"#{panel["tag"]}\""
-  if panel.include? "subtag"
-    ret["subtag"] = "\"#{panel["subtag"]}\""
-  end
-  if panel.include? "link"
-    ret["link"] = "\"#{panel["link"]}\""
+config.each do |room_name, room_data|
+  if room_data.include? "panels"
+    room_data["panels"].each do |panel_name, panel|
+      full_name = "#{room_name} - #{panel_name}"
+      panel_to_id[full_name] = panel["id"]
+
+      ret = {}
+      ret["id"] = "\"#{panel["id"]}\""
+      if panel.include? "colors"
+        if panel["colors"].kind_of? String
+          ret["color"] = "[\"#{panel["colors"]}\"]"
+        else
+          ret["color"] = "[\"" + panel["colors"].join("\",\"") + "\"]"
+        end
+      else
+        ret["color"] = "[\"white\"]"
+      end
+      ret["tag"] = "\"#{panel["tag"]}\""
+      if panel.include? "subtag"
+        ret["subtag"] = "\"#{panel["subtag"]}\""
+      end
+      if panel.include? "link"
+        ret["link"] = "\"#{panel["link"]}\""
+      end
+      if panel.include? "copy_to_sign"
+        copytos = []
+        if panel["copy_to_sign"].kind_of? String
+          copytos = [panel["copy_to_sign"]]
+        else
+          copytos = panel["copy_to_sign"]
+        end
+        ret["copy_to_sign"] = "[\"" + copytos.join("\",\"") + "\"]"
+      end
+      if panel.include? "achievement"
+        ret["achievement"] = "\"#{panel["achievement"]}\""
+      end
+      panel_output << ret
+
+      if panel.include? "check" and panel["check"]
+        panel_ids_by_location_id[full_name] = [panel["id"]]
+      end
+    end
   end
-  if panel.include? "copy_to_sign"
-    copytos = []
-    if panel["copy_to_sign"].kind_of? String
-      copytos = [panel["copy_to_sign"]]
-    else
-      copytos = panel["copy_to_sign"]
+
+  if room_data.include? "paintings"
+    room_data["paintings"].each do |painting|
+      painting_output[painting["id"]] = painting
     end
-    ret["copy_to_sign"] = "[\"" + copytos.join("\",\"") + "\"]"
   end
-  if panel.include? "achievement"
-    ret["achievement"] = "\"#{panel["achievement"]}\""
+end
+
+config.each do |room_name, room_data|
+  if room_data.include? "doors"
+    room_data["doors"].each do |door_name, door|
+      full_name = "#{room_name} - #{door_name}"
+
+      if not (door.include? "skip_location" and door["skip_location"]) and
+         not (door.include? "event" and door["event"]) and
+         door.include? "panels" then
+
+        chosen_name = full_name
+        if door.include? "location_name"
+          chosen_name = door["location_name"]
+        else
+          panels_per_room = {}
+          door["panels"].each do |panel_identifier|
+            if panel_identifier.kind_of? String
+              panels_per_room[room_name] ||= []
+              panels_per_room[room_name] << panel_identifier
+            else
+              panels_per_room[panel_identifier["room"]] ||= []
+              panels_per_room[panel_identifier["room"]] << panel_identifier["panel"]
+            end
+          end
+
+          chosen_name = panels_per_room.map do |room_name, panels|
+            room_name + " - " + panels.join(", ")
+          end.join(" and ")
+        end
+
+        panel_ids_by_location_id[chosen_name] = door["panels"].map do |panel_identifier|
+          other_name = ""
+          if panel_identifier.kind_of? String
+            other_name = "#{room_name} - #{panel_identifier}"
+          else
+            other_name = "#{panel_identifier["room"]} - #{panel_identifier["panel"]}"
+          end
+          panel_to_id[other_name]
+        end
+      end
+
+      if not (door.include? "skip_item" and door["skip_item"]) and
+         not (door.include? "event" and door["event"]) then
+
+        chosen_name = full_name
+        if door.include? "item_name"
+          chosen_name = door["item_name"]
+        end
+
+        if door.include? "id"
+          internal_door_ids = []
+          if door["id"].kind_of? String
+            internal_door_ids = [door["id"]]
+          else
+            internal_door_ids = door["id"]
+          end
+          
+          if door.include? "group"
+            door_groups[door["group"]] ||= Set[]
+            door_groups[door["group"]].merge(internal_door_ids)
+          end
+
+          door_ids_by_item_id[chosen_name] = internal_door_ids
+          mentioned_doors.merge(internal_door_ids)
+        end
+
+        if door.include? "painting_id"
+          internal_painting_ids = []
+          if door["painting_id"].kind_of? String
+            internal_painting_ids = [door["painting_id"]]
+          else
+            internal_painting_ids = door["painting_id"]
+          end
+
+          painting_ids_by_item_id[chosen_name] = internal_painting_ids
+          mentioned_paintings.merge(internal_painting_ids)
+        end
+      end
+    end
   end
-    ret
-end.map do |panel|
-  "{" + panel.to_a.map do |element|
-    "\"#{element[0]}\":#{element[1]}"
-  end.join(",") + "}"
-end.join(",")
+end
 
-header = "extends Node\n\nvar panels = ["
-footer = "]"
+door_groups.each do |group_name, door_ids|
+  door_ids_by_item_id[group_name] = door_ids.to_a
+end
 
-File.write(outputpath, header + output + footer)
\ No newline at end of file
+File.open(outputpath, "w") do |f|
+  f.write "extends Node\n\nvar panels = ["
+  f.write(panel_output.map do |panel|
+    "{" + panel.to_a.map do |element|
+      "\"#{element[0]}\":#{element[1]}"
+    end.join(",") + "}"
+  end.join(","))
+  f.write "]\nvar door_ids_by_item_id = {"
+  f.write(door_ids_by_item_id.map do |item_id, door_ids|
+    "\"#{item_id}\":[" + door_ids.map do |door_id|
+      "\"#{door_id}\""
+    end.join(",") + "]"
+  end.join(","))
+  f.write "}\nvar painting_ids_by_item_id = {"
+  f.write(painting_ids_by_item_id.map do |item_id, painting_ids|
+    "\"#{item_id}\":[" + painting_ids.map do |painting_id|
+      "\"#{painting_id}\""
+    end.join(",") + "]"
+  end.join(","))
+  f.write "}\nvar panel_ids_by_location_id = {"
+  f.write(panel_ids_by_location_id.map do |location_id, panel_ids|
+    "\"#{location_id}\":[" + panel_ids.map do |panel_id|
+      "\"#{panel_id}\""
+    end.join(",") + "]"
+  end.join(","))
+  f.write "}\nvar mentioned_doors = ["
+  f.write(mentioned_doors.map do |door_id|
+    "\"#{door_id}\""
+  end.join(","))
+  f.write "]\nvar mentioned_paintings = ["
+  f.write(mentioned_paintings.map do |painting_id|
+    "\"#{painting_id}\""
+  end.join(","))
+  f.write "]\nvar paintings = {"
+  f.write(painting_output.map do |painting_id, painting|
+    "\"#{painting_id}\":{\"orientation\":\"#{painting["orientation"]}\",\"move\":#{painting.include? "move" and painting["move"]}}"
+  end.join(","))
+  f.write "}"
+end