#include <mbgl/map/vector_tile.hpp>
#include <mbgl/geometry/pbf_geometry.hpp>
#include <mbgl/style/filter_expression_private.hpp>
#include <mbgl/util/variant_io.hpp>

#include <type_traits>
#include <ostream>

using namespace mbgl;

VectorTileFeature::VectorTileFeature(pbf feature_pbf, const util::ptr<GeometryTileLayer> layer) {
    while (feature_pbf.next()) {
        if (feature_pbf.tag == 1) { // id
            id = feature_pbf.varint<uint64_t>();
        } else if (feature_pbf.tag == 2) { // tags
            // tags are packed varints. They should have an even length.
            pbf tags = feature_pbf.message();
            while (tags) {
                uint32_t tag_key = tags.varint();

                if (layer->keys.size() <= tag_key) {
                    throw std::runtime_error("feature referenced out of range key");
                }

                if (tags) {
                    uint32_t tag_val = tags.varint();
                    if (layer->values.size() <= tag_val) {
                        throw std::runtime_error("feature referenced out of range value");
                    }

                    properties.emplace(layer->keys[tag_key], layer->values[tag_val]);
                } else {
                    throw std::runtime_error("uneven number of feature tag ids");
                }
            }
        } else if (feature_pbf.tag == 3) { // type
            type = (GeometryFeatureType)feature_pbf.varint();
        } else if (feature_pbf.tag == 4) { // geometry
            geometry_pbf = feature_pbf.message();
        } else {
            feature_pbf.skip();
        }
    }
}

Geometry VectorTileFeature::nextGeometry() {
    pbf current_geometry_pbf = geometry_pbf.message();
    PBFGeometry current_geometry(current_geometry_pbf);
    int32_t x, y;
    Geometry result;
    result.set<std::false_type>();

    if (multigeometry || geometry_pbf.next(4)) {
        if (type == GeometryFeatureType::Point) {
            current_geometry.next(x, y);
            GeometryPoint point(x, y);
            result.set<GeometryPoint>(point);
        } else if (type == GeometryFeatureType::LineString) {
            GeometryLine line;
            PBFGeometry::command cmd;
            while ((cmd = current_geometry.next(x, y)) != PBFGeometry::end) {
                if (cmd == PBFGeometry::move_to) {
                    multigeometry = true;
                    result.set<GeometryLine>(line);
                    break;
                } else {
                    line.emplace_back(x, y);
                }
            }
            if (!multigeometry) {
                result.set<GeometryLine>(line);
            }
        } else if (type == GeometryFeatureType::Polygon) {
            GeometryPolygon polygon;
            PBFGeometry::command cmd;
            while ((cmd = current_geometry.next(x, y)) != PBFGeometry::end) {
                if (cmd == PBFGeometry::move_to) {
                    multigeometry = true;
                    result.set<GeometryPolygon>(polygon);
                    break;
                } else {
//                    polygon.emplace_back(x, y);
                }
            }
            if (!multigeometry) {
                result.set<GeometryPolygon>(polygon);
            }
        } else {
            throw std::runtime_error("unrecognized geometry type");
        }
    }

    return std::move(result);
}

VectorTile::VectorTile(pbf tile_pbf) {
    while (tile_pbf.next()) {
        if (tile_pbf.tag == 3) { // layer
            util::ptr<VectorTileLayer> layer = std::make_shared<VectorTileLayer>(tile_pbf.message());
            layers.emplace(layer->name, std::forward<util::ptr<VectorTileLayer>>(layer));
        } else {
            tile_pbf.skip();
        }
    }
}

void VectorTile::logDebug() const {
    printf("got here for VectorTile\n");
}

VectorTileLayer::VectorTileLayer(pbf layer_pbf)
    : feature_pbf(layer_pbf) {
    while (layer_pbf.next()) {
        if (layer_pbf.tag == 1) { // name
            name = layer_pbf.string();
        } else if (layer_pbf.tag == 3) { // keys
            keys.emplace_back(layer_pbf.string());
            key_index.emplace(keys.back(), keys.size() - 1);
        } else if (layer_pbf.tag == 4) { // values
            values.emplace_back(std::move(parseValue(layer_pbf.message())));
        } else if (layer_pbf.tag == 5) { // extent
            extent = layer_pbf.varint();
        } else {
            layer_pbf.skip();
        }
    }
}

FilteredVectorTileLayer::FilteredVectorTileLayer(const util::ptr<VectorTileLayer> layer_, const FilterExpression& filterExpression_)
    : GeometryFilteredTileLayer(layer_, filterExpression_) {
    feature_pbf = layer_->feature_pbf;
}

template <>
GeometryTileTagExtractor<pbf>::GeometryTileTagExtractor(const util::ptr<GeometryTileLayer> layer_)
    : layer(layer_) {}

template <>
mapbox::util::optional<Value> GeometryTileTagExtractor<pbf>::getValue(const std::string& key) const {
    if (key == "$type") {
        return Value(uint64_t(type));
    }

    mapbox::util::optional<Value> value;

    auto field_it = layer->key_index.find(key);
    if (field_it != layer->key_index.end()) {
        const uint32_t filter_key = field_it->second;

        // Now loop through all the key/value pair tags.
        // tags are packed varints. They should have an even length.
        pbf tags_pbf_ = tags;
        uint32_t tag_key, tag_val;
        while (tags_pbf_) {
            tag_key = tags_pbf_.varint();
            if (!tags_pbf_) {
                // This should not happen; otherwise the vector tile is invalid.
                fprintf(stderr, "[WARNING] uneven number of feature tag ids\n");
                return value;
            }
            // Note: We need to run this command in all cases, even if the keys don't match.
            tag_val = tags_pbf_.varint();

            if (tag_key == filter_key) {
                if (layer->values.size() > tag_val) {
                    value = layer->values[tag_val];
                } else {
                    fprintf(stderr, "[WARNING] feature references out of range value\n");
                    break;
                }
            }
        }
    }

    return value;
}

template bool mbgl::evaluate(const FilterExpression&, const GeometryTileTagExtractor<pbf>&);

util::ptr<GeometryTileFeature> FilteredVectorTileLayer::nextMatchingFeature() {
    while (feature_pbf.next(2)) {
        pbf matching_feature_pbf = feature_pbf.message();
        pbf current_feature_pbf = matching_feature_pbf;

        GeometryTileTagExtractor<pbf> extractor(layer);

        while (current_feature_pbf.next()) {
            if (current_feature_pbf.tag == 2) { // tags
                extractor.setTags(current_feature_pbf.message());
            } else if (current_feature_pbf.tag == 3) { // geometry type
                extractor.setType(GeometryFeatureType(current_feature_pbf.varint()));
            } else {
                current_feature_pbf.skip();
            }
        }

        if (evaluate(filterExpression, extractor)) {
            return std::move(std::make_shared<VectorTileFeature>(matching_feature_pbf, layer));
        }
    }

    return std::move(std::make_shared<VectorTileFeature>(pbf(), layer));
}

util::ptr<GeometryFilteredTileLayer> VectorTileLayer::createFilter(const FilterExpression& filterExpression) {
    return std::move(std::make_shared<FilteredVectorTileLayer>(shared_from_this(), filterExpression));
}

util::ptr<GeometryTileFeature> VectorTileLayer::nextFeature() {
    if (feature_pbf.next(2)) {
        return std::move(std::make_shared<VectorTileFeature>(feature_pbf.message(), shared_from_this()));
    }

    return std::move(std::make_shared<VectorTileFeature>(pbf(), shared_from_this()));
}