use faker with UseMessageListener hook

Webby1015 · Oct 21, 2023 · fb3dea8 · fb3dea8
1 parent 05bfd9d
commit fb3dea8
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 1 deletion.
diff --git a/chat/src/Hooks/UseMessageListener/index.ts b/chat/src/Hooks/UseMessageListener/index.ts
@@ -1,7 +1,8 @@
 import { useEffect, useState } from "react";
 import { IMessage } from "../../types";
+import DummyMessageGenerator from "../../utils/faker";
 
-export const useMessageListener = (): IMessage[] => {
+const useMessageListenerProd = (): IMessage[] => {
   const [messages, setMessages] = useState<IMessage[]>([]);
   let intervalId: ReturnType<typeof setInterval> | null = null;
 
@@ -38,3 +39,25 @@ export const useMessageListener = (): IMessage[] => {
 
   return messages;
 };
+
+const useMessageListenerDev = (): IMessage[] => {
+  const [messages, setMessages] = useState<IMessage[]>([]);
+
+  useEffect(() => {
+    const intervalId = setInterval(() => {
+      const newMessage = DummyMessageGenerator.generate();
+      setMessages((prevMessages) => [...prevMessages, newMessage]);
+    }, 1000);
+
+    return () => {
+      clearInterval(intervalId);
+    };
+  }, []);
+
+  return messages;
+};
+
+export const useMessageListener =
+  import.meta.env.VITE_USE_DUMMY_DATA === "true"
+    ? useMessageListenerDev
+    : useMessageListenerProd;
diff --git a/chat/src/env.d.ts b/chat/src/env.d.ts
@@ -0,0 +1,9 @@
+/// <reference types="vite/client" />
+
+interface ImportMetaEnv {
+  readonly VITE_USE_DUMMY_DATA: string;
+}
+
+interface ImportMeta {
+  readonly env: ImportMetaEnv;
+}