store quota status in session

[cloudstore.git] / lib / CloudStore / Couchbase.pm

diff --git a/lib/CloudStore/Couchbase.pm b/lib/CloudStore/Couchbase.pm
index 897e9db..cc62539 100644 (file)
--- a/lib/CloudStore/Couchbase.pm
+++ b/lib/CloudStore/Couchbase.pm
@@ -67,7 +67,22 @@ sub user_set {
 
 sub user_get {
        my ($self,$login) = @_;
-       $self->json_get( 'users', $login );
+       $login = $login->{login} if ref $login;
+       my $user = $self->json_get( 'users', $login );
+       $user->{usage} = $self->usage( $login );
+       $user->{status} = $self->status( $login );
+       warn "## user ",dump($user);
+       return $user;
+}
+
+sub status {
+       my ($self,$login,$message) = @_;
+       $login = $login->{login} if ref $login;
+       if ( $message ) {
+               $self->{session}->set( "$login:status" => $message );
+       } else {
+               $self->{session}->get( "$login:status" );
+       }
 }
 
 sub usage_decr {
@@ -81,8 +96,9 @@ sub usage_incr {
 }
 
 sub usage {
-       my ($self,$data) = @_;
-       $self->{session}->get( $data->{login} . ':usage' );
+       my ($self,$login) = @_;
+       $login = $login->{login} if ref $login;
+       $self->{session}->get( $login . ':usage' );
 }
 
 sub usage_init {