fix: ask-ai-click-test (#2909)

## Description 

Provide a concise summary of the changes made in this pull request
- 

## Pull request type

Check the appropriate box:

- [ ] Review Fixes
- [ ] Documentation Overhaul
- [ ] Feature/Story
    - Link one or more Engineering Tickets
        * 
- [ ] A-Force
- [ ] Error in documentation
- [ ] Maintenance

## Documentation tickets

 Link to one or more documentation tickets:
 - 

## Checklist

From the below options, select the ones that are applicable:

- [ ] Checked for Grammarly suggestions.
- [ ] Adhered to the writing checklist.
- [ ] Adhered to the media checklist.
- [ ] Verified and updated cross-references or added redirect rules.
- [ ] Tested the redirect rules on deploy preview.
- [ ] Validated the modifications made to the content on the deploy
preview.
- [ ] Validated the CSS modifications on different screen sizes.

Author: harshilp24

website/src/components/custom-search/CustomSearchBar.jsx

@@ -7,6 +7,25 @@ import '@site/src/components/custom-search/css/CustomSearch.css';
 const CustomSearchBar = () => {
     const [searchType, setSearchType] = useState('');
 
+    // Identify the user once on mount
+    useEffect(() => {
+        if (ExecutionEnvironment.canUseDOM && typeof window.analytics !== 'undefined') {
+            const user = JSON.parse(localStorage.getItem('user') || '{}');
+
+            if (user?.id && user?.email) {
+                window.analytics.identify(user.id, {
+                    name: user.name,
+                    email: user.email,
+                    plan: user.plan || 'Free',
+                });
+            }
+
+            // Optional: Enable debug logging
+            // window.analytics.debug();
+        }
+    }, []);
+
+    // Trigger search input clicks
     useEffect(() => {
         if (ExecutionEnvironment.canUseDOM) {
             const searchInput = document.querySelector('.DocSearch-Button');
@@ -18,8 +37,10 @@ const CustomSearchBar = () => {
                 searchInput.click();
                 setTimeout(() => {
                     const searchTerm = document.querySelector('.DocSearch-Input');
-                    searchTerm.focus();
-                    searchTerm.click();
+                    if (searchTerm) {
+                        searchTerm.focus();
+                        searchTerm.click();
+                    }
                 }, 100);
             }
 
@@ -29,11 +50,15 @@ const CustomSearchBar = () => {
 
     const handleClick = (type) => {
         setSearchType(type);
-        if (ExecutionEnvironment.canUseDOM) {
+
+        if (ExecutionEnvironment.canUseDOM && typeof window.analytics !== 'undefined') {
             const eventName = type === 'ai' ? 'Ask AI Button Click' : 'Search Button Click';
-            if (typeof window.analytics !== 'undefined') {
-                window.analytics.track(eventName, { searchType: type });
-            }
+
+            window.analytics.track(eventName, {
+                searchType: type,
+                page: window.location.pathname,
+                timestamp: new Date().toISOString(),
+            });
         }
     };
 
@@ -44,13 +69,13 @@ const CustomSearchBar = () => {
                     className={`custom-search-option ${searchType === 'ai' ? 'selected' : ''}`}
                     onClick={() => handleClick('ai')}
                 >
-                    <img src="/img/ask-ai-robot-icon.svg" alt="Ask AI" className='ai-search-icon'></img> Ask AI
+                    <img src="/img/ask-ai-robot-icon.svg" alt="Ask AI" className="ai-search-icon" /> Ask AI
                 </div>
                 <div
                     className={`custom-search-option ${searchType === 'docs' ? 'selected' : ''}`}
                     onClick={() => handleClick('docs')}
                 >
-                    <img src="/img/search-in-docs-icon.svg" alt="Search" className='doc-search-icon'></img> Search
+                    <img src="/img/search-in-docs-icon.svg" alt="Search" className="doc-search-icon" /> Search
                 </div>
             </div>
             <AISearchButton />
@@ -61,4 +86,4 @@ const CustomSearchBar = () => {
     );
 };
 
-export default CustomSearchBar;
+export default CustomSearchBar;