fix some bugs in code

Lib/OAuth2StorageCake.php

@@ -177,12 +177,15 @@ public function checkRestrictedGrantType($client_id, $grant_type) {
  */
 	protected function setToken($token, $client_id, $user_id, $expires, $scope, $isRefresh = TRUE) {
 		$model = 'Token';
-		if ($isRefresh == true) {
+		if ($isRefresh) {
 			$model = 'RefreshToken';
+			$refresh_token = $token;
+		} else {
+			$oauth_token = $token;
 		}
 
 		$this->{$model}->save(array(
-			$this->{$model}->alias => compact('token', 'client_id', 'user_id', 'expires',
+			$this->{$model}->alias => compact('refresh_token', 'oauth_token', 'client_id', 'user_id', 'expires',
 				'scope')));
 	}
 
@@ -204,7 +207,7 @@ protected function getToken($token, $isRefresh = true) {
 		if (empty($result)) {
 			return null;
 		}
-		return $result[$this->AuthCode->alias];
+		return $result[$this->{$model}->alias];
 	}
 
 	/**

Model/Oauth2AccessToken.php

@@ -9,4 +9,11 @@ class Oauth2AccessToken extends Oauth2AppModel {
  */
 	public $name = 'Oauth2AccessToken';
 
+/**
+ * Primary key
+ *
+ * @var string
+ */
+	public $primaryKey = 'oauth_token';
+
 }

Model/Oauth2AuthCode.php

@@ -9,4 +9,12 @@ class Oauth2AuthCode extends Oauth2AppModel {
  */
 	public $name = 'Oauth2AuthCode';
 
+/**
+ * Primary key
+ *
+ * @var string
+ */
+	public $primaryKey = 'code';
+
+
 }

Model/Oauth2RefreshToken.php

@@ -9,4 +9,10 @@ class Oauth2RefreshToken extends Oauth2AppModel {
  */
 	public $name = 'Oauth2RefreshToken';
 
+/**
+ * Primary key
+ *
+ * @var string
+ */
+	public $primaryKey = 'refresh_token'; 	
 }