// Tencent is pleased to support the open source community by making RapidJSON available.
//
// Copyright (C) 2015 THL A29 Limited, a Tencent company, and Milo Yip. All rights reserved.
//
// Licensed under the MIT License (the "License"); you may not use this file except
// in compliance with the License. You may obtain a copy of the License at
//
// http://opensource.org/licenses/MIT
//
// Unless required by applicable law or agreed to in writing, software distributed
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.

#include "unittest.h"
#include "rapidjson/filereadstream.h"
#include "rapidjson/filewritestream.h"
#include "rapidjson/encodedstream.h"
#include "rapidjson/stringbuffer.h"
#include "rapidjson/memorystream.h"
#include "rapidjson/memorybuffer.h"

using namespace rapidjson;

class EncodedStreamTest : public ::testing::Test
{
public:
  EncodedStreamTest() : json_(), length_() {}
  virtual ~EncodedStreamTest();

  virtual void SetUp()
  { json_ = ReadFile("utf8.json", true, &length_);
  }

  virtual void TearDown()
  { free(json_);
    json_ = 0;
  }

private:
  EncodedStreamTest(const EncodedStreamTest&);
  EncodedStreamTest& operator=(const EncodedStreamTest&);

protected:
  static FILE* Open(const char* filename)
  { const char *paths[] =
    { "encodings",
      "bin/encodings",
      "../bin/encodings",
      "../../bin/encodings",
      "../../../bin/encodings"
    };
    char buffer[1024];
    for (size_t i = 0; i < sizeof(paths) / sizeof(paths[0]); i++)
    { sprintf(buffer, "%s/%s", paths[i], filename);
      FILE *fp = fopen(buffer, "rb");
      if (fp)
        return fp;
    }
    return 0;
  }

  static char *ReadFile(const char* filename, bool appendPath, size_t* outLength)
  { FILE *fp = appendPath ? Open(filename) : fopen(filename, "rb");

    if (!fp)
    { *outLength = 0;
      return 0;
    }

    fseek(fp, 0, SEEK_END);
    *outLength = static_cast<size_t>(ftell(fp));
    fseek(fp, 0, SEEK_SET);
    char* buffer = static_cast<char*>(malloc(*outLength + 1));
    size_t readLength = fread(buffer, 1, *outLength, fp);
    buffer[readLength] = '\0';
    fclose(fp);
    return buffer;
  }

  template <typename FileEncoding, typename MemoryEncoding>
  void TestEncodedInputStream(const char* filename)
  { // Test FileReadStream
    { char buffer[16];
      FILE *fp = Open(filename);
      ASSERT_TRUE(fp != 0);
      FileReadStream fs(fp, buffer, sizeof(buffer));
      EncodedInputStream<FileEncoding, FileReadStream> eis(fs);
      StringStream s(json_);

      while (eis.Peek() != '\0')
      { unsigned expected, actual;
        EXPECT_TRUE(UTF8<>::Decode(s, &expected));
        EXPECT_TRUE(MemoryEncoding::Decode(eis, &actual));
        EXPECT_EQ(expected, actual);
      }
      EXPECT_EQ('\0', s.Peek());
      fclose(fp);
    }

    // Test MemoryStream
    { size_t size;
      char* data = ReadFile(filename, true, &size);
      MemoryStream ms(data, size);
      EncodedInputStream<FileEncoding, MemoryStream> eis(ms);
      StringStream s(json_);

      while (eis.Peek() != '\0')
      { unsigned expected, actual;
        EXPECT_TRUE(UTF8<>::Decode(s, &expected));
        EXPECT_TRUE(MemoryEncoding::Decode(eis, &actual));
        EXPECT_EQ(expected, actual);
      }
      EXPECT_EQ('\0', s.Peek());
      free(data);
      EXPECT_EQ(size, eis.Tell());
    }
  }

