zenstack 1.0.0 → 1.0.2

package/language-server/validator/zmodel-validator.js

@@ -0,0 +1,83 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ZModelValidator = exports.ZModelValidationRegistry = void 0;
+const langium_1 = require("langium");
+const attribute_validator_1 = __importDefault(require("./attribute-validator"));
+const datamodel_validator_1 = __importDefault(require("./datamodel-validator"));
+const datasource_validator_1 = __importDefault(require("./datasource-validator"));
+const enum_validator_1 = __importDefault(require("./enum-validator"));
+const expression_validator_1 = __importDefault(require("./expression-validator"));
+const function_decl_validator_1 = __importDefault(require("./function-decl-validator"));
+const function_invocation_validator_1 = __importDefault(require("./function-invocation-validator"));
+const schema_validator_1 = __importDefault(require("./schema-validator"));
+/**
+ * Registry for validation checks.
+ */
+class ZModelValidationRegistry extends langium_1.ValidationRegistry {
+    constructor(services) {
+        super(services);
+        const validator = services.validation.ZModelValidator;
+        const checks = {
+            Model: validator.checkModel,
+            DataSource: validator.checkDataSource,
+            DataModel: validator.checkDataModel,
+            Enum: validator.checkEnum,
+            Attribute: validator.checkAttribute,
+            Expression: validator.checkExpression,
+            InvocationExpr: validator.checkFunctionInvocation,
+            FunctionDecl: validator.checkFunctionDecl,
+        };
+        this.register(checks, validator);
+    }
+}
+exports.ZModelValidationRegistry = ZModelValidationRegistry;
+/**
+ * Implementation of custom validations.
+ */
+class ZModelValidator {
+    constructor(services) {
+        this.services = services;
+    }
+    shouldCheck(node) {
+        let doc;
+        let currNode = node;
+        while (currNode) {
+            if (currNode.$document) {
+                doc = currNode.$document;
+                break;
+            }
+            currNode = currNode.$container;
+        }
+        return (doc === null || doc === void 0 ? void 0 : doc.parseResult.lexerErrors.length) === 0 && (doc === null || doc === void 0 ? void 0 : doc.parseResult.parserErrors.length) === 0;
+    }
+    checkModel(node, accept) {
+        this.shouldCheck(node) &&
+            new schema_validator_1.default(this.services.shared.workspace.LangiumDocuments).validate(node, accept);
+    }
+    checkDataSource(node, accept) {
+        this.shouldCheck(node) && new datasource_validator_1.default().validate(node, accept);
+    }
+    checkDataModel(node, accept) {
+        this.shouldCheck(node) && new datamodel_validator_1.default().validate(node, accept);
+    }
+    checkEnum(node, accept) {
+        this.shouldCheck(node) && new enum_validator_1.default().validate(node, accept);
+    }
+    checkAttribute(node, accept) {
+        this.shouldCheck(node) && new attribute_validator_1.default().validate(node, accept);
+    }
+    checkExpression(node, accept) {
+        this.shouldCheck(node) && new expression_validator_1.default().validate(node, accept);
+    }
+    checkFunctionInvocation(node, accept) {
+        this.shouldCheck(node) && new function_invocation_validator_1.default().validate(node, accept);
+    }
+    checkFunctionDecl(node, accept) {
+        this.shouldCheck(node) && new function_decl_validator_1.default().validate(node, accept);
+    }
+}
+exports.ZModelValidator = ZModelValidator;
+//# sourceMappingURL=zmodel-validator.js.map

