/* 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/. */

#ifndef mozilla_logging_h
#define mozilla_logging_h

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

#include "mozilla/Assertions.h"
#include "mozilla/Atomics.h"
#include "mozilla/Attributes.h"
#include "mozilla/Likely.h"
#include "mozilla/LoggingCore.h"

#include "fmt/format.h"

#define MOZ_LOGGING_ENABLED 1

// The mandatory extension we add to log files.  Note that rotate will append
// the file piece number still at the end.
#define MOZ_LOG_FILE_EXTENSION ".moz_log"

// Token for Process ID substitution.
#define MOZ_LOG_PID_TOKEN "%PID"

namespace mozilla {

class TimeStamp;

class LogModule {
 public:
  ~LogModule() { ::free(mName); }

  /**
   * Retrieves the module with the given name. If it does not already exist
   * it will be created.
   *
   * @param aName The name of the module.
   * @return A log module for the given name. This may be shared.
   */
  static LogModule* Get(const char* aName);

  /**
   * Logging processes -MOZ_LOG and -MOZ_LOG_FILE command line arguments
   * to override or set modules and the file as if passed through MOZ_LOG
   * and MOZ_LOG_FILE env vars.  It's fine to pass (0, nullptr) if args
   * are not accessible in the caller's context, it will just do nothing.
   * Note that the args take effect (are processed) only when this function
   * is called the first time.
   */
  static void Init(int argc, char* argv[]);

  /**
   * Sets the log file to the given filename.
   */
  static void SetLogFile(const char* aFilename);

  /**
   * @param aBuffer - pointer to a buffer
   * @param aLength - the length of the buffer
   *
   * @return the actual length of the filepath.
   */
  static uint32_t GetLogFile(char* aBuffer, size_t aLength);

  /**
   * @param aAddTimestamp If we should log a time stamp with every message.
   */
  static void SetAddTimestamp(bool aAddTimestamp);

  /**
   * @param aIsSync If we should flush the file after every logged message.
   */
  static void SetIsSync(bool aIsSync);

  /**
   * @param aCaptureStacks If we should capture stacks for the Firefox
   * Profiler markers that are recorded for for each log entry.
   */
  static void SetCaptureStacks(bool aCaptureStacks);

  /**
   * @return If we should log JS Stacks when logging js console messages.
   * (may be extended outside of just the js console)
   */
  static bool GetLogJSStacks();

  /**
   * Indicates whether or not the given log level is enabled.
   */
  bool ShouldLog(LogLevel aLevel) const { return mLevel >= aLevel; }

  /**
   * Retrieves the log module's current level.
   */
  LogLevel Level() const { return mLevel; }

  /**
   * Sets the log module's level.
   */
  void SetLevel(LogLevel level);

  /**
   * Print a log message for this module.
   */
  void Printv(LogLevel aLevel, const char* aFmt, va_list aArgs) const
      MOZ_FORMAT_PRINTF(3, 0);

  void Printv(LogLevel aLevel, const TimeStamp* aStart, const char* aFmt,
              va_list aArgs) const MOZ_FORMAT_PRINTF(4, 0);

  /**
   * Use {fmt} for specifying format
   */
  void PrintvFmt(LogLevel aLevel, fmt::string_view aFmt,
                 fmt::format_args aArgs) const;

  /**
   * Retrieves the module name.
   */
  const char* Name() const { return mName; }

  AtomicLogLevel& LevelRef() { return mLevel; }

 private:
  friend class LogModuleManager;

  explicit LogModule(const char* aName, LogLevel aLevel)
      : mName(strdup(aName)), mLevel(aLevel) {}

  LogModule(LogModule&) = delete;
  LogModule& operator=(const LogModule&) = delete;

  char* mName;

  AtomicLogLevel mLevel;
};

/**
 * Helper class that lazy loads the given log module. This is safe to use for
 * declaring static references to log modules and can be used as a replacement
 * for accessing a LogModule directly.
 *
 * Example usage:
 *   static LazyLogModule sLayoutLog("layout");
 *
 *   void Foo() {
 *     MOZ_LOG(sLayoutLog, LogLevel::Verbose, ("Entering foo"));
 *   }
 */
class LazyLogModule final {
 public:
  explicit constexpr LazyLogModule(const char* aLogName)
      : mLogName(aLogName), mLog(nullptr) {}

