diff --git a/src/main.m b/src/main.m index 020bddb..d6e015b 100644 --- a/src/main.m +++ b/src/main.m @@ -43,7 +43,7 @@ void set_default_handler(NSString *url_scheme, NSString *handler) { } int main(int argc, const char *argv[]) { - const char *target = (argc == 1) ? '\0' : argv[1]; + NSString *target = (argc > 1) ? [NSString stringWithUTF8String:argv[1]] : nil; @autoreleasepool { // Get all HTTP handlers @@ -52,17 +52,17 @@ int main(int argc, const char *argv[]) { // Get current HTTP handler NSString *current_handler_name = get_current_http_handler(); - if (target == '\0') { + if (target == nil) { // List all HTTP handlers, marking the current one with a star for (NSString *key in handlers) { char *mark = [key caseInsensitiveCompare:current_handler_name] == NSOrderedSame ? "* " : " "; printf("%s%s\n", mark, [key UTF8String]); } } else { - NSString *target_handler_name = [NSString stringWithUTF8String:target]; + NSString *target_handler_name = target; if ([target_handler_name caseInsensitiveCompare:current_handler_name] == NSOrderedSame) { - printf("%s is already set as the default HTTP handler\n", target); + printf("%s is already set as the default HTTP handler\n", [target UTF8String]); } else { NSString *target_handler = handlers[target_handler_name]; @@ -71,7 +71,7 @@ int main(int argc, const char *argv[]) { set_default_handler(@"http", target_handler); set_default_handler(@"https", target_handler); } else { - printf("%s is not available as an HTTP handler\n", target); + printf("%s is not available as an HTTP handler\n", [target UTF8String]); return 1; }