package/language-server/validator/zmodel-validator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zmodel-validator.js","sourceRoot":"","sources":["../../../src/language-server/validator/zmodel-validator.ts"],"names":[],"mappings":";;;;;;AAWA,qCAA6G;AAE7G,gFAAuD;AACvD,gFAAuD;AACvD,kFAAyD;AACzD,sEAA6C;AAC7C,kFAAyD;AACzD,wFAA8D;AAC9D,oGAA0E;AAC1E,0EAAiD;AAEjD;;GAEG;AACH,MAAa,wBAAyB,SAAQ,4BAAkB;IAC5D,YAAY,QAAwB;QAChC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChB,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,eAAe,CAAC;QACtD,MAAM,MAAM,GAAoC;YAC5C,KAAK,EAAE,SAAS,CAAC,UAAU;YAC3B,UAAU,EAAE,SAAS,CAAC,eAAe;YACrC,SAAS,EAAE,SAAS,CAAC,cAAc;YACnC,IAAI,EAAE,SAAS,CAAC,SAAS;YACzB,SAAS,EAAE,SAAS,CAAC,cAAc;YACnC,UAAU,EAAE,SAAS,CAAC,eAAe;YACrC,cAAc,EAAE,SAAS,CAAC,uBAAuB;YACjD,YAAY,EAAE,SAAS,CAAC,iBAAiB;SAC5C,CAAC;QACF,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACrC,CAAC;CACJ;AAhBD,4DAgBC;AAED;;GAEG;AACH,MAAa,eAAe;IACxB,YAA+B,QAAwB;QAAxB,aAAQ,GAAR,QAAQ,CAAgB;IAAG,CAAC;IACnD,WAAW,CAAC,IAAa;QAC7B,IAAI,GAAgC,CAAC;QACrC,IAAI,QAAQ,GAAwB,IAAI,CAAC;QACzC,OAAO,QAAQ,EAAE;YACb,IAAI,QAAQ,CAAC,SAAS,EAAE;gBACpB,GAAG,GAAG,QAAQ,CAAC,SAAS,CAAC;gBACzB,MAAM;aACT;YACD,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC;SAClC;QAED,OAAO,CAAA,GAAG,aAAH,GAAG,uBAAH,GAAG,CAAE,WAAW,CAAC,WAAW,CAAC,MAAM,MAAK,CAAC,IAAI,CAAA,GAAG,aAAH,GAAG,uBAAH,GAAG,CAAE,WAAW,CAAC,YAAY,CAAC,MAAM,MAAK,CAAC,CAAC;IACnG,CAAC;IAED,UAAU,CAAC,IAAW,EAAE,MAA0B;QAC9C,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;YAClB,IAAI,0BAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACpG,CAAC;IAED,eAAe,CAAC,IAAgB,EAAE,MAA0B;QACxD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,IAAI,8BAAmB,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,cAAc,CAAC,IAAe,EAAE,MAA0B;QACtD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,IAAI,6BAAkB,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC9E,CAAC;IAED,SAAS,CAAC,IAAU,EAAE,MAA0B;QAC5C,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,IAAI,wBAAa,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACzE,CAAC;IAED,cAAc,CAAC,IAAe,EAAE,MAA0B;QACtD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,IAAI,6BAAkB,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC9E,CAAC;IAED,eAAe,CAAC,IAAgB,EAAE,MAA0B;QACxD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,IAAI,8BAAmB,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC/E,CAAC;IAED,uBAAuB,CAAC,IAAoB,EAAE,MAA0B;QACpE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,IAAI,uCAA2B,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACvF,CAAC;IAED,iBAAiB,CAAC,IAAkB,EAAE,MAA0B;QAC5D,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,IAAI,iCAAqB,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACjF,CAAC;CACJ;AAhDD,0CAgDC"}

package/language-server/zmodel-code-action.d.ts

@@ -0,0 +1,15 @@
+import { AstReflection, CodeActionProvider, IndexManager, LangiumDocument, LangiumDocuments, LangiumServices, MaybePromise } from 'langium';
+import { CodeAction, CodeActionParams, Command } from 'vscode-languageserver';
+import { ZModelFormatter } from './zmodel-formatter';
+export declare class ZModelCodeActionProvider implements CodeActionProvider {
+    protected readonly reflection: AstReflection;
+    protected readonly indexManager: IndexManager;
+    protected readonly formatter: ZModelFormatter;
+    protected readonly documents: LangiumDocuments;
+    constructor(services: LangiumServices);
+    getCodeActions(document: LangiumDocument, params: CodeActionParams): MaybePromise<Array<Command | CodeAction> | undefined>;
+    private createCodeActions;
+    private fixMissingOppositeRelation;
+    private lowerCaseFirstLetter;
+    private upperCaseFirstLetter;
+}

package/language-server/zmodel-code-action.js

