ClassiCube/Launcher2/Patcher/ResourceFetcher.cs

// ClassicalSharp copyright 2014-2016 UnknownShadow200 | Licensed under MIT
using System;
using System.IO;
using ClassicalSharp.Network;

namespace Launcher.Patcher {
	
	public sealed class ResourceFetcher {
		
		public bool Done = false;
		internal AsyncDownloader downloader;
		SoundPatcher digPatcher, stepPatcher;
		
		const string jarClassicUri = "http://s3.amazonaws.com/Minecraft.Download/versions/c0.30_01c/c0.30_01c.jar";
		const string jar162Uri = "http://s3.amazonaws.com/Minecraft.Download/versions/1.6.2/1.6.2.jar";
		const string pngTerrainPatchUri = "http://static.classicube.net/terrain-patch.png";
		const string pngGuiPatchUri = "http://static.classicube.net/gui.png";
		const string digSoundsUri = "http://s3.amazonaws.com/MinecraftResources/sound3/dig/";
		const string altDigSoundsUri = "http://s3.amazonaws.com/MinecraftResources/sound3/random/";
		const string stepSoundsUri = "http://s3.amazonaws.com/MinecraftResources/newsound/step/";
		const string altStepSoundsUri = "http://s3.amazonaws.com/MinecraftResources/sound3/step/";
		const string musicUri = "http://s3.amazonaws.com/MinecraftResources/music/";
		const string newMusicUri = "http://s3.amazonaws.com/MinecraftResources/newmusic/";
		
		ushort flags;
		public void DownloadItems(AsyncDownloader downloader, Action<string> setStatus) {
			this.downloader = downloader;
			DownloadMusicFiles();
			digPatcher = new SoundPatcher(ResourceList.DigSounds, "dig_", "step_cloth1");
			digPatcher.FetchFiles(digSoundsUri, altDigSoundsUri, this, DigSoundsExist);
			stepPatcher = new SoundPatcher(ResourceList.StepSounds, "step_", "classic jar");
			stepPatcher.FetchFiles(stepSoundsUri, altStepSoundsUri, this, StepSoundsExist);
			
			flags = 0;
			foreach (var entry in ResourceList.Files)
				flags |= entry.Value;
			
			if ((flags & ResourceList.cMask) != 0)
				downloader.DownloadData(jarClassicUri, false, "classic_jar");
			if ((flags & ResourceList.mMask) != 0)
				downloader.DownloadData(jar162Uri, false, "162_jar");
			if ((flags & ResourceList.gMask) != 0)
				downloader.DownloadData(pngGuiPatchUri, false, "gui_patch");
			if ((flags & ResourceList.tMask) != 0)
				downloader.DownloadData(pngTerrainPatchUri, false, "terrain_patch");
			SetFirstStatus(setStatus);
		}
		
		void DownloadMusicFiles() {
			string[] files = ResourceList.MusicFiles;
			for (int i = 0; i < files.Length; i++) {
				if (musicExists[i]) continue;
				string baseUri = i < 3 ? musicUri : newMusicUri;
				string url = baseUri + files[i] + ".ogg";
				downloader.DownloadData(url, false, files[i]);
			}
		}
		
		public void AddDownload(string url, string identifier) {
			downloader.DownloadData(url, false, identifier);
		}
		
		void SetFirstStatus(Action<string> setStatus) {
			for (int i = 0; i < musicExists.Length; i++) {
				if (musicExists[i]) continue;
				setStatus(MakeNext(ResourceList.MusicFiles[i]));
				return;
			}
			setStatus(MakeNext(FirstItem()));
		}
		
		
		internal byte[] jarClassic, jar162, pngTerrainPatch, pngGuiPatch;
		public bool Check(Action<string> setStatus) {
			if (Done) return true;
			
			if (!CheckMusicFiles(setStatus))
				return false;
			if (!digPatcher.CheckDownloaded(this, setStatus))
				return false;
			if (!stepPatcher.CheckDownloaded(this, setStatus))
				return false;
			
			if (!Download("classic_jar", "classic jar", "1.6.2 jar", ref jarClassic, setStatus))
				return false;
			if (!Download("162_jar", "1.6.2 jar", "gui.png", ref jar162, setStatus))
				return false;
			if (!Download("gui_patch", "gui.png patch", "terrain.png patch", ref pngGuiPatch, setStatus))
				return false;
			if (!Download("terrain_patch", "terrain.png patch", null, ref pngTerrainPatch, setStatus))
				return false;

			Done |= IsDone();
			return true;
		}
		
		string FirstItem() {
			if (!DigSoundsExist) return "dig_cloth1";
			if (!StepSoundsExist) return "step_cloth1";
			
			if ((flags & ResourceList.cMask) != 0) 
				return "classic jar";
			if ((flags & ResourceList.mMask) != 0) 
				return "1.6.2 jar";
			if ((flags & ResourceList.gMask) != 0) 
				return "gui.png patch";
			if ((flags & ResourceList.tMask) != 0) 
				return "terrain.png patch";
			return "(unknown)";
		}
		
		bool IsDone() {
			if (flags == 0) return stepPatcher.Done;
			if ((flags & ResourceList.tMask) != 0) 
				return pngTerrainPatch != null;
			if ((flags & ResourceList.gMask) != 0) 
				return pngGuiPatch != null;
			if ((flags & ResourceList.mMask) != 0) 
				return jar162 != null;
			if ((flags & ResourceList.cMask) != 0) 
				return jarClassic != null;
			return true;
		}
		
		bool Download(string identifier, string name, string next,
		              ref byte[] data, Action<string> setStatus) {
			DownloadedItem item;
			if (downloader.TryGetItem(identifier, out item)) {
				Console.WriteLine("got resource " + identifier);
				if (item.Data == null) {
					setStatus("&cFailed to download " + name);
					return false;
				}
				
				if (next != null)
					setStatus(MakeNext(next));
				else
					setStatus("&eCreating default.zip..");
				data = (byte[])item.Data;
				return true;
			}
			return true;
		}
		
		const string lineFormat = "&eFetching {0}.. ({1}/{2})";
		public string MakeNext(string next) {
			CurrentResource++;
			return String.Format(lineFormat, next,
			                     CurrentResource, ResourcesCount);
		}
		
		bool CheckMusicFiles(Action<string> setStatus) {
			string[] files = ResourceList.MusicFiles;
			for (int i = 0; i < files.Length; i++) {
				string next = i < files.Length - 1 ? files[i + 1] : "dig_cloth1";
				string name = files[i];
				byte[] data = null;
				if (!Download(name, name, next, ref data, setStatus))
					return false;
				
				if (data == null) continue;
				string path = Path.Combine(Program.AppDirectory, "audio");
				path = Path.Combine(path, name + ".ogg");
				File.WriteAllBytes(path, data);
			}
			return true;
		}
		
		public void CheckResourceExistence() {
			ResourceChecker checker = new ResourceChecker();
			checker.CheckResourceExistence();
			
			AllResourcesExist = checker.AllResourcesExist;
			DigSoundsExist = checker.DigSoundsExist;
			StepSoundsExist = checker.StepSoundsExist;
			
			DownloadSize = checker.DownloadSize;
			ResourcesCount = checker.ResourcesCount;
			musicExists = checker.musicExists;
		}
		
		public bool AllResourcesExist, DigSoundsExist, StepSoundsExist;
		public float DownloadSize;
		public int ResourcesCount, CurrentResource;
		bool[] musicExists = new bool[7];
	}
}