diff --git a/KSHTMLWriter.h b/KSHTMLWriter.h
index 0919311..e05ddd9 100644
--- 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
index c2ab394..2297736 100644
--- 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];
}