@@ -0,0 +1,118 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ZModelCodeActionProvider = void 0;
+const ast_1 = require("@zenstackhq/language/ast");
+const langium_1 = require("langium");
+const vscode_languageserver_1 = require("vscode-languageserver");
+const constants_1 = require("./constants");
+class ZModelCodeActionProvider {
+    constructor(services) {
+        this.reflection = services.shared.AstReflection;
+        this.indexManager = services.shared.workspace.IndexManager;
+        this.formatter = services.lsp.Formatter;
+        this.documents = services.shared.workspace.LangiumDocuments;
+    }
+    getCodeActions(document, params) {
+        const result = [];
+        const acceptor = (ca) => ca && result.push(ca);
+        for (const diagnostic of params.context.diagnostics) {
+            this.createCodeActions(diagnostic, document, acceptor);
+        }
+        return result;
+    }
+    createCodeActions(diagnostic, document, accept) {
+        switch (diagnostic.code) {
+            case constants_1.IssueCodes.MissingOppositeRelation:
+                accept(this.fixMissingOppositeRelation(diagnostic, document));
+        }
+        return undefined;
+    }
+    fixMissingOppositeRelation(diagnostic, document) {
+        var _a, _b;
+        const data = diagnostic.data;
+        const rootCst = data.relationFieldDocUri == document.textDocument.uri
+            ? document.parseResult.value
+            : (_a = this.documents.all.find((doc) => doc.textDocument.uri === data.relationFieldDocUri)) === null || _a === void 0 ? void 0 : _a.parseResult.value;
+        if (rootCst) {
+            const fieldModel = rootCst;
+            const fieldAstNode = (_b = fieldModel.declarations.find((x) => (0, ast_1.isDataModel)(x) && x.name === data.relationDataModelName)) === null || _b === void 0 ? void 0 : _b.fields.find((x) => x.name === data.relationFieldName);
+            if (!fieldAstNode)
+                return undefined;
+            // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+            const oppositeModel = fieldAstNode.type.reference.ref;
+            const lastField = oppositeModel.fields[oppositeModel.fields.length - 1];
+            const currentModel = document.parseResult.value;
+            const container = currentModel.declarations.find((decl) => decl.name === data.dataModelName && (0, ast_1.isDataModel)(decl));
+            if (container && container.$cstNode) {
+                // indent
+                let indent = '\t';
+                const formatOptions = this.formatter.getFormatOptions();
+                if (formatOptions === null || formatOptions === void 0 ? void 0 : formatOptions.insertSpaces) {
+                    indent = ' '.repeat(formatOptions.tabSize);
+                }
+                indent = indent.repeat(this.formatter.getIndent());
+                let newText = '';
+                if (fieldAstNode.type.array) {
+                    //post Post[]
+                    const idField = container.$resolvedFields.find((f) => f.attributes.find((attr) => { var _a; return ((_a = attr.decl.ref) === null || _a === void 0 ? void 0 : _a.name) === '@id'; }));
+                    // if no id field, we can't generate reference
+                    if (!idField) {
+                        return undefined;
+                    }
+                    const typeName = container.name;
+                    const fieldName = this.lowerCaseFirstLetter(typeName);
+                    // might already exist
+                    let referenceField = '';
+                    const idFieldName = idField.name;
+                    const referenceIdFieldName = fieldName + this.upperCaseFirstLetter(idFieldName);
+                    if (!oppositeModel.$resolvedFields.find((f) => f.name === referenceIdFieldName)) {
+                        referenceField = '\n' + indent + `${referenceIdFieldName} ${idField.type.type}`;
+                    }
+                    newText =
+                        '\n' +
+                            indent +
+                            `${fieldName} ${typeName} @relation(fields: [${referenceIdFieldName}], references: [${idFieldName}])` +
+                            referenceField;
+                }
+                else {
+                    // user User @relation(fields: [userAbc], references: [id])
+                    const typeName = container.name;
+                    const fieldName = this.lowerCaseFirstLetter(typeName);
+                    newText = '\n' + indent + `${fieldName} ${typeName}[]`;
+                }
+                // the opposite model might be in the imported file
+                const targetDocument = (0, langium_1.getDocument)(oppositeModel);
+                return {
+                    title: `Add opposite relation fields on ${oppositeModel.name}`,
+                    kind: vscode_languageserver_1.CodeActionKind.QuickFix,
+                    diagnostics: [diagnostic],
+                    isPreferred: false,
+                    edit: {
+                        changes: {
+                            [targetDocument.textDocument.uri]: [
+                                {
+                                    range: {
+                                        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                                        start: lastField.$cstNode.range.end,
+                                        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                                        end: lastField.$cstNode.range.end,
+                                    },
+                                    newText,
+                                },
+                            ],
+                        },
+                    },
+                };
+            }
+        }
+        return undefined;
+    }
+    lowerCaseFirstLetter(str) {
+        return str.charAt(0).toLowerCase() + str.slice(1);
+    }
+    upperCaseFirstLetter(str) {
+        return str.charAt(0).toUpperCase() + str.slice(1);
+    }
+}
+exports.ZModelCodeActionProvider = ZModelCodeActionProvider;
+//# sourceMappingURL=zmodel-code-action.js.map

