1/*
2 * Copyright (C) 2017 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 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23 * THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#pragma once
27
28#if ENABLE(SERVICE_WORKER)
29
30#include "SecurityOrigin.h"
31#include <pal/SessionID.h>
32#include <wtf/ThreadSafeRefCounted.h>
33#include <wtf/WeakPtr.h>
34#include <wtf/WorkQueue.h>
35#include <wtf/text/WTFString.h>
36
37namespace WebCore {
38
39class RegistrationStore;
40class SQLiteDatabase;
41struct ServiceWorkerContextData;
42
43WEBCORE_EXPORT String serviceWorkerRegistrationDatabaseFilename(const String& databaseDirectory);
44
45class RegistrationDatabase : public ThreadSafeRefCounted<RegistrationDatabase, WTF::DestructionThread::Main> {
46WTF_MAKE_FAST_ALLOCATED;
47public:
48 static Ref<RegistrationDatabase> create(RegistrationStore& store, String&& databaseDirectory)
49 {
50 return adoptRef(*new RegistrationDatabase(store, WTFMove(databaseDirectory)));
51 }
52
53 ~RegistrationDatabase();
54
55
56 void pushChanges(Vector<ServiceWorkerContextData>&&, CompletionHandler<void()>&&);
57 void clearAll(CompletionHandler<void()>&&);
58 void close(CompletionHandler<void()>&&);
59
60private:
61 RegistrationDatabase(RegistrationStore&, String&& databaseDirectory);
62
63 void postTaskToWorkQueue(Function<void()>&&);
64
65 // Methods to be run on the work queue.
66 void openSQLiteDatabase(const String& fullFilename);
67 String ensureValidRecordsTable();
68 String importRecords();
69 void importRecordsIfNecessary();
70 void doPushChanges(Vector<ServiceWorkerContextData>&&);
71 void doClearOrigin(const SecurityOrigin&);
72
73 // Replies to the main thread.
74 void addRegistrationToStore(ServiceWorkerContextData&&);
75 void databaseFailedToOpen();
76 void databaseOpenedAndRecordsImported();
77
78 Ref<WorkQueue> m_workQueue;
79 WeakPtr<RegistrationStore> m_store;
80 PAL::SessionID m_sessionID;
81 String m_databaseDirectory;
82 String m_databaseFilePath;
83 std::unique_ptr<SQLiteDatabase> m_database;
84};
85
86} // namespace WebCore
87
88#endif // ENABLE(SERVICE_WORKER)
89