fix tests for clear_flash (mocking was bad)
[catagits/Catalyst-Plugin-Session.git] / lib / Catalyst / Plugin / Session.pm
index ca267c9..d8ce67e 100644 (file)
@@ -12,7 +12,7 @@ use Digest              ();
 use overload            ();
 use Object::Signature   ();
 
-our $VERSION = "0.05";
+our $VERSION = "0.13";
 
 my @session_data_accessors; # used in delete_session
 BEGIN {
@@ -22,6 +22,7 @@ BEGIN {
           _sessionid
           _session
           _session_expires
+          _extended_session_expires
           _session_data_sig
           _flash
           _flash_keep_keys
@@ -90,22 +91,39 @@ sub prepare_action {
 sub finalize {
     my $c = shift;
 
-    $c->_save_session_expires;
+    $c->finalize_session;
+    
+    $c->NEXT::finalize(@_);
+}
+
+sub finalize_session {
+    my $c = shift;
+
+    $c->NEXT::finalize_session;
+
+    $c->_save_session_id;
     $c->_save_session;
     $c->_save_flash;
+    $c->_save_session_expires;
 
-    $c->NEXT::finalize(@_);
+    $c->_clear_session_instance_data;
+}
+
+sub _save_session_id {
+    my $c = shift;
+
+    # we already called set when allocating
+    # no need to tell the state plugins anything new
 }
 
 sub _save_session_expires {
     my $c = shift;
 
-    if ( defined(my $expires = $c->_session_expires) ) {
+    if ( defined($c->_session_expires) ) {
+        my $expires = $c->session_expires; # force extension
+
         my $sid = $c->sessionid;
         $c->store_session_data( "expires:$sid" => $expires );
-
-        $c->_session_expires(undef);
-        $c->_tried_loading_session_expires(undef);
     }
 }
 
@@ -122,9 +140,6 @@ sub _save_session {
             my $sid = $c->sessionid;
             $c->store_session_data( "session:$sid" => $session_data );
         }
-
-        $c->_session(undef);
-        $c->_tried_loading_session_data(undef);
     }
 }
 
@@ -149,22 +164,20 @@ sub _save_flash {
         else {
             $c->delete_session_data("flash:$sid");
         }
-        
-        $c->_flash(undef);
-        $c->_tried_loading_flash_data(undef);
     }
 }
 
 sub _load_session_expires {
     my $c = shift;
-    return if $c->_tried_loading_session_expires;
+    return $c->_session_expires if $c->_tried_loading_session_expires;
     $c->_tried_loading_session_expires(1);
 
     if ( my $sid = $c->sessionid ) {
         my $expires = $c->get_session_data("expires:$sid") || 0;
 
         if ( $expires >= time() ) {
-            return $c->extend_session_expires( $expires );
+            $c->_session_expires( $expires );
+            return $expires;
         } else {
             $c->delete_session( "session expired" );
             return 0;
@@ -176,11 +189,11 @@ sub _load_session_expires {
 
 sub _load_session {
     my $c = shift;
-    return if $c->_tried_loading_session_data;
+    return $c->_session if $c->_tried_loading_session_data;
     $c->_tried_loading_session_data(1);
 
     if ( my $sid = $c->sessionid ) {
-        if ( $c->session_expires ) {    # > 0
+        if ( $c->_load_session_expires ) {    # > 0
 
             my $session_data = $c->get_session_data("session:$sid") || return;
             $c->_session($session_data);
@@ -211,7 +224,7 @@ sub _load_session {
 
 sub _load_flash {
     my $c = shift;
-    return if $c->_tried_loading_flash_data;
+    return $c->_flash if $c->_tried_loading_flash_data;
     $c->_tried_loading_flash_data(1);
 
     if ( my $sid = $c->sessionid ) {
@@ -239,18 +252,26 @@ sub _expire_session_keys {
     }
 }
 
+sub _clear_session_instance_data {
+    my $c = shift;
+    $c->$_(undef) for @session_data_accessors;
+    $c->NEXT::_clear_session_instance_data; # allow other plugins to hook in on this
+}
+
 sub delete_session {
     my ( $c, $msg ) = @_;
 
-    # delete the session data
-    my $sid = $c->sessionid || return;
-    $c->delete_session_data("${_}:${sid}") for qw/session expires flash/;
+    $c->log->debug("Deleting session" . ( defined($msg) ? "($msg)" : '(no reason given)') ) if $c->debug;
 
-    $c->delete_session_id;
+    # delete the session data
+    if ( my $sid = $c->sessionid ) {
+        $c->delete_session_data("${_}:${sid}") for qw/session expires flash/;
+        $c->delete_session_id($sid);
+    }
 
     # reset the values in the context object
     # see the BEGIN block
-    $c->$_(undef) for @session_data_accessors;
+    $c->_clear_session_instance_data;
 
     $c->_session_delete_reason($msg);
 }
@@ -266,11 +287,10 @@ sub session_delete_reason {
 sub session_expires {
     my $c = shift;
 
-    if ( defined( my $expires = $c->_session_expires ) ) {
+    if ( defined( my $expires = $c->_extended_session_expires ) ) {
         return $expires;
     } elsif ( defined( $expires = $c->_load_session_expires ) ) {
-        $c->_session_expires($expires);
-        return $expires;
+        return $c->extend_session_expires( $expires );
     } else {
         return 0;
     }
@@ -278,7 +298,7 @@ sub session_expires {
 
 sub extend_session_expires {
     my ( $c, $expires ) = @_;
-    $c->_session_expires( my $updated = $c->calculate_extended_session_expires( $expires ) );
+    $c->_extended_session_expires( my $updated = $c->calculate_extended_session_expires( $expires ) );
     $c->extend_session_id( $c->sessionid, $updated );
     return $updated;
 }
@@ -295,7 +315,10 @@ sub calculate_extended_session_expires {
 
 sub reset_session_expires {
     my ( $c, $sid ) = @_;
-    $c->_session_expires( my $exp = $c->calculate_initial_session_expires );
+    
+    my $exp = $c->calculate_initial_session_expires;
+    $c->_session_expires( $exp );
+    $c->_extended_session_expires( $exp );
     $exp;
 }
 
@@ -312,7 +335,8 @@ sub _load_sessionid {
 
     if ( defined( my $sid = $c->get_session_id ) ) {
         if ( $c->validate_session_id($sid) ) {
-            $c->_sessionid( $sid );
+            # temporarily set the inner key, so that validation will work
+            $c->_sessionid($sid);
             return $sid;
         } else {
             my $err = "Tried to set invalid session ID '$sid'";
@@ -327,9 +351,8 @@ sub _load_sessionid {
 sub session_is_valid {
     my $c = shift;
 
-    $c->_load_session; # check expiry and also __address, etc
-
-    if ( $c->_session ) {
+    # force a check for expiry, but also __address, etc
+    if ( $c->_load_session ) {
         return 1;
     } else {
         return;
@@ -365,6 +388,15 @@ sub flash {
     }
 }
 
+sub clear_flash {
+    my $c = shift;
+    
+    #$c->delete_session_data("flash:" . $c->sessionid); # should this be in here? or delayed till finalization?
+    $c->_flash_key_hashes({});
+    $c->_flash_keep_keys({});
+    $c->_flash({});
+}
+
 sub session_expire_key {
     my ( $c, %keys ) = @_;
 
@@ -616,6 +648,10 @@ of every request.
         }
     }
 
+=item clear_flash
+
+Zap all the keys in the flash regardless of their current state.
+
 =item keep_flash @keys
 
 If you wawnt to keep a flash key for the next request too, even if it hasn't
@@ -772,6 +808,12 @@ Or even more directly, replace C<generate_session_id>:
 Also have a look at L<Crypt::Random> and the various openssl bindings - these
 modules provide APIs for cryptographically secure random data.
 
+=item finalize_session
+
+Clean up the session during C<finalize>.
+
+This clears the various accessors after saving to the store.
+
 =item dump_these
 
 See L<Catalyst/dump_these> - ammends the session data structure to the list of