formatType.m

// -*- mode: ObjC -*-

//  This file is part of class-dump, a utility for examining the Objective-C segment of Mach-O files.
//  Copyright (C) 1997-1998, 2000-2001, 2004-2011 Steve Nygard.

#include <stdio.h>
#include <libc.h>
#include <unistd.h>
#include <getopt.h>
#include <stdlib.h>

#import <Foundation/Foundation.h>
#import "NSString-Extensions.h"

#import "CDClassDump.h"
#import "CDTypeFormatter.h"
#import "CDSymbolReferences.h"
#import "CDBalanceFormatter.h"

void print_usage(void)
{
    fprintf(stderr,
            "formatType %s\n"
            "Usage: formatType [options] <input file>\n"
            "\n"
            "  where options are:\n"
            "        -m        format method (default is to format ivars)\n"
            ,
            CLASS_DUMP_VERSION
       );
}

enum {
    CDFormatIvar = 0,
    CDFormatMethod = 1,
    CDFormatBalance = 2,
};

int main(int argc, char *argv[])
{
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];

    CDTypeFormatter *ivarTypeFormatter = [[CDTypeFormatter alloc] init];
    [ivarTypeFormatter setShouldExpand:YES];
    [ivarTypeFormatter setShouldAutoExpand:YES];
    [ivarTypeFormatter setBaseLevel:0];
    //[ivarTypeFormatter setDelegate:self];

    CDTypeFormatter *methodTypeFormatter = [[CDTypeFormatter alloc] init];
    [methodTypeFormatter setShouldExpand:NO];
    [methodTypeFormatter setShouldAutoExpand:NO];
    [methodTypeFormatter setBaseLevel:0];
    //[methodTypeFormatter setDelegate:self];

    struct option longopts[] = {
        { "balance", no_argument, NULL, 'b' },
        { "method", no_argument, NULL, 'm' },
        { NULL, 0, NULL, 0 },
    };

    if (argc == 1) {
        print_usage();
        exit(0);
    }

    NSUInteger formatType = CDFormatIvar;
    
    int ch;
    BOOL errorFlag = NO;

    while ( (ch = getopt_long(argc, argv, "bm", longopts, NULL)) != -1) {
        switch (ch) {
            case 'b':
                formatType = CDFormatBalance;
                break;
                
            case 'm':
                formatType = CDFormatMethod;
                break;
                
            case '?':
            default:
                errorFlag = YES;
                break;
        }
    }

    if (errorFlag) {
        print_usage();
        exit(2);
    }

    switch (formatType) {
        case CDFormatIvar:    printf("Format as ivars\n"); break;
        case CDFormatMethod:  printf("Format as methods\n"); break;
        case CDFormatBalance: printf("Format as balance\n"); break;
    }

    for (NSUInteger index = optind; index < argc; index++) {

        NSString *arg = [NSString stringWithFileSystemRepresentation:argv[index]];
        printf("======================================================================\n");
        printf("File: %s\n", argv[index]);

        NSError *error = nil;
        NSString *input = [[NSString alloc] initWithContentsOfFile:arg encoding:NSUTF8StringEncoding error:&error];
        if (error != nil) {
            NSLog(@"input error: %@", error);
            NSLog(@"localizedFailureReason: %@", [error localizedFailureReason]);
        }

        NSArray *lines = [input componentsSeparatedByString:@"\n"];

        NSString *name = nil;
        NSString *type = nil;
        for (NSString *line in lines) {
            if ([line hasPrefix:@"//"] || [line length] == 0) {
                printf("%s\n", [line UTF8String]);
                continue;
            }

            if (name == nil) {
                name = line;
            } else if (type == nil) {
                NSString *str;

                type = line;

                switch (formatType) {
                    case CDFormatIvar:
                        str = [ivarTypeFormatter formatVariable:name type:type symbolReferences:nil];
                        break;
                        
                    case CDFormatMethod:
                        str = [methodTypeFormatter formatMethodName:name type:type symbolReferences:nil];
                        break;
                        
                    case CDFormatBalance: {
                        CDBalanceFormatter *balance = [[CDBalanceFormatter alloc] initWithString:type];
                        str = [balance format];
                        [balance release];
                    }
                }
                if (str == nil)
                    printf("Error formatting type.\n");
                else
                    printf("%s\n", [str UTF8String]);
                printf("----------------------------------------------------------------------\n");

                name = type = nil;
            }

        }

        [input release];
    }

    [ivarTypeFormatter release];
    [methodTypeFormatter release];

    [pool release];

    return 0;
}