#include <gtest/gtest-message.h>  // for Message
#include <gtest/gtest-test-part.h>  // for TestPartResult, SuiteApiResolver, TestFactoryImpl
#include <string>                   // for allocator, string

#include "ftxui/screen/string.hpp"
#include "gtest/gtest_pred_impl.h"  // for EXPECT_EQ, Test, TEST

using namespace ftxui;

TEST(StringTest, StringWidth) {
  // Basic:
  EXPECT_EQ(0, string_width(""));
  EXPECT_EQ(1, string_width("a"));
  EXPECT_EQ(2, string_width("ab"));
  // Fullwidth glyphs:
  EXPECT_EQ(2, string_width("测"));
  EXPECT_EQ(4, string_width("测试"));
  // Combining characters:
  EXPECT_EQ(1, string_width("ā"));
  EXPECT_EQ(1, string_width("a⃒"));
  EXPECT_EQ(1, string_width("a̗"));
  // Control characters:
  EXPECT_EQ(0, string_width("\1"));
  EXPECT_EQ(2, string_width("a\1a"));
}

TEST(StringTest, Utf8ToGlyphs) {
  using T = std::vector<std::string>;
  // Basic:
  EXPECT_EQ(Utf8ToGlyphs(""), T({}));
  EXPECT_EQ(Utf8ToGlyphs("a"), T({"a"}));
  EXPECT_EQ(Utf8ToGlyphs("ab"), T({"a", "b"}));
  // Fullwidth glyphs:
  EXPECT_EQ(Utf8ToGlyphs("测"), T({"测", ""}));
  EXPECT_EQ(Utf8ToGlyphs("测试"), T({"测", "", "试", ""}));
  // Combining characters:
  EXPECT_EQ(Utf8ToGlyphs("ā"), T({"ā"}));
  EXPECT_EQ(Utf8ToGlyphs("a⃒"), T({"a⃒"}));
  EXPECT_EQ(Utf8ToGlyphs("a̗"), T({"a̗"}));
  // Control characters:
  EXPECT_EQ(Utf8ToGlyphs("\1"), T({}));
  EXPECT_EQ(Utf8ToGlyphs("a\1a"), T({"a", "a"}));
}

TEST(StringTest, GlyphCount) {
  // Basic:
  EXPECT_EQ(GlyphCount(""), 0);
  EXPECT_EQ(GlyphCount("a"), 1);
  EXPECT_EQ(GlyphCount("ab"), 2);
  // Fullwidth glyphs:
  EXPECT_EQ(GlyphCount("测"), 1);
  EXPECT_EQ(GlyphCount("测试"), 2);
  // Combining characters:
  EXPECT_EQ(GlyphCount("ā"), 1);
  EXPECT_EQ(GlyphCount("a⃒"), 1);
  EXPECT_EQ(GlyphCount("a̗"), 1);
  // Control characters:
  EXPECT_EQ(GlyphCount("\1"), 0);
  EXPECT_EQ(GlyphCount("a\1a"), 2);
}


TEST(StringTest, GlyphPosition) {
  // Basic:
  EXPECT_EQ(GlyphPosition("", -1), 0);
  EXPECT_EQ(GlyphPosition("", 0), 0);
  EXPECT_EQ(GlyphPosition("", 1), 0);
  EXPECT_EQ(GlyphPosition("a", 0), 0);
  EXPECT_EQ(GlyphPosition("a", 1), 1);
  EXPECT_EQ(GlyphPosition("ab", 0), 0);
  EXPECT_EQ(GlyphPosition("ab", 1), 1);
  EXPECT_EQ(GlyphPosition("ab", 2), 2);
  EXPECT_EQ(GlyphPosition("abc", 0), 0);
  EXPECT_EQ(GlyphPosition("abc", 1), 1);
  EXPECT_EQ(GlyphPosition("abc", 2), 2);
  EXPECT_EQ(GlyphPosition("abc", 3), 3);
  // Fullwidth glyphs:
  EXPECT_EQ(GlyphPosition("测", 0), 0);
  EXPECT_EQ(GlyphPosition("测", 1), 3);
  EXPECT_EQ(GlyphPosition("测试", 0), 0);
  EXPECT_EQ(GlyphPosition("测试", 1), 3);
  EXPECT_EQ(GlyphPosition("测试", 2), 6);
  EXPECT_EQ(GlyphPosition("测试", 1, 3), 6);
  EXPECT_EQ(GlyphPosition("测试", 1, 0), 3);
  // Combining characters:
  EXPECT_EQ(GlyphPosition("ā", 0), 0);
  EXPECT_EQ(GlyphPosition("ā", 1), 3);
  EXPECT_EQ(GlyphPosition("a⃒a̗ā", 0), 0);
  EXPECT_EQ(GlyphPosition("a⃒a̗ā", 1), 4);
  EXPECT_EQ(GlyphPosition("a⃒a̗ā", 2), 7);
  EXPECT_EQ(GlyphPosition("a⃒a̗ā", 3), 10);
  // Control characters:
  EXPECT_EQ(GlyphPosition("\1", 0), 0);
  EXPECT_EQ(GlyphPosition("\1", 1), 1);
  EXPECT_EQ(GlyphPosition("a\1a", 0), 0);
  EXPECT_EQ(GlyphPosition("a\1a", 1), 2);
  EXPECT_EQ(GlyphPosition("a\1a", 2), 3);
}

TEST(StringTest, CellToGlyphIndex) {
  // Basic:
  auto basic = CellToGlyphIndex("abc");
  ASSERT_EQ(basic.size(), 3);
  EXPECT_EQ(basic[0], 0);
  EXPECT_EQ(basic[1], 1);
  EXPECT_EQ(basic[2], 2);

  // Fullwidth glyphs:
  auto fullwidth = CellToGlyphIndex("测试");
  ASSERT_EQ(fullwidth.size(), 4);
  EXPECT_EQ(fullwidth[0], 0);
  EXPECT_EQ(fullwidth[1], 0);
  EXPECT_EQ(fullwidth[2], 1);
  EXPECT_EQ(fullwidth[3], 1);

  // Combining characters:
  auto combining = CellToGlyphIndex("a⃒a̗ā");
  ASSERT_EQ(combining.size(), 3);
  EXPECT_EQ(combining[0], 0);
  EXPECT_EQ(combining[1], 1);
  EXPECT_EQ(combining[2], 2);
}

// Copyright 2020 Arthur Sonzogni. All rights reserved.
// Use of this source code is governed by the MIT license that can be found in
// the LICENSE file.