Borg Knowledge Assimilator

using System;
using System.Runtime.InteropServices;

namespace Borg {
	public class ConsoleEx {
      [DllImport("kernel32.dll", SetLastError=true)]
      public static extern bool SetConsoleTextAttribute(int hConsoleOutput, short wAttributes);
      [DllImport("kernel32.dll", EntryPoint="GetStdHandle", SetLastError=true, CharSet=CharSet.Auto, CallingConvention=CallingConvention.StdCall)]
      private static extern int  GetStdHandle(Int32 stdHandle);

      private  const	   Int32	   STD_OUTPUT_HANDLE = -11;
      private  static   int      consoleHandle     = int.MinValue;

      /// <summary>
      /// Colour primitives enumeration.
      /// </summary>
      [Flags]
      private enum CP : short {
         Black   = 0,
         Blue        = 1,
         Green       = 2,
         Red         = 4,
         Intensity   = 8
      }

      /// <summary>
      /// An enumeration of colours available for console output.  The decimal
      /// numbers represent binary bit flags.
      /// </summary>
      public enum Colour : short {
   //                  Black        Blue        Green       Red      Intensity
   /*00*/Black       = CP.Black, 
   /*01*/DarkBlue    =              CP.Blue,
   /*02*/DarkGreen   =                          CP.Green,
   /*03*/DarkCyan    =              CP.Blue |   CP.Green,
   /*04*/DarkRed     =                                      CP.Red,
   /*05*/DarkMagenta =              CP.Blue |               CP.Red,
   /*06*/DarkYellow  =                          CP.Green |  CP.Red,
   /*07*/Gray        =              CP.Blue |   CP.Green |  CP.Red,
   /*08*/DarkGray    =                                               CP.Intensity,
   /*09*/Blue        =              CP.Blue |                        CP.Intensity,
   /*10*/Green       =                          CP.Green |           CP.Intensity,
   /*11*/Cyan        =              CP.Blue |   CP.Green |           CP.Intensity,
   /*12*/Red         =                                      CP.Red | CP.Intensity,    
   /*13*/Magenta     =              CP.Blue |               CP.Red | CP.Intensity,
   /*14*/Yellow      =                          CP.Green |  CP.Red | CP.Intensity,
   /*15*/White       =              CP.Blue |   CP.Green |  CP.Red | CP.Intensity 
      }

      /// <summary>
      /// Type constructor called to determine the handle to the output
      /// window.  The handle is needed for all subsequent API calls
      /// to console output functions.
      /// </summary>
      static ConsoleEx() {
         consoleHandle = GetStdHandle(STD_OUTPUT_HANDLE);
      }

      /// <summary>
      /// Sets the colour of the text sent to the console back to white.
      /// </summary>
      private static void ResetColour() {
         SetConsoleTextAttribute(consoleHandle, (short)Colour.White);
      }

      /// <summary>
      /// Reads a line of text from the user in a particular colour.
      /// </summary>
      /// <param name="colour">The Colour to use in which to display
      /// the solicited data.</param>
      /// <returns>The string entered by the user.</returns>
      public static string ReadLine(Colour colour) {
         SetConsoleTextAttribute(consoleHandle, (short)colour);
         string data = Console.ReadLine();
         ResetColour();

         return data;
      }

      public static void WriteLine(string message, Colour colour) {
         SetConsoleTextAttribute(consoleHandle, (short)colour);
         System.Console.WriteLine(message);
         ResetColour();
      }

      public static void Write(string message, Colour colour) {
         SetConsoleTextAttribute(consoleHandle, (short)colour);
         System.Console.Write(message);
         ResetColour();
      }
   }
}