...

3 files changed, 28 insertions, 5 deletions

diff --git a/src/ui/controls/text_block.cpp b/src/ui/controls/text_block.cpp
index 0a243d2e..1c20540d 100644
--- a/src/ui/controls/text_block.cpp
+++ b/src/ui/controls/text_block.cpp
@@ -56,7 +56,7 @@ render::CanvasRenderObject* TextBlock::GetCaretRenderObject() {
   return caret_render_object_.get();
 }
 
-platform::graph::IBrush* TextBlock::GetCaretBrush() {
-  return caret_brush_.get();
+std::shared_ptr<platform::graph::IBrush> TextBlock::GetCaretBrush() {
+  return caret_brush_;
 }
 }  // namespace cru::ui::controls
diff --git a/src/ui/controls/text_box.cpp b/src/ui/controls/text_box.cpp
index 9c6fcbe5..1e0890b8 100644
--- a/src/ui/controls/text_box.cpp
+++ b/src/ui/controls/text_box.cpp
@@ -5,6 +5,7 @@
 #include "cru/ui/render/stack_layout_render_object.hpp"
 #include "cru/ui/render/text_render_object.hpp"
 #include "cru/ui/ui_manager.hpp"
+#include "text_control_service.hpp"
 
 namespace cru::ui::controls {
 using render::BorderRenderObject;
@@ -16,9 +17,13 @@ TextBox::TextBox()
     : border_render_object_(new BorderRenderObject()),
       stack_layout_render_object_(new StackLayoutRenderObject()),
       text_render_object_(),
-      caret_render_object_(new CanvasRenderObject()) {
+      caret_render_object_(new CanvasRenderObject()),
+      service_(new TextControlService<TextBox>(this)) {
   const auto theme_resources = UiManager::GetInstance()->GetThemeResources();
 
+  caret_brush_ = theme_resources->caret_brush;
+  border_style_ = theme_resources->text_box_border_style;
+
   text_render_object_ = std::make_unique<TextRenderObject>(
       theme_resources->text_brush, theme_resources->default_font,
       theme_resources->text_selection_brush);
@@ -34,4 +39,22 @@ TextBox::TextBox()
 }
 
 TextBox::~TextBox() {}
+
+render::TextRenderObject* TextBox::GetTextRenderObject() {
+  return text_render_object_.get();
+}
+
+render::CanvasRenderObject* TextBox::GetCaretRenderObject() {
+  return caret_render_object_.get();
+}
+
+std::shared_ptr<platform::graph::IBrush> TextBox::GetCaretBrush() {
+  return caret_brush_;
+}
+
+const TextBoxBorderStyle& TextBox::GetBorderStyle() { return border_style_; }
+
+void TextBox::SetBorderStyle(TextBoxBorderStyle border_style) {
+  border_style_ = std::move(border_style_);
+}
 }  // namespace cru::ui::controls
diff --git a/src/ui/controls/text_control_service.hpp b/src/ui/controls/text_control_service.hpp
index 3885242a..d4f3ecf9 100644
--- a/src/ui/controls/text_control_service.hpp
+++ b/src/ui/controls/text_control_service.hpp
@@ -16,7 +16,7 @@ constexpr long long caret_blink_duration = 500;
 // ```
 // render::TextRenderObject* GetTextRenderObject();
 // render::CanvasRenderObject* GetCaretRenderObject();
-// platform::graph::IBrush* GetCaretBrush();
+// std::shared_ptr<platform::graph::IBrush> GetCaretBrush();
 // ```
 template <typename TControl>
 class TextControlService : public Object {
@@ -125,7 +125,7 @@ void TextControlService<TControl>::DrawCaret(
     painter->FillRectangle(
         Rect{point,
              Size{caret_width, text_render_object->GetFont()->GetFontSize()}},
-        control_->GetCaretBrush());
+        control_->GetCaretBrush().get());
   }
 }