PhotonVision/Main/src/main/java/com/chameleonvision/web/Server.java

package com.chameleonvision.web;

import com.chameleonvision.NoCameraException;
import com.chameleonvision.settings.SettingsManager;
import com.chameleonvision.vision.GeneralSettings;
import com.chameleonvision.vision.Pipeline;
import com.google.gson.JsonObject;
import io.javalin.Javalin;
import io.javalin.websocket.WsContext;

import java.lang.reflect.Field;
import java.util.*;

import org.json.JSONArray;
import org.json.JSONObject;
import org.springframework.beans.BeanUtils;


public class Server {
    private static List<WsContext> users = new ArrayList<WsContext>();

    public static void main(int port) {
        Javalin app = Javalin.create();
        app.config.addStaticFiles("web");
        app.ws("/websocket", ws -> {
            ws.onConnect(ctx -> {
                users.add(ctx);
                System.out.println("Socket Connected");
                sendFullSettings();
            });
            ws.onClose(ctx -> {
                users.remove(ctx);
                System.out.println("Socket Disconnected");
                SettingsManager.getInstance().SaveSettings();
            });
            ws.onMessage(ctx -> {
//                System.out.println(SettingsManager.getInstance().GetCurrentPipeline().);
                broadcastMessage(ctx, ctx.message());
                JSONObject jsonObject = new JSONObject(ctx.message());
                String key = jsonObject.keySet().toArray()[0].toString();
                Object value = jsonObject.get(key);
                if (!setField(SettingsManager.getInstance().GetCurrentPipeline(), key, value)) {
                    //If field not in pipeline
                    switch (key) {
                        case "change_general_settings_values":
                            JSONObject newSettings = (JSONObject) value;
                            setFields(SettingsManager.getInstance().GeneralSettings, newSettings);
                            break;
                        case "curr_camera":
                            SettingsManager.getInstance().SetCurrentCamera((String) value);
                            //broadcastMessage((Map<String, Object>) new HashMap<String, Object>(){}.put("port",SettingsManager.CameraPorts.get(SettingsManager.GeneralSettings.curr_camera)));
                            //broadcastMessage(SettingsManager.getInstance().GetCurrentCamera());//TODO CHECK JSON FOR CAMERA CHANGE
                            break;
                        case "curr_pipeline":
                            System.out.println("change pipeline");
                            SettingsManager.getInstance().SetCurrentPipeline((String) value);
                            SettingsManager.CamerasCurrentPipeline.put(SettingsManager.GeneralSettings.curr_camera, (String) value);
//                            broadcastMessage(SettingsManager.getInstance().GetCurrentPipeline());//TODO CHECK JSON FOR PIPELINE CHANGE
                            break;
                        case "resolution":
                            System.out.println("change res");
                            SettingsManager.getInstance().GetCurrentCamera().resolution = (int) value;
                            SettingsManager.getInstance().SetCameraSettings(SettingsManager.GeneralSettings.curr_camera, "resolution", value);
                            SettingsManager.getInstance().SaveSettings();
                            break;
                        case "fov":
                            System.out.println("change fov");
                            SettingsManager.getInstance().GetCurrentCamera().FOV = (double) value;
                            SettingsManager.getInstance().SaveSettings();
                            break;
                        default:
                            System.out.println("Unexpected value");
                            break;
                    }
                }
            });
        });
        app.start(port);
    }


    public static boolean setField(Object obj, String fieldName, Object value) {
        boolean successful = false;
        try {
            Field[] fields = obj.getClass().getFields();
            for (Field f : fields) {
                if (f.getName().equals(fieldName)) {
                    successful = true;
                    if (BeanUtils.isSimpleValueType(value.getClass())) {
                        f.set(obj, value);
                    } else if (value.getClass() == JSONArray.class) {
                        f.set(obj, ((JSONArray) value).toList());
                    }
                }
            }
        } catch (IllegalAccessException e) {
            return false;
        }
        return successful;
    }

    public static boolean setFields(Object obj, JSONObject data) {
        boolean successful = false;
        try {
            Field[] fields = obj.getClass().getFields();
            for (Field f : fields) {
                if (data.has(f.getName())) {
                    Object value = data.get(f.getName());
                    if (BeanUtils.isSimpleValueType(value.getClass())) {
                        successful = true;
                        f.set(obj, value);
                    }
                }
            }
        } catch (IllegalAccessException e) {
            return false;
        }
        return successful;
    }

    private static void broadcastMessage(WsContext sendingUser, String message) {
        for (var user : users) {
            if (user != sendingUser) {
                user.send(message);
            }
        }
    }

    private static void broadcastMessage(Map<String, Object> map) {
        for (var user : users) {
            user.send(new JSONObject(map).toString());
        }

    }

    private static void addAllFieldsToMap(Map<String, Object> map, Object obj) {
        try {
            Field[] fields = obj.getClass().getFields();
            for (Field field : fields)
                map.put(field.getName(), field.get(obj));
        } catch (IllegalAccessException e) {
            System.err.println("Illegal Access error:" + e.getStackTrace().toString());
        }
    }

    private static void sendFullSettings() {
        Map<String, Object> fullSettings = new HashMap<>();
        //General settings
        addAllFieldsToMap(fullSettings, SettingsManager.GeneralSettings);
        fullSettings.put("cameraList", SettingsManager.Cameras.keySet());
        try {
            addAllFieldsToMap(fullSettings, SettingsManager.getInstance().GetCurrentPipeline());
            fullSettings.put("pipelineList", SettingsManager.getInstance().GetCurrentCamera().pipelines.keySet());
            fullSettings.put("resolutionList", SettingsManager.getInstance().GetResolutionList());
            fullSettings.put("resolution", SettingsManager.getInstance().GetCurrentCamera().resolution);
            fullSettings.put("FOV", SettingsManager.getInstance().GetCurrentCamera().FOV);
//            fullSettings.put("port", SettingsManager.CameraPorts.get(SettingsManager.GeneralSettings.curr_camera));
        } catch (NoCameraException e) {
            System.err.println("No camera found!");
            //TODO: add message to ui to inform that there are no cameras
        }
        broadcastMessage(fullSettings);
    }


}