/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*-
 * vim: set ts=8 sts=2 et sw=2 tw=80:
 */
/* 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/. */

#include <cstdio>  // fclose, fflush, open_memstream

#include "jsapi.h"           // JS_{Clear,Get}PendingException
#include "jsfriendapi.h"     // js::ErrorReport
#include "js/ErrorReport.h"  // JS::PrintError
#include "js/Warnings.h"     // JS::SetWarningReporter, JS::WarnUTF8

#include "jsapi-tests/tests.h"

class AutoStreamBuffer {
  char* buffer;
  size_t size;
  FILE* fp;

 public:
  AutoStreamBuffer() { fp = open_memstream(&buffer, &size); }

  ~AutoStreamBuffer() {
    fclose(fp);
    free(buffer);
  }

  FILE* stream() { return fp; }

  bool contains(const char* str) {
    if (fflush(fp) != 0) {
      fprintf(stderr, "Error flushing stream\n");
      return false;
    }
    if (strcmp(buffer, str) != 0) {
      fprintf(stderr, "Expected |%s|, got |%s|\n", str, buffer);
      return false;
    }
    return true;
  }
};

BEGIN_TEST(testPrintError_Works) {
  AutoStreamBuffer buf;

  CHECK(!execDontReport("throw null;", "testPrintError_Works.js", 3));

  JS::ExceptionStack exnStack(cx);
  CHECK(JS::StealPendingExceptionStack(cx, &exnStack));

  JS::ErrorReportBuilder builder(cx);
  CHECK(builder.init(cx, exnStack, JS::ErrorReportBuilder::NoSideEffects));
  JS::PrintError(cx, buf.stream(), builder, false);

  CHECK(buf.contains("testPrintError_Works.js:3:1 uncaught exception: null\n"));

  return true;
}
END_TEST(testPrintError_Works)

BEGIN_TEST(testPrintError_SkipWarning) {
  JS::SetWarningReporter(cx, warningReporter);
  CHECK(JS::WarnUTF8(cx, "warning message"));
  CHECK(warningSuccess);
  return true;
}

static bool warningSuccess;

static void warningReporter(JSContext* cx, JSErrorReport* report) {
  AutoStreamBuffer buf;
  JS::PrintError(cx, buf.stream(), report, false);
  warningSuccess = buf.contains("");
}
END_TEST(testPrintError_SkipWarning)

bool cls_testPrintError_SkipWarning::warningSuccess = false;

BEGIN_TEST(testPrintError_PrintWarning) {
  JS::SetWarningReporter(cx, warningReporter);
  CHECK(JS::WarnUTF8(cx, "warning message"));
  CHECK(warningSuccess);
  return true;
}

static bool warningSuccess;

static void warningReporter(JSContext* cx, JSErrorReport* report) {
  AutoStreamBuffer buf;
  JS::PrintError(cx, buf.stream(), report, true);
  warningSuccess = buf.contains("warning: warning message\n");
}
END_TEST(testPrintError_PrintWarning)

bool cls_testPrintError_PrintWarning::warningSuccess = false;

#define BURRITO "\xF0\x9F\x8C\xAF"

BEGIN_TEST(testPrintError_UTF16CodePoints) {
  AutoStreamBuffer buf;

  static const char utf8code[] =
      "function f() {\n  var x = `\n" BURRITO "`; " BURRITO "; } f();";

  CHECK(!execDontReport(utf8code, "testPrintError_UTF16CodePoints.js", 1));

  JS::ExceptionStack exnStack(cx);
  CHECK(JS::StealPendingExceptionStack(cx, &exnStack));

  JS::ErrorReportBuilder builder(cx);
  CHECK(builder.init(cx, exnStack, JS::ErrorReportBuilder::NoSideEffects));
  JS::PrintError(cx, buf.stream(), builder, false);

  CHECK(
      buf.contains("testPrintError_UTF16CodePoints.js:3:4 SyntaxError: illegal "
                   "character U+1F32F:\n"
                   "testPrintError_UTF16CodePoints.js:3:4 " BURRITO
                   "`; " BURRITO "; } f();\n"
                   "testPrintError_UTF16CodePoints.js:3:4 .....^\n"));

  return true;
}
END_TEST(testPrintError_UTF16CodePoints)

#undef BURRITO