// Copyright 2017 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "xfa/fxfa/fm2js/cxfa_fmparser.h"

#include <vector>

#include "core/fxcrt/cfx_widetextbuf.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/test_support.h"
#include "third_party/base/ptr_util.h"
#include "xfa/fxfa/fm2js/cxfa_fmtojavascriptdepth.h"

TEST(CXFA_FMParserTest, Empty) {
  auto parser = pdfium::MakeUnique<CXFA_FMParser>(L"");
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(ast);
  EXPECT_FALSE(parser->HasError());

  CXFA_FMToJavaScriptDepth::Reset();
  CFX_WideTextBuf buf;
  EXPECT_TRUE(ast->ToJavaScript(&buf));
  // TODO(dsinclair): This is a little weird .....
  EXPECT_EQ(L"// comments only", buf.AsStringView());
}

TEST(CXFA_FMParserTest, CommentOnlyIsError) {
  auto parser = pdfium::MakeUnique<CXFA_FMParser>(L"; Just comment");
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(ast);
  // TODO(dsinclair): This isn't allowed per the spec.
  EXPECT_FALSE(parser->HasError());
  // EXPECT_TRUE(parser->HasError());

  CXFA_FMToJavaScriptDepth::Reset();
  CFX_WideTextBuf buf;
  EXPECT_TRUE(ast->ToJavaScript(&buf));
  EXPECT_EQ(L"// comments only", buf.AsStringView());
}

TEST(CXFA_FMParserTest, CommentThenValue) {
  const wchar_t ret[] =
      L"(function() {\n"
      L"let pfm_method_runner = function(obj, cb) {\n"
      L"  if (pfm_rt.is_ary(obj)) {\n"
      L"    let pfm_method_return = null;\n"
      L"    for (var idx = obj.length -1; idx > 1; idx--) {\n"
      L"      pfm_method_return = cb(obj[idx]);\n"
      L"    }\n"
      L"    return pfm_method_return;\n"
      L"  }\n"
      L"  return cb(obj);\n"
      L"};\n"
      L"var pfm_ret = null;\n"
      L"pfm_ret = 12;\n"
      L"return pfm_rt.get_val(pfm_ret);\n"
      L"}).call(this);";

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(L"; Just comment\n12");
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(ast);
  EXPECT_FALSE(parser->HasError());

  CXFA_FMToJavaScriptDepth::Reset();
  CFX_WideTextBuf buf;
  EXPECT_TRUE(ast->ToJavaScript(&buf));
  EXPECT_EQ(ret, buf.AsStringView());
}

