add new exit behavior

docopt · Aug 23, 2012 · 379976d · 379976d
1 parent 9cdde2d
commit 379976d
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/docopt.rb b/lib/docopt.rb
@@ -6,9 +6,6 @@ class DocoptLanguageError < SyntaxError
   end
 
   class Exit < RuntimeError
-    # TODO: try and derive this from SystemExit
-    @@usage = ''
-
     def self.usage
       @@usage
     end
@@ -17,8 +14,12 @@ def self.set_usage(usage)
       @@usage = usage ? usage : ''
     end
 
+    def message
+      @@message
+    end
+
     def initialize(message='')
-      super((message && message != '' ? (message + "\n") : '') + @@usage)
+      @@message = ((message && message != '' ? (message + "\n") : '') + @@usage)
     end
   end