AutoWinInstaller.cpp

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <cstdlib>
#include <conio.h>
#include <sstream>
#include <string>
#include <fstream>
#include <vector>
#include <fileapi.h>

using namespace std;

// Global variables
void WriteImageFile();
void DiskPartionSetup();
string writeDiskScript(string content);
void commandRun(string cmd);

string bcdboot = "%systemroot%\\System32\\bcdboot.exe";
string diskpart = "%systemroot%\\System32\\diskpart.exe";
string dism = "%systemroot%\\System32\\dism.exe";
string app_title = "Automated Windows Installer";
string efiPartitionLetter = "W";
string bootPartitionLetter = "V";
string windowsPartitionLetter = "R";



int main() {
	SetConsoleTitleA(app_title.c_str());
	cout << "Menu selection: \n";
	cout << "[1] - Setup the disk\n";
	cout << "[2] - Write the image file\n";
	cout << "[3] - Quit the application\n";
	cout << "Enter the number: ";
	while (true) {
		int option = _getch();
		switch (option) {
		case '1':
			system("cls");
			DiskPartionSetup();
			break;
		case '2':
			system("cls");
			WriteImageFile();
			break;
		case '3':
			exit(-1);
			break;
		default:
			cout << "Invalid option! Please try again!\n";
			system("pause>nul");
			system("cls");
			main();
			break;
		}
	}
}

void DiskPartionSetup() {
	// Variables
	int disk_number;

	// Warning
	cout << "Make sure that you have backed up any important data from your hard disk before you continue forward!\n";
	cout << "If you are ready to begin, press the enter key!\n";
	if (cin.ignore()) {
		// Step 1
        commandRun(diskpart + " /s ./" + writeDiskScript(
            "list disk\n"
            "exit"
        ));
		cout << "Enter the disk number from above: ";
		cin >> disk_number;
		if (disk_number == NULL) {
			cout << "The number you typed in is either invaild or is not a number! Please try again! Exiting the program!";
			system("pause>nul");
			exit(-1);
		}

		// Step 2
		cout << "Will you be using GPT or MBR for your partition table? [m/g]";
		while (true) {
			int disk_type = _getch();
			switch (disk_type) {
			case 'm':
			case 'M':
				commandRun(diskpart + " /s " + writeDiskScript(
					"select disk " + to_string(disk_number) + "\n"
					"clean\n"
					"convert mbr\n"
					"exit"
				));
				cout << "\n";
				cout << "Creating the required partitons! Please wait a moment!";
				commandRun(diskpart + " /s " + writeDiskScript(
					"select disk " + to_string(disk_number) + "\n"
					"create part primary size=100\n"
					"format fs=ntfs label=SYSTEM quick\n"
					"assign letter " + bootPartitionLetter + "\n"
					"active\n"
					"create par primary\n"
					"format fs=ntfs label=Windows quick\n"
					"assign letter " + windowsPartitionLetter + "\n"
					"exit"
				));
				system("cls");
				cout << "All done with this step! You can now write the image file!";
				remove("./diskpart_script.txt");
				system("pause>nul");
				system("cls");
				main();
				break;
			case 'g':
			case 'G':
				commandRun(diskpart + " /s " + writeDiskScript(
					"select disk " + to_string(disk_number) + "\n"
					"clean\n"
					"convert gpt\n"
					"exit"
				));
				cout << "\n";
				cout << "Creating the required partitons! Please wait a moment!";
				commandRun(diskpart + " /s " + writeDiskScript(
					"select disk " + to_string(disk_number) + "\n"
					"create part EFI size=500\n"
					"format fs=fat32 label=EFI quick\n"
					"assign letter " + efiPartitionLetter + "\n"
					"create par primary\n"
					"format fs=ntfs label=Windows quick\n"
					"assign letter " + windowsPartitionLetter + "\n"
					"exit"
				));
				system("cls");
				cout << "All done with this step! You can now write the image file!";
				remove("./diskpart_script.txt");
				system("pause>nul");
				system("cls");
				main();
				break;
			default:
				system("cls");
				main();
				break;
			}
		}
	}
}

