1 files changed, 16 insertions, 18 deletions
diff --git a/lib/Net/HTTP/Console/Dispatcher/Set.pm b/lib/Net/HTTP/Console/Dispatcher/Set.pm
index 9f23447..f97474b 100644
--- a/lib/Net/HTTP/Console/Dispatcher/Set.pm
+++ b/lib/Net/HTTP/Console/Dispatcher/Set.pm
@@ -5,24 +5,22 @@ use MooseX::Declare;
 class Net::HTTP::Console::Dispatcher::Set with Net::HTTP::Console::Dispatcher {
 
     method dispatch($input) {
-        (my $command, my $header, my $value) =
-          $input =~ /^([\w_]+)(?:\s([\w-]+))?(?:\s(.*))?$/;
+        (my $command, my $type, my $key, my $value) =
+          $input =~ /^([\w_]+)(?:\s([\w_]+))(?:\s([\w_]+))(?:\s(.*))?$/;
 
-        if ($command eq 'unset_header') {
-            $self->_unset_header($header);
-        }
-        elsif ($command eq 'set_header') {
-            $self->_set_header($header, $value);
-        }
-        elsif ($command eq 'show_defined_headers') {
-            $self->_show_defined_headers();
+        if ($command eq 'set') {
+            $self->_set_header($key, $value) if $type eq 'header';
+        }elsif($command eq 'unset') {
+            $self->_unset_header($key) if $type eq 'header';
         }
+
+        # elsif ($command eq 'show_defined_headers') {
+        #     $self->_show_defined_headers();
+        # }
     }
 
     method pattern($input) {
-        $input =~ /(un)?set_header|show_defined_headers/
-          ? return $input
-          : return 0;
+        $input =~ /^(un)?set/ ? return $input : return 0;
     }
 
     method _set_header($header, $value) {
@@ -35,11 +33,11 @@ class Net::HTTP::Console::Dispatcher::Set with Net::HTTP::Console::Dispatcher {
         print "header $header unset\n";
     }
 
-    method _show_defined_headers {
-        foreach ($self->application->all_headers) {
-            print $_->[0].": ".$_->[1]."\n";
-        }
-    }
+    # method _show_defined_headers {
+    #     foreach ($self->application->all_headers) {
+    #         print $_->[0].": ".$_->[1]."\n";
+    #     }
+    # }
 }
 
 1;