1 files changed, 17 insertions, 0 deletions
diff --git a/src/EditModel.cxx b/src/EditModel.cxx
index 83cade5ef..db8682efb 100644
--- a/src/EditModel.cxx
+++ b/src/EditModel.cxx
@@ -88,3 +88,20 @@ bool EditModel::BidirectionalEnabled() const {
 bool EditModel::BidirectionalR2L() const {
 	return bidirectional == Bidirectional::bidiR2L;
 }
+
+void EditModel::SetDefaultFoldDisplayText(const char *text) {
+	defaultFoldDisplayText = IsNullOrEmpty(text) ? UniqueString() : UniqueStringCopy(text);
+}
+
+const char *EditModel::GetDefaultFoldDisplayText() const noexcept {
+	return defaultFoldDisplayText.get();
+}
+
+const char *EditModel::GetFoldDisplayText(Sci::Line lineDoc) const {
+	if (foldDisplayTextStyle == SC_FOLDDISPLAYTEXT_HIDDEN || pcs->GetExpanded(lineDoc)) {
+		return nullptr;
+	}
+
+	const char *text = pcs->GetFoldDisplayText(lineDoc);
+	return text ? text : defaultFoldDisplayText.get();
+}