package/language-server/zmodel-code-action.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zmodel-code-action.js","sourceRoot":"","sources":["../../src/language-server/zmodel-code-action.ts"],"names":[],"mappings":";;;AAAA,kDAAyF;AACzF,qCASiB;AAEjB,iEAA0G;AAC1G,2CAAyC;AAIzC,MAAa,wBAAwB;IAMjC,YAAY,QAAyB;QACjC,IAAI,CAAC,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC;QAChD,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,SAA4B,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC;IAChE,CAAC;IAED,cAAc,CACV,QAAyB,EACzB,MAAwB;QAExB,MAAM,MAAM,GAAiB,EAAE,CAAC;QAChC,MAAM,QAAQ,GAAG,CAAC,EAA0B,EAAE,EAAE,CAAC,EAAE,IAAI,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvE,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE;YACjD,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;SAC1D;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAEO,iBAAiB,CACrB,UAAsB,EACtB,QAAyB,EACzB,MAA4C;QAE5C,QAAQ,UAAU,CAAC,IAAI,EAAE;YACrB,KAAK,sBAAU,CAAC,uBAAuB;gBACnC,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC;SACrE;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,0BAA0B,CAAC,UAAsB,EAAE,QAAyB;;QAChF,MAAM,IAAI,GAAG,UAAU,CAAC,IAAmC,CAAC;QAE5D,MAAM,OAAO,GACT,IAAI,CAAC,mBAAmB,IAAI,QAAQ,CAAC,YAAY,CAAC,GAAG;YACjD,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK;YAC5B,CAAC,CAAC,MAAA,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,KAAK,IAAI,CAAC,mBAAmB,CAAC,0CAAE,WAAW,CAC3F,KAAK,CAAC;QAErB,IAAI,OAAO,EAAE;YACT,MAAM,UAAU,GAAG,OAAgB,CAAC;YACpC,MAAM,YAAY,GAAG,MACjB,UAAU,CAAC,YAAY,CAAC,IAAI,CACxB,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,iBAAW,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,qBAAqB,CAErE,0CAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,iBAAiB,CAAmB,CAAC;YAE3E,IAAI,CAAC,YAAY;gBAAE,OAAO,SAAS,CAAC;YAEpC,oEAAoE;YACpE,MAAM,aAAa,GAAG,YAAY,CAAC,IAAI,CAAC,SAAU,CAAC,GAAiB,CAAC;YAErE,MAAM,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAExE,MAAM,YAAY,GAAG,QAAQ,CAAC,WAAW,CAAC,KAAc,CAAC;YAEzD,MAAM,SAAS,GAAG,YAAY,CAAC,YAAY,CAAC,IAAI,CAC5C,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,aAAa,IAAI,IAAA,iBAAW,EAAC,IAAI,CAAC,CACrD,CAAC;YAEf,IAAI,SAAS,IAAI,SAAS,CAAC,QAAQ,EAAE;gBACjC,SAAS;gBACT,IAAI,MAAM,GAAG,IAAI,CAAC;gBAClB,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC;gBACxD,IAAI,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,YAAY,EAAE;oBAC7B,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;iBAC9C;gBACD,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;gBAEnD,IAAI,OAAO,GAAG,EAAE,CAAC;gBACjB,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE;oBACzB,aAAa;oBACb,MAAM,OAAO,GAAG,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACjD,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,WAAC,OAAA,CAAA,MAAA,IAAI,CAAC,IAAI,CAAC,GAAG,0CAAE,IAAI,MAAK,KAAK,CAAA,EAAA,CAAC,CAC3C,CAAC;oBAEpB,8CAA8C;oBAC9C,IAAI,CAAC,OAAO,EAAE;wBACV,OAAO,SAAS,CAAC;qBACpB;oBAED,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC;oBAChC,MAAM,SAAS,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;oBAEtD,sBAAsB;oBACtB,IAAI,cAAc,GAAG,EAAE,CAAC;oBAExB,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;oBACjC,MAAM,oBAAoB,GAAG,SAAS,GAAG,IAAI,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;oBAEhF,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAAC,EAAE;wBAC7E,cAAc,GAAG,IAAI,GAAG,MAAM,GAAG,GAAG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;qBACnF;oBAED,OAAO;wBACH,IAAI;4BACJ,MAAM;4BACN,GAAG,SAAS,IAAI,QAAQ,uBAAuB,oBAAoB,mBAAmB,WAAW,IAAI;4BACrG,cAAc,CAAC;iBACtB;qBAAM;oBACH,2DAA2D;oBAC3D,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC;oBAChC,MAAM,SAAS,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;oBACtD,OAAO,GAAG,IAAI,GAAG,MAAM,GAAG,GAAG,SAAS,IAAI,QAAQ,IAAI,CAAC;iBAC1D;gBAED,mDAAmD;gBACnD,MAAM,cAAc,GAAG,IAAA,qBAAW,EAAC,aAAa,CAAC,CAAC;gBAElD,OAAO;oBACH,KAAK,EAAE,mCAAmC,aAAa,CAAC,IAAI,EAAE;oBAC9D,IAAI,EAAE,sCAAc,CAAC,QAAQ;oBAC7B,WAAW,EAAE,CAAC,UAAU,CAAC;oBACzB,WAAW,EAAE,KAAK;oBAClB,IAAI,EAAE;wBACF,OAAO,EAAE;4BACL,CAAC,cAAc,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE;gCAC/B;oCACI,KAAK,EAAE;wCACH,oEAAoE;wCACpE,KAAK,EAAE,SAAS,CAAC,QAAS,CAAC,KAAK,CAAC,GAAG;wCACpC,oEAAoE;wCACpE,GAAG,EAAE,SAAS,CAAC,QAAS,CAAC,KAAK,CAAC,GAAG;qCACrC;oCACD,OAAO;iCACV;6BACJ;yBACJ;qBACJ;iBACJ,CAAC;aACL;SACJ;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,oBAAoB,CAAC,GAAW;QACpC,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC;IAEO,oBAAoB,CAAC,GAAW;QACpC,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC;CACJ;AAvJD,4DAuJC"}

