feat: extend CloudPlan type and update TOC handling for premium plan

- Added 'premium' option to the CloudPlan type for enhanced plan support.
- Updated getTidbCloudFilesFromTocs function to include premium in the TOC type determination and file mapping logic.
- Enhanced the handling of TOC nodes to accommodate the new premium plan, ensuring accurate categorization of documentation files.

Author: shhdgit

gatsby/cloud-plan.ts

@@ -1,6 +1,7 @@
 import { mdxAstToToc, TocQueryData } from "./toc";
 import { generateConfig } from "./path";
 import { extractFilesFromToc } from "./toc-filter";
+import { CloudPlan } from "shared/useCloudPlan";
 
 /**
  * Get files from different TOC types for tidbcloud
@@ -40,7 +41,12 @@ export async function getTidbCloudFilesFromTocs(
   const tocNodes = tocQuery.data!.allMdx.nodes;
   const tidbCloudTocFilesMap = new Map<
     string,
-    { dedicated: Set<string>; starter: Set<string>; essential: Set<string> }
+    {
+      dedicated: Set<string>;
+      starter: Set<string>;
+      essential: Set<string>;
+      premium: Set<string>;
+    }
   >();
 
   tocNodes.forEach((node: TocQueryData["allMdx"]["nodes"][0]) => {
@@ -55,12 +61,14 @@ export async function getTidbCloudFilesFromTocs(
 
     // Determine TOC type based on filename
     const relativePath = node.parent.relativePath;
-    let tocType: "dedicated" | "starter" | "essential" = "dedicated";
+    let tocType: CloudPlan = "dedicated";
 
     if (relativePath.includes("TOC-tidb-cloud-starter")) {
       tocType = "starter";
     } else if (relativePath.includes("TOC-tidb-cloud-essential")) {
       tocType = "essential";
+    } else if (relativePath.includes("TOC-tidb-cloud-premium")) {
+      tocType = "premium";
     }
 
     // Initialize the entry if it doesn't exist
@@ -69,6 +77,7 @@ export async function getTidbCloudFilesFromTocs(
         dedicated: new Set(),
         starter: new Set(),
         essential: new Set(),
+        premium: new Set(),
       });
     }
 

src/shared/useCloudPlan.ts

@@ -12,8 +12,9 @@ import { Repo } from "./interface";
 export const CLOUD_MODE_KEY = "plan";
 export const CLOUD_MODE_VALUE_STARTER = "starter";
 export const CLOUD_MODE_VALUE_ESSENTIAL = "essential";
+export const CLOUD_MODE_VALUE_PREMIUM = "premium";
 
-export type CloudPlan = "dedicated" | "starter" | "essential";
+export type CloudPlan = "dedicated" | "starter" | "essential" | "premium";
 
 const CloudPlanContext = createContext<{
   repo: Repo;
@@ -35,6 +36,7 @@ export const useCloudPlan = () => {
   const isTidbcloud = repo === Repo.tidbcloud;
   const [isStarter, setIsStarter] = useState<boolean>(false);
   const [isEssential, setIsEssential] = useState<boolean>(false);
+  const [isPremium, setIsPremium] = useState<boolean>(false);
   const [isClassic, setIsClassic] = useState<boolean>(true);
 
   const setCloudPlan = (cloudPlan: CloudPlan) => {
@@ -50,12 +52,14 @@ export const useCloudPlan = () => {
       _cloudPlan) as CloudPlan;
     const isStarter = isTidbcloud && cloudPlan === CLOUD_MODE_VALUE_STARTER;
     const isEssential = isTidbcloud && cloudPlan === CLOUD_MODE_VALUE_ESSENTIAL;
+    const isPremium = isTidbcloud && cloudPlan === CLOUD_MODE_VALUE_PREMIUM;
     const isClassic =
       !isTidbcloud || !cloudPlan || (!isStarter && !isEssential);
 
     _setCloudPlan(cloudPlan);
     setIsStarter(isStarter);
     setIsEssential(isEssential);
+    setIsPremium(isPremium);
     setIsClassic(isClassic);
   }, []);
 
@@ -64,6 +68,7 @@ export const useCloudPlan = () => {
     setCloudPlan,
     isStarter,
     isEssential,
+    isPremium,
     isClassic,
   };
 };