// MicrophoneMuter.cpp : Defines the entry point for the application.
//

#include "framework.h"
#include "windows.h"
#include "mmdeviceapi.h"
#include "endpointvolume.h"
#include "shellapi.h"
#include "MicrophoneMuter.h"

#define MUTED 1
#define UNMUTED 2
#define NO_DEVICE 3

#define MAX_LOADSTRING 100
#define MUTE_BUTTON 4
#define UNMUTE_BUTTON 5

// Global Variables:
HINSTANCE hInst;                                // current instance
WCHAR szTitle[MAX_LOADSTRING];                  // The title bar text
WCHAR szWindowClass[MAX_LOADSTRING];            // the main window class name
TCHAR appName[] = _T("Microphone Muter");
TCHAR successfulMuteMsg[] = _T("Sucessfully muted default microphone!");
TCHAR successfulUnmuteMsg[] = _T("Sucessfully unmuted default microphone!");
TCHAR noDefaultDevice[] = _T("No default microphone was found.");
TCHAR unknownStatus[] = _T("Unknown status. Contact the developer");
HRESULT hResult = NULL;
UINT16 status = NULL;

// Forward declarations of functions included in this code module:
ATOM                MyRegisterClass(HINSTANCE hInstance);
BOOL                InitInstance(HINSTANCE, int);
LRESULT CALLBACK    WndProc(HWND, UINT, WPARAM, LPARAM);
INT_PTR CALLBACK    About(HWND, UINT, WPARAM, LPARAM);

int APIENTRY wWinMain(_In_ HINSTANCE hInstance,
                     _In_opt_ HINSTANCE hPrevInstance,
                     _In_ LPWSTR    lpCmdLine,
                     _In_ int       nCmdShow)
{
    UNREFERENCED_PARAMETER(hPrevInstance);
    UNREFERENCED_PARAMETER(lpCmdLine);

    hResult = CoInitialize(NULL);

    int args;
    LPWSTR* argList = CommandLineToArgvW(GetCommandLineW(), &args);
    if (args != NULL && args > 1)
    {
        OutputDebugStringW(L"There are args!\n");

        for (int i = 0; i < args; i++)
        {
            if (wcscmp(argList[i], L"unmute") == 0)
            {
                UnmuteDefaultMic();
                return 0;
            }
            else if (wcscmp(argList[i], L"mute") == 0)
            {
                MuteDefaultMic();
                return 0;
            }
        }
    }

    // Initialize global strings
    LoadStringW(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
    LoadStringW(hInstance, IDC_MICROPHONEMUTER, szWindowClass, MAX_LOADSTRING);
    MyRegisterClass(hInstance);

    // Perform application initialization:
    if (!InitInstance (hInstance, nCmdShow))
    {
        return FALSE;
    }

    HACCEL hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_MICROPHONEMUTER));

    MSG msg;

    // Main message loop:
    while (GetMessage(&msg, nullptr, 0, 0))
    {
        if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))
        {
            TranslateMessage(&msg);
            DispatchMessage(&msg);
        }
    }

    return (int) msg.wParam;
}



//
//  FUNCTION: MyRegisterClass()
//
//  PURPOSE: Registers the window class.
//
ATOM MyRegisterClass(HINSTANCE hInstance)
{
    WNDCLASSEXW wcex;

    wcex.cbSize = sizeof(WNDCLASSEX);

    wcex.style          = CS_HREDRAW | CS_VREDRAW;
    wcex.lpfnWndProc    = WndProc;
    wcex.cbClsExtra     = 0;
    wcex.cbWndExtra     = 0;
    wcex.hInstance      = hInstance;
    wcex.hIcon          = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_MICROPHONEMUTER));
    wcex.hCursor        = LoadCursor(nullptr, IDC_ARROW);
    wcex.hbrBackground  = (HBRUSH)(COLOR_WINDOW+1);
    wcex.lpszMenuName   = MAKEINTRESOURCEW(IDC_MICROPHONEMUTER);
    wcex.lpszClassName  = szWindowClass;
    wcex.hIconSm        = LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_SMALL));

    return RegisterClassExW(&wcex);
}

//
//   FUNCTION: InitInstance(HINSTANCE, int)
//
//   PURPOSE: Saves instance handle and creates main window
//
//   COMMENTS:
//
//        In this function, we save the instance handle in a global variable and
//        create and display the main program window.
//
BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
{
   hInst = hInstance; // Store instance handle in our global variable

   HWND hWnd = CreateWindowW(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,
      CW_USEDEFAULT, 0, 450, 350, nullptr, nullptr, hInstance, nullptr);

   if (!hWnd)
   {
      return FALSE;
   }

   ShowWindow(hWnd, nCmdShow);
   UpdateWindow(hWnd);

   return TRUE;
}

