lib/filegen/runner.rb in filegen-0.2.2 vs lib/filegen/runner.rb in filegen-0.2.3

@@ -33,9 +33,12 @@
 
         exitstatus = 0
       rescue RuntimeError => e
         Filegen::Ui.error e.message
         exitstatus = 1
+      rescue Interrupt => e
+        Filegen::Ui.warning 'You told me to stop command execution.'
+        exitstatus = 2
       end
 
       @kernel.exit(exitstatus)
     end
   end