Added icon support to source (#166)

src/app/(docs)/not-found.tsx

@@ -10,7 +10,8 @@ export default function NotFound() {
             404
           </h1>
           <p className="text-gray-500">
-            Looks like you&apos;ve ventured into the unknown digital realm.
+            Oops! You&apos;ve wandered off the Zen path. Let&apos;s guide you
+            back to tranquility.
           </p>
         </div>
         <Link

src/lib/source.ts

@@ -1,9 +1,19 @@
 import { docs } from "@/.source";
 import { loader } from "fumadocs-core/source";
+import { icons } from "lucide-react";
+import { createElement } from "react";
 
 // `loader()` also assign a URL to your pages
 // See https://fumadocs.vercel.app/docs/headless/source-api for more info
 export const source = loader({
+  icon(icon) {
+    if (!icon) {
+      // You may set a default icon
+      return;
+    }
+
+    if (icon in icons) return createElement(icons[icon as keyof typeof icons]);
+  },
   baseUrl: "/",
   source: docs.toFumadocsSource(),
 });