package/language-server/zmodel-definition.d.ts

@@ -0,0 +1,7 @@
+import { DefaultDefinitionProvider, LangiumDocuments, LangiumServices, LeafCstNode, MaybePromise } from 'langium';
+import { DefinitionParams, LocationLink } from 'vscode-languageserver';
+export declare class ZModelDefinitionProvider extends DefaultDefinitionProvider {
+    protected documents: LangiumDocuments;
+    constructor(services: LangiumServices);
+    protected collectLocationLinks(sourceCstNode: LeafCstNode, _params: DefinitionParams): MaybePromise<LocationLink[] | undefined>;
+}

package/language-server/zmodel-definition.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ZModelDefinitionProvider = void 0;
+const langium_1 = require("langium");
+const vscode_languageserver_1 = require("vscode-languageserver");
+const ast_utils_1 = require("../utils/ast-utils");
+const ast_1 = require("@zenstackhq/language/ast");
+class ZModelDefinitionProvider extends langium_1.DefaultDefinitionProvider {
+    constructor(services) {
+        super(services);
+        this.documents = services.shared.workspace.LangiumDocuments;
+    }
+    collectLocationLinks(sourceCstNode, _params) {
+        var _a, _b, _c, _d;
+        if ((0, ast_1.isModelImport)(sourceCstNode.element)) {
+            const importedModel = (0, ast_utils_1.resolveImport)(this.documents, sourceCstNode.element);
+            if (importedModel === null || importedModel === void 0 ? void 0 : importedModel.$document) {
+                const targetObject = importedModel;
+                const selectionRange = (_b = (_a = this.nameProvider.getNameNode(targetObject)) === null || _a === void 0 ? void 0 : _a.range) !== null && _b !== void 0 ? _b : vscode_languageserver_1.Range.create(0, 0, 0, 0);
+                const previewRange = (_d = (_c = targetObject.$cstNode) === null || _c === void 0 ? void 0 : _c.range) !== null && _d !== void 0 ? _d : vscode_languageserver_1.Range.create(0, 0, 0, 0);
+                return [
+                    vscode_languageserver_1.LocationLink.create(importedModel.$document.uri.toString(), previewRange, selectionRange, sourceCstNode.range),
+                ];
+            }
+            return undefined;
+        }
+        return super.collectLocationLinks(sourceCstNode, _params);
+    }
+}
+exports.ZModelDefinitionProvider = ZModelDefinitionProvider;
+//# sourceMappingURL=zmodel-definition.js.map

