// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.

#pragma once

typedef REQUEST_NOTIFICATION_STATUS(WINAPI * PFN_REQUEST_HANDLER) (IN_PROCESS_HANDLER* pInProcessHandler, void* pvRequestHandlerContext);
typedef BOOL(WINAPI * PFN_SHUTDOWN_HANDLER) (void* pvShutdownHandlerContext);
typedef REQUEST_NOTIFICATION_STATUS(WINAPI * PFN_MANAGED_CONTEXT_HANDLER)(void *pvManagedHttpContext, HRESULT hrCompletionStatus, DWORD cbCompletion);

class IN_PROCESS_APPLICATION : public APPLICATION
{
public:
    IN_PROCESS_APPLICATION(IHttpServer* pHttpServer, ASPNETCORE_CONFIG* pConfig);

    ~IN_PROCESS_APPLICATION();

    __override
        VOID
        ShutDown();

    VOID
        SetCallbackHandles(
            _In_ PFN_REQUEST_HANDLER request_callback,
            _In_ PFN_SHUTDOWN_HANDLER shutdown_callback,
            _In_ PFN_MANAGED_CONTEXT_HANDLER managed_context_callback,
            _In_ VOID* pvRequstHandlerContext,
            _In_ VOID* pvShutdownHandlerContext
        );

    VOID
        Recycle(
            VOID
        );

    // Executes the .NET Core process
    HRESULT
        ExecuteApplication(
            VOID
        );

    VOID
        ReadStdErrHandleInternal(
            VOID
        );

    VOID
        CloseStdErrHandles(
            VOID
        );

    HRESULT
        LoadManagedApplication(
            VOID
        );

    REQUEST_NOTIFICATION_STATUS
        OnAsyncCompletion(
            DWORD                   cbCompletion,
            HRESULT                 hrCompletionStatus,
            IN_PROCESS_HANDLER*     pInProcessHandler
        );

    REQUEST_NOTIFICATION_STATUS
        OnExecuteRequest
        (
            IHttpContext* pHttpContext,
            IN_PROCESS_HANDLER* pInProcessHandler
        );

    static
        IN_PROCESS_APPLICATION*
        GetInstance(
            VOID
        )
    {
        return s_Application;
    }

private:
    // Thread executing the .NET Core process
    HANDLE                          m_hThread;

    // The request handler callback from managed code
    PFN_REQUEST_HANDLER             m_RequestHandler;
    VOID*                           m_RequestHandlerContext;

    // The shutdown handler callback from managed code
    PFN_SHUTDOWN_HANDLER            m_ShutdownHandler;
    VOID*                           m_ShutdownHandlerContext;

    PFN_MANAGED_CONTEXT_HANDLER     m_AsyncCompletionHandler;

    // The event that gets triggered when managed initialization is complete
    HANDLE                          m_pInitalizeEvent;

    // The std log file handle
    HANDLE                          m_hLogFileHandle;
    HANDLE                          m_hErrReadPipe;
    HANDLE                          m_hErrWritePipe;
    STRU                            m_struLogFilePath;

    // The exit code of the .NET Core process
    INT                             m_ProcessExitCode;

    BOOL                            m_fManagedAppLoaded;
    BOOL                            m_fLoadManagedAppError;
    BOOL                            m_fInitialized;
    BOOL                            m_fIsWebSocketsConnection;
    BOOL                            m_fDoneStdRedirect;
    BOOL                            m_fRecycleProcessCalled;

    FILE*                           m_pStdFile;
    STTIMER                         m_Timer;
    SRWLOCK                         m_srwLock;

    // Thread for capturing startup stderr logs when logging is disabled
    HANDLE                          m_hErrThread;
    CHAR                            m_pzFileContents[4096] = { 0 };
    DWORD                           m_dwStdErrReadTotal;
    static IN_PROCESS_APPLICATION*  s_Application;

    VOID
        SetStdOut(
            VOID
        );

    static
        VOID
        ExecuteAspNetCoreProcess(
            _In_ LPVOID pContext
        );

    static
        VOID
        ReadStdErrHandle
        (
            _In_ LPVOID pContext
        );

    HRESULT
    SetEnvironementVariablesOnWorkerProcess(
        VOID
    );

    static
        INT
        FilterException(unsigned int code, struct _EXCEPTION_POINTERS *ep);

    HRESULT
        RunDotnetApplication(
            DWORD argc,
            CONST PCWSTR* argv,
            hostfxr_main_fn pProc
        );
};