Account for indentation level in -writeHTMLString:

karelia · Jun 1, 2012 · 6bfd659 · 6bfd659
1 parent 12b9584
commit 6bfd659
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/KSHTMLWriter.h b/KSHTMLWriter.h
@@ -71,6 +71,7 @@ extern NSString *KSHTMLWriterDocTypeHTML_5;
 
 
 #pragma mark HTML Fragments
+// Any newlines in the HTML will be adjusted to account for current indentation level, but that's all
 - (void)writeHTMLString:(NSString *)html;
 - (void)writeHTMLFormat:(NSString *)format , ...;
 

diff --git a/KSHTMLWriter.m b/KSHTMLWriter.m
@@ -162,6 +162,16 @@ - (BOOL)hasCurrentAttributes;
 
 - (void)writeHTMLString:(NSString *)html;
 {
+    NSUInteger indent = [self indentationLevel];
+    if (indent)
+    {
+        NSString *indentedNewline = [@"\n" stringByPaddingToLength:indent + 1
+                                                        withString:@"\t"
+                                                   startingAtIndex:0];
+
+        html = [html stringByReplacingOccurrencesOfString:@"\n" withString:indentedNewline];
+    }
+
     [self writeString:html];
 }