//
//  FUNCTION: WndProc(HWND, UINT, WPARAM, LPARAM)
//
//  PURPOSE: Processes messages for the main window.
//
//  WM_COMMAND  - process the application menu
//  WM_PAINT    - Paint the main window
//  WM_DESTROY  - post a quit message and return
//
//
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
    switch (message)
    {
    case WM_COMMAND:
        {
            switch (wParam)
            {
                case MUTE_BUTTON:
                {
                    MuteDefaultMic();
                    RedrawWindow(hWnd, NULL, NULL, RDW_ERASE | RDW_INVALIDATE | RDW_ERASENOW | RDW_ALLCHILDREN);
                }
                break;

                case UNMUTE_BUTTON:
                {
                    UnmuteDefaultMic();
                    RedrawWindow(hWnd, NULL, NULL, RDW_ERASE | RDW_INVALIDATE | RDW_ERASENOW | RDW_ALLCHILDREN);
                }
                break;
            }

            int wmId = LOWORD(wParam);
            // Parse the menu selections:
            switch (wmId)
            {
            case IDM_ABOUT:
                DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
                break;
            case IDM_EXIT:
                DestroyWindow(hWnd);
                break;
            default:
                return DefWindowProc(hWnd, message, wParam, lParam);
            }
        }
        break;
    case WM_PAINT:
        {
            PAINTSTRUCT ps;
            HDC hdc = BeginPaint(hWnd, &ps);
            TextOut(hdc, 5, 5, appName, _tcslen(appName));
            

            switch (status)
            {
            case NULL:
                {
                    HWND hwndButton = CreateWindow(
                        L"BUTTON",
                        L"Mute Microphone",
                        WS_VISIBLE | WS_CHILD | BS_DEFPUSHBUTTON,
                        5, 35,
                        150, 25,
                        hWnd,
                        (HMENU)MUTE_BUTTON,
                        (HINSTANCE)GetWindowLongPtr(hWnd, GWLP_HINSTANCE),
                        NULL);

                    HWND hwndButton1 = CreateWindow(
                        L"BUTTON",
                        L"Unmute Microphone",
                        WS_VISIBLE | WS_CHILD | BS_DEFPUSHBUTTON,
                        165, 35,
                        150, 25,
                        hWnd,
                        (HMENU)UNMUTE_BUTTON,
                        (HINSTANCE)GetWindowLongPtr(hWnd, GWLP_HINSTANCE),
                        NULL);
                }
                break;
            case MUTED:
                {
                    TextOut(hdc, 5, 65, successfulMuteMsg, _tcslen(successfulMuteMsg));
                }
                break;
            case UNMUTED:
                {
                    TextOut(hdc, 5, 65, successfulUnmuteMsg, _tcslen(successfulUnmuteMsg));
                }
                break;
            case NO_DEVICE:
                {
                    TextOut(hdc, 5, 65, noDefaultDevice, _tcslen(noDefaultDevice));
                }
                break;
            default:
                {
                    TextOut(hdc, 5, 65, unknownStatus, _tcslen(unknownStatus));
                }
                break;
            }

            EndPaint(hWnd, &ps);
        }
        break;
    case WM_DESTROY:
        PostQuitMessage(0);
        break;
    default:
        return DefWindowProc(hWnd, message, wParam, lParam);
    }
    return 0;
}

// Message handler for about box.
INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
    UNREFERENCED_PARAMETER(lParam);
    switch (message)
    {
    case WM_INITDIALOG:
        return (INT_PTR)TRUE;

    case WM_COMMAND:
        if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
        {
            EndDialog(hDlg, LOWORD(wParam));
            return (INT_PTR)TRUE;
        }
        break;
    }
    return (INT_PTR)FALSE;
}

void MuteDefaultMic() {
    IMMDeviceEnumerator* deviceEnumerator = NULL;
    hResult = CoCreateInstance(__uuidof(MMDeviceEnumerator), NULL, CLSCTX_INPROC_SERVER, __uuidof(IMMDeviceEnumerator), (LPVOID*)&deviceEnumerator);
    IMMDevice* defaultDevice = NULL;

    hResult = deviceEnumerator->GetDefaultAudioEndpoint(eCapture, eCommunications, &defaultDevice);
    deviceEnumerator->Release();
    deviceEnumerator = NULL;

    if (defaultDevice == NULL) {
        OutputDebugStringW(L"Default device not found!");
        status = NO_DEVICE;
        return;
    }

    IAudioEndpointVolume* endpointVolume = NULL;
    hResult = defaultDevice->Activate(__uuidof(IAudioEndpointVolume), CLSCTX_INPROC_SERVER, NULL, (LPVOID*)&endpointVolume);
    defaultDevice->Release();
    defaultDevice = NULL;

    hResult = endpointVolume->SetMasterVolumeLevelScalar(0.0, NULL);
    hResult = endpointVolume->SetMute(true, NULL);
    endpointVolume->Release();

    OutputDebugStringW(L"Done muting\n");
    status = MUTED;
}

void UnmuteDefaultMic() {
    IMMDeviceEnumerator* deviceEnumerator = NULL;
    hResult = CoCreateInstance(__uuidof(MMDeviceEnumerator), NULL, CLSCTX_INPROC_SERVER, __uuidof(IMMDeviceEnumerator), (LPVOID*)&deviceEnumerator);
    IMMDevice* defaultDevice = NULL;

    hResult = deviceEnumerator->GetDefaultAudioEndpoint(eCapture, eCommunications, &defaultDevice);
    deviceEnumerator->Release();
    deviceEnumerator = NULL;

    if (defaultDevice == NULL) {
        OutputDebugStringW(L"Default device not found!");
        status = NO_DEVICE;
        return;
    }

    IAudioEndpointVolume* endpointVolume = NULL;
    hResult = defaultDevice->Activate(__uuidof(IAudioEndpointVolume), CLSCTX_INPROC_SERVER, NULL, (LPVOID*)&endpointVolume);
    defaultDevice->Release();
    defaultDevice = NULL;

    hResult = endpointVolume->SetMasterVolumeLevelScalar(1.0, NULL);
    hResult = endpointVolume->SetMute(false, NULL);
    endpointVolume->Release();

    OutputDebugStringW(L"Done unmuting\n");
    status = UNMUTED;
}