UT_ORMOperation.h

Go to the documentation of this file.

/*
 * PROPRIETARY INFORMATION.  This software is proprietary to
 * Side Effects Software Inc., and is not to be reproduced,
 * transmitted, or disclosed in any way without written permission.
 *
 * NAME:        UT_ORMOperation.h
 *
 * COMMENTS:
 *
 *
 */

#ifndef __UT_ORMOPERATION_H__
#define __UT_ORMOPERATION_H__

#include "UT_API.h"

#include "UT_Array.h"
#include "UT_ErrorCode.h"
#include "UT_ORMColumn.h"
#include "UT_StringHolder.h"
#include "UT_UniquePtr.h"

class UT_SqlStatement;
class UT_WorkBuffer;
class UT_JSONValue;
class UT_ORMMigrationChangeNotifier;

class UT_API UT_IORMOperation
{
public:
    virtual ~UT_IORMOperation() = default;
    UT_IORMOperation(const UT_IORMOperation&) = default;
    UT_IORMOperation& operator=(const UT_IORMOperation&) = default;

    virtual const UT_StringHolder& name() const = 0;
    virtual void sql(const UT_SqlStatement& stmt, UT_WorkBuffer& wbuf) = 0;
    virtual void upgrade(UT_SqlStatement& stmt, UT_ErrorCode& ec) = 0;
    virtual void apply(
            UT_ORMMigrationChangeNotifier& notifier,
            UT_ErrorCode& ec)
            = 0;

protected:
    UT_IORMOperation() = default;

private:
};

class UT_API UT_ORMCreateTableOperation : public UT_IORMOperation
{
public:
    UT_ORMCreateTableOperation() = default;
    explicit UT_ORMCreateTableOperation(const UT_StringHolder& table_name)
        : myTableName(table_name)
    {
    }

    const UT_StringHolder& name() const override;
    void sql(const UT_SqlStatement& stmt, UT_WorkBuffer& wbuf) override;
    void upgrade(UT_SqlStatement& stmt, UT_ErrorCode& ec) override;
    void apply(UT_ORMMigrationChangeNotifier& notifier, UT_ErrorCode& ec)
            override;

    UT_ORMCreateTableOperation& setTableName(const UT_StringHolder& table_name)
    {
        myTableName = table_name;
        return *this;
    }
    UT_ORMCreateTableOperation& addField(UT_ORMColumn&& field)
    {
        myFields.emplace_back(std::move(field));
        return *this;
    }
    UT_ORMColumn& addForeignField(
            const UT_StringHolder& name,
            UT_ORMColumn::Type type,
            const UT_StringHolder& foreign_table,
            const UT_StringHolder& foreign_col,
            unsigned props = UT_ORMColumn::Properties::Empty)
    {
        UT_ORMColumn field(name, type, props);
        field.setAsForeignKey(foreign_table, foreign_col);
        exint i = myFields.emplace_back(std::move(field));
        return myFields[i];
    }
    UT_ORMColumn& addField(
            const UT_StringHolder& name,
            UT_ORMColumn::Type type,
            UT_ORMColumn::Properties props = UT_ORMColumn::Properties::Empty)
    {
        exint i = myFields.emplace_back(name, type, props);
        return myFields[i];
    }
    UT_ORMColumn& addAutoField(
            const UT_StringHolder& name,
            UT_ORMColumn::Type type,
            UT_ORMColumn::Properties props = UT_ORMColumn::Properties::Empty)
    {
        exint i = myFields.emplace_back(
                name, type, props | UT_ORMColumn::Properties::AutoIncrement);
        return myFields[i];
    }

private:
    UT_StringHolder myTableName;
    UT_Array<UT_ORMColumn> myFields;
};

class UT_API UT_ORMDropTableOperation : public UT_IORMOperation
{
public:
    UT_ORMDropTableOperation() = default;
    explicit UT_ORMDropTableOperation(const UT_StringHolder& table_name)
        : myTableName(table_name)
    {
    }

    const UT_StringHolder& name() const override;
    void sql(const UT_SqlStatement& stmt, UT_WorkBuffer& wbuf) override;
    void upgrade(UT_SqlStatement& stmt, UT_ErrorCode& ec) override;
    void apply(UT_ORMMigrationChangeNotifier& notifier, UT_ErrorCode& ec)
            override;

private:
    UT_StringHolder myTableName;
};

class UT_API UT_ORMAddFieldOperation : public UT_IORMOperation
{
public:
    UT_ORMAddFieldOperation() = default;
    explicit UT_ORMAddFieldOperation(
            const UT_StringHolder& table_name,
            UT_ORMColumn&& field)
        : myTableName(table_name), myField(std::move(field))
    {
    }

    const UT_StringHolder& name() const override;
    void sql(const UT_SqlStatement& stmt, UT_WorkBuffer& wbuf) override;
    void upgrade(UT_SqlStatement& stmt, UT_ErrorCode& ec) override;
    void apply(UT_ORMMigrationChangeNotifier& notifier, UT_ErrorCode& ec)
            override;

private:
    UT_StringHolder myTableName;
    UT_ORMColumn myField;
};

class UT_API UT_ORMRemoveFieldOperation : public UT_IORMOperation
{
public:
    UT_ORMRemoveFieldOperation() = default;
    explicit UT_ORMRemoveFieldOperation(
            const UT_StringHolder& table_name,
            const UT_StringHolder& field_name)
        : myTableName(table_name), myFieldName(field_name)
    {
    }

    const UT_StringHolder& name() const override;
    void sql(const UT_SqlStatement& stmt, UT_WorkBuffer& wbuf) override;
    void upgrade(UT_SqlStatement& stmt, UT_ErrorCode& ec) override;
    void apply(UT_ORMMigrationChangeNotifier& notifier, UT_ErrorCode& ec)
            override;

private:
    UT_StringHolder myTableName;
    UT_StringHolder myFieldName;
};

class UT_API UT_ORMRenameFieldOperation : public UT_IORMOperation
{
public:
    UT_ORMRenameFieldOperation() = default;
    explicit UT_ORMRenameFieldOperation(
            const UT_StringHolder& table_name,
            const UT_StringHolder& from_name,
            const UT_StringHolder& to_name)
        : myTableName(table_name), myFromName(from_name), myToName(to_name)
    {
    }

    const UT_StringHolder& name() const override;
    void sql(const UT_SqlStatement& stmt, UT_WorkBuffer& wbuf) override;
    void upgrade(UT_SqlStatement& stmt, UT_ErrorCode& ec) override;
    void apply(UT_ORMMigrationChangeNotifier& notifier, UT_ErrorCode& ec)
            override;

private:
    UT_StringHolder myTableName;
    UT_StringHolder myFromName;
    UT_StringHolder myToName;
};

#endif // __UT_ORMOPERATION_H__