shadcn-studio-extension-cli 0.1.10 → 0.2.0

package/dist/toolbar-app/index.js

@@ -2,7 +2,7 @@ import * as React from "react";
 import React__default, { createContext, useState, useEffect, useContext, useCallback, useMemo, useRef, forwardRef, createElement, useLayoutEffect, Fragment as Fragment$1, isValidElement, cloneElement, useId as useId$1, useReducer, useSyncExternalStore, createRef, useImperativeHandle, StrictMode } from "react";
 import { createRoot } from "react-dom/client";
 import { jsx, jsxs, Fragment } from "react/jsx-runtime";
-import { A as AgentStateType, u as useAgents, a as useAppState, b as usePlugins, c as useAgentMessaging, d as usePanels, g as getSelectedElementInfo, e as getSelectedDocInfo, f as getSelectedBlockInfo, h as getSelectedThemeInfo, i as collectUserMessageMetadata, j as generateId, C as ConfigProvider, k as AgentProvider, l as AgentAvailabilityProvider, m as AgentMessagingProvider, P as PanelsProvider, n as PluginProvider, H as HotkeyActions, o as hotkeyActionDefinitions, p as getIFrameWindow, q as cn, r as getElementAtPoint, s as getXPathForElement, B as Button, t as glassyBoxClassName, v as AgentAvailabilityError, w as useAgentAvailability, x as Panel, y as PanelHeader, z as PanelContent, D as PanelFooter, E as cva, F as getLicenseDataFromStorage, G as Glassy, I as AppStateProvider } from "index-CcQR8qhj.js";
+import { A as AgentStateType, u as useAgents, a as useAppState, b as usePlugins, c as useAgentMessaging, d as usePanels, g as getSelectedElementInfo, e as getSelectedDocInfo, f as getSelectedBlockInfo, h as getSelectedThemeInfo, i as collectUserMessageMetadata, j as generateId, C as ConfigProvider, k as AgentProvider, l as AgentAvailabilityProvider, m as AgentMessagingProvider, P as PanelsProvider, n as PluginProvider, H as HotkeyActions, o as hotkeyActionDefinitions, p as getIFrameWindow, q as cn, r as getElementAtPoint, s as getXPathForElement, B as Button, t as glassyBoxClassName, v as AgentAvailabilityError, w as useAgentAvailability, x as Panel, y as PanelHeader, z as PanelContent, D as PanelFooter, E as cva, F as getLicenseDataFromStorage, G as Glassy, I as AppStateProvider } from "index-B0k7lxog.js";
 import * as ReactDOM from "react-dom";
 import { createPortal, flushSync } from "react-dom";
 import config from "@stagewise/toolbar/config";
