DefaultURLHandler: Add `saveDefaultURLHandler`

This method is supposed to set CopyMailto as the default application for
"mailto" URLs. But it keeps giving me a `-54` error, or `202` status
code on the shell. WTF is going on?

1 files changed, 4 insertions, 1 deletions

diff --git a/CopyMailto/main.m b/CopyMailto/main.m
index 8dfe2c1..e61f04e 100644
--- a/CopyMailto/main.m
+++ b/CopyMailto/main.m
@@ -11,6 +11,9 @@
 #import "DefaultURLHandler.h"
 
 int main(int argc, const char * argv[]) {
-    [DefaultURLHandler saveDefaultURLHandler];
+    if (argc == 2 && strcmp(argv[1], "--set-url-handler") == 0) {
+        return [DefaultURLHandler setDefaultURLHandler];
+    }
+
     return NSApplicationMain(argc, argv);
 }