migrate.h

Go to the documentation of this file.

/**********************************************************************
 * LeechCraft - modular cross-platform feature rich internet client.
 * Copyright (C) 2006-2014  Georg Rudoy
 *
 * Distributed under the Boost Software License, Version 1.0.
 * (See accompanying file LICENSE or copy at https://www.boost.org/LICENSE_1_0.txt)
 **********************************************************************/
 
#pragma once
 
#include "util/db/oral/oral.h"
 
namespace LC
{
namespace Util
{
namespace oral
{
    namespace detail
    {
        bool MatchesSchema (const QString& baseName, const QString& schema, QSqlDatabase& db)
        {
            auto result = Util::RunTextQuery (db,
                    QString { "SELECT sql FROM sqlite_master WHERE type = 'table' AND name = '%1'" }
                            .arg (baseName));
            if (!result.next ())
                return true;
 
            const auto& existingDDL = result.value (0).toString ();
 
            auto figureOutFields = [] (const QString& str)
            {
                auto firstOpen = str.indexOf ('(');
                auto lastClose = str.lastIndexOf (')');
                return str.midRef (firstOpen, lastClose - firstOpen);
            };
            auto existing = figureOutFields (existingDDL);
            auto suggested = figureOutFields (schema);
            return existing == suggested;
        }
    }
 
    template<typename Record, typename ImplFactory = SQLiteImplFactory>
    void Migrate (QSqlDatabase& db)
    {
        const auto& baseName = Record::ClassName ();
 
        const auto& thisName = "copy" + baseName;
        const auto& schema = detail::AdaptCreateTable<ImplFactory, Record> (detail::BuildCachedFieldsData<Record> (thisName));
 
        if (detail::MatchesSchema (baseName, schema, db))
        {
            qDebug () << Q_FUNC_INFO
                    << "not migrating"
                    << db.connectionName ();
            return;
        }
 
        qDebug () << Q_FUNC_INFO
                << "migrating"
                << db.connectionName ();
 
        Util::DBLock lock { db };
        lock.Init ();
 
        Util::RunTextQuery (db, schema);
 
        const auto& fields = detail::GetFieldsNames<Record> {} ().join (", ");
 
        Util::RunTextQuery (db,
                QString { "INSERT INTO %2 (%1) SELECT %1 FROM %3;" }
                        .arg (fields)
                        .arg (thisName)
                        .arg (baseName));
 
        Util::RunTextQuery (db,
                QString { "DROP TABLE %1;" }
                        .arg (baseName));
        Util::RunTextQuery (db,
                QString { "ALTER TABLE %1 RENAME TO %2;" }
                        .arg (thisName)
                        .arg (baseName));
 
        lock.Good ();
    }
}
}
}