#import "VVDocumenter.h"

#import "NSString+VVSyntax.h"

#import "VVCommenter.h"

@interface VVDocumenter()

@property (nonatomic, copy) NSString *code;

@property (nonatomic, assign) BOOL isEnum;

@implementation VVDocumenter

-(id) initWithCode:(NSString *)code

{

self = [super init];

if (self) {

self.isEnum = NO;

if ([code vv_isEnum]) {

self.code = code;

self.isEnum = YES;

} else {

//Trim the space around the braces

//Then trim the new line character

self.code = [[code vv_stringByReplacingRegexPattern:@"\\s*(\\(.*\?\\))\\s*" withString:@"$1"]

vv_stringByReplacingRegexPattern:@"\\s*\n\\s*" withString:@" "];

}

}

return self;

}

-(NSString *) baseIndentation

{

NSArray *matchedSpaces = [self.code vv_stringsByExtractingGroupsUsingRegexPattern:@"^(\\s*)"];

if (matchedSpaces.count > 0) {

return matchedSpaces[0];

} else {

return @"";

}

}

-(NSString *) document

{

NSString *trimCode = [self.code stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceCharacterSet]];

NSString *baseIndent = [self baseIndentation];

VVBaseCommenter *commenter = nil;

if (self.isEnum) {

commenter = [[VVEnumCommenter alloc] initWithIndentString:baseIndent codeString:trimCode];

} else if ([trimCode vv_isProperty]) {

commenter = [[VVPropertyCommenter alloc] initWithIndentString:baseIndent codeString:trimCode];

} else if ([trimCode vv_isCFunction]) {

commenter = [[VVFunctionCommenter alloc] initWithIndentString:baseIndent codeString:trimCode];

} else if ([trimCode vv_isMacro]) {

commenter = [[VVMacroCommenter alloc] initWithIndentString:baseIndent codeString:trimCode];

} else if ([trimCode vv_isStruct]) {

commenter = [[VVStructCommenter alloc] initWithIndentString:baseIndent codeString:trimCode];

} else if ([trimCode vv_isUnion]) {

commenter = [[VVStructCommenter alloc] initWithIndentString:baseIndent codeString:trimCode];

} else if ([trimCode vv_isObjCMethod]) {

commenter = [[VVMethodCommenter alloc] initWithIndentString:baseIndent codeString:trimCode];

} else {

commenter = [[VVVariableCommenter alloc] initWithIndentString:baseIndent codeString:trimCode];

}

return [commenter document];

}