105 lines
2.9 KiB
TypeScript
105 lines
2.9 KiB
TypeScript
import { getPublicUser, prisma } from "@repo/db";
|
|
import { Server, Socket } from "socket.io";
|
|
|
|
export const handleConnectPilot =
|
|
(socket: Socket, io: Server) =>
|
|
async ({
|
|
logoffTime,
|
|
stationId,
|
|
}: {
|
|
logoffTime: string;
|
|
stationId: string;
|
|
}) => {
|
|
try {
|
|
const user = socket.data.user; // User ID aus dem JWT-Token
|
|
const userId = socket.data.user.id; // User ID aus dem JWT-Token
|
|
|
|
console.log("Pilot connected:", userId);
|
|
|
|
if (!user) return Error("User not found");
|
|
|
|
const existingConnection = await prisma.connectedAircraft.findFirst({
|
|
where: {
|
|
userId: user.id,
|
|
logoutTime: null,
|
|
},
|
|
});
|
|
|
|
if (existingConnection) {
|
|
await io
|
|
.to(`user:${user.id}`)
|
|
.emit("force-disconnect", "double-connection");
|
|
await prisma.connectedAircraft.updateMany({
|
|
where: {
|
|
userId: user.id,
|
|
logoutTime: null,
|
|
},
|
|
data: {
|
|
logoutTime: new Date().toISOString(),
|
|
},
|
|
});
|
|
}
|
|
|
|
let parsedLogoffDate = null;
|
|
if (logoffTime.length > 0) {
|
|
const now = new Date();
|
|
const [hours, minutes] = logoffTime.split(":").map(Number);
|
|
if (!hours || !minutes) {
|
|
throw new Error("Invalid logoffTime format");
|
|
}
|
|
parsedLogoffDate = new Date(now);
|
|
parsedLogoffDate.setHours(hours, minutes, 0, 0);
|
|
|
|
// If the calculated time is earlier than now, add one day to make it tomorrow
|
|
if (parsedLogoffDate <= now) {
|
|
parsedLogoffDate.setDate(parsedLogoffDate.getDate() + 1);
|
|
}
|
|
}
|
|
|
|
const connectedAircraftEntry = await prisma.connectedAircraft.create({
|
|
data: {
|
|
publicUser: getPublicUser(user) as any,
|
|
esimatedLogoutTime: parsedLogoffDate?.toISOString() || null,
|
|
lastHeartbeat: new Date().toISOString(),
|
|
userId: userId,
|
|
loginTime: new Date().toISOString(),
|
|
stationId: parseInt(stationId),
|
|
},
|
|
});
|
|
|
|
socket.join("dispatchers"); // Join the dispatchers room
|
|
socket.join(`user:${userId}`); // Join the user-specific room
|
|
socket.join(`station:${stationId}`); // Join the station-specific room
|
|
|
|
console.log(`Pilot in: station:${stationId}`);
|
|
|
|
io.to("dispatchers").emit("pilots-update");
|
|
io.to("pilots").emit("pilots-update");
|
|
|
|
// Add a listener for station-specific events
|
|
socket.on(`station:${stationId}:event`, async (data) => {
|
|
console.log(`Received event for station ${stationId}:`, data);
|
|
// Handle station-specific logic here
|
|
io.to(`station:${stationId}`).emit("station-event-update", data);
|
|
});
|
|
|
|
socket.on("disconnect", async () => {
|
|
console.log("Disconnected from dispatch server");
|
|
await prisma.connectedAircraft
|
|
.update({
|
|
where: {
|
|
id: connectedAircraftEntry.id,
|
|
},
|
|
data: {
|
|
logoutTime: new Date().toISOString(),
|
|
},
|
|
})
|
|
.catch(console.error);
|
|
io.to("dispatchers").emit("pilots-update");
|
|
io.to("pilots").emit("pilots-update");
|
|
});
|
|
} catch (error) {
|
|
console.error("Error connecting to dispatch server:", error);
|
|
}
|
|
};
|