feat: add visit cookie

src/components/GenerateRoadmap/GenerateRoadmap.tsx

@@ -12,7 +12,7 @@ import { generateAIRoadmapFromText } from '../../../editor/utils/roadmap-generat
 import { renderFlowJSON } from '../../../editor/renderer/renderer';
 import { replaceChildren } from '../../lib/dom';
 import { readAIRoadmapStream } from '../../helper/read-stream';
-import { isLoggedIn, removeAuthToken } from '../../lib/jwt';
+import { isLoggedIn, removeAuthToken, visitAIRoadmap } from '../../lib/jwt';
 import { RoadmapSearch } from './RoadmapSearch.tsx';
 import { Spinner } from '../ReactIcons/Spinner.tsx';
 import { Ban, Download, PenSquare, Wand } from 'lucide-react';
@@ -176,7 +176,7 @@ export function GenerateRoadmap() {
     setIsLoading(false);
   };
 
-  const editGeneratedRoadmapContent = async () => {
+  const addToCustomRoadmap = async () => {
     if (!isLoggedIn()) {
       showLoginPopup();
       return;
@@ -188,7 +188,9 @@ export function GenerateRoadmap() {
 
     const { response, error } = await httpPost<{
       roadmapId: string;
-    }>(`${import.meta.env.PUBLIC_API_URL}/v1-edit-ai-generated-roadmap`, {
+    }>(
+      `${import.meta.env.PUBLIC_API_URL}/v1-edit-ai-generated-roadmap/${currentRoadmap?.id}`,
+      {
         title: roadmapTopic,
         nodes: nodes.map((node) => ({
           ...node,
@@ -204,7 +206,8 @@ export function GenerateRoadmap() {
           },
         })),
         edges,
-    });
+      },
+    );
 
     if (error || !response) {
       toast.error(error?.message || 'Something went wrong');
@@ -215,10 +218,7 @@ export function GenerateRoadmap() {
 
     setIsLoading(false);
     pageProgressMessage.set('');
-    window.open(
+    return response.roadmapId;
-      `${import.meta.env.PUBLIC_EDITOR_APP_URL}/${response.roadmapId}`,
-      '_blank',
-    );
   };
 
   const downloadGeneratedRoadmapContent = async () => {
@@ -279,6 +279,7 @@ export function GenerateRoadmap() {
     });
     setRoadmapTopic(topic);
     setGeneratedRoadmapContent(data);
+    visitAIRoadmap(roadmapId);
   };
 
   const handleNodeClick = useCallback(
@@ -372,7 +373,10 @@ export function GenerateRoadmap() {
           )}
           {!isLoading && (
             <div className="flex max-w-[600px] flex-grow flex-col items-center px-5">
-              <div className="mt-2 flex w-full items-center justify-between text-sm">
+              <h1 className="relative self-start text-2xl font-bold sm:text-3xl">
+                AI Roadmap Generator
+              </h1>
+              <div className="mt-6 flex w-full items-center justify-between text-sm">
                 <span className="text-gray-800">
                   <span
                     className={cn(
@@ -459,9 +463,33 @@ export function GenerateRoadmap() {
                     />
                   )}
                 </div>
+
+                <div className="flex items-center justify-between gap-2">
+                  <button
+                    className="inline-flex items-center justify-center gap-2 rounded-md bg-gray-200 py-1.5 pl-2.5 pr-3 text-xs font-medium text-black transition-colors duration-300 hover:bg-gray-300 sm:text-sm"
+                    onClick={async () => {
+                      const roadmapId = await addToCustomRoadmap();
+                      if (roadmapId) {
+                        window.location.href = `/r?id=${roadmapId}`;
+                      }
+                    }}
+                    disabled={isLoading}
+                  >
+                    <PenSquare size={15} />
+                    Track your progress
+                  </button>
+
                   <button
                     className="inline-flex items-center justify-center gap-2 rounded-md bg-gray-200 py-1.5 pl-2.5 pr-3 text-xs font-medium text-black transition-colors duration-300 hover:bg-gray-300 sm:text-sm"
-                  onClick={editGeneratedRoadmapContent}
+                    onClick={async () => {
+                      const roadmapId = await addToCustomRoadmap();
+                      if (roadmapId) {
+                        window.open(
+                          `${import.meta.env.PUBLIC_EDITOR_APP_URL}/${roadmapId}`,
+                          '_blank',
+                        );
+                      }
+                    }}
                     disabled={isLoading}
                   >
                     <PenSquare size={15} />
@@ -469,6 +497,7 @@ export function GenerateRoadmap() {
                   </button>
                 </div>
               </div>
+            </div>
           )}
         </div>
         <div

src/lib/jwt.ts

@@ -48,3 +48,18 @@ export function removeAuthToken() {
     domain: import.meta.env.DEV ? 'localhost' : '.roadmap.sh',
   });
 }
+
+export function visitAIRoadmap(roadmapId: string) {
+  const isAlreadyVisited = Number(Cookies.get(`crv-${roadmapId}`) || 0) === 1;
+  if (isAlreadyVisited) {
+    return;
+  }
+
+  Cookies.set(`crv-${roadmapId}`, '1', {
+    path: '/',
+    expires: 30,
+    sameSite: 'lax',
+    secure: !import.meta.env.DEV,
+    domain: import.meta.env.DEV ? 'localhost' : '.roadmap.sh',
+  });
+}