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

#include "InProcessApplicationBase.h"
#include "SRWExclusiveLock.h"

hostfxr_main_fn InProcessApplicationBase::s_fMainCallback = NULL;

InProcessApplicationBase::InProcessApplicationBase(
    IHttpServer& pHttpServer,
    IHttpApplication& pHttpApplication)
    : AppOfflineTrackingApplication(pHttpApplication),
      m_fRecycleCalled(FALSE),
      m_srwLock(),
      m_pHttpServer(pHttpServer)
{
    InitializeSRWLock(&m_srwLock);
}

VOID
InProcessApplicationBase::Stop(bool fServerInitiated)
{
    // We need to guarantee that recycle is only called once, as calling pHttpServer->RecycleProcess
    // multiple times can lead to AVs.
    if (m_fRecycleCalled)
    {
        return;
    }

    {
        SRWExclusiveLock lock(m_srwLock);

        if (m_fRecycleCalled)
        {
            return;
        }

        m_fRecycleCalled = true;
    }

    // Stop was initiated by server no need to do anything, server would stop on it's own
    if (fServerInitiated)
    {
        return;
    }

    if (!m_pHttpServer.IsCommandLineLaunch())
    {
        // IIS scenario.
        // We don't actually handle any shutdown logic here.
        // Instead, we notify IIS that the process needs to be recycled, which will call
        // ApplicationManager->Shutdown(). This will call shutdown on the application.
        m_pHttpServer.RecycleProcess(L"AspNetCore InProcess Recycle Process on Demand");
    }
    else
    {
        exit(0);
    }
}