@@ -68,7 +68,7 @@ const useAgentState = () => useContext(agentStateContext), ChatContext = createC
   },
   isSending: !1
 }), ChatStateProvider = ({ children }) => {
-  const [chatInput, setChatInput] = useState(""), [isPromptCreationMode, setIsPromptCreationMode] = useState(!1), [isSending, setIsSending] = useState(!1), [domContextElements, setDomContextElements] = useState([]), [selectedDocs, setSelectedDocs] = useState([]), [selectedBlocks, setSelectedBlocks] = useState([]), [selectedThemes, setSelectedThemes] = useState([]), { minimized } = useAppState(), { plugins } = usePlugins(), { sendMessage: sendAgentMessage } = useAgentMessaging(), { isChatOpen } = usePanels(), agentState = useAgentState(), { promptAction, cliVersion } = useAppState(), startPromptCreation = useCallback(() => {
+  const [chatInput, setChatInput] = useState(""), [isPromptCreationMode, setIsPromptCreationMode] = useState(!1), [isSending, setIsSending] = useState(!1), [domContextElements, setDomContextElements] = useState([]), [selectedDocs, setSelectedDocs] = useState([]), [selectedBlocks, setSelectedBlocks] = useState([]), [selectedThemes, setSelectedThemes] = useState([]), { minimized } = useAppState(), { plugins } = usePlugins(), { sendMessage: sendAgentMessage } = useAgentMessaging(), { isChatOpen } = usePanels(), agentState = useAgentState(), { promptAction } = useAppState(), startPromptCreation = useCallback(() => {
     setIsPromptCreationMode(!0), plugins.forEach((plugin) => {
       var _a;
       (_a = plugin.onPromptingStart) == null || _a.call(plugin);
@@ -143,16 +143,15 @@ const useAgentState = () => useContext(agentStateContext), ChatContext = createC
         ), selectedDocsInfo = selectedDocs.map(
           (doc) => getSelectedDocInfo(doc)
         ), selectedBlocksInfo = await Promise.all(
-          selectedBlocks.map((block) => getSelectedBlockInfo(block, cliVersion))
+          selectedBlocks.map((block) => getSelectedBlockInfo(block))
         ), selectedThemesInfo = selectedThemes.map(
-          (theme) => getSelectedThemeInfo(theme, cliVersion)
+          (theme) => getSelectedThemeInfo(theme)
         ), metadata = collectUserMessageMetadata(
           selectedElementsInfo,
           selectedDocsInfo,
           selectedBlocksInfo,
           selectedThemesInfo,
-          promptAction,
-          cliVersion
+          promptAction
         ), baseUserMessage = {
           id: generateId(),
           createdAt: /* @__PURE__ */ new Date(),
@@ -7120,23 +7119,34 @@ const performLocalSearch = (blocks, query) => {
   return new Fuse(blocks, fuseLocalSearchOptions).search(query).map((result) => result.item);
 }, filterProBlocks = (blocks) => blocks.filter((block) => {
   var _a;
-  return ((_a = block.meta) == null ? void 0 : _a.isPro) === !1;
-}), fetchBlocksFromAPI = async (isValidated) => {
-  const fetchBlocksUrl = "https://shadcnstudio.com/r/blocks/registry.json?is_extension=true";
+  return ((_a = block.meta) == null ? void 0 : _a.isBasic) === !1;
+}), checkPlan$1 = async (licenseKey, email) => {
+  const fetchPlanUrl = `http://localhost:3000/api/ide-extension/plan-variant?email=${email}&license_key=${licenseKey}`;
+  try {
+    return (await (await fetch(fetchPlanUrl, { method: "GET" })).json()).plan_variant;
+  } catch (error2) {
+    console.error("Failed to fetch plan variant:", error2);
+  }
+}, fetchBlocksFromAPI = async (isValidated, licenseKey, email) => {
+  const fetchBlocksUrl = "https://shadcnstudio.com/r/blocks/radix-vega/registry.json?is_extension=true";
   try {
     const response = await fetch(fetchBlocksUrl, { method: "GET" });
     if (!response.ok)
       return console.warn(
         `Failed to fetch blocks: ${response.status} ${response.statusText}`
       ), [];
-    let blocks = (await response.json()).items;
-    return isValidated || (blocks = filterProBlocks(blocks)), blocks;
+    let blocks = (await response.json()).items, planVariant = "basic";
+    return isValidated && licenseKey && email && (planVariant = await checkPlan$1(licenseKey, email)), console.log("User plan variant:", planVariant, "isValidated:", isValidated), (!isValidated || isValidated && planVariant === "basic") && (blocks = filterProBlocks(blocks)), blocks;
   } catch (error2) {
     return console.warn("Error fetching blocks:", error2), [];
   }
-}, fetchAndSearchBlocks = async (query, isValidated) => {
+}, fetchAndSearchBlocks = async (query, isValidated, licenseKey, email) => {
   try {
-    const searchResults = await fetchBlocksFromAPI(isValidated);
+    const searchResults = await fetchBlocksFromAPI(
+      isValidated,
+      licenseKey,
+      email
+    );
     if (searchResults.length === 0)
       return [];
     const fuseBlocksOptions = {
@@ -7161,7 +7171,9 @@ const performLocalSearch = (blocks, query) => {
     isValidated,
     debounceMs = 200,
     minScore = 0.35,
-    maxResults = 20
+    maxResults = 20,
+    licenseKey,
+    email
   } = options2, [searchResults, setSearchResults] = useState([]), [isSearching, setIsSearching] = useState(!1), [searchError, setSearchError] = useState(null);
   return useEffect(() => {
     if (!(searchQuery != null && searchQuery.trim())) {
@@ -7173,7 +7185,12 @@ const performLocalSearch = (blocks, query) => {
       try {
         const query = searchQuery.trim().toLowerCase(), localResults = performLocalSearch(localBlocks, query);
         setSearchResults(localResults);
-        const apiResults = await fetchAndSearchBlocks(query, isValidated);
+        const apiResults = await fetchAndSearchBlocks(
+          query,
+          isValidated,
+          licenseKey,
+          email
+        );
         if (apiResults.length > 0) {
           const combinedResults = [...apiResults, ...localResults], uniqueResults = removeDuplicateBlocks(combinedResults);
           setSearchResults(uniqueResults);
@@ -7186,7 +7203,16 @@ const performLocalSearch = (blocks, query) => {
       }
     }, debounceMs);
     return () => clearTimeout(searchTimeout);
-  }, [searchQuery, localBlocks, isValidated, debounceMs, minScore, maxResults]), {
+  }, [
+    searchQuery,
+    localBlocks,
+    isValidated,
+    debounceMs,
+    minScore,
+    maxResults,
+    licenseKey,
+    email
+  ]), {
     searchResults,
     isSearching,
     searchError
@@ -7227,18 +7253,26 @@ function useLicenseKey() {
   }, [loadLicenseKey]);
   const validateLicenseKey = useCallback(
     async (licensekey, email) => {
-      if (!licensekey || typeof licensekey != "string" || !email || typeof email != "string" || !email.includes("@"))
-        return !1;
-      const trimmedLicenseKey = licensekey.trim(), validateUrl = `https://shadcnstudio.com/api/validate-user?email=${email.trim()}&license_key=${trimmedLicenseKey}`, response = await fetch(validateUrl, {
+      if (!licensekey || typeof licensekey != "string")
+        return { isValid: !1, message: "Invalid license key" };
+      if (!email || typeof email != "string" || !email.includes("@"))
+        return { isValid: !1, message: "Invalid email" };
+      const trimmedLicenseKey = licensekey.trim(), validateUrl = `http://localhost:3000/api/ide-extension/validate-user?email=${email.trim()}&license_key=${trimmedLicenseKey}`, response = await fetch(validateUrl, {
         method: "GET",
         headers: {
           "Content-Type": "application/json"
         }
-      });
-      return response.ok ? !0 : (console.error(
+      }), data = await response.json();
+      return response.ok ? {
+        isValid: !0,
+        message: data.message || "License key validation successful"
+      } : (console.error(
         "License key validation request failed:",
         response.status
-      ), !1);
+      ), {
+        isValid: !1,
+        message: data.message
+      });
     },
     []
   ), saveLicenseKey = useCallback(
@@ -7294,7 +7328,7 @@ function useLicenseKey() {
     }
   }, []), refreshLicenseValidation = useCallback(async () => {
     if (!licenseState.licenseKey)
-      return !1;
+      return { isValid: !1, message: "No license key found" };
     try {
       const isValid = await validateLicenseKey(
         licenseState.licenseKey,
@@ -7320,7 +7354,10 @@ function useLicenseKey() {
         removeLicenseKey();
       return isValid;
     } catch (error2) {
-      return console.error("Failed to refresh license validation:", error2), !1;
+      return console.error("Failed to refresh license validation:", error2), {
+        isValid: !1,
+        message: "Failed to refresh license validation"
+      };
     }
   }, [licenseState.licenseKey, validateLicenseKey, removeLicenseKey]), needsRevalidation = useCallback(() => licenseState.lastValidated ? (Date.now() - licenseState.lastValidated.getTime()) / (1e3 * 60 * 60) > 24 : !0, [licenseState.lastValidated]);
   return {
@@ -7365,7 +7402,7 @@ const RECENT_BLOCKS_KEY = "shadcnstudio-toolbar-blocks-recent", getRecentBlocks
     onCloseBlocks,
     onReady
   }, ref) => {
-    const { isValidated } = useLicenseKey(), [recentBlocks, setRecentBlocks] = useState([]);
+    const { isValidated, licenseKey, email } = useLicenseKey(), [recentBlocks, setRecentBlocks] = useState([]);
     useEffect(() => {
       setRecentBlocks(getRecentBlocks());
     }, []);
@@ -7379,7 +7416,7 @@ const RECENT_BLOCKS_KEY = "shadcnstudio-toolbar-blocks-recent", getRecentBlocks
     }, [recentBlocks]), { searchResults, isSearching, searchError } = useBlockSearch(
       searchQuery || "",
       localBlocks,
-      { isValidated, debounceMs: 500 }
+      { isValidated, debounceMs: 500, licenseKey, email }
     ), filteredBlocks = useMemo(() => searchQuery != null && searchQuery.trim() ? searchResults : localBlocks, [searchQuery, searchResults, localBlocks]), [activeIndex, setActiveIndex] = useState(-1), [startIndex, setStartIndex] = useState(0), [isFocused, setIsFocused] = useState(!1), containerRef = useRef(null), visibleBlocks = useMemo(() => filteredBlocks.slice(startIndex, startIndex + 3), [filteredBlocks, startIndex]);
     useEffect(() => {
       activeIndex >= filteredBlocks.length && filteredBlocks.length > 0 ? (setActiveIndex(filteredBlocks.length - 1), setStartIndex(Math.max(0, filteredBlocks.length - 3))) : filteredBlocks.length === 0 && activeIndex !== -1 && (setActiveIndex(-1), setStartIndex(0));
@@ -8007,25 +8044,44 @@ const fetchGenericThemesFromAPI = async () => {
   const fetchThemesUrl = "https://shadcnstudio.com/r/themes/registry.json?is_extension=true";
   try {
     const response = await fetch(fetchThemesUrl, { method: "GET" });
-    return response.ok ? (await response.json()).items : (console.warn(
+    return response.ok ? (await response.json()).items.map((theme) => ({
+      ...theme,
+      type: "generic"
+    })) : (console.warn(
       `Failed to fetch themes: ${response.status} ${response.statusText}`
     ), []);
   } catch (error2) {
     return console.warn("Error fetching themes:", error2), [];
   }
+}, checkPlan = async () => {
+  const { email, licenseKey } = getLicenseDataFromStorage();
+  if (!email || !licenseKey)
+    return "basic";
+  const fetchPlanUrl = `http://localhost:3000/api/ide-extension/plan-variant?email=${email}&license_key=${licenseKey}`;
+  try {
+    return (await (await fetch(fetchPlanUrl, { method: "GET" })).json()).plan_variant;
+  } catch (error2) {
+    console.error("Failed to fetch plan variant:", error2);
+  }
 }, fetchUserThemesFromAPI = async () => {
-  const { email, licenseKey } = getLicenseDataFromStorage(), fetchThemesUrl = `https://shadcnstudio.com/api/user-themes?email=${email}&license_key=${licenseKey}&is_extension=true`;
+  const { email, licenseKey } = getLicenseDataFromStorage(), fetchThemesUrl = `http://localhost:3000/api/ide-extension/user-themes?email=${email}&license_key=${licenseKey}&is_extension=true`;
   try {
     const response = await fetch(fetchThemesUrl, { method: "GET" });
-    return response.ok ? (await response.json()).themes : (console.warn(
+    return response.ok ? (await response.json()).themes.map((theme) => ({
+      ...theme,
+      type: "user"
+    })) : (console.warn(
       `Failed to fetch themes: ${response.status} ${response.statusText}`
     ), []);
   } catch (error2) {
     return console.warn("Error fetching themes:", error2), [];
   }
 }, fetchThemesFromAPI = async () => {
-  const userThemes = await fetchUserThemesFromAPI(), genericThemes = await fetchGenericThemesFromAPI();
-  return userThemes.length > 0 ? [...userThemes, ...genericThemes] : genericThemes;
+  const userThemes = await fetchUserThemesFromAPI(), genericThemes = await fetchGenericThemesFromAPI(), userPlan = await checkPlan(), allThemes = userThemes.length > 0 ? [...userThemes, ...genericThemes] : genericThemes;
+  return userPlan === "basic" ? allThemes.filter((theme) => {
+    var _a;
+    return ((_a = theme.meta) == null ? void 0 : _a.isPro) !== !0;
+  }) : allThemes;
 }, fetchAndSearchThemes = async (searchQuery) => {
   try {
     const searchResults = await fetchThemesFromAPI();
@@ -10534,9 +10590,9 @@ function LicenseKeyDialog({
     }
     setIsValidating(!0), setValidationError(null);
     try {
-      console.log("In handle Save function Validating license key:", inputKey), console.log("In handle save function With email:", email), await validateLicenseKey(inputKey.trim(), email.trim()) ? (await saveLicenseKey(inputKey.trim(), email.trim()), setInputKey(""), onClose()) : setValidationError(
-        "Invalid license key or Email. Please check and try again."
-      );
+      console.log("In handle Save function Validating license key:", inputKey), console.log("In handle save function With email:", email);
+      const isValid = await validateLicenseKey(inputKey.trim(), email.trim());
+      console.log("License key validation result:", isValid), isValid.isValid ? (await saveLicenseKey(inputKey.trim(), email.trim()), setInputKey(""), onClose()) : (console.log("License key validation failed:", isValid), setValidationError(isValid.message || "Invalid license key or Email"));
     } catch (error2) {
       setValidationError(
         error2 instanceof Error ? error2.message : "Failed to validate license key"
@@ -10889,47 +10945,6 @@ function PromptActionSelector() {
     ] })
   ] }) });
 }
-function ShadcnCliVersionSelector() {
-  const { cliVersion, setCliVersion } = useAppState();
-  return /* @__PURE__ */ jsx("div", { children: /* @__PURE__ */ jsxs("div", { className: "flex items-center justify-between", children: [
-    /* @__PURE__ */ jsxs("div", { className: "flex flex-col", children: [
-      /* @__PURE__ */ jsx("div", { className: "font-medium text-foreground text-sm", children: "Shadcn CLI version" }),
-      /* @__PURE__ */ jsx("div", { className: "text-muted-foreground text-xs", children: "Set your preferred shadcn CLI version." })
-    ] }),
-    /* @__PURE__ */ jsxs(DropdownMenu, { children: [
-      /* @__PURE__ */ jsx(DropdownMenuButton, { children: /* @__PURE__ */ jsxs(
-        Button,
-        {
-          size: "sm",
-          variant: "ghost",
-          className: "flex items-center gap-2 rounded-md text-xs",
-          children: [
-            cliVersion === "v2" && "v2",
-            cliVersion === "v3" && "v3"
-          ]
-        }
-      ) }),
-      /* @__PURE__ */ jsxs(DropdownMenuContent, { children: [
-        /* @__PURE__ */ jsx(
-          DropdownMenuButtonItem,
-          {
-            onClick: () => setCliVersion("v2"),
-            className: "flex items-center gap-2",
-            children: "v2"
-          }
-        ),
-        /* @__PURE__ */ jsx(
-          DropdownMenuButtonItem,
-          {
-            onClick: () => setCliVersion("v3"),
-            className: "flex items-center gap-2",
-            children: "v3"
-          }
-        )
-      ] })
-    ] })
-  ] }) });
-}
 function SettingsPanel() {
   const { closeSettings } = usePanels();
   return /* @__PURE__ */ jsxs(Panel, { children: [
@@ -10959,8 +10974,6 @@ function SettingsPanel() {
       /* @__PURE__ */ jsx("hr", { className: "-mx-4 text-zinc-500/15 dark:text-zinc-500/30" }),
       /* @__PURE__ */ jsx(CompactThemeToggle, {}),
       /* @__PURE__ */ jsx("hr", { className: "-mx-4 text-zinc-500/15 dark:text-zinc-500/30" }),
-      /* @__PURE__ */ jsx(ShadcnCliVersionSelector, {}),
-      /* @__PURE__ */ jsx("hr", { className: "-mx-4 text-zinc-500/15 dark:text-zinc-500/30" }),
       /* @__PURE__ */ jsxs("div", { children: [
         /* @__PURE__ */ jsxs("div", { className: "flex items-center justify-between", children: [
           /* @__PURE__ */ jsx("div", { className: "font-medium text-foreground text-sm", children: "Toggle chat" }),

package/dist/toolbar-app/plugin-sdk.js

@@ -1,5 +1,5 @@
-import { q as cn, E as cva, b as usePlugins } from "index-CcQR8qhj.js";
-import { B, x, z, D, y } from "index-CcQR8qhj.js";
+import { q as cn, E as cva, b as usePlugins } from "index-B0k7lxog.js";
+import { B, x, z, D, y } from "index-B0k7lxog.js";
 import { jsx } from "react/jsx-runtime";
 const badgeVariants = cva("rounded-md p-2", {
   variants: {

package/package.json

@@ -2,7 +2,7 @@
   "name": "shadcn-studio-extension-cli",
   "module": "src/index.ts",
   "type": "module",
-  "version": "0.1.10",
+  "version": "0.2.0",
   "description": "Shadcn/Studio Extension CLI",
   "author": "stagewise GmbH",
   "license": "AGPL-3.0-only",