  MOZ_NEVER_INLINE_DEBUG operator LogModule*() {
    // NB: The use of an atomic makes the reading and assignment of mLog
    //     thread-safe. There is a small chance that mLog will be set more
    //     than once, but that's okay as it will be set to the same LogModule
    //     instance each time. Also note LogModule::Get is thread-safe.
    LogModule* tmp = mLog;
    if (MOZ_UNLIKELY(!tmp)) {
      tmp = LogModule::Get(mLogName);
      mLog = tmp;
    }

    return tmp;
  }

 private:
  const char* const mLogName;

  Atomic<LogModule*, ReleaseAcquire> mLog;
};

namespace detail {

inline bool log_test(const LogModule* module, LogLevel level) {
  MOZ_ASSERT(level != LogLevel::Disabled);
  return module && module->ShouldLog(level);
}

void log_print(const LogModule* aModule, LogLevel aLevel, const char* aFmt, ...)
    MOZ_FORMAT_PRINTF(3, 4);

template <typename... T>
inline void log_print_fmt(const LogModule* aModule, LogLevel aLevel,
                          fmt::format_string<T...> aFmt, T&&... aArgs) {
  aModule->PrintvFmt(aLevel, aFmt, fmt::make_format_args(aArgs...));
}

void log_print(const LogModule* aModule, LogLevel aLevel, TimeStamp* aStart,
               const char* aFmt, ...) MOZ_FORMAT_PRINTF(4, 5);
}  // namespace detail

}  // namespace mozilla

// Helper macro used convert MOZ_LOG's third parameter, |_args|, from a
// parenthesized form to a varargs form. For example:
//   ("%s", "a message") => "%s", "a message"
#define MOZ_LOG_EXPAND_ARGS(...) __VA_ARGS__

#if MOZ_LOGGING_ENABLED
#  define MOZ_LOG_TEST(_module, _level) \
    MOZ_UNLIKELY(mozilla::detail::log_test(_module, _level))
#else
// Define away MOZ_LOG_TEST here so the compiler will fold away entire
// logging blocks via dead code elimination, e.g.:
//
//   if (MOZ_LOG_TEST(...)) {
//     ...compute things to log and log them...
//   }
#  define MOZ_LOG_TEST(_module, _level) false
#endif

// The natural definition of the MOZ_LOG macro would expand to:
//
//   do {
//     if (MOZ_LOG_TEST(_module, _level)) {
//       mozilla::detail::log_print(_module, ...);
//     }
//   } while (0)
//
// However, since _module is a LazyLogModule, and we need to call
// LazyLogModule::operator() to get a LogModule* for the MOZ_LOG_TEST
// macro and for the logging call, we'll wind up doing *two* calls, one
// for each, rather than a single call.  The compiler is not able to
// fold the two calls into one, and the extra call can have a
// significant effect on code size.  (Making LazyLogModule::operator() a
// `const` function does not have any effect.)
//
// Therefore, we will have to make a single call ourselves.  But again,
// the natural definition:
//
//   do {
//     ::mozilla::LogModule* real_module = _module;
//     if (MOZ_LOG_TEST(real_module, _level)) {
//       mozilla::detail::log_print(real_module, ...);
//     }
//   } while (0)
//
// also has a problem: if logging is disabled, then we will call
// LazyLogModule::operator() unnecessarily, and the compiler will not be
// able to optimize away the call as dead code.  We would like to avoid
// such a scenario, as the whole point of disabling logging is for the
// logging statements to not generate any code.
//
// Therefore, we need different definitions of MOZ_LOG, depending on
// whether logging is enabled or not.  (We need an actual definition of
// MOZ_LOG even when logging is disabled to ensure the compiler sees that
// variables only used during logging code are actually used, even if the
// code will never be executed.)  Hence, the following code.
//
// MOZ_LOG_DURATION takes a start time, and will generate a time range
// in the logs.  Also, if the Firefox Profiler is running,
// MOZ_LOG_DURATION will generate a marker with a time duration
// instead of a single point in time.
#if MOZ_LOGGING_ENABLED
#  define MOZ_LOG(_module, _level, _args)                      \
    do {                                                       \
      const ::mozilla::LogModule* moz_real_module = _module;   \
      if (MOZ_LOG_TEST(moz_real_module, _level)) {             \
        mozilla::detail::log_print(moz_real_module, _level,    \
                                   MOZ_LOG_EXPAND_ARGS _args); \
      }                                                        \
    } while (0)
#  define MOZ_LOG_DURATION(_module, _level, start, _args)          \
    do {                                                           \
      const ::mozilla::LogModule* moz_real_module = _module;       \
      if (MOZ_LOG_TEST(moz_real_module, _level)) {                 \
        mozilla::detail::log_print(moz_real_module, _level, start, \
                                   MOZ_LOG_EXPAND_ARGS _args);     \
      }                                                            \
    } while (0)