  void TestAutoUTFInputStream(const char *filename, bool expectHasBOM)
  { // Test FileReadStream
    { char buffer[16];
      FILE *fp = Open(filename);
      ASSERT_TRUE(fp != 0);
      FileReadStream fs(fp, buffer, sizeof(buffer));
      AutoUTFInputStream<unsigned, FileReadStream> eis(fs);
      EXPECT_EQ(expectHasBOM, eis.HasBOM());
      StringStream s(json_);
      while (eis.Peek() != '\0')
      { unsigned expected, actual;
        EXPECT_TRUE(UTF8<>::Decode(s, &expected));
        EXPECT_TRUE(AutoUTF<unsigned>::Decode(eis, &actual));
        EXPECT_EQ(expected, actual);
      }
      EXPECT_EQ('\0', s.Peek());
      fclose(fp);
    }

    // Test MemoryStream
    { size_t size;
      char* data = ReadFile(filename, true, &size);
      MemoryStream ms(data, size);
      AutoUTFInputStream<unsigned, MemoryStream> eis(ms);
      EXPECT_EQ(expectHasBOM, eis.HasBOM());
      StringStream s(json_);

      while (eis.Peek() != '\0')
      { unsigned expected, actual;
        EXPECT_TRUE(UTF8<>::Decode(s, &expected));
        EXPECT_TRUE(AutoUTF<unsigned>::Decode(eis, &actual));
        EXPECT_EQ(expected, actual);
      }
      EXPECT_EQ('\0', s.Peek());
      free(data);
      EXPECT_EQ(size, eis.Tell());
    }
  }

  template <typename FileEncoding, typename MemoryEncoding>
  void TestEncodedOutputStream(const char* expectedFilename, bool putBOM)
  { // Test FileWriteStream
    { char filename[L_tmpnam];
      FILE* fp = TempFile(filename);
      char buffer[16];
      FileWriteStream os(fp, buffer, sizeof(buffer));
      EncodedOutputStream<FileEncoding, FileWriteStream> eos(os, putBOM);
      StringStream s(json_);
      while (s.Peek() != '\0')
      { bool success = Transcoder<UTF8<>, MemoryEncoding>::Transcode(s, eos);
        EXPECT_TRUE(success);
      }
      eos.Flush();
      fclose(fp);
      EXPECT_TRUE(CompareFile(filename, expectedFilename));
      remove(filename);
    }

    // Test MemoryBuffer
    { MemoryBuffer mb;
      EncodedOutputStream<FileEncoding, MemoryBuffer> eos(mb, putBOM);
      StringStream s(json_);
      while (s.Peek() != '\0')
      { bool success = Transcoder<UTF8<>, MemoryEncoding>::Transcode(s, eos);
        EXPECT_TRUE(success);
      }
      eos.Flush();
      EXPECT_TRUE(CompareBufferFile(mb.GetBuffer(), mb.GetSize(), expectedFilename));
    }
  }

  void TestAutoUTFOutputStream(UTFType type, bool putBOM, const char *expectedFilename)
  { // Test FileWriteStream
    { char filename[L_tmpnam];
      FILE* fp = TempFile(filename);

      char buffer[16];
      FileWriteStream os(fp, buffer, sizeof(buffer));
      AutoUTFOutputStream<unsigned, FileWriteStream> eos(os, type, putBOM);
      StringStream s(json_);
      while (s.Peek() != '\0')
      { bool success = Transcoder<UTF8<>, AutoUTF<unsigned> >::Transcode(s, eos);
        EXPECT_TRUE(success);
      }
      eos.Flush();
      fclose(fp);
      EXPECT_TRUE(CompareFile(filename, expectedFilename));
      remove(filename);
    }

    // Test MemoryBuffer
    { MemoryBuffer mb;
      AutoUTFOutputStream<unsigned, MemoryBuffer> eos(mb, type, putBOM);
      StringStream s(json_);
      while (s.Peek() != '\0')
      { bool success = Transcoder<UTF8<>, AutoUTF<unsigned> >::Transcode(s, eos);
        EXPECT_TRUE(success);
      }
      eos.Flush();
      EXPECT_TRUE(CompareBufferFile(mb.GetBuffer(), mb.GetSize(), expectedFilename));
    }
  }

  bool CompareFile(const char* filename, const char* expectedFilename)
  { size_t actualLength, expectedLength;
    char* actualBuffer = ReadFile(filename, false, &actualLength);
    char* expectedBuffer = ReadFile(expectedFilename, true, &expectedLength);
    bool ret = (expectedLength == actualLength) && memcmp(expectedBuffer, actualBuffer, actualLength) == 0;
    free(actualBuffer);
    free(expectedBuffer);
    return ret;
  }

  bool CompareBufferFile(const char* actualBuffer, size_t actualLength, const char* expectedFilename)
  { size_t expectedLength;
    char* expectedBuffer = ReadFile(expectedFilename, true, &expectedLength);
    bool ret = (expectedLength == actualLength) && memcmp(expectedBuffer, actualBuffer, actualLength) == 0;
    free(expectedBuffer);
    return ret;
  }

