qt-creator/src/libs/sqlite/constraints.h

/****************************************************************************
**
** Copyright (C) 2020 The Qt Company Ltd.
** Contact: https://www.qt.io/licensing/
**
** This file is part of Qt Creator.
**
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and The Qt Company. For licensing terms
** and conditions see https://www.qt.io/terms-conditions. For further
** information use the contact form at https://www.qt.io/contact-us.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3 as published by the Free Software
** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
** included in the packaging of this file. Please review the following
** information to ensure the GNU General Public License requirements will
** be met: https://www.gnu.org/licenses/gpl-3.0.html.
**
****************************************************************************/

#pragma once

#include "sqliteglobal.h"

#include <sqlitevalue.h>
#include <utils/smallstring.h>
#include <utils/variant.h>

namespace Sqlite {

class Unique
{
    friend bool operator==(Unique, Unique) { return true; }
};

class PrimaryKey
{
    friend bool operator==(PrimaryKey, PrimaryKey) { return true; }
};

class NotNull
{
    friend bool operator==(NotNull, NotNull) { return true; }
};

class Check
{
public:
    Check(Utils::SmallStringView expression)
        : expression(expression)
    {}

    friend bool operator==(const Check &first, const Check &second)
    {
        return first.expression == second.expression;
    }

public:
    Utils::SmallString expression;
};

class DefaultValue
{
public:
    DefaultValue(long long value)
        : value(value)
    {}

    DefaultValue(double value)
        : value(value)
    {}

    DefaultValue(Utils::SmallStringView value)
        : value(value)
    {}

    friend bool operator==(const DefaultValue &first, const DefaultValue &second)
    {
        return first.value == second.value;
    }

public:
    Sqlite::Value value;
};

class DefaultExpression
{
public:
    DefaultExpression(Utils::SmallStringView expression)
        : expression(expression)
    {}

    friend bool operator==(const DefaultExpression &first, const DefaultExpression &second)
    {
        return first.expression == second.expression;
    }

public:
    Utils::SmallString expression;
};

class Collate
{
public:
    Collate(Utils::SmallStringView collation)
        : collation(collation)
    {}

    friend bool operator==(const Collate &first, const Collate &second)
    {
        return first.collation == second.collation;
    }

public:
    Utils::SmallString collation;
};

class ForeignKey
{
public:
    ForeignKey() = default;
    ForeignKey(Utils::SmallStringView table,
               Utils::SmallStringView column,
               ForeignKeyAction updateAction = {},
               ForeignKeyAction deleteAction = {},
               Enforment enforcement = {})
        : table(table)
        , column(column)
        , updateAction(updateAction)
        , deleteAction(deleteAction)
        , enforcement(enforcement)
    {}

    friend bool operator==(const ForeignKey &first, const ForeignKey &second)
    {
        return first.table == second.table && first.column == second.column
               && first.updateAction == second.updateAction
               && first.deleteAction == second.deleteAction;
    }

public:
    Utils::SmallString table;
    Utils::SmallString column;
    ForeignKeyAction updateAction = {};
    ForeignKeyAction deleteAction = {};
    Enforment enforcement = {};
};

enum class GeneratedAlwaysStorage { Stored, Virtual };

class GeneratedAlways
{
public:
    GeneratedAlways(Utils::SmallStringView expression, GeneratedAlwaysStorage storage)
        : expression(expression)
        , storage(storage)
    {}

    friend bool operator==(const GeneratedAlways &first, const GeneratedAlways &second)
    {
        return first.expression == second.expression;
    }

public:
    Utils::SmallString expression;
    GeneratedAlwaysStorage storage = {};
};

using Constraint = Utils::variant<Unique,
                                  PrimaryKey,
                                  ForeignKey,
                                  NotNull,
                                  Check,
                                  DefaultValue,
                                  DefaultExpression,
                                  Collate,
                                  GeneratedAlways>;
using Constraints = std::vector<Constraint>;

} // namespace Sqlite