Synnefo » Pithos iOS Client

+ (unsigned long long)sizeOfDirectory:(NSString *)directoryPath {

    NSFileManager *fileManager = [NSFileManager defaultManager];

    NSDirectoryEnumerator *directoryEnumerator = [fileManager enumeratorAtPath:directoryPath];

    unsigned long long directorySize = 0;

    NSString *file;

    while (file = [directoryEnumerator nextObject]) {

        NSDictionary *fileAttributes = [fileManager attributesOfItemAtPath:[NSString stringWithFormat:@"%@/%@",directoryPath,file] error:nil];

        directorySize += [fileAttributes fileSize];

    }

    return directorySize;

}

+ (NSString *)humanReadableSize:(unsigned long long)bytes {

    if (bytes <= 999) {

        return [NSString stringWithFormat:@"%d B", (int)bytes];

    } else if (bytes <= 999000) {

        return [NSString stringWithFormat:@"%d KB", (int)ceil(bytes / 1000)];

    } else if (bytes <= 999900000) {

        double megabytes = floor(bytes / 1000000);

        double hundredkilobytes = ceil((bytes - megabytes * 1000000) / 100000);

        if (hundredkilobytes == 10) {

            megabytes++;

            hundredkilobytes = 0;

        }

        if (hundredkilobytes == 0) {

            return [NSString stringWithFormat:@"%d MB", (int)megabytes];

        } else {

            return [NSString stringWithFormat:@"%d.%d MB", (int)megabytes, (int)hundredkilobytes];

        }

    } else {

        double gigabytes = floor(bytes / 1000000000);

        double hundredmegabytes = ceil((bytes - gigabytes * 1000000000) / 100000000);

        if (hundredmegabytes == 10) {

            gigabytes++;

            hundredmegabytes = 0;

        }

        if (hundredmegabytes == 0) {

            return [NSString stringWithFormat:@"%d GB", (int)gigabytes];

        } else {

            return [NSString stringWithFormat:@"%d.%d GB", (int)gigabytes, (int)hundredmegabytes];

        }

    }

}