|
1 | 1 | using Our.Umbraco.AuthU.Interfaces; |
2 | 2 | using Our.Umbraco.AuthU.Models; |
3 | | -using NPoco; |
| 3 | +using Umbraco.Core.Scoping; |
4 | 4 |
|
5 | 5 | namespace Our.Umbraco.AuthU.Data |
6 | 6 | { |
7 | 7 | public class UmbracoDbOAuthRefreshTokenStore : IOAuthRefreshTokenStore |
8 | | - { |
9 | | - internal const string CurrentVersion = "1.0.1"; |
10 | | - internal const string SubProductName = "AuthU_UmbracoDbOAuthRefreshTokenStore"; |
| 8 | + { |
| 9 | + internal const string CurrentVersion = "1.0.1"; |
| 10 | + internal const string SubProductName = "AuthU_UmbracoDbOAuthRefreshTokenStore"; |
11 | 11 |
|
12 | | - protected Database Db => new Database("umbracoDbDsn"); |
| 12 | + private readonly IScopeProvider _scopeProvider; |
| 13 | + |
| 14 | + public UmbracoDbOAuthRefreshTokenStore(IScopeProvider scopeProvider) |
| 15 | + { |
| 16 | + _scopeProvider = scopeProvider; |
| 17 | + } |
13 | 18 |
|
14 | 19 | public void AddRefreshToken(OAuthRefreshToken token) |
15 | 20 | { |
16 | | - Db.Execute("DELETE FROM [OAuthRefreshToken] WHERE [Subject] = @0 AND [UserType] = @1 AND [Realm] = @2 AND [ClientId] = @3 AND [DeviceId] = @4", |
| 21 | + using (var scope = _scopeProvider.CreateScope(autoComplete: true)) |
| 22 | + { |
| 23 | + scope.Database.Execute("DELETE FROM [OAuthRefreshToken] WHERE [Subject] = @0 AND [UserType] = @1 AND [Realm] = @2 AND [ClientId] = @3 AND [DeviceId] = @4", |
17 | 24 | token.Subject, |
18 | 25 | token.UserType, |
19 | 26 | token.Realm, |
20 | 27 | token.ClientId, |
21 | | - token.DeviceId); |
| 28 | + token.DeviceId); |
22 | 29 |
|
23 | | - Db.Save(token); |
| 30 | + scope.Database.Save(token); |
| 31 | + } |
24 | 32 | } |
25 | 33 |
|
26 | 34 | public void RemoveRefreshToken(string refreshTokenId) |
27 | 35 | { |
28 | | - Db.Execute("DELETE FROM [OAuthRefreshToken] WHERE [Key] = @0", |
29 | | - refreshTokenId); |
| 36 | + using (var scope = _scopeProvider.CreateScope(autoComplete: true)) |
| 37 | + { |
| 38 | + scope.Database.Execute("DELETE FROM [OAuthRefreshToken] WHERE [Key] = @0", refreshTokenId); |
| 39 | + } |
30 | 40 | } |
31 | 41 |
|
32 | 42 | public OAuthRefreshToken FindRefreshToken(string refreshTokenId) |
33 | 43 | { |
34 | | - return Db.SingleOrDefault<OAuthRefreshToken>("SELECT * FROM [OAuthRefreshToken] WHERE [Key] = @0", |
35 | | - refreshTokenId); |
| 44 | + using (var scope = _scopeProvider.CreateScope(autoComplete: true)) |
| 45 | + { |
| 46 | + return scope.Database.SingleOrDefault<OAuthRefreshToken>("SELECT * FROM [OAuthRefreshToken] WHERE [Key] = @0", refreshTokenId); |
| 47 | + } |
36 | 48 | } |
37 | 49 | } |
38 | 50 | } |
0 commit comments