feat: More convenient inits for rich text.

laosb · Feb 21, 2024 · 6eeb716 · 6eeb716
1 parent a098c5e
commit 6eeb716
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/Sources/EasyRichText/Models/ERTRichText.swift b/Sources/EasyRichText/Models/ERTRichText.swift
@@ -38,6 +38,11 @@ public extension ERTRichText {
         self.init(attributedString: attributedString)
     }
 
+    init(string: String) {
+        self.init()
+        self.segments = [.init(text: string)]
+    }
+
     func nsAttributedString(defaultFont: CTFont) -> NSAttributedString {
         ERTAttributedStringBridge.default.nsAttributedString(for: attributedString(defaultFont: defaultFont))
     }

diff --git a/Sources/EasyRichText/Models/ERTSegment.swift b/Sources/EasyRichText/Models/ERTSegment.swift
@@ -15,6 +15,12 @@ public protocol ERTSegment: Codable, Hashable {
     func attributedString(defaultFont: CTFont) -> AttributedString
 }
 
+public extension ERTSegment {
+    init(text: String) {
+        self.init(text: text, attributeContainer: .init())
+    }
+}
+
 public protocol ERTForegroundColorSegment: Codable, Hashable {
     associatedtype ForegroundColor: ERTColor