void WriteImageFile() {
	int indexImg;
	string imgFile;

	cout << "Enter the WIM file path: ";
	cin >> imgFile;
	if (imgFile == "") {
		cout << "You didn't enter anything to this input section! Please try again!";
		system("pause>nul");
		system("cls");
		WriteImageFile();
	}
	ifstream file(imgFile);

	// Checks if the image file exists
	if (!file.is_open()) {
		cout << "The image file does not exist! Please try again!";
		system("pause>nul");
		system("cls");
		WriteImageFile();
	}
	// Grabs the data from the image file and asks for the index number
	commandRun(dism + " /Get-WimInfo /WimFile:" + imgFile + "");
	cout << "Input the index number from above: ";
	cin >> indexImg;
	if (indexImg == NULL) {
		cout << "The number you typed in is either invaild or is not a number! Please try again! Exiting the program!";
		system("pause>nul");
		exit(-1);
	}
	// Asks if the disk partition scheme is MBR or GPT
	cout << "Are you using either MBR or GPT for the disk partition scheme? [m/g] ";

	while (true) {
		int win_type = _getch();
		switch (win_type) {
		case 'm':
		case 'M': {
			// The partition letters are already defined, so nothing has be adding as a input section here!
			system("cls");
			cout << "Starting the image writing! Please do not close this window while the writing is in progress!\n";
			commandRun(dism + " /apply-image /imagefile:" + imgFile + " /index:" + to_string(indexImg) + " /applydir:" + windowsPartitionLetter + ":\\");
			cout << "\n";
			cout << "Image write has completed! Please wait for at least 10 seconds before the next step can begin!";
			Sleep(10000);
			cout << "\n";
			cout << "Adding the required boot files. Please wait!\n";
			commandRun(bcdboot + " " + windowsPartitionLetter + ":""\\Windows"" /s " + bootPartitionLetter + ":"" /f BIOS");
			cout << "\n";
			cout << "Image write was successful! Press the enter key to go back to the main menu!\n";

			system("pause>nul");
			system("cls");
			main();
			break;
		}
		case 'g':
		case 'G': {
			// The partition letters are already defined, so nothing has be adding as a input section here!
			system("cls");
			cout << "Starting the image writing! Please do not close this window while the writing is in progress!\n";
			commandRun(dism + " /apply-image /imagefile:" + imgFile + " /index:" + to_string(indexImg) + " /applydir:" + windowsPartitionLetter + ":\\");
			cout << "\n";
			cout << "Image write has completed! Please wait for at least 10 seconds before the next step can begin!";
			Sleep(10000);
			cout << "\n";
			cout << "Adding the required boot files. Please wait!\n";
			commandRun(bcdboot + " " + windowsPartitionLetter + ":""\\Windows"" /s " + efiPartitionLetter + ":"" /f UEFI");
			cout << "\n";
			cout << "Image write was successful! Press the enter key to go back to the main menu!\n";

			system("pause>nul");
			system("cls");
			main();
			break;
		}
		default:
			system("cls");
			main();
			break;
		}
	}
}

// Helper functions
void commandRun(string cmd) {
	int runCmd = system(cmd.c_str());
	if (runCmd != 0) {
		cout << "\n";
		cout << "Something went wrong while trying to run this command!\n";
		cout << "Press any key to exit the program\n";
		system("pause>nul");
		exit(-1);
	}
}

string writeDiskScript(string content) {
	string file_name = "./diskpart_script.txt";
	ofstream scriptFile(file_name);
	if (scriptFile.is_open()) {
		scriptFile << content.c_str();
		scriptFile.close();
		return file_name;
	}
	else {
		perror("Something went wrong while trying to create the diskpart script! Exiting now!");
		system("pause>nul");
		exit(-1);
	}
}