TEST(CXFA_FMParserTest, Parse) {
  const wchar_t input[] =
      L"$ = Avg (-3, 5, -6, 12, -13);\n"
      L"$ = Avg (Table2..Row[*].Cell1);\n"
      L"\n"
      L"if ($ ne -1)then\n"
      L"  border.fill.color.value = \"255,64,64\";\n"
      L"elseif ($ ne -2) then\n"
      L"  border.fill.color.value = \"128,128,128\";\n"
      L"else\n"
      L"  border.fill.color.value = \"20,170,13\";\n"
      L"endif\n"
      L"\n"
      L"$";

  const wchar_t ret[] =
      L"(function() {\n"
      L"let pfm_method_runner = function(obj, cb) {\n"
      L"  if (pfm_rt.is_ary(obj)) {\n"
      L"    let pfm_method_return = null;\n"
      L"    for (var idx = obj.length -1; idx > 1; idx--) {\n"
      L"      pfm_method_return = cb(obj[idx]);\n"
      L"    }\n"
      L"    return pfm_method_return;\n"
      L"  }\n"
      L"  return cb(obj);\n"
      L"};\n"
      L"var pfm_ret = null;\n"
      L"if (pfm_rt.is_obj(this))\n{\n"
      L"pfm_rt.asgn_val_op(this, pfm_rt.Avg(pfm_rt.neg_op(3), 5, "
      L"pfm_rt.neg_op(6), 12, pfm_rt.neg_op(13)));\n"
      L"}\n"
      L"if (pfm_rt.is_obj(this))\n{\n"
      L"pfm_rt.asgn_val_op(this, pfm_rt.Avg(pfm_rt.dot_acc(pfm_rt.dotdot_acc("
      L"Table2, \"Table2\", \"Row\", 1), \"\", \"Cell1\", 0, 0)));\n"
      L"}\n"
      L"if (pfm_rt.get_val(pfm_rt.neq_op(this, pfm_rt.neg_op(1))))\n{\n"
      L"if (pfm_rt.is_obj(pfm_rt.dot_acc(pfm_rt.dot_acc(pfm_rt.dot_acc("
      L"border, \"border\", \"fill\", 0, 0), \"\", \"color\", 0, 0), \"\", "
      L"\"value\", 0, 0)))\n{\n"
      L"pfm_rt.asgn_val_op(pfm_rt.dot_acc(pfm_rt.dot_acc("
      L"pfm_rt.dot_acc(border, \"border\", \"fill\", 0, 0), \"\", "
      L"\"color\", 0, 0), \"\", \"value\", 0, 0), \"255,64,64\");\n"
      L"}\n"
      L"}\nelse if (pfm_rt.get_val(pfm_rt.neq_op(this, pfm_rt.neg_op(2))))\n"
      L"{\n"
      L"if (pfm_rt.is_obj(pfm_rt.dot_acc(pfm_rt.dot_acc(pfm_rt.dot_acc(border, "
      L"\"border\", \"fill\", 0, 0), \"\", \"color\", 0, 0), \"\", \"value\", "
      L"0, 0)))\n"
      L"{\n"
      L"pfm_rt.asgn_val_op(pfm_rt.dot_acc(pfm_rt.dot_acc(pfm_rt.dot_acc(border,"
      L" \"border\", \"fill\", 0, 0), \"\", \"color\", 0, 0), \"\", \"value\", "
      L"0, 0), \"128,128,128\");\n"
      L"}\n"
      L"}\n"
      L"else {\n"
      L"if (pfm_rt.is_obj(pfm_rt.dot_acc(pfm_rt.dot_acc(pfm_rt.dot_acc("
      L"border, \"border\", \"fill\", 0, 0), \"\", \"color\", 0, 0), \"\", "
      L"\"value\", 0, 0)))\n{\n"
      L"pfm_rt.asgn_val_op(pfm_rt.dot_acc(pfm_rt.dot_acc("
      L"pfm_rt.dot_acc(border, \"border\", \"fill\", 0, 0), \"\", "
      L"\"color\", 0, 0), \"\", \"value\", 0, 0), \"20,170,13\");\n"
      L"}\n"
      L"}\n"
      L"pfm_ret = this;\n"
      L"return pfm_rt.get_val(pfm_ret);\n"
      L"}).call(this);";

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(ast);
  EXPECT_FALSE(parser->HasError());

  CXFA_FMToJavaScriptDepth::Reset();
  CFX_WideTextBuf buf;
  EXPECT_TRUE(ast->ToJavaScript(&buf));
  EXPECT_EQ(ret, buf.AsStringView());
}

TEST(CXFA_FMParserTest, MaxParseDepth) {
  auto parser = pdfium::MakeUnique<CXFA_FMParser>(L"foo(bar[baz(fizz[0])])");
  parser->SetMaxParseDepthForTest(5);
  EXPECT_EQ(nullptr, parser->Parse());
  EXPECT_TRUE(parser->HasError());
}

TEST(CFXA_FMParserTest, chromium752201) {
  auto parser = pdfium::MakeUnique<CXFA_FMParser>(
      L"fTep a\n"
      L".#\n"
      L"fo@ =[=l");
  EXPECT_EQ(nullptr, parser->Parse());
  EXPECT_TRUE(parser->HasError());
}

TEST(CXFA_FMParserTest, MultipleAssignmentIsNotAllowed) {
  auto parser = pdfium::MakeUnique<CXFA_FMParser>(L"(a=(b=t))=u");

  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(!ast);
  EXPECT_TRUE(parser->HasError());
}

