Merge pull request #40 from zjffun/8-cooperation-with-profiles

zjffun · web-flow · commit 608954ec2a24 · 2025-07-20T19:18:07.000+08:00
feat: manage user snippets of profiles
diff --git a/src/core/getUserSnippetsFilesInfo.ts b/src/core/getUserSnippetsFilesInfo.ts
@@ -1,15 +1,58 @@
 import * as vscode from "vscode";
-import { getUserFolderUri } from "../share";
+import {
+  getProfilesFolderUri,
+  getStorageJsonUri,
+  getUserFolderUri,
+} from "../share";
+import logger from "../utils/logger";
 
-export default async () => {
+interface IUserDataProfile {
+  location: string;
+  name: string;
+}
+
+interface IsnippetsFileInfo {
+  fileName: string;
+  uri: vscode.Uri;
+}
+
+async function getProfileUserSnippetsFilesInfo() {
+  try {
+    const storageJsonUri = getStorageJsonUri();
+    const snippetsTextDoc =
+      await vscode.workspace.openTextDocument(storageJsonUri);
+    const storageJsonContent = snippetsTextDoc.getText();
+    const storageJson = JSON.parse(storageJsonContent);
+    const userDataProfiles: IUserDataProfile[] =
+      storageJson?.userDataProfiles || [];
+    const userSnippetsFolders = userDataProfiles
+      .filter((profile) => profile.location)
+      .map((profile) => {
+        return {
+          name: profile.name,
+          uri: vscode.Uri.joinPath(
+            getProfilesFolderUri(),
+            profile.location,
+            "snippets",
+          ),
+        };
+      });
+
+    return userSnippetsFolders;
+  } catch (error: any) {
+    logger.error(error?.message);
+    return [];
+  }
+}
+
+async function getUserSnippetsFilesInfoByFolderUri(uri: vscode.Uri) {
   const userSnippetsFileUris = [];
 
-  const userFolderUri = getUserFolderUri();
   let userSnippetFiles: [string, vscode.FileType][] = [];
   try {
-    userSnippetFiles = await vscode.workspace.fs.readDirectory(userFolderUri);
+    userSnippetFiles = await vscode.workspace.fs.readDirectory(uri);
   } catch (error) {
-    // havn no .vscode folder, do noting
+    // havn no folder, do noting
   }
 
   for (const [fileName, fileType] of userSnippetFiles) {
@@ -18,11 +61,45 @@ export default async () => {
       fileType & vscode.FileType.File &&
       (fileName.endsWith(".code-snippets") || fileName.endsWith(".json"))
     ) {
-      const snippetUri = vscode.Uri.joinPath(userFolderUri, fileName);
+      const snippetUri = vscode.Uri.joinPath(uri, fileName);
 
       userSnippetsFileUris.push({ fileName, uri: snippetUri });
     }
   }
 
   return userSnippetsFileUris;
+}
+
+export default async () => {
+  const userFolderUri = getUserFolderUri();
+  const profileUserSnippetsFolders = await getProfileUserSnippetsFilesInfo();
+
+  if (!profileUserSnippetsFolders?.length) {
+    const userSnippetsFileInfos =
+      await getUserSnippetsFilesInfoByFolderUri(userFolderUri);
+
+    return userSnippetsFileInfos;
+  }
+
+  const userSnippetsFolders = [
+    {
+      name: "default",
+      uri: userFolderUri,
+    },
+    ...profileUserSnippetsFolders,
+  ];
+
+  const userSnippetsFileInfos: IsnippetsFileInfo[] = [];
+  for (const { uri, name } of userSnippetsFolders) {
+    const rawFileInfos = await getUserSnippetsFilesInfoByFolderUri(uri);
+
+    const fileInfos = rawFileInfos.map(({ fileName, uri }) => ({
+      fileName: `${fileName} (${name})`,
+      uri,
+    }));
+
+    userSnippetsFileInfos.push(...fileInfos);
+  }
+
+  return userSnippetsFileInfos;
 };
diff --git a/src/share.ts b/src/share.ts
@@ -17,6 +17,15 @@ export let context: vscode.ExtensionContext;
 export const setContext = (c: vscode.ExtensionContext) => (context = c);
 
 export const getUserFolderUri = () =>
-  vscode.Uri.joinPath(context.globalStorageUri, "../../snippets");
+  vscode.Uri.joinPath(context?.globalStorageUri, "../../snippets");
+
+export const getStorageJsonUri = () =>
+  vscode.Uri.joinPath(
+    context?.globalStorageUri,
+    "../../globalStorage/storage.json",
+  );
+
+export const getProfilesFolderUri = () =>
+  vscode.Uri.joinPath(context?.globalStorageUri, "../../profiles");
 
 export const isBrowser = vscode.env.appHost !== "desktop";
