UNPKG

@dbml/core

Version:
141 lines (140 loc) 7.75 kB
"use strict"; function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); } Object.defineProperty(exports, "__esModule", { value: true }); exports["default"] = void 0; var _lodash = require("lodash"); var _element = _interopRequireDefault(require("./element")); var _utils = require("./utils"); function _interopRequireDefault(e) { return e && e.__esModule ? e : { "default": e }; } function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; } function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; } function _defineProperty(e, r, t) { return (r = _toPropertyKey(r)) in e ? Object.defineProperty(e, r, { value: t, enumerable: !0, configurable: !0, writable: !0 }) : e[r] = t, e; } function _classCallCheck(a, n) { if (!(a instanceof n)) throw new TypeError("Cannot call a class as a function"); } function _defineProperties(e, r) { for (var t = 0; t < r.length; t++) { var o = r[t]; o.enumerable = o.enumerable || !1, o.configurable = !0, "value" in o && (o.writable = !0), Object.defineProperty(e, _toPropertyKey(o.key), o); } } function _createClass(e, r, t) { return r && _defineProperties(e.prototype, r), t && _defineProperties(e, t), Object.defineProperty(e, "prototype", { writable: !1 }), e; } function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : i + ""; } function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); } function _callSuper(t, o, e) { return o = _getPrototypeOf(o), _possibleConstructorReturn(t, _isNativeReflectConstruct() ? Reflect.construct(o, e || [], _getPrototypeOf(t).constructor) : o.apply(t, e)); } function _possibleConstructorReturn(t, e) { if (e && ("object" == _typeof(e) || "function" == typeof e)) return e; if (void 0 !== e) throw new TypeError("Derived constructors may only return object or undefined"); return _assertThisInitialized(t); } function _assertThisInitialized(e) { if (void 0 === e) throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); return e; } function _isNativeReflectConstruct() { try { var t = !Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); } catch (t) {} return (_isNativeReflectConstruct = function _isNativeReflectConstruct() { return !!t; })(); } function _getPrototypeOf(t) { return _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf.bind() : function (t) { return t.__proto__ || Object.getPrototypeOf(t); }, _getPrototypeOf(t); } function _inherits(t, e) { if ("function" != typeof e && null !== e) throw new TypeError("Super expression must either be null or a function"); t.prototype = Object.create(e && e.prototype, { constructor: { value: t, writable: !0, configurable: !0 } }), Object.defineProperty(t, "prototype", { writable: !1 }), e && _setPrototypeOf(t, e); } function _setPrototypeOf(t, e) { return _setPrototypeOf = Object.setPrototypeOf ? Object.setPrototypeOf.bind() : function (t, e) { return t.__proto__ = e, t; }, _setPrototypeOf(t, e); } var TableGroup = /*#__PURE__*/function (_Element) { function TableGroup(_ref) { var _this; var name = _ref.name, token = _ref.token, _ref$tables = _ref.tables, tables = _ref$tables === void 0 ? [] : _ref$tables, _ref$schema = _ref.schema, schema = _ref$schema === void 0 ? {} : _ref$schema, note = _ref.note, color = _ref.color, _ref$noteToken = _ref.noteToken, noteToken = _ref$noteToken === void 0 ? null : _ref$noteToken; _classCallCheck(this, TableGroup); _this = _callSuper(this, TableGroup, [token]); _this.name = name; _this.tables = []; _this.schema = schema; _this.dbState = _this.schema.dbState; _this.note = note ? (0, _lodash.get)(note, 'value', note) : null; _this.noteToken = note ? (0, _lodash.get)(note, 'token', noteToken) : null; _this.color = color; _this.generateId(); _this.processTables(tables); return _this; } _inherits(TableGroup, _Element); return _createClass(TableGroup, [{ key: "generateId", value: function generateId() { this.id = this.dbState.generateId('tableGroupId'); } }, { key: "processTables", value: function processTables(rawTables) { var _this2 = this; rawTables.forEach(function (rawTable) { var table = _this2.schema.database.findTable(rawTable.schemaName, rawTable.name); if (!table) { _this2.error("Table ".concat(rawTable.schemaName ? "\"".concat(rawTable.schemaName, "\".") : '').concat(rawTable.name, " don't exist")); } _this2.pushTable(table); }); } }, { key: "pushTable", value: function pushTable(table) { this.checkTable(table); this.tables.push(table); table.group = this; } }, { key: "checkTable", value: function checkTable(table) { if (this.tables.some(function (t) { return t.id === table.id; })) { this.error("Table ".concat((0, _utils.shouldPrintSchema)(table.schema) ? "\"".concat(table.schema.name, "\".") : '', ".").concat(table.name, " is already in the group")); } if (table.group) { this.error("Table ".concat((0, _utils.shouldPrintSchema)(table.schema) ? "\"".concat(table.schema.name, "\".") : '', ".").concat(table.name, " is already in group ").concat((0, _utils.shouldPrintSchema)(table.group.schema) ? "\"".concat(table.group.schema.name, "\".") : '').concat(table.group.name)); } } }, { key: "export", value: function _export() { return _objectSpread(_objectSpread({}, this.shallowExport()), this.exportChild()); } }, { key: "exportChild", value: function exportChild() { return { tables: this.tables.map(function (t) { return { tableName: t.name, schemaName: t.schema.name }; }) }; } }, { key: "exportChildIds", value: function exportChildIds() { return { tableIds: this.tables.map(function (t) { return t.id; }) }; } }, { key: "exportParentIds", value: function exportParentIds() { return { schemaId: this.schema.id }; } }, { key: "shallowExport", value: function shallowExport() { return { name: this.name, note: this.note, color: this.color }; } }, { key: "normalize", value: function normalize(model) { model.tableGroups[this.id] = _objectSpread(_objectSpread(_objectSpread({ id: this.id }, this.shallowExport()), this.exportChildIds()), this.exportParentIds()); } }]); }(_element["default"]); var _default = exports["default"] = TableGroup;