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

namespace Launcher.Patcher {
	
	public sealed class SoundPatcher {

		string[] files, identifiers;
		string prefix, nextAction;
		public bool Done;
		
		public SoundPatcher( string[] files, string prefix, string nextAction ) {
			this.files = files;
			this.prefix = prefix;
			this.nextAction = nextAction;
		}
		
		const StringComparison comp = StringComparison.OrdinalIgnoreCase;
		public void FetchFiles( string baseUrl, string altBaseUrl, 
		                       ResourceFetcher fetcher, bool allExist ) {
			if( allExist ) { Done = true; return; }
			
			identifiers = new string[files.Length];
			for( int i = 0; i < files.Length; i++ )
				identifiers[i] = prefix + files[i].Substring( 1 );
			
			for( int i = 0; i < files.Length; i++ ) {
				string loc = files[i][0] == 'A' ? baseUrl : altBaseUrl;
				string url = loc + files[i].Substring( 1 ) + ".ogg";
				fetcher.downloader.DownloadData( url, false, identifiers[i] );
			}
		}
		
		public bool CheckDownloaded( ResourceFetcher fetcher, Action<string> setStatus ) {
			if( Done ) return true;
			for( int i = 0; i < identifiers.Length; i++ ) {
				DownloadedItem item;
				if( fetcher.downloader.TryGetItem( identifiers[i], out item ) ) {
					Console.WriteLine( "got sound " + identifiers[i] );
					if( item.Data == null ) {
						setStatus( "&cFailed to download " + identifiers[i] );
						return false;
					}
					DecodeSound( files[i].Substring( 1 ), (byte[])item.Data );
					
					// TODO: setStatus( next );
					if( i == identifiers.Length - 1 ) {
						Done = true;
						setStatus( fetcher.MakeNext( nextAction ) );
					} else {
						setStatus( fetcher.MakeNext( identifiers[i + 1] ) );
					}
				}
			}
			return true;
		}
		
		void DecodeSound( string name, byte[] rawData ) {
			string path = Path.Combine( Program.AppDirectory, "audio" );
			path = Path.Combine( path, prefix + name + ".wav" );
			
			using( FileStream dst = File.Create( path ) )
				using ( MemoryStream src = new MemoryStream( rawData ) ) 
			{
				dst.SetLength( 44 );
				RawOut output = new RawOut( dst, true );
				OggContainer container = new OggContainer( src );
				output.PlayStreaming( container );
				
				dst.Position = 0;
				BinaryWriter w = new BinaryWriter( dst );
				WriteWaveHeader( w, dst, output );
			}
		}
		
		void WriteWaveHeader( BinaryWriter w, Stream stream, RawOut data ) {
			WriteFourCC( w, "RIFF" );
			w.Write( (int)(stream.Length - 8) );
			WriteFourCC( w, "WAVE" );
			
			WriteFourCC( w, "fmt " );
			w.Write( 16 );
			w.Write( (ushort)1 ); // audio format, PCM
			w.Write( (ushort)data.Channels );
			w.Write( data.Frequency );
			w.Write((data.Frequency * data.Channels * data.BitsPerSample) / 8 ); // byte rate
			w.Write( (ushort)((data.Channels * data.BitsPerSample) / 8) ); // block align
			w.Write( (ushort)data.BitsPerSample );
			
			WriteFourCC( w, "data" );
			w.Write( (int)(stream.Length - 44) );
		}
		
		void WriteFourCC( BinaryWriter w, string fourCC ) {
			w.Write( (byte)fourCC[0] ); w.Write( (byte)fourCC[1] );
			w.Write( (byte)fourCC[2] ); w.Write( (byte)fourCC[3] );
		}
	}
}