Reviewed by Darin
[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 NSString *WebDatabaseDisplayNameKey = @"WebDatabaseDisplayNameKey";
43 NSString *WebDatabaseExpectedSizeKey = @"WebDatabaseExpectedSizeKey";
44 NSString *WebDatabaseUsageKey = @"WebDatabaseUsageKey";
45
46 NSString *WebDatabaseDidModifyOriginNotification = @"WebDatabaseDidModifyOriginNotification";
47 NSString *WebDatabaseDidModifyDatabaseNotification = @"WebDatabaseDidModifyDatabaseNotification";
48 NSString *WebDatabaseIdentifierKey = @"WebDatabaseIdentifierKey";
49
50 @implementation WebDatabaseManager
51
52 + (WebDatabaseManager *) sharedWebDatabaseManager
53 {
54     static WebDatabaseManager *sharedManager = [[WebDatabaseManager alloc] init];
55     return sharedManager;
56 }
57
58 - (NSArray *)origins
59 {
60     Vector<WebCoreSecurityOriginData> coreOrigins;
61     DatabaseTracker::tracker().origins(coreOrigins);
62     NSMutableArray *webOrigins = [[NSMutableArray alloc] initWithCapacity:coreOrigins.size()];
63
64     for (unsigned i = 0; i < coreOrigins.size(); ++i) {
65         WebSecurityOrigin *webOrigin = [[WebSecurityOrigin alloc] _initWithWebCoreSecurityOriginData:&coreOrigins[i]];
66         [webOrigins addObject:webOrigin];
67         [webOrigin release];
68     }
69
70     return [webOrigins autorelease];
71 }
72
73 - (NSArray *)databasesWithOrigin:(WebSecurityOrigin *)origin
74 {
75     Vector<String> nameVector;
76     if (!DatabaseTracker::tracker().databaseNamesForOrigin(*[origin _core], nameVector))
77         return nil;
78     
79     NSMutableArray *names = [[NSMutableArray alloc] initWithCapacity:nameVector.size()];
80
81     for (unsigned i = 0; i < nameVector.size(); ++i)
82         [names addObject:(NSString *)nameVector[i]];
83
84     return [names autorelease];
85 }
86
87 - (NSDictionary *)detailsForDatabase:(NSString *)databaseIdentifier withOrigin:(WebSecurityOrigin *)origin
88 {
89     static id keys[3] = {WebDatabaseDisplayNameKey, WebDatabaseExpectedSizeKey, WebDatabaseUsageKey};
90     
91     DatabaseDetails details = DatabaseTracker::tracker().detailsForNameAndOrigin(databaseIdentifier, *[origin _core]);
92     if (!details.isValid())
93         return nil;
94         
95     id objects[3];
96     objects[0] = details.displayName().isEmpty() ? databaseIdentifier : (NSString *)details.displayName();
97     objects[1] = [NSNumber numberWithUnsignedLongLong:details.expectedUsage()];
98     objects[2] = [NSNumber numberWithUnsignedLongLong:details.currentUsage()];
99     
100     return [[[NSDictionary alloc] initWithObjects:objects forKeys:keys count:3] autorelease];
101 }
102
103 - (void)deleteAllDatabases
104 {
105     DatabaseTracker::tracker().deleteAllDatabases();
106 }
107
108 - (void)deleteOrigin:(WebSecurityOrigin *)origin
109 {
110     // FIXME: this needs to delete the origin as well as the databases therein
111     [self deleteDatabasesWithOrigin:origin];
112 }
113
114 - (void)deleteDatabasesWithOrigin:(WebSecurityOrigin *)origin
115 {
116     DatabaseTracker::tracker().deleteDatabasesWithOrigin(*[origin _core]);
117 }
118
119 - (void)deleteDatabase:(NSString *)databaseIdentifier withOrigin:(WebSecurityOrigin *)origin
120 {
121     DatabaseTracker::tracker().deleteDatabase(*[origin _core], databaseIdentifier);
122 }
123
124 @end
125
126 void WebKitInitializeDatabasesIfNecessary()
127 {
128     static BOOL initialized = NO;
129     if (initialized)
130         return;
131
132     // Set the database root path in WebCore
133     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
134
135     NSString *databasesDirectory = [defaults objectForKey:WebDatabaseDirectoryDefaultsKey];
136     if (!databasesDirectory || ![databasesDirectory isKindOfClass:[NSString class]])
137         databasesDirectory = @"~/Library/WebKit/Databases";
138
139     DatabaseTracker::tracker().setDatabasePath([databasesDirectory stringByStandardizingPath]);
140
141     // Set the DatabaseTrackerClient
142     DatabaseTracker::tracker().setClient(WebDatabaseTrackerClient::sharedWebDatabaseTrackerClient());
143     
144     initialized = YES;
145 }