ClassiCube/ClassicalSharp/Game/ChatLog.cs

// ClassicalSharp copyright 2014-2016 UnknownShadow200 | Licensed under MIT
using System;
using System.Collections.Generic;
using System.IO;
using System.Runtime.InteropServices;
using ClassicalSharp.Commands;

namespace ClassicalSharp {

	public sealed class Chat : IGameComponent {
		
		public ChatLine Status1, Status2, Status3, BottomRight1,
		BottomRight2, BottomRight3, Announcement, ClientClock;
		public ChatLine[] ClientStatus = new ChatLine[6];
		
		Game game;
		public void Init( Game game ) {
			this.game = game;
		}

		public void Ready( Game game ) { }			
		public void Reset( Game game ) { logName = null; }
		public void OnNewMap( Game game ) { }
		public void OnNewMapLoaded( Game game ) { }
		
		/// <summary> List of chat messages received from the server and added by client commands. </summary>
		/// <remarks> index 0 is the oldest chat message, last index is newest. </remarks>
		public List<ChatLine> Log = new List<ChatLine>();
		
		/// <summary> List of chat messages sent by the user to the server. </summary>
		public List<string> InputLog = new List<string>();
		
		public void Send( string text, bool partial ) {
			text = text.TrimEnd( trimChars );
			if( String.IsNullOrEmpty( text ) ) return;
			
			if( game.CommandList.IsCommandPrefix( text ) ) {
				game.CommandList.Execute( text );
				return;
			}
			game.Server.SendChat( text, partial );
		}
		
		static char[] trimChars = new char[] { ' ', '\0' };
		StringBuffer logBuffer = new StringBuffer( 128 );
		public void Add( string text ) {
			Log.Add( text );
			LogChatToFile( text );
			game.Events.RaiseChatReceived( text, MessageType.Normal );
		}
		
		public void Add( string text, MessageType type ) {
			if( type == MessageType.Normal ) {
				Log.Add( text );
				LogChatToFile( text );
			} else if( type == MessageType.Status1 ) {
				Status1 = text;
			} else if( type == MessageType.Status2 ) {
				Status2 = text;
			} else if( type == MessageType.Status3 ) {
				Status3 = text;
			} else if( type == MessageType.BottomRight1 ) {
				BottomRight1 = text;
			} else if( type == MessageType.BottomRight2 ) {
				BottomRight2 = text;
			} else if( type == MessageType.BottomRight3 ) {
				BottomRight3 = text;
			} else if( type == MessageType.Announcement ) {
				Announcement = text;
			} else if( type >= MessageType.ClientStatus1 && type <= MessageType.ClientStatus6 ) {
				ClientStatus[(int)(type - MessageType.ClientStatus1)] = text;
			} else if( type == MessageType.ClientClock ) {
				ClientClock = text;
			}
			game.Events.RaiseChatReceived( text, type );
		}
		
		public void Dispose() {
			if( writer == null ) return;
			writer.Dispose();
			writer = null;
		}
		
		#region Chat logger
		string logName;
		
		public void SetLogName( string name ) {
			if( logName != null ) return;
			StringBuffer buffer = new StringBuffer( name.Length );
			int len = 0;
			
			for( int i = 0; i < name.Length; i++ ) {
				if( Allowed( name[i] ) )
					buffer.Append( ref len, name[i] );
			}
			logName = buffer.ToString();
		}
		
		static bool Allowed( char c ) {
			return c == '[' || c == ']' || c == '(' || c == ')' ||
				(c >= '0' && c <= '9') || (c >= 'a' && c <= 'z') || 
				(c >= 'A' && c <= 'Z');
		}
		
		DateTime last;
		StreamWriter writer = null;
		void LogChatToFile( string text ) {
			if( logName == null ) return;
			DateTime now = DateTime.Now;
			
			if( now.Day != last.Day || now.Month != last.Month || now.Year != last.Year ) {
				Dispose();
				OpenChatFile( now );
			}
			
			last = now;
			if( writer == null ) return;
			
			if( 32 + text.Length > logBuffer.Capacity )
				logBuffer = new StringBuffer( 32 + text.Length );
			int index = 0;
			logBuffer.Clear() // [HH:mm:ss] text
				.Append( ref index, '[' ).AppendPaddedNum( ref index, 2, now.Hour )
				.Append( ref index, ':' ).AppendPaddedNum( ref index, 2, now.Minute )
				.Append( ref index, ':' ).AppendPaddedNum( ref index, 2, now.Second )
				.Append( ref index, "] " ).AppendColourless( ref index, text )
				.Append( ref index, Environment.NewLine );
			writer.Write( logBuffer.value, 0, logBuffer.Length );
		}
		
		void OpenChatFile( DateTime now ) {
			string basePath = Path.Combine( Program.AppDirectory, "logs" );
			if( !Directory.Exists( basePath ) )
				Directory.CreateDirectory( basePath );

			string date = now.ToString( "yyyy-MM-dd" );
			// Ensure multiple instances do not end up overwriting each other's log entries.
			for( int i = 0; i < 20; i++ ) {
				string id = i == 0 ? "" : " _" + i;
				string fileName = date + " " + logName + id + ".log";
				string path = Path.Combine( basePath, fileName );
				
				FileStream stream = null;
				try {
					stream = File.Open( path, FileMode.Append, FileAccess.Write, FileShare.Read );
				} catch( IOException ex ) {
					int hresult = Marshal.GetHRForException(ex);
					uint errorCode = (uint)hresult & 0xFFFF;
					if( errorCode != 32 ) // ERROR_SHARING_VIOLATION
						throw;
					continue;
				}
				
				writer = new StreamWriter( stream );
				writer.AutoFlush = true;
				return;
			}
			ErrorHandler.LogError( "creating chat log",
			                      "Failed to open or create a chat log file after 20 tries, giving up." );
		}
		#endregion
	}
	
	public struct ChatLine {
		public string Text;
		public DateTime Received;
		
		public ChatLine( string text ) {
			Text = text;
			Received = DateTime.UtcNow;
		}
		
		public static implicit operator ChatLine( string text ) {
			return new ChatLine( text );
		}
	}
}