Expose -className.

karelia · Jul 27, 2010 · ecfabe0 · ecfabe0
1 parent 8783f1f
commit ecfabe0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/KSHTMLWriter.h b/KSHTMLWriter.h
@@ -36,6 +36,7 @@
 #pragma mark CSS Class Name
 // Class names are accumulated and written automatically as an attribute of the next element started
 - (void)addClassName:(NSString *)className;
+- (NSString *)className;
 
 
 #pragma mark HTML Fragments

diff --git a/KSHTMLWriter.m b/KSHTMLWriter.m
@@ -59,6 +59,16 @@ - (void)addClassName:(NSString *)className;
     [_classNames addObject:className];
 }
 
+- (NSString *)className;
+{
+    NSString *result = nil;
+    if ([_classNames count])
+    {
+        result = [_classNames componentsJoinedByString:@" "];
+    }
+    return result;
+}
+
 #pragma mark HTML Fragments
 
 - (void)writeHTMLString:(NSString *)html;
@@ -294,9 +304,9 @@ - (BOOL)canWriteElementInline:(NSString *)tagName;
 - (void)startElement:(NSString *)elementName writeInline:(BOOL)writeInline; // for more control
 {
     // Add in any pre-written classes
-    if ([_classNames count])
+    NSString *class = [self className];
+    if (class)
     {
-        NSString *class = [_classNames componentsJoinedByString:@" "];
         [_classNames removeAllObjects];
         [self addElementAttribute:@"class" value:class];
     }