/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#if defined(XP_WIN)
#include <windows.h>
#endif


#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <stdarg.h>

#include "updatelogging.h"

UpdateLog::UpdateLog() : logFP(nullptr)
{
}

void UpdateLog::Init(NS_tchar* sourcePath,
                     const NS_tchar* fileName,
                     const NS_tchar* alternateFileName,
                     bool append)
{
  if (logFP)
    return;

#ifdef XP_WIN
  GetTempFileNameW(sourcePath, L"log", 0, mTmpFilePath);
  int dstFilePathLen = NS_tsnprintf(mDstFilePath,
    sizeof(mDstFilePath)/sizeof(mDstFilePath[0]),
    NS_T("%s/%s"), sourcePath, append ? alternateFileName : fileName);

  // If the destination path was over the length limit,
  // disable logging by skipping opening the file and setting logFP.
  if ((dstFilePathLen > 0) &&
      (dstFilePathLen <
         static_cast<int>(sizeof(mDstFilePath)/sizeof(mDstFilePath[0])))) {
    if (append) {
      CopyFileW(mDstFilePath, mTmpFilePath, FALSE);
    }
    logFP = NS_tfopen(mTmpFilePath, append ? NS_T("a") : NS_T("w"));

    // Delete this file now so it is possible to tell from the unelevated
    // updater process if the elevated updater process has written the log.
    DeleteFileW(mDstFilePath);
  }
#else
  int dstFilePathLen = 0;
  if (alternateFileName && NS_taccess(mDstFilePath, F_OK)) {
    dstFilePathLen = NS_tsnprintf(mDstFilePath,
      sizeof(mDstFilePath)/sizeof(mDstFilePath[0]),
      NS_T("%s/%s"), sourcePath, alternateFileName);
  } else {
    dstFilePathLen = NS_tsnprintf(mDstFilePath,
      sizeof(mDstFilePath)/sizeof(mDstFilePath[0]),
      NS_T("%s/%s"), sourcePath, fileName);
  }

  // If the destination path was over the length limit,
  // disable logging by skipping opening the file and setting logFP.
  if ((dstFilePathLen > 0) &&
      (dstFilePathLen <
         static_cast<int>(sizeof(mDstFilePath)/sizeof(mDstFilePath[0])))) {
    logFP = NS_tfopen(mDstFilePath, append ? NS_T("a") : NS_T("w"));
  }
#endif
}

void UpdateLog::Finish()
{
  if (!logFP)
    return;

  fclose(logFP);
  logFP = nullptr;

#ifdef XP_WIN
  // When the log file already exists then the elevated updater has already
  // written the log file and the temp file for the log should be discarded.
  if (!NS_taccess(mDstFilePath, F_OK)) {
    DeleteFileW(mTmpFilePath);
  } else {
    MoveFileW(mTmpFilePath, mDstFilePath);
  }
#endif
}

void UpdateLog::Flush()
{
  if (!logFP)
    return;

  fflush(logFP);
}

void UpdateLog::Printf(const char *fmt, ... )
{
  if (!logFP)
    return;

  va_list ap;
  va_start(ap, fmt);
  vfprintf(logFP, fmt, ap);
  fprintf(logFP, "\n");
  va_end(ap);
}

void UpdateLog::WarnPrintf(const char *fmt, ... )
{
  if (!logFP)
    return;

  va_list ap;
  va_start(ap, fmt);
  fprintf(logFP, "*** Warning: ");
  vfprintf(logFP, fmt, ap);
  fprintf(logFP, "***\n");
  va_end(ap);
}