/*
 * Copyright (C) 2015 ScyllaDB
 */

/*
 * This file is part of Scylla.
 *
 * Scylla is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Scylla is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Scylla.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "frozen_mutation.hh"
#include "mutation_partition.hh"
#include "mutation.hh"
#include "partition_builder.hh"
#include "mutation_partition_serializer.hh"
#include "utils/UUID.hh"
#include "utils/data_input.hh"
#include "query-result-set.hh"
#include "utils/UUID.hh"
#include "serializer.hh"
#include "idl/uuid.dist.hh"
#include "idl/keys.dist.hh"
#include "idl/mutation.dist.hh"
#include "serializer_impl.hh"
#include "serialization_visitors.hh"
#include "idl/uuid.dist.impl.hh"
#include "idl/keys.dist.impl.hh"
#include "idl/mutation.dist.impl.hh"

//
// Representation layout:
//
// <mutation> ::= <column-family-id> <schema-version> <partition-key> <partition>
//

using namespace db;

utils::UUID
frozen_mutation::column_family_id() const {
    auto in = ser::as_input_stream(_bytes);
    auto mv = ser::deserialize(in, boost::type<ser::mutation_view>());
    return mv.table_id();
}

utils::UUID
frozen_mutation::schema_version() const {
    auto in = ser::as_input_stream(_bytes);
    auto mv = ser::deserialize(in, boost::type<ser::mutation_view>());
    return mv.schema_version();
}

partition_key_view
frozen_mutation::key(const schema& s) const {
    return _pk;
}

dht::decorated_key
frozen_mutation::decorated_key(const schema& s) const {
    return dht::global_partitioner().decorate_key(s, key(s));
}

partition_key frozen_mutation::deserialize_key() const {
    auto in = ser::as_input_stream(_bytes);
    auto mv = ser::deserialize(in, boost::type<ser::mutation_view>());
    return mv.key();
}

frozen_mutation::frozen_mutation(bytes&& b)
    : _bytes(std::move(b))
    , _pk(deserialize_key())
{ }

frozen_mutation::frozen_mutation(bytes_view bv, partition_key pk)
    : _bytes(bytes(bv.begin(), bv.end()))
    , _pk(std::move(pk))
{ }

frozen_mutation::frozen_mutation(const mutation& m)
    : _pk(m.key())
{
    mutation_partition_serializer part_ser(*m.schema(), m.partition());

    bytes_ostream out;
    ser::writer_of_mutation wom(out);
    std::move(wom).write_table_id(m.schema()->id())
                  .write_schema_version(m.schema()->version())
                  .write_key(m.key())
                  .partition([&] (auto wr) {
                      part_ser.write(std::move(wr));
                  }).end_mutation();

    auto bv = out.linearize();
    _bytes = bytes(bv.begin(), bv.end()); // FIXME: avoid copy
}

mutation
frozen_mutation::unfreeze(schema_ptr schema) const {
    mutation m(key(*schema), schema);
    partition_builder b(*schema, m.partition());
    partition().accept(*schema, b);
    return m;
}

frozen_mutation freeze(const mutation& m) {
    return { m };
}

mutation_partition_view frozen_mutation::partition() const {
    auto in = ser::as_input_stream(_bytes);
    auto mv = ser::deserialize(in, boost::type<ser::mutation_view>());
    return mutation_partition_view::from_view(mv.partition());
}

std::ostream& operator<<(std::ostream& out, const frozen_mutation::printer& pr) {
    return out << pr.self.unfreeze(pr.schema);
}

frozen_mutation::printer frozen_mutation::pretty_printer(schema_ptr s) const {
    return { *this, std::move(s) };
}

stop_iteration streamed_mutation_freezer::consume(tombstone pt) {
    _partition_tombstone = pt;
    return stop_iteration::no;
}

stop_iteration streamed_mutation_freezer::consume(static_row&& sr) {
    _sr = std::move(sr);
    return stop_iteration::no;
}

stop_iteration streamed_mutation_freezer::consume(clustering_row&& cr) {
    if (_reversed) {
        _crs.emplace_front(std::move(cr));
    } else {
        _crs.emplace_back(std::move(cr));
    }
    return stop_iteration::no;
}

stop_iteration streamed_mutation_freezer::consume(range_tombstone_begin&& rtb) {
    assert(!_range_tombstone_begin);
    _range_tombstone_begin = std::move(rtb);
    return stop_iteration::no;
}

stop_iteration streamed_mutation_freezer::consume(range_tombstone_end&& rte) {
    assert(_range_tombstone_begin);
    if (_reversed) {
        _rts.apply(_schema, std::move(rte.key()), flip_bound_kind(rte.kind()),
                   std::move(_range_tombstone_begin->key()), flip_bound_kind(_range_tombstone_begin->kind()),
                   _range_tombstone_begin->tomb());
    } else {
        _rts.apply(_schema, std::move(_range_tombstone_begin->key()), _range_tombstone_begin->kind(),
                   std::move(rte.key()), rte.kind(), _range_tombstone_begin->tomb());
    }
    _range_tombstone_begin = { };
    return stop_iteration::no;
}

frozen_mutation streamed_mutation_freezer::consume_end_of_stream() {
    bytes_ostream out;
    ser::writer_of_mutation wom(out);
    std::move(wom).write_table_id(_schema.id())
                  .write_schema_version(_schema.version())
                  .write_key(_key)
                  .partition([&] (auto wr) {
                      serialize_mutation_fragments(_schema, _partition_tombstone,
                                                   std::move(_sr), std::move(_rts),
                                                   std::move(_crs), std::move(wr));
                  }).end_mutation();
    return frozen_mutation(out.linearize(), std::move(_key));
}

future<frozen_mutation> freeze(streamed_mutation sm) {
    return do_with(streamed_mutation(std::move(sm)), [] (auto& sm) mutable {
        return consume(sm, streamed_mutation_freezer(*sm.schema(), sm.key()));
    });
}