// HomeWork1.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <stdio.h>
#include <tchar.h>
#include <iostream>
#include <Windows.h>

using namespace std;

int PressAnyKey(char *prompt) // msdn library
{
	DWORD mode;
	HANDLE hstdin;
	INPUT_RECORD inrec;
	DWORD count;
	char default_prompt[] = "Press a key to continue...";

	// Set the console mode to no-echo, raw input,
	// and no window or mouse events.
	hstdin = GetStdHandle(STD_INPUT_HANDLE);
	if (hstdin == INVALID_HANDLE_VALUE || !GetConsoleMode(hstdin, &mode) || !SetConsoleMode(hstdin, 0))
		return 0;

	if (!prompt) 
		prompt = default_prompt;

	// little fix on the example code by me: convert const char* to WCHAR*
	int buffer_size = MultiByteToWideChar(0, 0, prompt, -1, NULL, NULL) * sizeof(wchar_t);
	WCHAR *w_prompt = new WCHAR[buffer_size]; // allocate memory
	MultiByteToWideChar(0, 0, prompt, -1, w_prompt, buffer_size);

	// Instruct the user 
	WriteConsole(GetStdHandle(STD_OUTPUT_HANDLE), w_prompt, lstrlen(w_prompt), &count, NULL);

	FlushConsoleInputBuffer(hstdin);

	// Get a single key RELEASE
	do ReadConsoleInput(hstdin, &inrec, 1, &count);
	while ((inrec.EventType != KEY_EVENT) || inrec.Event.KeyEvent.bKeyDown);

	// Restore the original console mode
	SetConsoleMode(hstdin, mode);

	return inrec.Event.KeyEvent.wVirtualKeyCode;
}

int _tmain(int argc, _TCHAR* argv[])
{
	cout << "Countback program." << endl;

	for(int i = 0; i <= 10; i++)
	{
		cout << i << endl;

		if(i == 10)
		{
			cout << "Thunderbirds are go!" << endl;
			break;
		}

		Sleep(1000);
	}

	PressAnyKey("Press any key to continue...");

	return 0;
}