  char *json_;
  size_t length_;
};

EncodedStreamTest::~EncodedStreamTest() {}

TEST_F(EncodedStreamTest, EncodedInputStream)
{ TestEncodedInputStream<UTF8<>,    UTF8<>  >("utf8.json");
  TestEncodedInputStream<UTF8<>,    UTF8<>  >("utf8bom.json");
  TestEncodedInputStream<UTF16LE<>, UTF16<> >("utf16le.json");
  TestEncodedInputStream<UTF16LE<>, UTF16<> >("utf16lebom.json");
  TestEncodedInputStream<UTF16BE<>, UTF16<> >("utf16be.json");
  TestEncodedInputStream<UTF16BE<>, UTF16<> >("utf16bebom.json");
  TestEncodedInputStream<UTF32LE<>, UTF32<> >("utf32le.json");
  TestEncodedInputStream<UTF32LE<>, UTF32<> >("utf32lebom.json");
  TestEncodedInputStream<UTF32BE<>, UTF32<> >("utf32be.json");
  TestEncodedInputStream<UTF32BE<>, UTF32<> >("utf32bebom.json");
}

TEST_F(EncodedStreamTest, AutoUTFInputStream)
{ TestAutoUTFInputStream("utf8.json",      false);
  TestAutoUTFInputStream("utf8bom.json",   true);
  TestAutoUTFInputStream("utf16le.json",   false);
  TestAutoUTFInputStream("utf16lebom.json",true);
  TestAutoUTFInputStream("utf16be.json",   false);
  TestAutoUTFInputStream("utf16bebom.json",true);
  TestAutoUTFInputStream("utf32le.json",   false);
  TestAutoUTFInputStream("utf32lebom.json",true);
  TestAutoUTFInputStream("utf32be.json",   false);
  TestAutoUTFInputStream("utf32bebom.json", true);

  { // Auto detection fail, use user defined UTF type
    const char json[] = "{ }";
    MemoryStream ms(json, sizeof(json));
    AutoUTFInputStream<unsigned, MemoryStream> eis(ms, kUTF8);
    EXPECT_FALSE(eis.HasBOM());
    EXPECT_EQ(kUTF8, eis.GetType());
  }
}

TEST_F(EncodedStreamTest, EncodedOutputStream)
{ TestEncodedOutputStream<UTF8<>,     UTF8<>  >("utf8.json",      false);
  TestEncodedOutputStream<UTF8<>,     UTF8<>  >("utf8bom.json",   true);
  TestEncodedOutputStream<UTF16LE<>,  UTF16<> >("utf16le.json",   false);
  TestEncodedOutputStream<UTF16LE<>,  UTF16<> >("utf16lebom.json",true);
  TestEncodedOutputStream<UTF16BE<>,  UTF16<> >("utf16be.json",   false);
  TestEncodedOutputStream<UTF16BE<>,  UTF16<> >("utf16bebom.json",true);
  TestEncodedOutputStream<UTF32LE<>,  UTF32<> >("utf32le.json",   false);
  TestEncodedOutputStream<UTF32LE<>,  UTF32<> >("utf32lebom.json",true);
  TestEncodedOutputStream<UTF32BE<>,  UTF32<> >("utf32be.json",   false);
  TestEncodedOutputStream<UTF32BE<>,  UTF32<> >("utf32bebom.json",true);
}

TEST_F(EncodedStreamTest, AutoUTFOutputStream)
{ TestAutoUTFOutputStream(kUTF8,      false,  "utf8.json");
  TestAutoUTFOutputStream(kUTF8,      true,   "utf8bom.json");
  TestAutoUTFOutputStream(kUTF16LE,   false,  "utf16le.json");
  TestAutoUTFOutputStream(kUTF16LE,   true,   "utf16lebom.json");
  TestAutoUTFOutputStream(kUTF16BE,   false,  "utf16be.json");
  TestAutoUTFOutputStream(kUTF16BE,   true,   "utf16bebom.json");
  TestAutoUTFOutputStream(kUTF32LE,   false,  "utf32le.json");
  TestAutoUTFOutputStream(kUTF32LE,   true,   "utf32lebom.json");
  TestAutoUTFOutputStream(kUTF32BE,   false,  "utf32be.json");
  TestAutoUTFOutputStream(kUTF32BE,   true,   "utf32bebom.json");
}