TEST(CXFA_FMParserTest, ParseFuncWithParams) {
  const wchar_t input[] = {
      L"func MyFunction(param1, param2) do\n"
      L"  param1 * param2\n"
      L"endfunc"};

  const wchar_t ret[] = {
      L"(function() {\n"
      L"let pfm_method_runner = function(obj, cb) {\n"
      L"  if (pfm_rt.is_ary(obj)) {\n"
      L"    let pfm_method_return = null;\n"
      L"    for (var idx = obj.length -1; idx > 1; idx--) {\n"
      L"      pfm_method_return = cb(obj[idx]);\n"
      L"    }\n"
      L"    return pfm_method_return;\n"
      L"  }\n"
      L"  return cb(obj);\n"
      L"};\n"
      L"var pfm_ret = null;\n"
      L"function MyFunction(param1, param2) {\n"
      L"var pfm_ret = null;\n"
      L"pfm_ret = pfm_rt.mul_op(param1, param2);\n"
      L"return pfm_ret;\n"
      L"}\n"
      L"return pfm_rt.get_val(pfm_ret);\n"
      L"}).call(this);"};

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(ast);
  EXPECT_FALSE(parser->HasError());

  CXFA_FMToJavaScriptDepth::Reset();
  CFX_WideTextBuf buf;
  EXPECT_TRUE(ast->ToJavaScript(&buf));
  EXPECT_EQ(ret, buf.AsStringView());
}

TEST(CXFA_FMParserTest, ParseFuncWithoutParams) {
  const wchar_t input[] = {
      L"func MyFunction() do\n"
      L"  42\n"
      L"endfunc"};

  const wchar_t ret[] = {
      L"(function() {\n"
      L"let pfm_method_runner = function(obj, cb) {\n"
      L"  if (pfm_rt.is_ary(obj)) {\n"
      L"    let pfm_method_return = null;\n"
      L"    for (var idx = obj.length -1; idx > 1; idx--) {\n"
      L"      pfm_method_return = cb(obj[idx]);\n"
      L"    }\n"
      L"    return pfm_method_return;\n"
      L"  }\n"
      L"  return cb(obj);\n"
      L"};\n"
      L"var pfm_ret = null;\n"
      L"function MyFunction() {\n"
      L"var pfm_ret = null;\n"
      L"pfm_ret = 42;\n"
      L"return pfm_ret;\n"
      L"}\n"
      L"return pfm_rt.get_val(pfm_ret);\n"
      L"}).call(this);"};

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(ast);
  EXPECT_FALSE(parser->HasError());

  CXFA_FMToJavaScriptDepth::Reset();
  CFX_WideTextBuf buf;
  EXPECT_TRUE(ast->ToJavaScript(&buf));
  EXPECT_EQ(ret, buf.AsStringView());
}

TEST(CXFA_FMParserTest, ParseFuncWithBadParamsList) {
  const wchar_t input[] = {
      L"func MyFunction(param1,) do\n"
      L"  param1 * param2\n"
      L"endfunc"};

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(ast == nullptr);
  EXPECT_TRUE(parser->HasError());
}

TEST(CXFA_FMParserTest, ParseBadIfExpression) {
  const wchar_t input[] = {L"if ( then"};

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(ast == nullptr);
  EXPECT_TRUE(parser->HasError());
}

TEST(CXFA_FMParserTest, ParseBadElseIfExpression) {
  const wchar_t input[] = {
      L"if ($ ne -1) then\n"
      L"elseif( then"};

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  ASSERT_TRUE(ast == nullptr);
  EXPECT_TRUE(parser->HasError());
}

TEST(CXFA_FMParserTest, ParseDepthWithWideTree) {
  const wchar_t input[] = {L"a <> b <> c <> d <> e <> f <> g <> h <> i <> j"};

  {
    auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
    std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
    ASSERT_TRUE(ast);
    EXPECT_TRUE(!parser->HasError());
  }

  {
    auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
    parser->SetMaxParseDepthForTest(5);
    std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
    ASSERT_TRUE(ast == nullptr);
    EXPECT_TRUE(parser->HasError());
  }
}

