luaut-language-server 1.1.0 → 1.1.1

package/dist/cli.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 import {
   startServer
-} from "./chunk-HK7PKBDB.js";
+} from "./chunk-OQWE7ISD.js";
 
 // src/cli.ts
 startServer();

package/dist/index.cjs

@@ -889,6 +889,8 @@ var IDENTIFIER_CHAR = /[A-Za-z0-9_]/;
 function completion(analyzer, document, position) {
   const inImport = importCompletion(analyzer, document, position);
   if (inImport) return inImport;
+  const inString = stringCompletion(analyzer, document, position);
+  if (inString) return inString;
   const source = document.getText();
   const offset = document.offsetAt(position);
   let start = offset;
@@ -929,6 +931,43 @@ function completion(analyzer, document, position) {
   }
   return valueItems(first.analysis, at);
 }
+function stringCompletion(analyzer, document, position) {
+  const analysis = analyzer.get(document);
+  const path = pathAt(analysis.program, position, false);
+  const literal = [...path].reverse().find((n) => n.type === "StringLiteral");
+  if (!literal) return void 0;
+  const expected = analysis.types.expectedTypeOf.get(literal);
+  const values = stringLiterals(expected, analysis.types.aliases);
+  if (!values.length) return [];
+  const line = literal.line.start - 1;
+  const range = literal.line.start === literal.line.end ? {
+    start: { line, character: literal.column.start },
+    end: { line, character: literal.column.end - 2 }
+  } : void 0;
+  return values.map((value) => ({
+    label: value,
+    kind: import_vscode_languageserver4.CompletionItemKind.Constant,
+    ...range ? { textEdit: { range, newText: value } } : {}
+  }));
+}
+function stringLiterals(type, aliases, seen = /* @__PURE__ */ new Set()) {
+  if (!type || seen.has(type)) return [];
+  seen.add(type);
+  switch (type.kind) {
+    case "literal":
+      return typeof type.value === "string" ? [type.value] : [];
+    case "union":
+      return [...new Set(type.types.flatMap((t) => stringLiterals(t, aliases, seen)))];
+    case "genericRef": {
+      const alias = aliases.get(type.name);
+      return alias ? stringLiterals(alias, aliases, seen) : [];
+    }
+    case "typeParam":
+      return stringLiterals(type.constraint, aliases, seen);
+    default:
+      return [];
+  }
+}
 function memberOperator(source, wordStart) {
   const ch = source[wordStart - 1];
   if (ch === ":") return source[wordStart - 2] === ":" ? void 0 : ":";
@@ -1203,7 +1242,7 @@ var TOKEN_TYPES = [
   "method",
   "keyword"
 ];
-var TOKEN_MODIFIERS = ["declaration", "readonly", "defaultLibrary"];
+var TOKEN_MODIFIERS = ["declaration", "readonly", "defaultLibrary", "control"];
 var semanticTokensLegend = {
   tokenTypes: [...TOKEN_TYPES],
   tokenModifiers: [...TOKEN_MODIFIERS]
@@ -1218,8 +1257,10 @@ var SOFT_KEYWORDS = /* @__PURE__ */ new Set([
   "is",
   "asserts",
   "satisfies",
-  "typeof"
+  "typeof",
+  "default"
 ]);
+var CONTROL_KEYWORDS = /* @__PURE__ */ new Set(["default"]);
 var PRIMITIVES3 = /* @__PURE__ */ new Set(["any", "unknown", "never", "nil", "boolean", "number", "string", "thread", "buffer"]);
 function semanticTokens(analysis) {
   const entries = /* @__PURE__ */ new Map();
@@ -1245,12 +1286,8 @@ function semanticTokens(analysis) {
   walk2(analysis.program);
   for (const token of tokens) {
     const value = token.value;
-    if (typeof value !== "string") continue;
-    if (token.type === "Keyword" && value !== "true" && value !== "false" && value !== "nil") {
-      add(token, value.length, "keyword");
-    } else if (token.type === "Identifier" && SOFT_KEYWORDS.has(value)) {
-      add(token, value.length, "keyword");
-    }
+    if (token.type !== "Identifier" || typeof value !== "string" || !SOFT_KEYWORDS.has(value)) continue;
+    add(token, value.length, "keyword", CONTROL_KEYWORDS.has(value) ? ["control"] : []);
   }
   return { data: encode([...entries.values()]) };
 }