// Filename: PPLogger.cpp
// Created by:  atrestman (14Sept09)
//
////////////////////////////////////////////////////////////////////
//
// PANDA 3D SOFTWARE
// Copyright (c) Carnegie Mellon University.  All rights reserved.
//
// All use of this software is subject to the terms of the revised BSD
// license.  You should have received a copy of this license along
// with this source code in a file named "LICENSE."
//
////////////////////////////////////////////////////////////////////

#include "stdafx.h"

#include "windows.h"
#include "PPLogger.h"

std::ofstream PPLogger::m_logfile;
bool PPLogger::m_isOpen = false;

PPLogger::PPLogger( )
{
}

PPLogger::~PPLogger( )
{
}

int PPLogger::CreateNewFolder( const std::string& dirname ) 
{
    int error( 0 );
    if ( CreateDirectory( dirname.c_str( ), NULL ) != 0 ) 
    {
        // Success!
        return error;
    }
    // Failed.
    DWORD lastError = GetLastError( );
    if ( lastError == ERROR_ALREADY_EXISTS ) 
    {
        // Not really an error: the directory is already there.
        return error;
    }
    if ( lastError == ERROR_PATH_NOT_FOUND ) 
    {
        // We need to make the parent directory first.
        std::string parent = dirname;
        if ( !parent.empty() && CreateNewFolder( parent ) ) 
        {
            // Parent successfully created.  Try again to make the child.
            if ( CreateDirectory( dirname.c_str(), NULL ) != 0) 
            {
                // Got it!
                return error;
            }
            m_logfile << "Couldn't create " << dirname << "\n";
        }
    }
    return ( error = 1 );
}

int PPLogger::CreateNewFile(const std::string& dirname, const std::string& filename) 
{
    int error( 0 );
    std::string logfilename = dirname + filename;
    HANDLE file = CreateFile( logfilename.c_str(), GENERIC_READ | GENERIC_WRITE,
        FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
        NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL );
    if ( file == INVALID_HANDLE_VALUE ) 
    {
        // Try to make the parent directory first.
        std::string parent = dirname;
        if ( !parent.empty( ) && CreateNewFolder( parent ) ) 
        {
            // Parent successfully created.  Try again to make the file.
            file = CreateFile( logfilename.c_str(), GENERIC_READ | GENERIC_WRITE,
                FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
                NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL );
        }
        if ( file == INVALID_HANDLE_VALUE ) 
        {
            m_logfile << "Couldn't create " << filename << "\n";
            return ( error = 1 );
        }
    }
    CloseHandle( file );
    return error;
}

void PPLogger::Open( const std::string& logdir, const std::string& filename ) 
{
    if ( !m_isOpen ) 
    {
        int error = CreateNewFile( logdir, filename );
        // Note that this logfile name may not be specified at runtime.  It
        // must be compiled in if it is specified at all.

        std::string logBasename = filename;

#ifdef P3D_DEFAULT_PLUGIN_LOG_FILENAME
        if ( logBasename.empty( ) )
        {
            logBasename = P3D_DEFAULT_PLUGIN_LOG_FILENAME;
        }
#endif
        if ( !logBasename.empty( ) ) 
        {
            // Get the log directory.
            std::string logDirectory = logdir;
            if ( logDirectory.empty( ) ) 
            {
                char buffer[MAX_PATH];
                if ( GetTempPath( MAX_PATH, buffer ) != 0 ) 
                {
                    logDirectory = buffer;
                }
            }
            // Construct the full logfile pathname.
            std::string logPathname = logDirectory;
            logPathname += logBasename;

            m_logfile.open( logPathname.c_str( ) );
            m_logfile.setf( std::ios::unitbuf );
        }

        // If we didn't have a logfile name compiled in, we throw away log
        // output by the simple expedient of never actually opening the
        // ofstream.
        m_isOpen = true;
    }
}