feat(new features): adds features, read pr for details

package.json

@@ -32,6 +32,39 @@
         "typecheck": "tsc --noEmit"
     },
     "contributes": {
+        "configuration": {
+            "title": "Vencord Companion configuration",
+            "properties": {
+                "vencord-companion.delModulesOnExit": {
+                    "default": false,
+                    "type": "boolean",
+                    "description": "delete extracted modules on exit"
+                },
+                "vencord-companion.moduleSavePath": {
+                    "type": "string",
+                    "description": "The path to save extracted modules, only automatically works if delModulesOnExit is false",
+                    "default": null
+                }
+            }
+        },
+        "commands": [
+            {
+                "command": "vencord-companion.diffModuleSearch",
+                "title": "Diff Module Search"
+            },
+            {
+                "command": "vencord-companion.diffModule",
+                "title": "Diff Module"
+            },
+            {
+                "command": "vencord-companion.extract",
+                "title": "Extract"
+            },
+            {
+                "command": "vencord-companion.extractSearch",
+                "title": "Extract With Search"
+            }
+        ],
         "snippets": [
             {
                 "language": "typescript",
@@ -52,6 +85,7 @@
         "onLanguage:typescriptreact"
     ],
     "dependencies": {
+        "prettier": "^3.3.3",
         "typescript": "^4.9.5",
         "ws": "^8.12.1"
     },
@@ -79,5 +113,6 @@
     "sponsor": {
         "url": "https://github.com/sponsors/Vendicated"
     },
-    "homepage": "https://github.com/Vencord/Companion"
-}
+    "homepage": "https://github.com/Vencord/Companion",
+    "packageManager": "[email protected]+sha512.9df9cf27c91715646c7d675d1c9c8e41f6fce88246f1318c1aa6a1ed1aeb3c4f032fcdf4ba63cc69c4fe6d634279176b5358727d8f2cc1e65b65f43ce2f8bfb0"
+}

src/PatchCodeLensProvider.ts

@@ -154,13 +154,24 @@ export class PatchCodeLensProvider implements CodeLensProvider {
                 if (!data) continue;
 
                 const range = new Range(document.positionAt(patch.properties.pos), document.positionAt(patch.properties.end));
-                const lens = new CodeLens(range, {
+                lenses.push(new CodeLens(range, {
+                    title: "View Module",
+                    command: "vencord-companion.extractSearch",
+                    arguments: [data.find],
+                    tooltip: "View Module"
+                }))
+                lenses.push(new CodeLens(range, {
+                    title: "Diff Module",
+                    command: "vencord-companion.diffModuleSearch",
+                    arguments: [data.find],
+                    tooltip: "Diff Module"
+                }))
+                lenses.push(new CodeLens(range, {
                     title: "Test Patch",
                     command: "vencord-companion.testPatch",
                     arguments: [data],
-                    tooltip: "Test Patch",
-                });
-                lenses.push(lens);
+                    tooltip: "Test Patch"
+                }));
             }
             return lenses;
         }

src/WebpackCodeLensProvider.ts

