OpenRA/OpenRA.Mods.RA/Widgets/Logic/ReplayBrowserLogic.cs

#region Copyright & License Information
/*
 * Copyright 2007-2013 The OpenRA Developers (see AUTHORS)
 * This file is part of OpenRA, which is free software. It is made
 * available to you under the terms of the GNU General Public License
 * as published by the Free Software Foundation. For more information,
 * see COPYING.
 */
#endregion

using System;
using System.Collections.Generic;
using System.Drawing;
using System.IO;
using System.Linq;
using OpenRA.FileFormats;
using OpenRA.Network;
using OpenRA.Widgets;

namespace OpenRA.Mods.RA.Widgets.Logic
{
	public class ReplayBrowserLogic
	{
		Widget panel;
		ScrollPanelWidget playerList;
		ScrollItemWidget playerTemplate, playerHeader;

		MapPreview selectedMap = MapCache.UnknownMap;
		Dictionary<CPos, Session.Client> selectedSpawns;
		string selectedFilename;
		string selectedDuration;
		bool selectedValid;

		[ObjectCreator.UseCtor]
		public ReplayBrowserLogic(Widget widget, Action onExit, Action onStart)
		{
			panel = widget;

			playerList = panel.Get<ScrollPanelWidget>("PLAYER_LIST");
			playerHeader = playerList.Get<ScrollItemWidget>("HEADER");
			playerTemplate = playerList.Get<ScrollItemWidget>("TEMPLATE");
			playerList.RemoveChildren();

			panel.Get<ButtonWidget>("CANCEL_BUTTON").OnClick = () => { Ui.CloseWindow(); onExit(); };

			var rl = panel.Get<ScrollPanelWidget>("REPLAY_LIST");
			var template = panel.Get<ScrollItemWidget>("REPLAY_TEMPLATE");

			var mod = Game.modData.Manifest.Mod;
			var dir = new[] { Platform.SupportDir, "Replays", mod.Id, mod.Version }.Aggregate(Path.Combine);

			rl.RemoveChildren();
			if (Directory.Exists(dir))
			{
				List<ReplayMetadata> replays;

				using (new Support.PerfTimer("Load replays"))
				{
					replays = Directory
						.GetFiles(dir, "*.rep")
						.Select((filename) => ReplayMetadata.Read(filename))
						.Where((r) => r != null)
						.OrderByDescending((r) => Path.GetFileName(r.FilePath))
						.ToList();
				}

				foreach (var replay in replays)
					AddReplay(rl, replay, template);

				SelectReplay(replays.FirstOrDefault());
			}

			var watch = panel.Get<ButtonWidget>("WATCH_BUTTON");
			watch.IsDisabled = () => !selectedValid || selectedMap.Status != MapStatus.Available;
			watch.OnClick = () => { WatchReplay(); onStart(); };

			panel.Get("REPLAY_INFO").IsVisible = () => selectedFilename != null;

			var preview = panel.Get<MapPreviewWidget>("MAP_PREVIEW");
			preview.SpawnClients = () => selectedSpawns;
			preview.Preview = () => selectedMap;

			var title = panel.GetOrNull<LabelWidget>("MAP_TITLE");
			if (title != null)
				title.GetText = () => selectedMap.Title;

			var type = panel.GetOrNull<LabelWidget>("MAP_TYPE");
			if (type != null)
				type.GetText = () => selectedMap.Type;

			panel.Get<LabelWidget>("DURATION").GetText = () => selectedDuration;
		}

		void SelectReplay(ReplayMetadata replay)
		{
			if (replay == null)
				return;

			try
			{
				var lobby = replay.Session.Value;

				selectedFilename = replay.FilePath;
				selectedMap = Game.modData.MapCache[lobby.GlobalSettings.Map];
				selectedSpawns = LobbyUtils.GetSpawnClients(lobby, selectedMap);
				selectedDuration = WidgetUtils.FormatTimeSeconds((int)replay.Duration.TotalSeconds);
				selectedValid = true;

				var clients = lobby.Clients.Where(c => c.Slot != null)
					.GroupBy(c => c.Team)
					.OrderBy(g => g.Key);

				var teams = new Dictionary<string, IEnumerable<Session.Client>>();
				var noTeams = clients.Count() == 1;
				foreach (var c in clients)
				{
					var label = noTeams ? "Players" : c.Key == 0 ? "No Team" : "Team {0}".F(c.Key);
					teams.Add(label, c);
				}

				playerList.RemoveChildren();

				foreach (var kv in teams)
				{
					var group = kv.Key;
					if (group.Length > 0)
					{
						var header = ScrollItemWidget.Setup(playerHeader, () => true, () => {});
						header.Get<LabelWidget>("LABEL").GetText = () => group;
						playerList.AddChild(header);
					}

					foreach (var option in kv.Value)
					{
						var o = option;

						var color = o.Color.RGB;

						var item = ScrollItemWidget.Setup(playerTemplate, () => false, () => { });

						var label = item.Get<LabelWidget>("LABEL");
						label.GetText = () => o.Name;
						label.GetColor = () => color;

						var flag = item.Get<ImageWidget>("FLAG");
						flag.GetImageCollection = () => "flags";
						flag.GetImageName = () => o.Country;

						playerList.AddChild(item);
					}
				}
			}
			catch (Exception e)
			{
				Log.Write("debug", "Exception while parsing replay: {0}", e);
				selectedFilename = null;
				selectedValid = false;
				selectedMap = MapCache.UnknownMap;
			}
		}

		void WatchReplay()
		{
			if (selectedFilename != null)
			{
				Game.JoinReplay(selectedFilename);
				Ui.CloseWindow();
			}
		}

		void AddReplay(ScrollPanelWidget list, ReplayMetadata replay, ScrollItemWidget template)
		{
			var item = ScrollItemWidget.Setup(template,
				() => selectedFilename == replay.FilePath,
				() => SelectReplay(replay),
				() => WatchReplay());
			var f = Path.GetFileName(replay.FilePath);
			item.Get<LabelWidget>("TITLE").GetText = () => f;
			list.AddChild(item);
		}
	}
}