import express, { Request, Response } from "express";
import { registrar } from "../services/registrar";
import { adminMiddleware } from "../middleware/adminMiddleware";
import logger from "../services/logger";
import { MAX_PAYLOAD_SIZE } from "../init";

type ConnectionApiKeyParams = { apiKey: string };

const postConnection = (req: Request, res: Response) => {
  const apiKey = req.body.apiKey;
  if (!apiKey) {
    return res.status(400).json({ message: "API key required" });
  }
  try {
    registrar.setConnection(apiKey, req.body);
  } catch (e) {
    logger.error({ err: e }, "Malformed payload");
    return res.status(400).json({ message: "Malformed payload" });
  }
  return res.status(200).json({ message: "Success" });
};

const getConnection = (req: Request<ConnectionApiKeyParams>, res: Response) => {
  const apiKey = req.params.apiKey;
  if (!apiKey) {
    return res.status(400).json({ message: "API key required" });
  }
  const data = registrar.getConnection(apiKey);
  if (!data) {
    return res
      .status(404)
      .json({ message: "No connection found for that API key" });
  }
  return res.status(200).json(data);
};

const getAllConnections = (req: Request, res: Response) => {
  const data = registrar.getAllConnections();
  return res.status(200).json(data);
};

const deleteConnection = (req: Request<ConnectionApiKeyParams>, res: Response) => {
  const apiKey = req.params.apiKey;
  if (!apiKey) {
    return res.status(400).json({ message: "API key required" });
  }
  const status = registrar.deleteConnection(apiKey);
  if (!status) {
    return res
      .status(404)
      .json({ message: "No connection found for that API key" });
  }
  // todo: cleanup cache, sse
  return res.status(200).json({ message: "Success" });
};

export const adminRouter = express.Router();

adminRouter.post(
  "/connection",
  adminMiddleware,
  express.json({
    limit: process.env.MAX_PAYLOAD_SIZE ?? MAX_PAYLOAD_SIZE,
  }),
  postConnection,
);

adminRouter.get("/connection/:apiKey", adminMiddleware, getConnection);

adminRouter.get("/connections", adminMiddleware, getAllConnections);

adminRouter.delete("/connection/:apiKey", adminMiddleware, deleteConnection);