#include <windows.h>
#include <stdio.h>
#include <process.h>

#pragma comment(lib, "kernel32.lib")

// Thread function to drain ConPTY output
unsigned int __stdcall OutputThread(void* pParam) {
	HANDLE hRead = (HANDLE)pParam;
	char buffer[4096];
	DWORD bytesRead;

	while (ReadFile(hRead, buffer, sizeof(buffer), &bytesRead, NULL) && bytesRead > 0) {}
	return 0;
}

int main() {
	HRESULT hr = S_OK;
	HANDLE hPipeInRead = NULL, hPipeInWrite = NULL;
	HANDLE hPipeOutRead = NULL, hPipeOutWrite = NULL;

	CreatePipe(&hPipeInRead, &hPipeInWrite, NULL, 0);
	CreatePipe(&hPipeOutRead, &hPipeOutWrite, NULL, 0);
	HANDLE hThread = (HANDLE)_beginthreadex(NULL, 0, OutputThread, hPipeOutRead, 0, NULL);

	HPCON hPC = NULL;
	COORD consoleSize = { 1, 20 };
	hr = CreatePseudoConsole(consoleSize, hPipeInRead, hPipeOutWrite, 0, &hPC);
	if (FAILED(hr)) return (int)hr;

	CloseHandle(hPipeInRead);
	CloseHandle(hPipeOutWrite);

	STARTUPINFOEXW si = { 0 };
	si.StartupInfo.cb = sizeof(STARTUPINFOEXW);

	SIZE_T attrListSize = 0;
	InitializeProcThreadAttributeList(NULL, 1, 0, &attrListSize);
	si.lpAttributeList = (PPROC_THREAD_ATTRIBUTE_LIST)HeapAlloc(GetProcessHeap(), 0, attrListSize);
	InitializeProcThreadAttributeList(si.lpAttributeList, 1, 0, &attrListSize);
	UpdateProcThreadAttribute(si.lpAttributeList, 0, PROC_THREAD_ATTRIBUTE_PSEUDOCONSOLE, hPC, sizeof(HPCON), NULL, NULL);

	PROCESS_INFORMATION pi = { 0 };
	wchar_t cmdLine[] = L"cmd.exe";
	CreateProcessW(NULL, cmdLine, NULL, NULL, TRUE, EXTENDED_STARTUPINFO_PRESENT, NULL, NULL, &si.StartupInfo, &pi);

	const char* inputChar = "\xe3\x82\xad";	// "キ" in UTF-8
	DWORD bytesWritten;
	WriteFile(hPipeInWrite, inputChar, (DWORD)strlen(inputChar), &bytesWritten, NULL);

	Sleep(500);	// Give cmd.exe a moment to process before closing
	ClosePseudoConsole(hPC);

	WaitForSingleObject(hThread, INFINITE);

	CloseHandle(hThread);
	CloseHandle(pi.hProcess);
	CloseHandle(pi.hThread);
	CloseHandle(hPipeInWrite);
	CloseHandle(hPipeOutRead);
	HeapFree(GetProcessHeap(), 0, si.lpAttributeList);

	printf("Execution finished cleanly.\n");
	return 0;
}