@zenstackhq/language 3.0.0-alpha.14 → 3.0.0-alpha.16

package/dist/index.js

@@ -8,45 +8,12 @@ var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require
 });
 
 // src/index.ts
-import { URI as URI3 } from "langium";
+import { isAstNode, URI as URI3 } from "langium";
 import { NodeFileSystem } from "langium/node";
 import fs3 from "fs";
 import path3 from "path";
 import { fileURLToPath as fileURLToPath2 } from "url";
 
-// src/constants.ts
-var SUPPORTED_PROVIDERS = [
-  "sqlite",
-  "postgresql"
-];
-var SCALAR_TYPES = [
-  "String",
-  "Int",
-  "Float",
-  "Decimal",
-  "BigInt",
-  "Boolean",
-  "Bytes",
-  "DateTime"
-];
-var STD_LIB_MODULE_NAME = "stdlib.zmodel";
-var PLUGIN_MODULE_NAME = "plugin.zmodel";
-var IssueCodes = /* @__PURE__ */ function(IssueCodes2) {
-  IssueCodes2["MissingOppositeRelation"] = "miss-opposite-relation";
-  return IssueCodes2;
-}({});
-var ExpressionContext = /* @__PURE__ */ function(ExpressionContext2) {
-  ExpressionContext2["DefaultValue"] = "DefaultValue";
-  ExpressionContext2["AccessPolicy"] = "AccessPolicy";
-  ExpressionContext2["ValidationRule"] = "ValidationRule";
-  ExpressionContext2["Index"] = "Index";
-  return ExpressionContext2;
-}({});
-
-// src/module.ts
-import { inject } from "langium";
-import { createDefaultModule, createDefaultSharedModule } from "langium/lsp";
-
 // src/generated/ast.ts
 import * as langium from "langium";
 var AbstractDeclaration = "AbstractDeclaration";
@@ -1045,6 +1012,39 @@ var ZModelAstReflection = class extends langium.AbstractAstReflection {
 };
 var reflection = new ZModelAstReflection();
 
+// src/constants.ts
+var SUPPORTED_PROVIDERS = [
+  "sqlite",
+  "postgresql"
+];
+var SCALAR_TYPES = [
+  "String",
+  "Int",
+  "Float",
+  "Decimal",
+  "BigInt",
+  "Boolean",
+  "Bytes",
+  "DateTime"
+];
+var STD_LIB_MODULE_NAME = "stdlib.zmodel";
+var PLUGIN_MODULE_NAME = "plugin.zmodel";
+var IssueCodes = /* @__PURE__ */ function(IssueCodes2) {
+  IssueCodes2["MissingOppositeRelation"] = "miss-opposite-relation";
+  return IssueCodes2;
+}({});
+var ExpressionContext = /* @__PURE__ */ function(ExpressionContext2) {
+  ExpressionContext2["DefaultValue"] = "DefaultValue";
+  ExpressionContext2["AccessPolicy"] = "AccessPolicy";
+  ExpressionContext2["ValidationRule"] = "ValidationRule";
+  ExpressionContext2["Index"] = "Index";
+  return ExpressionContext2;
+}({});
+
+// src/module.ts
+import { inject } from "langium";
+import { createDefaultModule, createDefaultSharedModule } from "langium/lsp";
+
 // src/generated/grammar.ts
 import { loadGrammarFromJson } from "langium";
 var loadedZModelGrammar;