package/language-server/zmodel-definition.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zmodel-definition.js","sourceRoot":"","sources":["../../src/language-server/zmodel-definition.ts"],"names":[],"mappings":";;;AAAA,qCAAkH;AAClH,iEAA8E;AAC9E,kDAAmD;AACnD,kDAAyD;AAEzD,MAAa,wBAAyB,SAAQ,mCAAyB;IAGnE,YAAY,QAAyB;QACjC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChB,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC;IAChE,CAAC;IACkB,oBAAoB,CACnC,aAA0B,EAC1B,OAAyB;;QAEzB,IAAI,IAAA,mBAAa,EAAC,aAAa,CAAC,OAAO,CAAC,EAAE;YACtC,MAAM,aAAa,GAAG,IAAA,yBAAa,EAAC,IAAI,CAAC,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;YAC3E,IAAI,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,SAAS,EAAE;gBAC1B,MAAM,YAAY,GAAG,aAAa,CAAC;gBACnC,MAAM,cAAc,GAAG,MAAA,MAAA,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,YAAY,CAAC,0CAAE,KAAK,mCAAI,6BAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;gBACtG,MAAM,YAAY,GAAG,MAAA,MAAA,YAAY,CAAC,QAAQ,0CAAE,KAAK,mCAAI,6BAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC9E,OAAO;oBACH,oCAAY,CAAC,MAAM,CACf,aAAa,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,EACtC,YAAY,EACZ,cAAc,EACd,aAAa,CAAC,KAAK,CACtB;iBACJ,CAAC;aACL;YACD,OAAO,SAAS,CAAC;SACpB;QACD,OAAO,KAAK,CAAC,oBAAoB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;CACJ;AA9BD,4DA8BC"}

package/language-server/zmodel-formatter.d.ts

@@ -0,0 +1,9 @@
+import { AbstractFormatter, AstNode, LangiumDocument } from 'langium';
+import { FormattingOptions, Range, TextEdit } from 'vscode-languageserver';
+export declare class ZModelFormatter extends AbstractFormatter {
+    private formatOptions?;
+    protected format(node: AstNode): void;
+    protected doDocumentFormat(document: LangiumDocument<AstNode>, options: FormattingOptions, range?: Range | undefined): TextEdit[];
+    getFormatOptions(): FormattingOptions | undefined;
+    getIndent(): number;
+}

package/language-server/zmodel-formatter.js

@@ -0,0 +1,76 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ZModelFormatter = void 0;
+const langium_1 = require("langium");
+const ast = __importStar(require("@zenstackhq/language/ast"));
+class ZModelFormatter extends langium_1.AbstractFormatter {
+    format(node) {
+        const formatter = this.getNodeFormatter(node);
+        if (ast.isDataModelField(node)) {
+            formatter.property('type').prepend(langium_1.Formatting.oneSpace());
+            if (node.attributes.length > 0) {
+                formatter.properties('attributes').prepend(langium_1.Formatting.oneSpace());
+            }
+        }
+        else if (ast.isDataModelFieldAttribute(node)) {
+            formatter.keyword('(').surround(langium_1.Formatting.noSpace());
+            formatter.keyword(')').prepend(langium_1.Formatting.noSpace());
+            formatter.keyword(',').append(langium_1.Formatting.oneSpace());
+            if (node.args.length > 1) {
+                formatter.nodes(...node.args.slice(1)).prepend(langium_1.Formatting.oneSpace());
+            }
+        }
+        else if (ast.isAttributeArg(node)) {
+            formatter.keyword(':').prepend(langium_1.Formatting.noSpace());
+            formatter.keyword(':').append(langium_1.Formatting.oneSpace());
+        }
+        else if (ast.isAbstractDeclaration(node)) {
+            const bracesOpen = formatter.keyword('{');
+            const bracesClose = formatter.keyword('}');
+            // allow extra blank lines between declarations
+            formatter.interior(bracesOpen, bracesClose).prepend(langium_1.Formatting.indent({ allowMore: true }));
+            bracesOpen.prepend(langium_1.Formatting.oneSpace());
+            bracesClose.prepend(langium_1.Formatting.newLine());
+        }
+        else if (ast.isModel(node)) {
+            const model = node;
+            const nodes = formatter.nodes(...model.declarations);
+            nodes.prepend(langium_1.Formatting.noIndent());
+        }
+    }
+    doDocumentFormat(document, options, range) {
+        this.formatOptions = options;
+        return super.doDocumentFormat(document, options, range);
+    }
+    getFormatOptions() {
+        return this.formatOptions;
+    }
+    getIndent() {
+        return 1;
+    }
+}
+exports.ZModelFormatter = ZModelFormatter;
+//# sourceMappingURL=zmodel-formatter.js.map