#  define MOZ_LOG_FMT(_module, _level, _fmt, ...)                     \
    do {                                                              \
      const ::mozilla::LogModule* moz_real_module = _module;          \
      if (MOZ_LOG_TEST(moz_real_module, _level)) {                    \
        mozilla::detail::log_print_fmt(moz_real_module, _level, _fmt, \
                                       ##__VA_ARGS__);                \
      }                                                               \
    } while (0)
#else
#  define MOZ_LOG(_module, _level, _args)                      \
    do {                                                       \
      if (MOZ_LOG_TEST(_module, _level)) {                     \
        mozilla::detail::log_print(_module, _level,            \
                                   MOZ_LOG_EXPAND_ARGS _args); \
      }                                                        \
    } while (0)
#  define MOZ_LOG_DURATION(_module, _level, start, _args)      \
    do {                                                       \
      if (MOZ_LOG_TEST(_module, _level)) {                     \
        mozilla::detail::log_print(_module, _level, start,     \
                                   MOZ_LOG_EXPAND_ARGS _args); \
      }                                                        \
    } while (0)
#  define MOZ_LOG_FMT(_module, _level, _fmt, ...)                             \
    do {                                                                      \
      if (MOZ_LOG_TEST(_module, _level)) {                                    \
        mozilla::detail::log_print_fmt(_module, _level, _fmt, ##__VA_ARGS__); \
      }                                                                       \
    } while (0)
#endif

#ifdef DEBUG
   // MOZ_LOG_DEBUG_ONLY calls MOZ_LOG only in debug builds.
#  define MOZ_LOG_DEBUG_ONLY(...) MOZ_LOG(__VA_ARGS__)
   // MOZ_LOG_DURATION_DEBUG_ONLY calls MOZ_LOG_DURATION only in debug builds.
#  define MOZ_LOG_DURATION_DEBUG_ONLY(...) MOZ_LOG_DURATION(__VA_ARGS__)
   // MOZ_LOG_FMT_DEBUG_ONLY calls MOZ_LOG_FMT only in debug builds.
#  define MOZ_LOG_FMT_DEBUG_ONLY(...) MOZ_LOG_FMT(__VA_ARGS__)
#else
   // MOZ_LOG_DEBUG_ONLY is noop in non-debug builds.
#  define MOZ_LOG_DEBUG_ONLY(...)
   // MOZ_LOG_DURATION_DEBUG_ONLY is noop in non-debug builds.
#  define MOZ_LOG_DURATION_DEBUG_ONLY(...)
   // MOZ_LOG_FMT_DEBUG_ONLY is noop in non-debug builds.
#  define MOZ_LOG_FMT_DEBUG_ONLY(...)
#endif

// This #define is a Logging.h-only knob!  Don't encourage people to get fancy
// with their log definitions by exporting it outside of Logging.h.
#undef MOZ_LOGGING_ENABLED

/**
 * Define a utility method to pretty print a boolean value.
 * TrueOrFalse, YesOrNo, SucceededOrFailed, OkOrError, DoneOrIgnored,
 * HandledOrIgnored, DoneOrCanceled, HandledOrCanceled are defined in the
 * global scope.
 */
#define MOZ_DEFINE_BOOL_PRETTY_PRINTER(_method_name, _true_str, _false_str) \
  inline const char* _method_name(bool aBool) {                             \
    return aBool ? #_true_str : #_false_str;                                \
  }

namespace mozilla {
MOZ_DEFINE_BOOL_PRETTY_PRINTER(TrueOrFalse, true, false);
MOZ_DEFINE_BOOL_PRETTY_PRINTER(YesOrNo, Yes, No);
MOZ_DEFINE_BOOL_PRETTY_PRINTER(OnOrOff, On, Off);
MOZ_DEFINE_BOOL_PRETTY_PRINTER(SucceededOrFailed, Succeeded, Failed);
MOZ_DEFINE_BOOL_PRETTY_PRINTER(OkOrError, OK, Failed);
MOZ_DEFINE_BOOL_PRETTY_PRINTER(DoneOrIgnored, Done, Ignored);
MOZ_DEFINE_BOOL_PRETTY_PRINTER(HandledOrIgnored, Handled, Ignored);
MOZ_DEFINE_BOOL_PRETTY_PRINTER(DoneOrCanceled, Done, Canceled);
MOZ_DEFINE_BOOL_PRETTY_PRINTER(HandledOrCanceled, Handled, Canceled);
}  // namespace mozilla

#endif  // mozilla_logging_h