feat: 增加Java补全支持

DrizzleTime · DrizzleTime · commit e7e36d7e3ade · 2024-09-19T16:10:39.000+08:00
扩展了代码编辑器的功能，添加了对Java语言的自动补全支持。此改动涉及修改index.ts文件，包括导入和初始化Java补全功能。
diff --git a/src/index.ts b/src/index.ts
@@ -1,4 +1,5 @@
 import { initCppCompletion } from './cpp';
+import { initJavaCompletion } from './java';
 import * as monaco from 'monaco-editor';
 function waitForMonaco(): Promise<typeof monaco> {
     return new Promise((resolve) => {
@@ -15,6 +16,8 @@ function waitForMonaco(): Promise<typeof monaco> {
 async function init() {
     const monaco = await waitForMonaco();
     initCppCompletion(monaco);
+    initJavaCompletion(monaco);
+
 }
 
 window.addEventListener('load', init);
diff --git a/src/java/arrayList.ts b/src/java/arrayList.ts
@@ -0,0 +1,35 @@
+
+export function getArrayListSuggestions(monaco: typeof import('monaco-editor'), range: any) {
+    return [
+        {
+            label: 'add',
+            kind: monaco.languages.CompletionItemKind.Method,
+            insertText: 'add(${1:element});',
+            insertTextRules: monaco.languages.CompletionItemInsertTextRule.InsertAsSnippet,
+            documentation: 'Adds an element to the ArrayList.',
+            range: range
+        },
+        {
+            label: 'remove',
+            kind: monaco.languages.CompletionItemKind.Method,
+            insertText: 'remove(${1:index});',
+            documentation: 'Removes the element at the specified index.',
+            range: range
+        },
+        {
+            label: 'get',
+            kind: monaco.languages.CompletionItemKind.Method,
+            insertText: 'get(${1:index});',
+            documentation: 'Returns the element at the specified index.',
+            range: range
+        },
+        {
+            label: 'size',
+            kind: monaco.languages.CompletionItemKind.Method,
+            insertText: 'size();',
+            documentation: 'Returns the number of elements in the ArrayList.',
+            range: range
+        }
+        // 其他 ArrayList 方法
+    ];
+}
diff --git a/src/java/getGeneralSuggestions.ts b/src/java/getGeneralSuggestions.ts
@@ -0,0 +1,63 @@
+import type { languages } from 'monaco-editor';
+
+export function getGeneralSuggestions(monaco: typeof import('monaco-editor'), range: any) {
+    return [
+        {
+            label: 'System.out.println',
+            kind: monaco.languages.CompletionItemKind.Snippet,
+            insertText: 'System.out.println(${1:message});',
+            insertTextRules: monaco.languages.CompletionItemInsertTextRule.InsertAsSnippet,
+            documentation: 'Prints a message to the standard output',
+            range: range
+        },
+        {
+            label: 'for (int i = 0; i < n; i++)',
+            kind: monaco.languages.CompletionItemKind.Snippet,
+            insertText: 'for (int ${1:i} = 0; ${1:i} < ${2:n}; ${1:i}++) {\n\t$0\n}',
+            insertTextRules: monaco.languages.CompletionItemInsertTextRule.InsertAsSnippet,
+            documentation: 'Basic for loop',
+            range: range
+        },
+        {
+            label: 'while',
+            kind: monaco.languages.CompletionItemKind.Snippet,
+            insertText: 'while (${1:condition}) {\n\t$0\n}',
+            insertTextRules: monaco.languages.CompletionItemInsertTextRule.InsertAsSnippet,
+            documentation: 'While loop',
+            range: range
+        },
+        {
+            label: 'if',
+            kind: monaco.languages.CompletionItemKind.Snippet,
+            insertText: 'if (${1:condition}) {\n\t$0\n}',
+            insertTextRules: monaco.languages.CompletionItemInsertTextRule.InsertAsSnippet,
+            documentation: 'If statement',
+            range: range
+        },
+        {
+            label: 'ArrayList',
+            kind: monaco.languages.CompletionItemKind.Snippet,
+            insertText: 'ArrayList<${1:type}> ${2:list} = new ArrayList<>();',
+            insertTextRules: monaco.languages.CompletionItemInsertTextRule.InsertAsSnippet,
+            documentation: 'Creates an ArrayList',
+            range: range
+        },
+        {
+            label: 'HashMap',
+            kind: monaco.languages.CompletionItemKind.Snippet,
+            insertText: 'HashMap<${1:K}, ${2:V}> ${3:map} = new HashMap<>();',
+            insertTextRules: monaco.languages.CompletionItemInsertTextRule.InsertAsSnippet,
+            documentation: 'Creates a HashMap',
+            range: range
+        },
+        {
+            label: 'HashSet',
+            kind: monaco.languages.CompletionItemKind.Snippet,
+            insertText: 'HashSet<${1:type}> ${2:set} = new HashSet<>();',
+            insertTextRules: monaco.languages.CompletionItemInsertTextRule.InsertAsSnippet,
+            documentation: 'Creates a HashSet',
+            range: range
+        },
+        // 可以继续添加更多通用提示
+    ];
+}
diff --git a/src/java/hashMap.ts b/src/java/hashMap.ts
@@ -0,0 +1,42 @@
+
+export function getHashMapSuggestions(monaco: typeof import('monaco-editor'), range: any) {
+    return [
+        {
+            label: 'put',
+            kind: monaco.languages.CompletionItemKind.Method,
+            insertText: 'put(${1:key}, ${2:value});',
+            insertTextRules: monaco.languages.CompletionItemInsertTextRule.InsertAsSnippet,
+            documentation: 'Associates the specified value with the specified key in this map.',
+            range: range
+        },
+        {
+            label: 'get',
+            kind: monaco.languages.CompletionItemKind.Method,
+            insertText: 'get(${1:key});',
+            documentation: 'Returns the value to which the specified key is mapped.',
+            range: range
+        },
+        {
+            label: 'remove',
+            kind: monaco.languages.CompletionItemKind.Method,
+            insertText: 'remove(${1:key});',
+            documentation: 'Removes the mapping for the specified key from this map if present.',
+            range: range
+        },
+        {
+            label: 'containsKey',
+            kind: monaco.languages.CompletionItemKind.Method,
+            insertText: 'containsKey(${1:key});',
+            documentation: 'Returns true if this map contains a mapping for the specified key.',
+            range: range
+        },
+        {
+            label: 'keySet',
+            kind: monaco.languages.CompletionItemKind.Method,
+            insertText: 'keySet();',
+            documentation: 'Returns a Set view of the keys contained in this map.',
+            range: range
+        },
+        // 可以继续添加更多方法提示
+    ];
+}
diff --git a/src/java/index.ts b/src/java/index.ts
@@ -0,0 +1,71 @@
+import type { languages } from 'monaco-editor';
+import { getArrayListSuggestions } from './arrayList';
+import { getHashMapSuggestions } from './hashMap';
+import { getJavaRange, extractJavaVariables } from './utils';
+import { getGeneralSuggestions } from './getGeneralSuggestions';
+
+export function initJavaCompletion(monaco: typeof import('monaco-editor')) {
+    monaco.languages.register({ id: 'java' });
+
+    monaco.languages.registerCompletionItemProvider('java', {
+        provideCompletionItems: (model, position) => {
+            const word = model.getWordUntilPosition(position);
+            const range = getJavaRange(position, word); // 获取光标位置的范围
+
+            const code = model.getValue(); // 获取整个代码
+            const variables = extractJavaVariables(code, position); // 提取变量
+
+            // 提供变量补全提示
+            const variableSuggestions: { label: string; kind: languages.CompletionItemKind; insertText: string; documentation: string; range: any; }[] = [];
+            variables.forEach(({ type }, name) => {
+                variableSuggestions.push({
+                    label: name,
+                    kind: monaco.languages.CompletionItemKind.Variable,
+                    insertText: name,
+                    documentation: `Variable of type ${type}`,
+                    range: range
+                });
+            });
+
+            // 获取当前行的内容并判断是否有“.”
+            const lineContent = model.getLineContent(position.lineNumber);
+            const textBeforeCursor = lineContent.substring(0, position.column - 1).trim();
+            const lastDotIndex = textBeforeCursor.lastIndexOf('.');
+            let methodSuggestions: { label: string; kind: languages.CompletionItemKind; insertText: string; documentation: string; range: any; }[] = [];
+            let isDotAfterVariable = false;
+
+            // 检查是否有“.”和变量，并获取该变量类型的相应方法提示
+            if (lastDotIndex !== -1) {
+                const varName = textBeforeCursor.substring(0, lastDotIndex).trim();
+
+                if (variables.has(varName)) {
+                    const varInfo = variables.get(varName);
+                    isDotAfterVariable = true;
+                    // 根据变量类型返回相应的成员方法提示
+                    switch (varInfo.type) {
+                        case 'ArrayList':
+                            methodSuggestions = getArrayListSuggestions(monaco, range);
+                            break;
+                        case 'HashMap':
+                            methodSuggestions = getHashMapSuggestions(monaco, range);
+                            break;
+                        default:
+                            break;
+                    }
+                }
+            }
+
+            // 如果没有"."或者未识别出变量，返回通用的建议和变量提示
+            if (!isDotAfterVariable) {
+                const generalSuggestions = getGeneralSuggestions(monaco, range);
+                return {
+                    suggestions: [...generalSuggestions, ...variableSuggestions]
+                };
+            } else {
+                return {
+                    suggestions: [...methodSuggestions]
+                };
+            }
+        }
+    });
+}
diff --git a/src/java/utils.ts b/src/java/utils.ts
@@ -0,0 +1,50 @@
+export function extractJavaVariables(code: string, position: any) {
+    // 改进正则表达式，匹配泛型声明 (如 ArrayList<int> list)
+    const variableRegex = /\b(?:ArrayList|List|Set|Map|Queue|Stack|Deque|LinkedList|HashMap|HashSet|TreeMap|TreeSet|PriorityQueue)\s*<[^>]+>\s+(\w+)\s*=/g;
+    const basicTypeRegex = /\b(?:int|float|double|char|boolean|String|byte|short|long)\s+(\w+)\s*;/g; // 基本类型匹配
+    const funcParamRegex = /\(([^)]*)\)/g; // 提取函数参数列表
+
+    const variables = new Map<string, { type: string, kind: string }>();
+
+    // 仅解析光标之前的代码
+    const lines = code.split('\n').slice(0, position.lineNumber - 1);
+    let currentLine = code.split('\n')[position.lineNumber - 1].substring(0, position.column - 1);
+    lines.push(currentLine);
+    const codeBeforeCursor = lines.join('\n');
+
+    // 提取泛型类型变量（如 ArrayList<int> list = new ArrayList<>()）
+    let match;
+    while ((match = variableRegex.exec(codeBeforeCursor)) !== null) {
+        const varName = match[1]; // 获取变量名
+        variables.set(varName, { type: "ArrayList", kind: "variable" }); // 设置为泛型类型
+    }
+
+    // 提取基本类型变量
+    while ((match = basicTypeRegex.exec(codeBeforeCursor)) !== null) {
+        const varName = match[1]; // 获取变量名
+        variables.set(varName, { type: "basic", kind: "variable" }); // 设置为基本类型
+    }
+
+    // 提取函数形参
+    const funcMatch = funcParamRegex.exec(codeBeforeCursor);
+    if (funcMatch) {
+        const params = funcMatch[1].split(',');
+        params.forEach(param => {
+            const parts = param.trim().split(/\s+/);
+            const paramName = parts[parts.length - 1]; // 获取形参名
+            variables.set(paramName, { type: "parameter", kind: "parameter" }); // 设置为形参
+        });
+    }
+
+    return variables;
+}
+
+
+export function getJavaRange(position: any, word: any) {
+    return {
+        startLineNumber: position.lineNumber,
+        endLineNumber: position.lineNumber,
+        startColumn: word.startColumn,
+        endColumn: word.endColumn
+    };
+}
