From 359cc332ece713a732ce9f954e2fcc5d27dccdcd Mon Sep 17 00:00:00 2001
From: Kamran Ahmed <kamranahmed.se@gmail.com>
Date: Mon, 4 Mar 2024 16:01:31 +0000
Subject: [PATCH] Update UI for roadmap search

---
 .../GenerateRoadmap/GenerateRoadmap.tsx       | 53 +++++++++++++++----
 .../GenerateRoadmap/RoadmapSearch.tsx         | 33 ++++++++++--
 2 files changed, 71 insertions(+), 15 deletions(-)

diff --git a/src/components/GenerateRoadmap/GenerateRoadmap.tsx b/src/components/GenerateRoadmap/GenerateRoadmap.tsx
index e74580118..1b16d960f 100644
--- a/src/components/GenerateRoadmap/GenerateRoadmap.tsx
+++ b/src/components/GenerateRoadmap/GenerateRoadmap.tsx
@@ -8,7 +8,7 @@ import { readAIRoadmapStream } from '../../helper/read-stream';
 import { isLoggedIn, removeAuthToken } from '../../lib/jwt';
 import { RoadmapSearch } from './RoadmapSearch.tsx';
 import { Spinner } from '../ReactIcons/Spinner.tsx';
-import { Download, PenSquare, Wand } from 'lucide-react';
+import { Ban, Download, PenSquare, StopCircleIcon, Wand } from 'lucide-react';
 import { ShareRoadmapButton } from '../ShareRoadmapButton.tsx';
 import { httpGet, httpPost } from '../../lib/http.ts';
 import { pageProgressMessage } from '../../stores/page.ts';
@@ -19,6 +19,8 @@ import {
 } from '../../lib/browser.ts';
 import { downloadGeneratedRoadmapImage } from '../../helper/download-image.ts';
 import { showLoginPopup } from '../../lib/popup.ts';
+import { cn } from '../../lib/classname.ts';
+import { StopIcon } from '../ReactIcons/StopIcon.tsx';
 
 const ROADMAP_ID_REGEX = new RegExp('@ROADMAPID:(\\w+)@');
 
@@ -46,15 +48,13 @@ export function GenerateRoadmap() {
 
   const handleSubmit = async (e: FormEvent<HTMLFormElement>) => {
     e.preventDefault();
-    setIsLoading(true);
-    setHasSubmitted(true);
-
     if (!roadmapTopic) {
-      toast.error('Please enter a topic');
-      setIsLoading(false);
       return;
     }
 
+    setIsLoading(true);
+    setHasSubmitted(true);
+
     if (roadmapLimitUsed >= roadmapLimit) {
       toast.error('You have reached your limit of generating roadmaps');
       setIsLoading(false);
@@ -241,6 +241,7 @@ export function GenerateRoadmap() {
   }
 
   const pageUrl = `https://roadmap.sh/ai?id=${roadmapId}`;
+  const canGenerateMore = roadmapLimitUsed < roadmapLimit;
 
   return (
     <section className="flex flex-grow flex-col bg-gray-100">
@@ -255,7 +256,18 @@ export function GenerateRoadmap() {
           <div className="flex max-w-[600px] flex-grow flex-col items-center">
             <div className="mt-2 flex w-full items-center justify-between text-sm">
               <span className="text-gray-800">
-                {roadmapLimitUsed} of {roadmapLimit} roadmaps generated
+                <span
+                  className={cn(
+                    'inline-block w-[55px] rounded-md border px-0.5 text-center text-sm tabular-nums text-gray-800',
+                    {
+                      'animate-pulse border-zinc-300 bg-zinc-300 text-zinc-300':
+                        !roadmapLimit,
+                    },
+                  )}
+                >
+                  {roadmapLimitUsed} of {roadmapLimit}
+                </span>{' '}
+                roadmaps generated
                 {!isLoggedIn() && (
                   <>
                     {' '}
@@ -285,10 +297,31 @@ export function GenerateRoadmap() {
               />
               <button
                 type={'submit'}
-                className="flex flex-shrink-0 items-center gap-2 rounded-md border border-black bg-black px-4 py-2 text-white"
+                className={cn(
+                  'ml-2 flex min-w-[127px] flex-shrink-0 items-center gap-2 rounded-md bg-black px-4 py-2 text-white',
+                  {
+                    'cursor-not-allowed opacity-50':
+                      !roadmapLimit ||
+                      !roadmapTopic ||
+                      roadmapLimitUsed >= roadmapLimit,
+                  },
+                )}
               >
-                <Wand size={20} />
-                Generate
+                {roadmapLimit > 0 && canGenerateMore && (
+                  <>
+                    <Wand size={20} />
+                    Generate
+                  </>
+                )}
+
+                {roadmapLimit === 0 && <span>Please wait..</span>}
+
+                {roadmapLimit > 0 && !canGenerateMore && (
+                  <span className="flex items-center text-sm">
+                    <Ban size={15} className="mr-2" />
+                    Limit reached
+                  </span>
+                )}
               </button>
             </form>
             <div className="flex w-full items-center justify-between gap-2">
diff --git a/src/components/GenerateRoadmap/RoadmapSearch.tsx b/src/components/GenerateRoadmap/RoadmapSearch.tsx
index 1ae3d92a3..5db9a124e 100644
--- a/src/components/GenerateRoadmap/RoadmapSearch.tsx
+++ b/src/components/GenerateRoadmap/RoadmapSearch.tsx
@@ -21,6 +21,8 @@ export function RoadmapSearch(props: RoadmapSearchProps) {
     limitUsed = 0,
   } = props;
 
+  const canGenerateMore = limitUsed < limit;
+
   return (
     <div className="flex flex-grow flex-col items-center justify-center py-6">
       <div className="flex flex-col gap-2 text-center">
@@ -43,9 +45,29 @@ export function RoadmapSearch(props: RoadmapSearchProps) {
           value={roadmapTopic}
           onInput={(e) => setRoadmapTopic((e.target as HTMLInputElement).value)}
         />
-        <button className="ml-2 flex flex-shrink-0 items-center gap-2 rounded-md bg-black px-4 py-2 text-white">
-          <Wand size={20} />
-          Generate
+        <button
+          className={cn(
+            'ml-2 flex min-w-[127px] flex-shrink-0 items-center gap-2 rounded-md bg-black px-4 py-2 text-white',
+            {
+              'cursor-not-allowed opacity-50':
+                !limit || !roadmapTopic || limitUsed >= limit,
+            },
+          )}
+        >
+          {limit > 0 && canGenerateMore && (
+            <>
+              <Wand size={20} />
+              Generate
+            </>
+          )}
+
+          {limit === 0 && (
+            <>
+              <span>Please wait..</span>
+            </>
+          )}
+
+          {limit > 0 && !canGenerateMore && <span>Limit reached</span>}
         </button>
       </form>
       <div className="mb-36">
@@ -53,9 +75,10 @@ export function RoadmapSearch(props: RoadmapSearchProps) {
           You have generated{' '}
           <span
             className={cn(
-              'inline-block border w-[50px] text-sm rounded-md text-center tabular-nums text-gray-800',
+              'inline-block w-[55px] rounded-md border text-center text-sm tabular-nums text-gray-800 px-0.5',
               {
-                'bg-zinc-300 text-zinc-300 border-zinc-300 animate-pulse': !limit,
+                'animate-pulse border-zinc-300 bg-zinc-300 text-zinc-300':
+                  !limit,
               },
             )}
           >