1/*
2 * Copyright (C) 2007, 2013 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 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#include "config.h"
29#include "ChangeVersionWrapper.h"
30
31#if ENABLE(SQL_DATABASE)
32
33#include "Database.h"
34#include "SQLError.h"
35#include <wtf/PassRefPtr.h>
36#include <wtf/RefPtr.h>
37
38namespace WebCore {
39
40ChangeVersionWrapper::ChangeVersionWrapper(const String& oldVersion, const String& newVersion)
41    : m_oldVersion(oldVersion.isolatedCopy())
42    , m_newVersion(newVersion.isolatedCopy())
43{
44}
45
46bool ChangeVersionWrapper::performPreflight(SQLTransactionBackend* transaction)
47{
48    ASSERT(transaction && transaction->database());
49
50    DatabaseBackend* database = transaction->database();
51
52    String actualVersion;
53    if (!database->getVersionFromDatabase(actualVersion)) {
54        int sqliteError = database->sqliteDatabase().lastError();
55        m_sqlError = SQLError::create(SQLError::UNKNOWN_ERR, "unable to read the current version",
56                                      sqliteError, database->sqliteDatabase().lastErrorMsg());
57        return false;
58    }
59
60    if (actualVersion != m_oldVersion) {
61        m_sqlError = SQLError::create(SQLError::VERSION_ERR, "current version of the database and `oldVersion` argument do not match");
62        return false;
63    }
64
65    return true;
66}
67
68bool ChangeVersionWrapper::performPostflight(SQLTransactionBackend* transaction)
69{
70    ASSERT(transaction && transaction->database());
71
72    DatabaseBackend* database = transaction->database();
73
74    if (!database->setVersionInDatabase(m_newVersion)) {
75        int sqliteError = database->sqliteDatabase().lastError();
76        m_sqlError = SQLError::create(SQLError::UNKNOWN_ERR, "unable to set new version in database",
77                                      sqliteError, database->sqliteDatabase().lastErrorMsg());
78        return false;
79    }
80
81    database->setExpectedVersion(m_newVersion);
82    return true;
83}
84
85void ChangeVersionWrapper::handleCommitFailedAfterPostflight(SQLTransactionBackend* transaction)
86{
87    transaction->database()->setCachedVersion(m_oldVersion);
88}
89
90} // namespace WebCore
91
92#endif // ENABLE(SQL_DATABASE)
93