Name abbreviation

NSString+Abbreviation

#import "NSString+Abbreviation.h"

@implementation NSString (Abbreviation)

- (NSString *)abbreviatedWithMiddles:(BOOL)middles
{
    NSArray *nameParts = [self componentsSeparatedByString:@" "];
    
    if (nameParts && nameParts.count > 0)
    {
        if (nameParts.count == 1)
        {
            return nameParts.firstObject;
        }
        else
        {
            if (middles)
            {
                NSMutableString *abbreviatedName = NSMutableString.new;
                
                for (int i = 0; i < nameParts.count; i++)
                {
                    if (i == 0)
                    {
                        [abbreviatedName appendFormat:@"%@ ", [nameParts objectAtIndex:i]];
                    }
                    else
                    {
                        [abbreviatedName appendFormat:@"%@.", [[nameParts objectAtIndex:i] substringToIndex:1]];
                        
                        if (i < (nameParts.count - 1))
                        {
                            [abbreviatedName appendString:@" "];
                        }
                    }
                }
                
                return abbreviatedName;
            }
            else
            {
                return [NSString stringWithFormat:@"%@ %@.", nameParts.firstObject, [nameParts.lastObject substringToIndex:1]];
            }
        }
    }
    else
    {
        return @"";
    }
}

@end