package moe.ymc.acron.serialization;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonParseException;
import com.google.gson.typeadapters.RuntimeTypeAdapterFactory;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import moe.ymc.acron.c2s.ReqCmd;
import moe.ymc.acron.c2s.ReqSetConfig;
import moe.ymc.acron.c2s.Request;
import moe.ymc.acron.common.Vec2f;
import moe.ymc.acron.common.Vec3d;
import moe.ymc.acron.s2c.Event;
import moe.ymc.acron.s2c.event.*;
import moe.ymc.acron.s2c.response.EventCmdOut;
import moe.ymc.acron.s2c.response.EventCmdRes;
import moe.ymc.acron.s2c.response.EventError;
import moe.ymc.acron.s2c.response.EventOk;
import org.jetbrains.annotations.NotNull;

public class Serializer {
    @NotNull
    public static Request deserialize(@NotNull TextWebSocketFrame frame)
            throws JsonParseException, IllegalArgumentException, IllegalStateException {
        final String text = frame.text();
        final RuntimeTypeAdapterFactory<Request> adapter =
                RuntimeTypeAdapterFactory.of(Request.class, "type")
                        .registerSubtype(ReqSetConfig.class, "set_config")
                        .registerSubtype(ReqCmd.class, "cmd");
        final Gson gson = new GsonBuilder()
                .registerTypeAdapter(ReqSetConfig.class, new ReqSetConfig.ReqSetConfigDeserializer())
                .registerTypeAdapter(Vec3d.class, new Vec3d.Vec3dDeserializer())
                .registerTypeAdapter(Vec2f.class, new Vec2f.Vec2fDeserializer())
                .registerTypeAdapter(ReqCmd.class, new ReqCmd.ReqCmdDeserializer())
                .registerTypeAdapterFactory(adapter)
                .create();
        final Request request = gson.fromJson(text, Request.class);
        request.validate();
        return request;
    }

    @NotNull
    public static TextWebSocketFrame serialize(@NotNull Event message) {
        final RuntimeTypeAdapterFactory<Event> adapter =
                RuntimeTypeAdapterFactory.of(Event.class, "type")
                        .registerSubtype(EventDisconnected.class, "disconnect")
                        .registerSubtype(EventPlayerMessage.class, "message")
                        .registerSubtype(EventPlayerJoined.class, "join")
                        .registerSubtype(EventEntityDeath.class, "death")
                        .registerSubtype(EventCmdOut.class, "cmd_out")
                        .registerSubtype(EventCmdRes.class, "cmd_result")
                        .registerSubtype(EventLagging.class, "lagging")
                        .registerSubtype(EventError.class, "error")
                        .registerSubtype(EventOk.class, "ok");
        final Gson gson = new GsonBuilder()
                .registerTypeAdapterFactory(adapter)
                .create();
        return new TextWebSocketFrame(gson.toJson(message, message.getClass()));
    }
}