TEST(CXFA_FMParserTest, ParseCallSmall) {
  const wchar_t input[] = {L"i.f(O)"};
  const wchar_t ret[] = {
      L"(function() {\n"
      L"let pfm_method_runner = function(obj, cb) {\n"
      L"  if (pfm_rt.is_ary(obj)) {\n"
      L"    let pfm_method_return = null;\n"
      L"    for (var idx = obj.length -1; idx > 1; idx--) {\n"
      L"      pfm_method_return = cb(obj[idx]);\n"
      L"    }\n"
      L"    return pfm_method_return;\n"
      L"  }\n"
      L"  return cb(obj);\n"
      L"};\n"
      L"var pfm_ret = null;\n"
      L"pfm_ret = pfm_rt.get_val((function() {\n"
      L"  return pfm_method_runner(i, function(obj) {\n"
      L"    return obj.f(pfm_rt.get_val(O));\n"
      L"  });\n"
      L"}).call(this));\n"
      L"return pfm_rt.get_val(pfm_ret);\n"
      L"}).call(this);"};

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  EXPECT_FALSE(parser->HasError());

  CXFA_FMToJavaScriptDepth::Reset();
  CFX_WideTextBuf buf;
  EXPECT_TRUE(ast->ToJavaScript(&buf));
  EXPECT_EQ(ret, buf.AsStringView());
}

TEST(CXFA_FMParserTest, ParseCallBig) {
  const wchar_t input[] = {L"i.f(O.e(O.e(O)))"};
  const wchar_t ret[] = {
      L"(function() {\n"
      L"let pfm_method_runner = function(obj, cb) {\n"
      L"  if (pfm_rt.is_ary(obj)) {\n"
      L"    let pfm_method_return = null;\n"
      L"    for (var idx = obj.length -1; idx > 1; idx--) {\n"
      L"      pfm_method_return = cb(obj[idx]);\n"
      L"    }\n"
      L"    return pfm_method_return;\n"
      L"  }\n"
      L"  return cb(obj);\n"
      L"};\n"
      L"var pfm_ret = null;\n"
      L"pfm_ret = pfm_rt.get_val((function() {\n"
      L"  return pfm_method_runner(i, function(obj) {\n"
      L"    return obj.f(pfm_rt.get_val((function() {\n"
      L"  return pfm_method_runner(O, function(obj) {\n"
      L"    return obj.e(pfm_rt.get_val((function() {\n"
      L"  return pfm_method_runner(O, function(obj) {\n"
      L"    return obj.e(pfm_rt.get_val(O));\n"
      L"  });\n"
      L"}).call(this)));\n"
      L"  });\n"
      L"}).call(this)));\n"
      L"  });\n"
      L"}).call(this));\n"
      L"return pfm_rt.get_val(pfm_ret);\n"
      L"}).call(this);"};

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  EXPECT_FALSE(parser->HasError());

  CXFA_FMToJavaScriptDepth::Reset();
  CFX_WideTextBuf buf;
  EXPECT_TRUE(ast->ToJavaScript(&buf));
  EXPECT_EQ(ret, buf.AsStringView());
}

TEST(CXFA_FMParserTest, ParseVar) {
  const wchar_t input[] = L"var s = \"\"\n";
  const wchar_t ret[] =
      L"(function() {\n"
      L"let pfm_method_runner = function(obj, cb) {\n"
      L"  if (pfm_rt.is_ary(obj)) {\n"
      L"    let pfm_method_return = null;\n"
      L"    for (var idx = obj.length -1; idx > 1; idx--) {\n"
      L"      pfm_method_return = cb(obj[idx]);\n"
      L"    }\n"
      L"    return pfm_method_return;\n"
      L"  }\n"
      L"  return cb(obj);\n"
      L"};\n"
      L"var pfm_ret = null;\n"
      L"var s = \"\";\n"
      L"s = pfm_rt.var_filter(s);\n"
      L"pfm_ret = s;\n"
      L"return pfm_rt.get_val(pfm_ret);\n"
      L"}).call(this);";

  auto parser = pdfium::MakeUnique<CXFA_FMParser>(input);
  std::unique_ptr<CXFA_FMAST> ast = parser->Parse();
  EXPECT_FALSE(parser->HasError());

  CXFA_FMToJavaScriptDepth::Reset();
  CFX_WideTextBuf buf;
  EXPECT_TRUE(ast->ToJavaScript(&buf));
  EXPECT_STREQ(ret, buf.MakeString().c_str());
}