#include <Storages/System/StorageSystemDetachedParts.h>

#include <DataTypes/DataTypeString.h>
#include <DataTypes/DataTypesNumber.h>
#include <DataStreams/OneBlockInputStream.h>
#include <ext/shared_ptr_helper.h>
#include <Storages/IStorage.h>
#include <Storages/System/StorageSystemPartsBase.h>

namespace DB
{

/**
  * Implements system table 'detached_parts' which allows to get information
  * about detached data parts for tables of MergeTree family.
  * We don't use StorageSystemPartsBase, because it introduces virtual _state
  * column and column aliases which we don't need.
  */
class StorageSystemDetachedParts :
    public ext::shared_ptr_helper<StorageSystemDetachedParts>,
    public IStorage
{
public:
    std::string getName() const override { return "SystemDetachedParts"; }
    std::string getTableName() const override { return "detached_parts"; }
    std::string getDatabaseName() const override { return "system"; }

protected:
    explicit StorageSystemDetachedParts()
    {
        setColumns(ColumnsDescription{{
            {"database", std::make_shared<DataTypeString>()},
            {"table", std::make_shared<DataTypeString>()},
            {"partition_id", std::make_shared<DataTypeString>()},
            {"name", std::make_shared<DataTypeString>()},
            {"reason", std::make_shared<DataTypeString>()},
            {"min_block_number", std::make_shared<DataTypeInt64>()},
            {"max_block_number", std::make_shared<DataTypeInt64>()},
            {"level", std::make_shared<DataTypeUInt32>()}
        }});
    }

    BlockInputStreams read(
            const Names & /* column_names */,
            const SelectQueryInfo & query_info,
            const Context & context,
            QueryProcessingStage::Enum /*processed_stage*/,
            const size_t /*max_block_size*/,
            const unsigned /*num_streams*/) override
    {
        StoragesInfoStream stream(query_info, context);

        /// Create the result.
        Block block = getSampleBlock();
        MutableColumns columns = block.cloneEmptyColumns();

        while (StoragesInfo info = stream.next())
        {
            const auto parts = info.data->getDetachedParts();
            for (auto & p : parts)
            {
                size_t i = 0;
                columns[i++]->insert(info.database);
                columns[i++]->insert(info.table);
                columns[i++]->insert(p.partition_id);
                columns[i++]->insert(p.getPartName());
                columns[i++]->insert(p.prefix);
                columns[i++]->insert(p.min_block);
                columns[i++]->insert(p.max_block);
                columns[i++]->insert(p.level);
            }
        }

        return BlockInputStreams(1, std::make_shared<OneBlockInputStream>(
             block.cloneWithColumns(std::move(columns))));
    }
};

StoragePtr
createDetachedPartsTable()
{
    return StorageSystemDetachedParts::create();
}

}