4b5f42094978a139c156a41aff6b8f3e1c47d5ac
[grauphel.git] / lib / tokenstorage.php
1 <?php
2 /**
3  * Part of grauphel
4  *
5  * PHP version 5
6  *
7  * @category  Tools
8  * @package   Grauphel
9  * @author    Christian Weiske <cweiske@cweiske.de>
10  * @copyright 2014 Christian Weiske
11  * @license   http://www.gnu.org/licenses/agpl.html GNU AGPL v3
12  * @link      http://cweiske.de/grauphel.htm
13  */
14 namespace OCA\Grauphel\Lib;
15
16 /**
17  * Token store
18  *
19  * @category  Tools
20  * @package   Grauphel
21  * @author    Christian Weiske <cweiske@cweiske.de>
22  * @copyright 2014 Christian Weiske
23  * @license   http://www.gnu.org/licenses/agpl.html GNU AGPL v3
24  * @version   Release: @package_version@
25  * @link      http://cweiske.de/grauphel.htm
26  */
27 class TokenStorage
28 {
29     /**
30      * Store the given token
31      *
32      * @param OAuth_Token $token Token object to store
33      *
34      * @return void
35      */
36     public function store(OAuth_Token $token)
37     {
38         \OC_DB::executeAudited(
39             'INSERT INTO `*PREFIX*grauphel_oauth_tokens`'
40             . '(`token_user`, `token_type`, `token_key`, `token_secret`, `token_verifier`, `token_callback`)'
41             . ' VALUES(?, ?, ?, ?, ?, ?)',
42             array(
43                 $token->user,
44                 $token->type,
45                 $token->tokenKey,
46                 (string) $token->secret,
47                 (string) $token->verifier,
48                 (string) $token->callback
49             )
50         );
51     }
52
53     /**
54      * Load the token and destroy it.
55      *
56      * @param string $type     Token type: temp, access, verify
57      * @param string $tokenKey Random token string to load
58      *
59      * @return OAuth_Token Stored token
60      *
61      * @throws OAuthException When token does not exist
62      */
63     public function loadAndDelete($type, $tokenKey)
64     {
65         try {
66             $token = $this->load($type, $tokenKey);
67             \OC_DB::executeAudited(
68                 'DELETE FROM `*PREFIX*grauphel_oauth_tokens`'
69                 . ' WHERE `token_key` = ? AND `token_type` = ?',
70                 array($tokenKey, $type)
71             );
72             return $token;
73         } catch (OAuthException $e) {
74             throw $e;
75         }
76     }
77
78
79     /**
80      * Load the token.
81      *
82      * @param string $type     Token type: temp, access, verify
83      * @param string $tokenKey Random token string to load
84      *
85      * @return OAuth_Token Stored token
86      *
87      * @throws OAuthException When token does not exist or it is invalid
88      */
89     public function load($type, $tokenKey)
90     {
91         $tokenRow = \OC_DB::executeAudited(
92             'SELECT * FROM `*PREFIX*grauphel_oauth_tokens`'
93             . ' WHERE `token_key` = ? AND `token_type` = ?',
94             array($tokenKey, $type)
95         )->fetchRow();
96
97         if ($tokenRow === false) {
98             throw new OAuthException(
99                 'Unknown token: ' . $type . ' / ' . $tokenKey,
100                 OAUTH_TOKEN_REJECTED
101             );
102         }
103
104         $token = $this->fromDb($tokenRow);
105         if ($token->tokenKey != $tokenKey) {
106             throw new OAuthException('Invalid token', OAUTH_TOKEN_REJECTED);
107         }
108
109         return $token;
110     }
111
112     /**
113      * Load multiple tokens
114      *
115      * @param string $username User name
116      * @param string $type     Token type: temp, access, verify
117      *
118      * @return array Array of Token objects
119      */
120     public function loadForUser($username, $type)
121     {
122         $result = \OC_DB::executeAudited(
123             'SELECT * FROM `*PREFIX*grauphel_oauth_tokens`'
124             . ' WHERE `token_user` = ? AND `token_type` = ?',
125             array($username, $type)
126         );
127
128         $tokens = array();
129         while ($tokenRow = $result->fetchRow()) {
130             $tokens[] = $this->fromDb($tokenRow);
131         }
132
133         return $tokens;
134     }
135
136     protected function fromDb($tokenRow)
137     {
138         $token = new Token();
139         $token->type     = $tokenRow['token_user'];
140         $token->tokenKey = $tokenRow['token_key'];
141         $token->secret   = $tokenRow['token_secret'];
142         $token->user     = $tokenRow['token_user'];
143         $token->verifier = $tokenRow['token_verifier'];
144         $token->callback = $tokenRow['token_callback'];
145         return $token;
146     }
147 }
148 ?>