feature/IO-3357-Reynolds-and-Reynolds-DMS-API-Integration - Checkpoint

client/src/pages/dms/dms.container.jsx

@@ -1,7 +1,8 @@
+// DmsContainer updated
 import { useQuery } from "@apollo/client";
 import { Button, Card, Col, Result, Row, Select, Space } from "antd";
 import queryString from "query-string";
-import { useEffect, useRef, useState } from "react";
+import { useEffect, useMemo, useRef, useState } from "react";
 import { useTranslation } from "react-i18next";
 import { connect } from "react-redux";
 import { Link, useLocation, useNavigate } from "react-router-dom";
@@ -66,6 +67,10 @@ export function DmsContainer({ bodyshop, setBreadcrumbs, setSelectedHeader, inse
   // New unified wss socket (Fortellis, RR)
   const { socket: wsssocket } = useSocket();
 
+  const activeSocket = useMemo(() => {
+    return dms === "rr" || (dms === "cdk" && Fortellis.treatment === "on") ? wsssocket : socket;
+  }, [dms, Fortellis.treatment, wsssocket, socket]);
+
   const { loading, error, data } = useQuery(QUERY_JOB_EXPORT_DMS, {
     variables: { id: jobId },
     skip: !jobId,
@@ -229,12 +234,12 @@ export function DmsContainer({ bodyshop, setBreadcrumbs, setSelectedHeader, inse
                 } ${data.jobs_by_pk.v_make_desc || ""} ${data.jobs_by_pk.v_model_desc || ""}`}
               </span>
             }
-            socket={socket}
+            socket={activeSocket}
             jobId={jobId}
           />
         </Col>
         <Col md={24} lg={14}>
-          <DmsPostForm socket={socket} jobId={jobId} job={data?.jobs_by_pk} logsRef={logsRef} />
+          <DmsPostForm socket={activeSocket} jobId={jobId} job={data?.jobs_by_pk} logsRef={logsRef} />
         </Col>
 
         <DmsCustomerSelector jobid={jobId} />