C언어] COLOR 명령 확장: 명령프롬프트 글자색 변경 유틸리티 setclr.exe 만들기

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <Windows.h>

#define THIS_PROGRAM_VERSION "1.0"
char this_program_name[] = "SETCLR";

void setrgb(char *s);
int help(void);



int main(int argc, char *argv[]) {

   if (argc == 1) help();
   setrgb(argv[1]);

   return 0;
}




void setrgb(char *s) {
   WORD attr, previous_background_color;
   char fc, bc;
   HANDLE hConsole = GetStdHandle (STD_OUTPUT_HANDLE); // GetConsoleScreenBufferInfo에 필요
   CONSOLE_SCREEN_BUFFER_INFO csbi;                    // GetConsoleScreenBufferInfo에 필요

   GetConsoleScreenBufferInfo(hConsole, &csbi);
   // 전경색과 배경색이 혼합된 wAttributes에서 배경색만 추출
   // 11111000 등에서 앞의 4비트가 배경색이므로 4비트를 우측으로 이동시켰다가 원위치시키면
   // 뒤의 네자리 즉 전경색은 0000로 채워져 배경색만 남음.
   previous_background_color = csbi.wAttributes >> 0x4;
   previous_background_color = previous_background_color << 0x4;



   strupr(s);
   // 만약 두 색이 모두 지정되어 있으면 처음 숫자는 배경색, 다음 숫자는 전경색.
   // 한 색만 지정돼 있으면 전경색
   // 이것은 MS의 color 명령의 옵션 순서임
   // color 명령은, 한 색만 지정하면 전경색. 두 색을 지정하면 앞색을 배경색으로 인식해 버림.
   if (s[1] != 0) { // 만약 두 색이 모두 지정되어 있으면
      bc = s[0]; fc = s[1];
   } else {         // 한 색만 지정하면
      fc = s[0]; bc = 'X';
   }




   /* Foreground Color */
   switch (fc) {
      case '0': // Black
         attr = 0;
         break;
      case '1': // Blue
         attr = FOREGROUND_BLUE;
         break;
      case '2': // Green
         attr = FOREGROUND_GREEN;
         break;
      case '3': // Aqua (Cyan)
         attr = FOREGROUND_GREEN | FOREGROUND_BLUE;
         break;
      case '4': // Red
         attr = FOREGROUND_RED;
         break;
      case '5': // Purple (Magenta)
         attr = FOREGROUND_RED | FOREGROUND_BLUE;
         break;
      case '6': // Yellow
         attr = FOREGROUND_RED | FOREGROUND_GREEN;
         break;
      case '7': // White (Gray: System Default)
         attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE;
         break;
      case '8': // Gray (Light Black)
         attr = FOREGROUND_INTENSITY;
         break;
      case '9': // Light Blue
         attr = FOREGROUND_BLUE | FOREGROUND_INTENSITY;
         break;
      case 'A': // Light Green
         attr = FOREGROUND_GREEN | FOREGROUND_INTENSITY;
         break;
      case 'B': // Light Aqua (Cyan)
         attr = FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY;
         break;
      case 'C': // Light Red
         attr = FOREGROUND_RED | FOREGROUND_INTENSITY;
         break;
      case 'D': // Light Purple (Magenta)
         attr = FOREGROUND_RED | FOREGROUND_BLUE | FOREGROUND_INTENSITY;
         break;
      case 'E': // Light Yellow
         attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY;
         break;
      case 'F': // Bright White
         attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY;
         break;
      default : help();
         break;
   }




   /* Background Color */
   switch (bc) {
      case 'X': // 배경색이 지정돼 있지 않으면 기존의 배경색을 사용.
                // 전경색만 지정하면 배경색이 항상 검정으로 되어 버림.
         attr = attr | previous_background_color;
         break;
      case '0': // Black
         attr = attr | 0;
         break;
      case '1': // Blue
         attr = attr | BACKGROUND_BLUE;
         break;
      case '2': // Green
         attr = attr | BACKGROUND_GREEN;
         break;
      case '3': // Aqua (Cyan)
         attr = attr | BACKGROUND_GREEN | BACKGROUND_BLUE;
         break;
      case '4': // Red
         attr = attr | BACKGROUND_RED;
         break;
      case '5': // Purple (Magenta)
         attr = attr | BACKGROUND_RED | BACKGROUND_BLUE;
         break;
      case '6': // Yellow
         attr = attr | BACKGROUND_RED | BACKGROUND_GREEN;
         break;
      case '7': // White (Gray: System Default)
         attr = attr | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE;
         break;
      case '8': // Gray (Light Black)
         attr = attr | BACKGROUND_INTENSITY;
         break;
      case '9': // Light Blue
         attr = attr | BACKGROUND_BLUE | BACKGROUND_INTENSITY;
         break;
      case 'A': // Light Green
         attr = attr | BACKGROUND_GREEN | BACKGROUND_INTENSITY;
         break;
      case 'B': // Light Aqua (Cyan)
         attr = attr | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY;
         break;
      case 'C': // Light Red
         attr = attr | BACKGROUND_RED | BACKGROUND_INTENSITY;
         break;
      case 'D': // Light Purple (Magenta)
         attr = attr | BACKGROUND_RED | BACKGROUND_BLUE | BACKGROUND_INTENSITY;
         break;
      case 'E': // Light Yellow
         attr = attr | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_INTENSITY;
         break;
      case 'F': // Bright White
         attr = attr | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY;
         break;
      default : help();
         break;
   }

   SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), attr);
}




int help(void) {
   printf("\n"
      "%s " THIS_PROGRAM_VERSION "\n"
      "(c) 2006 mwultong.blogspot.com\n"
      "\n"
      "Set Console Letter Color (Microsoft COLOR command compatible)\n"
      "\n\n"
      "Usage   : %s < Foreground Color >\n"
      "          %s < Background Color + Foreground Color >\n\n"
      "Example : %s f  :: Bright White\n"
      "          %s 1e :: Bright Yellow on Blue\n"
      "          %s 07 :: White on Black (System Default)\n"
      "\n\n"
      "\t0 = Black (Invisible)               8 = Gray (Dark Gray)\n"
      "\t1 = Blue                            9 = Light Blue\n"
      "\t2 = Green                           A = Light Green\n"
      "\t3 = Aqua (Cyan)                     B = Light Aqua (Light Cyan)\n"
      "\t4 = Red                             C = Light Red\n"
      "\t5 = Purple (Magenta)                D = Light Purple (Light Magenta)\n"
      "\t6 = Yellow                          E = Light Yellow\n"
      "\t7 = White (Gray: System Default)    F = Bright White\n"
      "\n"
      , this_program_name,
        this_program_name, this_program_name,
        this_program_name, this_program_name, this_program_name
      );

   exit(EXIT_FAILURE);
}