@@ -5400,34 +5400,24 @@ function resolveImport(documents, imp) {
 }
 __name(resolveImport, "resolveImport");
 function resolveImportUri(imp) {
-  if (!imp.path) return void 0;
-  if (!imp.path.endsWith(".zmodel")) {
-    imp.path += ".zmodel";
-  }
-  if (!imp.path.startsWith(".") && // Respect relative paths
-  !path.isAbsolute(imp.path)) {
-    const contextPath = imp.$container.$document ? path.dirname(imp.$container.$document.uri.fsPath) : process.cwd();
-    imp.path = findNodeModulesFile(imp.path, contextPath) ?? imp.path;
+  if (!imp.path) {
+    return void 0;
   }
   const doc = AstUtils.getDocument(imp);
   const dir = path.dirname(doc.uri.fsPath);
-  return URI.file(path.resolve(dir, imp.path));
+  const importPath = imp.path.endsWith(".zmodel") ? imp.path : `${imp.path}.zmodel`;
+  return URI.file(path.resolve(dir, importPath));
 }
 __name(resolveImportUri, "resolveImportUri");
-function findNodeModulesFile(name, cwd = process.cwd()) {
-  if (!name) return void 0;
-  try {
-    const resolvedPath = __require.resolve(name, {
-      paths: [
-        cwd
-      ]
-    });
-    return resolvedPath;
-  } catch {
-    return void 0;
+function getDataModelAndTypeDefs(model, includeIgnored = false) {
+  const r = model.declarations.filter((d) => isDataModel(d) || isTypeDef(d));
+  if (includeIgnored) {
+    return r;
+  } else {
+    return r.filter((model2) => !hasAttribute(model2, "@@ignore"));
   }
 }
-__name(findNodeModulesFile, "findNodeModulesFile");
+__name(getDataModelAndTypeDefs, "getDataModelAndTypeDefs");
 function getAllDeclarationsIncludingImports(documents, model) {
   const imports = resolveTransitiveImports(documents, model);
   return model.declarations.concat(...imports.map((imp) => imp.declarations));
@@ -5528,6 +5518,22 @@ function getAllAttributes(decl, seen = /* @__PURE__ */ new Set()) {
   return attributes;
 }
 __name(getAllAttributes, "getAllAttributes");
+function getDocument(node) {
+  const rootNode = findRootNode(node);
+  const result = rootNode.$document;
+  if (!result) {
+    throw new Error("AST node has no document.");
+  }
+  return result;
+}
+__name(getDocument, "getDocument");
+function findRootNode(node) {
+  while (node.$container) {
+    node = node.$container;
+  }
+  return node;
+}
+__name(findRootNode, "findRootNode");
 
 // src/validators/attribute-application-validator.ts
 function _ts_decorate(decorators, target, key, desc) {
@@ -6884,8 +6890,8 @@ var SchemaValidator = class {
   validateImports(model, accept) {
     model.imports.forEach((imp) => {
       const importedModel = resolveImport(this.documents, imp);
-      const importPath = imp.path.endsWith(".zmodel") ? imp.path : `${imp.path}.zmodel`;
       if (!importedModel) {
+        const importPath = imp.path.endsWith(".zmodel") ? imp.path : `${imp.path}.zmodel`;
         accept("error", `Cannot find model file ${importPath}`, {
           node: imp
         });
@@ -7693,10 +7699,16 @@ async function loadDocument(fileName, pluginModelFiles = []) {
   const pluginDocs = await Promise.all(pluginModelFiles.map((file) => services.shared.workspace.LangiumDocuments.getOrCreateDocument(URI3.file(path3.resolve(file)))));
   const langiumDocuments = services.shared.workspace.LangiumDocuments;
   const document = await langiumDocuments.getOrCreateDocument(URI3.file(path3.resolve(fileName)));
+  const importedURIs = await loadImports(document, langiumDocuments);
+  const importedDocuments = [];
+  for (const uri of importedURIs) {
+    importedDocuments.push(await langiumDocuments.getOrCreateDocument(uri));
+  }
   await services.shared.workspace.DocumentBuilder.build([
     stdLib,
     ...pluginDocs,
-    document
+    document,
+    ...importedDocuments
   ], {
     validation: true
   });
@@ -7723,6 +7735,20 @@ async function loadDocument(fileName, pluginModelFiles = []) {
       warnings
     };
   }
+  const model = document.parseResult.value;
+  const imported = mergeImportsDeclarations(langiumDocuments, model);
+  imported.forEach((model2) => {
+    langiumDocuments.deleteDocument(model2.$document.uri);
+    services.shared.workspace.IndexManager.remove(model2.$document.uri);
+  });
+  const additionalErrors = validationAfterImportMerge(model);
+  if (additionalErrors.length > 0) {
+    return {
+      success: false,
+      errors: additionalErrors,
+      warnings
+    };
+  }
   return {
     success: true,
     model: document.parseResult.value,
@@ -7730,6 +7756,64 @@ async function loadDocument(fileName, pluginModelFiles = []) {
   };
 }
 __name(loadDocument, "loadDocument");
+async function loadImports(document, documents, uris = /* @__PURE__ */ new Set()) {
+  const uriString = document.uri.toString();
+  if (!uris.has(uriString)) {
+    uris.add(uriString);
+    const model = document.parseResult.value;
+    for (const imp of model.imports) {
+      const importedModel = resolveImport(documents, imp);
+      if (importedModel) {
+        const importedDoc = getDocument(importedModel);
+        await loadImports(importedDoc, documents, uris);
+      }
+    }
+  }
+  return Array.from(uris).filter((x) => uriString != x).map((e) => URI3.parse(e));
+}
+__name(loadImports, "loadImports");
+function mergeImportsDeclarations(documents, model) {
+  const importedModels = resolveTransitiveImports(documents, model);
+  const importedDeclarations = importedModels.flatMap((m) => m.declarations);
+  model.declarations.push(...importedDeclarations);
+  model.imports = [];
+  linkContentToContainer(model);
+  return importedModels;
+}
+__name(mergeImportsDeclarations, "mergeImportsDeclarations");
+function linkContentToContainer(node) {
+  for (const [name, value] of Object.entries(node)) {
+    if (!name.startsWith("$")) {
+      if (Array.isArray(value)) {
+        value.forEach((item, index) => {
+          if (isAstNode(item)) {
+            item.$container = node;
+            item.$containerProperty = name;
+            item.$containerIndex = index;
+          }
+        });
+      } else if (isAstNode(value)) {
+        value.$container = node;
+        value.$containerProperty = name;
+      }
+    }
+  }
+}
+__name(linkContentToContainer, "linkContentToContainer");
+function validationAfterImportMerge(model) {
+  const errors = [];
+  const dataSources = model.declarations.filter((d) => isDataSource(d));
+  if (dataSources.length > 1) {
+    errors.push("Validation error: Multiple datasource declarations are not allowed");
+  }
+  const decls = getDataModelAndTypeDefs(model, true);
+  const authDecls = decls.filter((d) => hasAttribute(d, "@@auth"));
+  if (authDecls.length > 1) {
+    errors.push("Validation error: Multiple `@@auth` declarations are not allowed");
+  }
+  return errors;
+}
+__name(validationAfterImportMerge, "validationAfterImportMerge");
 export {
   DocumentLoadError,
   ZModelLanguageMetaData,