@@ -26,6 +26,18 @@ export const WebpackCodeLensProvider: CodeLensProvider = {
                 const args = node.arguments.map(a => tryParseStringLiteral(a) ?? tryParseRegularExpressionLiteral(a) ?? tryParseFunction(document, a));
 
                 const range = new Range(document.positionAt(node.getStart()), document.positionAt(node.getEnd()));
+                lenses.push(new CodeLens(range, {
+                    title: "View Module",
+                    command: "vencord-companion.extractFind",
+                    arguments: [{
+                        type: "extractFind",
+                        data: {
+                            args: args.filter(isNotNull),
+                            type
+                        }
+                    }],
+                    tooltip: "View Module"
+                }))
                 lenses.push(new CodeLens(range, {
                     title: "Test Find",
                     command: "vencord-companion.testFind",

src/extension.ts

@@ -1,8 +1,9 @@
-import { commands, ExtensionContext, languages, window as vscWindow } from "vscode";
+import { commands, ExtensionContext, languages, QuickPickItem, window as vscWindow, window, workspace } from "vscode";
 import { PatchCodeLensProvider } from "./PatchCodeLensProvider";
-import { FindData, PatchData } from "./shared";
+import { ExtractSendData, FindData, PatchData } from "./shared";
 import { WebpackCodeLensProvider } from "./WebpackCodeLensProvider";
-import { sendToSockets, startWebSocketServer, stopWebSocketServer } from "./webSocketServer";
+import { moduleCache, sendToSockets, startWebSocketServer, stopWebSocketServer } from "./webSocketServer";
+import generateFinds from "./findGenerator";
 
 export function activate(context: ExtensionContext) {
 	startWebSocketServer();
@@ -16,6 +17,173 @@ export function activate(context: ExtensionContext) {
 		languages.registerCodeLensProvider({ language: "typescript" }, WebpackCodeLensProvider),
 		languages.registerCodeLensProvider({ language: "typescriptreact" }, WebpackCodeLensProvider),
 
+		workspace.registerTextDocumentContentProvider("vencord-companion", {
+			async provideTextDocumentContent(uri) {
+				console.log(uri)
+				//FIXME: full uri shows up in title bar
+				const newLocal = Buffer.from(uri.path.substring(1, uri.path.lastIndexOf(".")), "base64url");
+				return newLocal.toString()
+			},
+		}),
+
+		commands.registerCommand("vencord-companion.generateFinds", async (args) => {
+			console.log(generateFinds());
+		}),
+		commands.registerCommand("vencord-companion.diffModule", async args => {
+			if (args)
+				return sendToSockets({
+					type: "search",
+					data: {
+						extractType: "id",
+						idOrSearch: args
+					}
+				})
+			const quickPick = window.createQuickPick();
+			quickPick.placeholder = "module ID";
+			quickPick.canSelectMany = false;
+			const items: QuickPickItem[] = [{ label: "", alwaysShow: false }, { label: "", kind: -1 }, ...(moduleCache.map(m => { return { label: m }; }))];
+			quickPick.items = items;
+			quickPick.onDidChangeValue(() => {
+				if (!moduleCache.includes(quickPick.value)) {
+					items[0].label = quickPick.value;
+					items[0].alwaysShow = true;
+
+				} else {
+					items[0].alwaysShow = false;
+				}
+				quickPick.items = items
+			})
+			quickPick.show()
+			quickPick.onDidAccept(async () => {
+				const modId = quickPick.value;
+				quickPick.dispose();
+				if (!modId || isNaN(+modId))
+					return vscWindow.showErrorMessage("No Module ID provided")
+				try {
+					await sendToSockets({
+						type: "search",
+						data: {
+							extractType: "id",
+							idOrSearch: +modId
+						},
+					})
+				} catch (error) {
+					vscWindow.showErrorMessage(String(error))
+				}
+			})
+
+
+		}),
+		commands.registerCommand("vencord-companion.diffModuleSearch", async (args:string) => {
+			if (args)
+				return sendToSockets({
+					type: "diff",
+					data: {
+						extractType: "search",
+						idOrSearch: args
+					}
+				})
+			const input = await window.showInputBox();
+			if (!input)
+				return window.showErrorMessage("No Input Provided")
+			try {
+				await sendToSockets({
+					type: "diff",
+					data: {
+						extractType: "search",
+						idOrSearch: input
+					}
+				})
+			} catch (error) {
+				vscWindow.showErrorMessage(String(error))
+			}
+		}),
+		commands.registerCommand("vencord-companion.extractFind", async (args: {
+			type: string,
+			data: {
+				args: string[],
+				type: string
+			}
+		}) => {
+			if (!args)
+				return vscWindow.showErrorMessage("No Data Provided");
+			await sendToSockets({
+				type: "extract",
+				data: {
+					extractType: "find",
+					findType: args.data.type,
+					findArgs: args.data.args
+				}
+			})
+		}),
+		commands.registerCommand("vencord-companion.extract", async (args: number) => {
+			if (args)
+				return sendToSockets({
+					type: "extract",
+					data: {
+						extractType: "id",
+						idOrSearch: args
+					}
+				})
+			const quickPick = window.createQuickPick();
+			quickPick.placeholder = "module ID";
+			quickPick.canSelectMany = false;
+			const items: QuickPickItem[] = [{ label: "", alwaysShow: false }, { label: "", kind: -1 }, ...(moduleCache.map(m => { return { label: m }; }))];
+			quickPick.items = items;
+			quickPick.onDidChangeValue(() => {
+				if (!moduleCache.includes(quickPick.value)) {
+					items[0].label = quickPick.value;
+					items[0].alwaysShow = true;
+
+				} else {
+					items[0].alwaysShow = false;
+				}
+				quickPick.items = items
+			})
+			quickPick.show()
+			quickPick.onDidAccept(async () => {
+				const modId = quickPick.value;
+				quickPick.dispose();
+				if (!modId || isNaN(+modId))
+					return vscWindow.showErrorMessage("No Module ID provided")
+				try {
+					await sendToSockets({
+						type: "extract",
+						data: {
+							extractType: "id",
+							idOrSearch: +modId
+						} as ExtractSendData,
+					})
+				} catch (error) {
+					vscWindow.showErrorMessage(String(error))
+				}
+			})
+
+		}),
+		commands.registerCommand("vencord-companion.extractSearch", async (args: string) => {
+			if (args)
+				return sendToSockets({
+					type: "extract",
+					data: {
+						extractType: "search",
+						idOrSearch: args
+					}
+				})
+			const input = await window.showInputBox();
+			if (!input)
+				return window.showErrorMessage("No Input Provided")
+			try {
+				await sendToSockets({
+					type: "extract",
+					data: {
+						extractType: "search",
+						idOrSearch: input
+					}
+				})
+			} catch (error) {
+				vscWindow.showErrorMessage(String(error))
+			}
+		}),
 		commands.registerCommand("vencord-companion.testPatch", async (patch: PatchData) => {
 			try {
 				await sendToSockets({ type: "testPatch", data: patch });

src/format.ts

@@ -0,0 +1,13 @@
+import { format } from "prettier/standalone"
+//@ts-ignore for some fucking reason, the d.ts file shows no exports when the mjs file clearly has them
+import estree from "prettier/plugins/estree"
+//@ts-ignore d.ts file strikes again, shows no default export. usage follows their own docs
+import babel from "prettier/plugins/babel"
+
+export default async function (text: string) {
+    return await format(text, {
+        parser: "babel",
+        plugins: [babel, estree],
+        tabWidth: 4
+    })
+}

src/shared.ts

@@ -20,6 +20,16 @@ export interface FunctionNode {
     value: string;
 }
 
+export interface ExtractSendData {
+    extractType: string
+    idOrSearch: string | number,
+}
+export interface ExtraceRecieveData {
+    moduleNumber: number,
+    type: string,
+    data: string,
+    find?: boolean
+}
 export interface PatchData {
     find: string | null;
     replacement: {