package com.spring.restapi.serializers;

import java.lang.reflect.Type;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;
import com.google.gson.JsonSerializationContext;
import com.google.gson.JsonSerializer;
import org.bson.types.ObjectId;

public class ConnectionSerializer {

    public static class Connection {

        public ObjectId _id;
        public String connectionDate;
        public String sourceIP;
        public String destinationIP;
        public String destinationHostname;
        public String connectionFlag;
        public String connectionSequence;
        public Integer dataWindow;

        public Connection(final ObjectId _id, String connectionDate, String sourceIP, String destinationIP, String destinationHostname, String connectionFlag, String connectionSequence, Integer dataWindow) {
            super();
            this._id = _id;
            this.connectionDate = connectionDate;
            this.sourceIP = sourceIP;
            this.destinationIP = destinationIP;
            this.destinationHostname = destinationHostname;
            this.connectionFlag = connectionFlag;
            this.connectionSequence = connectionSequence;
            this.dataWindow = dataWindow;
        }

        public ObjectId getId() {
            return _id;
        }

        public void setId(final ObjectId _id) {
            this._id = _id;
        }

        public String getConnectionDate() {
            return connectionDate;
        }

        public void setConnectionDate(final String connectionDate) {
            this.connectionDate = connectionDate;
        }

        public String getSourceIP() {
            return sourceIP;
        }

        public void setSourceIP(final String sourceIP) {
            this.sourceIP = sourceIP;
        }

        public String getDestinationIP() {
            return destinationIP;
        }

        public void setDestinationIP(final String destinationIP) {
            this.destinationIP = destinationIP;
        }

        public String getDestinationHostname() {
            return destinationHostname;
        }

        public void setDestinationHostname(final String destinationHostname) {
            this.destinationHostname = destinationHostname;
        }

        public String getConnectionFlag() {
            return connectionFlag;
        }

        public void setConnectionFlag(final String connectionFlag) {
            this.connectionFlag = connectionFlag;
        }

        public String getConnectionSequence() {
            return connectionSequence;
        }

        public void setConnectionSequence(final String connectionSequence) {
            this.connectionSequence = connectionSequence;
        }

        public Integer getDataWindow() {
            return dataWindow;
        }

        public void setDataWindow(final Integer dataWindow) {
            this.dataWindow = dataWindow;
        }
    }

    public static class Serializer implements JsonSerializer<Connection> {
        public JsonElement serialize(final Connection connection, final Type type, final JsonSerializationContext context) {
            JsonObject result = new JsonObject();
            result.add("_id", new JsonPrimitive(connection.getId().toHexString()));
            result.add("connectionDate", new JsonPrimitive(connection.getConnectionDate()));
            result.add("sourceIP", new JsonPrimitive(connection.getSourceIP()));
            result.add("destinationIP", new JsonPrimitive(connection.getDestinationIP()));
            result.add("destinationHostname", new JsonPrimitive(connection.getDestinationHostname()));
            result.add("connectionFlag", new JsonPrimitive(connection.getConnectionFlag()));
            result.add("connectionSequence", new JsonPrimitive(connection.getConnectionSequence()));
            result.add("dataWindow", new JsonPrimitive(connection.getDataWindow()));
            return result;
        }
    }

    public static void main(tutasj potrzebuje przekazać obiekt) {
        Connection connection = przekazany_obiekt;
        com.google.gson.Gson gson = new GsonBuilder().registerTypeAdapter(Connection.class, new Serializer()).create();
        System.out.println(gson.toJson(connection));
    }

}