package/language-server/zmodel-formatter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zmodel-formatter.js","sourceRoot":"","sources":["../../src/language-server/zmodel-formatter.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,qCAAkF;AAElF,8DAAgD;AAGhD,MAAa,eAAgB,SAAQ,2BAAiB;IAExC,MAAM,CAAC,IAAa;QAC1B,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE;YAC5B,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,oBAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC1D,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC5B,SAAS,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,oBAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;aACrE;SACJ;aAAM,IAAI,GAAG,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE;YAC5C,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,oBAAU,CAAC,OAAO,EAAE,CAAC,CAAC;YACtD,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,oBAAU,CAAC,OAAO,EAAE,CAAC,CAAC;YACrD,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,oBAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;YACrD,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE;gBACtB,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,oBAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;aACzE;SACJ;aAAM,IAAI,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE;YACjC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,oBAAU,CAAC,OAAO,EAAE,CAAC,CAAC;YACrD,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,oBAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;SACxD;aAAM,IAAI,GAAG,CAAC,qBAAqB,CAAC,IAAI,CAAC,EAAE;YACxC,MAAM,UAAU,GAAG,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAC1C,MAAM,WAAW,GAAG,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAC3C,+CAA+C;YAC/C,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC,OAAO,CAAC,oBAAU,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAC5F,UAAU,CAAC,OAAO,CAAC,oBAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC1C,WAAW,CAAC,OAAO,CAAC,oBAAU,CAAC,OAAO,EAAE,CAAC,CAAC;SAC7C;aAAM,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;YAC1B,MAAM,KAAK,GAAG,IAAiB,CAAC;YAChC,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,KAAK,CAAC,YAAY,CAAC,CAAC;YACrD,KAAK,CAAC,OAAO,CAAC,oBAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;SACxC;IACL,CAAC;IAEkB,gBAAgB,CAC/B,QAAkC,EAClC,OAA0B,EAC1B,KAAyB;QAEzB,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;QAC7B,OAAO,KAAK,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;IAC5D,CAAC;IAEM,gBAAgB;QACnB,OAAO,IAAI,CAAC,aAAa,CAAC;IAC9B,CAAC;IAEM,SAAS;QACZ,OAAO,CAAC,CAAC;IACb,CAAC;CACJ;AAjDD,0CAiDC"}

package/language-server/zmodel-linker.d.ts

@@ -0,0 +1,32 @@
+import { DefaultLinker, LangiumDocument, LangiumServices } from 'langium';
+import { CancellationToken } from 'vscode-jsonrpc';
+/**
+ * Langium linker implementation which links references and resolves expression types
+ */
+export declare class ZModelLinker extends DefaultLinker {
+    private readonly descriptions;
+    constructor(services: LangiumServices);
+    link(document: LangiumDocument, cancelToken?: CancellationToken): Promise<void>;
+    private linkReference;
+    private resolveFromScopeProviders;
+    private resolve;
+    private resolveBinary;
+    private resolveUnary;
+    private resolveObject;
+    private resolveReference;
+    private resolveArray;
+    private resolveInvocation;
+    private getContainingDataModel;
+    private resolveLiteral;
+    private resolveMemberAccess;
+    private resolveCollectionPredicate;
+    private resolveThis;
+    private resolveNull;
+    private resolveAttributeArg;
+    private findAttrParamForArg;
+    private resolveDataModel;
+    private resolveDataModelField;
+    private resolveDefault;
+    private resolveToDeclaredType;
+    private resolveToBuiltinTypeOrDecl;
+}