869d2c930f2fcf05db33acb82c06ee7ffc6f3923
[WebKit-https.git] / WebKit / mac / Storage / WebDatabaseManager.mm
1 /*
2  * Copyright (C) 2007 Apple Inc.  All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  *
8  * 1.  Redistributions of source code must retain the above copyright
9  *     notice, this list of conditions and the following disclaimer.
10  * 2.  Redistributions in binary form must reproduce the above copyright
11  *     notice, this list of conditions and the following disclaimer in the
12  *     documentation and/or other materials provided with the distribution.
13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
14  *     its contributors may be used to endorse or promote products derived
15  *     from this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #import "WebDatabaseManagerPrivate.h"
30 #import "WebDatabaseManagerInternal.h"
31
32 #import "WebDatabaseTrackerClient.h"
33 #import "WebSecurityOriginPrivate.h"
34 #import "WebSecurityOriginInternal.h"
35
36 #import <WebCore/DatabaseTracker.h>
37
38 using namespace WebCore;
39
40 const NSString *WebDatabaseDirectoryDefaultsKey = @"WebDatabaseDirectory";
41
42 const NSString *WebDatabaseDisplayNameKey = @"WebDatabaseDisplayNameKey";
43 const NSString *WebDatabaseExpectedSizeKey = @"WebDatabaseExpectedSizeKey";
44 const NSString *WebDatabaseUsageKey = @"WebDatabaseUsageKey";
45
46 const NSString *WebDatabaseDidModifyOriginNotification = @"WebDatabaseDidModifyOriginNotification";
47 const NSString *WebDatabaseDidModifyDatabaseNotification = @"WebDatabaseDidModifyDatabaseNotification";
48
49 @implementation WebDatabaseManager
50
51 + (WebDatabaseManager *) sharedWebDatabaseManager
52 {
53     static WebDatabaseManager *sharedManager = [[WebDatabaseManager alloc] init];
54     return sharedManager;
55 }
56
57 - (NSArray *)origins
58 {
59     Vector<WebCoreSecurityOriginData> coreOrigins;
60     DatabaseTracker::tracker().origins(coreOrigins);
61     NSMutableArray *webOrigins = [[NSMutableArray alloc] initWithCapacity:coreOrigins.size()];
62
63     for (unsigned i = 0; i < coreOrigins.size(); ++i) {
64         WebSecurityOrigin *webOrigin = [[WebSecurityOrigin alloc] _initWithWebCoreSecurityOriginData:&coreOrigins[i]];
65         [webOrigins addObject:webOrigin];
66         [webOrigin release];
67     }
68
69     return [webOrigins autorelease];
70 }
71
72 - (NSArray *)databasesWithOrigin:(WebSecurityOrigin *)origin
73 {
74     Vector<String> nameVector;
75     if (!DatabaseTracker::tracker().databaseNamesForOrigin(*[origin _core], nameVector))
76         return nil;
77     
78     NSMutableArray *names = [[NSMutableArray alloc] initWithCapacity:nameVector.size()];
79
80     for (unsigned i = 0; i < nameVector.size(); ++i)
81         [names addObject:(NSString *)nameVector[i]];
82
83     return [names autorelease];
84 }
85
86 - (NSDictionary *)detailsForDatabase:(NSString *)databaseName withOrigin:(WebSecurityOrigin *)origin
87 {
88     static id keys[3] = {WebDatabaseDisplayNameKey, WebDatabaseExpectedSizeKey, WebDatabaseUsageKey};
89     
90     DatabaseDetails details = DatabaseTracker::tracker().detailsForNameAndOrigin(databaseName, *[origin _core]);
91     if (!details.isValid())
92         return nil;
93         
94     id objects[3];
95     objects[0] = details.displayName().isEmpty() ? databaseName : (NSString *)details.displayName();
96     objects[1] = [NSNumber numberWithUnsignedLongLong:details.expectedUsage()];
97     objects[2] = [NSNumber numberWithUnsignedLongLong:details.currentUsage()];
98     
99     return [[[NSDictionary alloc] initWithObjects:objects forKeys:keys count:3] autorelease];
100 }
101
102 - (void)deleteAllDatabases
103 {
104     DatabaseTracker::tracker().deleteAllDatabases();
105 }
106
107 - (void)deleteDatabasesWithOrigin:(WebSecurityOrigin *)origin
108 {
109     DatabaseTracker::tracker().deleteDatabasesWithOrigin(*[origin _core]);
110 }
111
112 - (void)deleteDatabase:(NSString *)databaseName withOrigin:(WebSecurityOrigin *)origin
113 {
114     DatabaseTracker::tracker().deleteDatabase(*[origin _core], databaseName);
115 }
116
117 @end
118
119 void WebKitInitializeDatabasesIfNecessary()
120 {
121     static BOOL initialized = NO;
122     if (initialized)
123         return;
124
125     // Set the database root path in WebCore
126     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
127
128     NSString *databasesDirectory = [defaults objectForKey:WebDatabaseDirectoryDefaultsKey];
129     if (!databasesDirectory || ![databasesDirectory isKindOfClass:[NSString class]])
130         databasesDirectory = @"~/Library/WebKit/Databases";
131
132     DatabaseTracker::tracker().setDatabasePath([databasesDirectory stringByStandardizingPath]);
133
134     // Set the DatabaseTrackerClient
135     DatabaseTracker::tracker().setClient(WebDatabaseTrackerClient::sharedWebDatabaseTrackerClient());
136     
137     initialized = YES;
138 }