"use strict";

function _typeof(obj) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (obj) { return typeof obj; } : function (obj) { return obj && "function" == typeof Symbol && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }, _typeof(obj); }
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.fetchSchemaJson = fetchSchemaJson;
var _promise = require("mysql2/promise");
var _lodash = require("lodash");
function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _iterableToArrayLimit(arr, i) { var _i = null == arr ? null : "undefined" != typeof Symbol && arr[Symbol.iterator] || arr["@@iterator"]; if (null != _i) { var _s, _e, _x, _r, _arr = [], _n = !0, _d = !1; try { if (_x = (_i = _i.call(arr)).next, 0 === i) { if (Object(_i) !== _i) return; _n = !1; } else for (; !(_n = (_s = _x.call(_i)).done) && (_arr.push(_s.value), _arr.length !== i); _n = !0); } catch (err) { _d = !0, _e = err; } finally { try { if (!_n && null != _i["return"] && (_r = _i["return"](), Object(_r) !== _r)) return; } finally { if (_d) throw _e; } } return _arr; } }
function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return exports; }; var exports = {}, Op = Object.prototype, hasOwn = Op.hasOwnProperty, defineProperty = Object.defineProperty || function (obj, key, desc) { obj[key] = desc.value; }, $Symbol = "function" == typeof Symbol ? Symbol : {}, iteratorSymbol = $Symbol.iterator || "@@iterator", asyncIteratorSymbol = $Symbol.asyncIterator || "@@asyncIterator", toStringTagSymbol = $Symbol.toStringTag || "@@toStringTag"; function define(obj, key, value) { return Object.defineProperty(obj, key, { value: value, enumerable: !0, configurable: !0, writable: !0 }), obj[key]; } try { define({}, ""); } catch (err) { define = function define(obj, key, value) { return obj[key] = value; }; } function wrap(innerFn, outerFn, self, tryLocsList) { var protoGenerator = outerFn && outerFn.prototype instanceof Generator ? outerFn : Generator, generator = Object.create(protoGenerator.prototype), context = new Context(tryLocsList || []); return defineProperty(generator, "_invoke", { value: makeInvokeMethod(innerFn, self, context) }), generator; } function tryCatch(fn, obj, arg) { try { return { type: "normal", arg: fn.call(obj, arg) }; } catch (err) { return { type: "throw", arg: err }; } } exports.wrap = wrap; var ContinueSentinel = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var IteratorPrototype = {}; define(IteratorPrototype, iteratorSymbol, function () { return this; }); var getProto = Object.getPrototypeOf, NativeIteratorPrototype = getProto && getProto(getProto(values([]))); NativeIteratorPrototype && NativeIteratorPrototype !== Op && hasOwn.call(NativeIteratorPrototype, iteratorSymbol) && (IteratorPrototype = NativeIteratorPrototype); var Gp = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(IteratorPrototype); function defineIteratorMethods(prototype) { ["next", "throw", "return"].forEach(function (method) { define(prototype, method, function (arg) { return this._invoke(method, arg); }); }); } function AsyncIterator(generator, PromiseImpl) { function invoke(method, arg, resolve, reject) { var record = tryCatch(generator[method], generator, arg); if ("throw" !== record.type) { var result = record.arg, value = result.value; return value && "object" == _typeof(value) && hasOwn.call(value, "__await") ? PromiseImpl.resolve(value.__await).then(function (value) { invoke("next", value, resolve, reject); }, function (err) { invoke("throw", err, resolve, reject); }) : PromiseImpl.resolve(value).then(function (unwrapped) { result.value = unwrapped, resolve(result); }, function (error) { return invoke("throw", error, resolve, reject); }); } reject(record.arg); } var previousPromise; defineProperty(this, "_invoke", { value: function value(method, arg) { function callInvokeWithMethodAndArg() { return new PromiseImpl(function (resolve, reject) { invoke(method, arg, resolve, reject); }); } return previousPromise = previousPromise ? previousPromise.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(innerFn, self, context) { var state = "suspendedStart"; return function (method, arg) { if ("executing" === state) throw new Error("Generator is already running"); if ("completed" === state) { if ("throw" === method) throw arg; return doneResult(); } for (context.method = method, context.arg = arg;;) { var delegate = context.delegate; if (delegate) { var delegateResult = maybeInvokeDelegate(delegate, context); if (delegateResult) { if (delegateResult === ContinueSentinel) continue; return delegateResult; } } if ("next" === context.method) context.sent = context._sent = context.arg;else if ("throw" === context.method) { if ("suspendedStart" === state) throw state = "completed", context.arg; context.dispatchException(context.arg); } else "return" === context.method && context.abrupt("return", context.arg); state = "executing"; var record = tryCatch(innerFn, self, context); if ("normal" === record.type) { if (state = context.done ? "completed" : "suspendedYield", record.arg === ContinueSentinel) continue; return { value: record.arg, done: context.done }; } "throw" === record.type && (state = "completed", context.method = "throw", context.arg = record.arg); } }; } function maybeInvokeDelegate(delegate, context) { var methodName = context.method, method = delegate.iterator[methodName]; if (undefined === method) return context.delegate = null, "throw" === methodName && delegate.iterator["return"] && (context.method = "return", context.arg = undefined, maybeInvokeDelegate(delegate, context), "throw" === context.method) || "return" !== methodName && (context.method = "throw", context.arg = new TypeError("The iterator does not provide a '" + methodName + "' method")), ContinueSentinel; var record = tryCatch(method, delegate.iterator, context.arg); if ("throw" === record.type) return context.method = "throw", context.arg = record.arg, context.delegate = null, ContinueSentinel; var info = record.arg; return info ? info.done ? (context[delegate.resultName] = info.value, context.next = delegate.nextLoc, "return" !== context.method && (context.method = "next", context.arg = undefined), context.delegate = null, ContinueSentinel) : info : (context.method = "throw", context.arg = new TypeError("iterator result is not an object"), context.delegate = null, ContinueSentinel); } function pushTryEntry(locs) { var entry = { tryLoc: locs[0] }; 1 in locs && (entry.catchLoc = locs[1]), 2 in locs && (entry.finallyLoc = locs[2], entry.afterLoc = locs[3]), this.tryEntries.push(entry); } function resetTryEntry(entry) { var record = entry.completion || {}; record.type = "normal", delete record.arg, entry.completion = record; } function Context(tryLocsList) { this.tryEntries = [{ tryLoc: "root" }], tryLocsList.forEach(pushTryEntry, this), this.reset(!0); } function values(iterable) { if (iterable) { var iteratorMethod = iterable[iteratorSymbol]; if (iteratorMethod) return iteratorMethod.call(iterable); if ("function" == typeof iterable.next) return iterable; if (!isNaN(iterable.length)) { var i = -1, next = function next() { for (; ++i < iterable.length;) if (hasOwn.call(iterable, i)) return next.value = iterable[i], next.done = !1, next; return next.value = undefined, next.done = !0, next; }; return next.next = next; } } return { next: doneResult }; } function doneResult() { return { value: undefined, done: !0 }; } return GeneratorFunction.prototype = GeneratorFunctionPrototype, defineProperty(Gp, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), defineProperty(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, toStringTagSymbol, "GeneratorFunction"), exports.isGeneratorFunction = function (genFun) { var ctor = "function" == typeof genFun && genFun.constructor; return !!ctor && (ctor === GeneratorFunction || "GeneratorFunction" === (ctor.displayName || ctor.name)); }, exports.mark = function (genFun) { return Object.setPrototypeOf ? Object.setPrototypeOf(genFun, GeneratorFunctionPrototype) : (genFun.__proto__ = GeneratorFunctionPrototype, define(genFun, toStringTagSymbol, "GeneratorFunction")), genFun.prototype = Object.create(Gp), genFun; }, exports.awrap = function (arg) { return { __await: arg }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, asyncIteratorSymbol, function () { return this; }), exports.AsyncIterator = AsyncIterator, exports.async = function (innerFn, outerFn, self, tryLocsList, PromiseImpl) { void 0 === PromiseImpl && (PromiseImpl = Promise); var iter = new AsyncIterator(wrap(innerFn, outerFn, self, tryLocsList), PromiseImpl); return exports.isGeneratorFunction(outerFn) ? iter : iter.next().then(function (result) { return result.done ? result.value : iter.next(); }); }, defineIteratorMethods(Gp), define(Gp, toStringTagSymbol, "Generator"), define(Gp, iteratorSymbol, function () { return this; }), define(Gp, "toString", function () { return "[object Generator]"; }), exports.keys = function (val) { var object = Object(val), keys = []; for (var key in object) keys.push(key); return keys.reverse(), function next() { for (; keys.length;) { var key = keys.pop(); if (key in object) return next.value = key, next.done = !1, next; } return next.done = !0, next; }; }, exports.values = values, Context.prototype = { constructor: Context, reset: function reset(skipTempReset) { if (this.prev = 0, this.next = 0, this.sent = this._sent = undefined, this.done = !1, this.delegate = null, this.method = "next", this.arg = undefined, this.tryEntries.forEach(resetTryEntry), !skipTempReset) for (var name in this) "t" === name.charAt(0) && hasOwn.call(this, name) && !isNaN(+name.slice(1)) && (this[name] = undefined); }, stop: function stop() { this.done = !0; var rootRecord = this.tryEntries[0].completion; if ("throw" === rootRecord.type) throw rootRecord.arg; return this.rval; }, dispatchException: function dispatchException(exception) { if (this.done) throw exception; var context = this; function handle(loc, caught) { return record.type = "throw", record.arg = exception, context.next = loc, caught && (context.method = "next", context.arg = undefined), !!caught; } for (var i = this.tryEntries.length - 1; i >= 0; --i) { var entry = this.tryEntries[i], record = entry.completion; if ("root" === entry.tryLoc) return handle("end"); if (entry.tryLoc <= this.prev) { var hasCatch = hasOwn.call(entry, "catchLoc"), hasFinally = hasOwn.call(entry, "finallyLoc"); if (hasCatch && hasFinally) { if (this.prev < entry.catchLoc) return handle(entry.catchLoc, !0); if (this.prev < entry.finallyLoc) return handle(entry.finallyLoc); } else if (hasCatch) { if (this.prev < entry.catchLoc) return handle(entry.catchLoc, !0); } else { if (!hasFinally) throw new Error("try statement without catch or finally"); if (this.prev < entry.finallyLoc) return handle(entry.finallyLoc); } } } }, abrupt: function abrupt(type, arg) { for (var i = this.tryEntries.length - 1; i >= 0; --i) { var entry = this.tryEntries[i]; if (entry.tryLoc <= this.prev && hasOwn.call(entry, "finallyLoc") && this.prev < entry.finallyLoc) { var finallyEntry = entry; break; } } finallyEntry && ("break" === type || "continue" === type) && finallyEntry.tryLoc <= arg && arg <= finallyEntry.finallyLoc && (finallyEntry = null); var record = finallyEntry ? finallyEntry.completion : {}; return record.type = type, record.arg = arg, finallyEntry ? (this.method = "next", this.next = finallyEntry.finallyLoc, ContinueSentinel) : this.complete(record); }, complete: function complete(record, afterLoc) { if ("throw" === record.type) throw record.arg; return "break" === record.type || "continue" === record.type ? this.next = record.arg : "return" === record.type ? (this.rval = this.arg = record.arg, this.method = "return", this.next = "end") : "normal" === record.type && afterLoc && (this.next = afterLoc), ContinueSentinel; }, finish: function finish(finallyLoc) { for (var i = this.tryEntries.length - 1; i >= 0; --i) { var entry = this.tryEntries[i]; if (entry.finallyLoc === finallyLoc) return this.complete(entry.completion, entry.afterLoc), resetTryEntry(entry), ContinueSentinel; } }, "catch": function _catch(tryLoc) { for (var i = this.tryEntries.length - 1; i >= 0; --i) { var entry = this.tryEntries[i]; if (entry.tryLoc === tryLoc) { var record = entry.completion; if ("throw" === record.type) { var thrown = record.arg; resetTryEntry(entry); } return thrown; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(iterable, resultName, nextLoc) { return this.delegate = { iterator: values(iterable), resultName: resultName, nextLoc: nextLoc }, "next" === this.method && (this.arg = undefined), ContinueSentinel; } }, exports; }
function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
var NUMBER_REGEX = '^-?[0-9]+(.[0-9]+)?$';

/**
 * @param {string} connection
 * @returns {Promise<import('mysql2').Connection>} client
 */
function connectMySQL(_x) {
  return _connectMySQL.apply(this, arguments);
}
function _connectMySQL() {
  _connectMySQL = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(connection) {
    var client, _client;
    return _regeneratorRuntime().wrap(function _callee$(_context) {
      while (1) switch (_context.prev = _context.next) {
        case 0:
          client = null;
          _context.prev = 1;
          _context.next = 4;
          return (0, _promise.createConnection)(connection);
        case 4:
          client = _context.sent;
          _context.next = 7;
          return client.connect();
        case 7:
          _context.next = 9;
          return client.query('SELECT 1');
        case 9:
          return _context.abrupt("return", client);
        case 12:
          _context.prev = 12;
          _context.t0 = _context["catch"](1);
          console.error('MySQL connection error:', _context.t0);
          _context.next = 17;
          return (_client = client) === null || _client === void 0 ? void 0 : _client.end();
        case 17:
          throw _context.t0;
        case 18:
        case "end":
          return _context.stop();
      }
    }, _callee, null, [[1, 12]]);
  }));
  return _connectMySQL.apply(this, arguments);
}
function getEnumName(tableName, columnName) {
  return "".concat(tableName, "_").concat(columnName, "_enum");
}
function getGenerationExpression(extraType, generationExpression) {
  if (extraType === 'VIRTUAL GENERATED') {
    return "GENERATED ALWAYS AS (".concat(generationExpression, ") VIRTUAL");
  }
  if (extraType === 'STORED GENERATED') {
    return "GENERATED ALWAYS AS (".concat(generationExpression, ") STORED");
  }

  // for timestamp data type
  if (extraType.includes('on update CURRENT_TIMESTAMP')) {
    return 'on update CURRENT_TIMESTAMP';
  }

  // others
  return '';
}
function getDbDefault(columnDefault, defaultValueType) {
  if (columnDefault === null) {
    return null;
  }
  return {
    value: columnDefault,
    type: defaultValueType
  };
}
function getFieldType(tableName, columnName, columnType, columnDataType, columnExtra, generationExpression) {
  if (columnDataType === 'enum') {
    // enum must have static value -> no need to check the generation expression
    return getEnumName(tableName, columnName);
  }
  var fieldGenerationExpression = getGenerationExpression(columnExtra, generationExpression);
  if (fieldGenerationExpression) {
    return "".concat(columnType, " ").concat(fieldGenerationExpression);
  }
  return columnType;
}
function generateField(row) {
  var tableName = row.tableName,
    columnName = row.columnName,
    columnDefault = row.columnDefault,
    defaultValueType = row.defaultValueType,
    columnIsNullable = row.columnIsNullable,
    columnType = row.columnType,
    columnDataType = row.columnDataType,
    columnComment = row.columnComment,
    columnExtra = row.columnExtra,
    generationExpression = row.generationExpression;
  var fieldType = getFieldType(tableName, columnName, columnType, columnDataType, columnExtra, generationExpression);
  var fieldDefaultValue = getDbDefault(columnDefault, defaultValueType);
  var isNullable = columnIsNullable === 'YES';
  return {
    name: columnName,
    type: {
      type_name: fieldType
    },
    dbdefault: fieldDefaultValue,
    not_null: !isNullable,
    increment: columnExtra === 'auto_increment',
    note: {
      value: columnComment || ''
    }
  };
}

// Do not get the index sub part since in DBML, it is impossible to create index on part of column.
function getIndexColumn(columnName, idxExpression, idxSubPart) {
  if (idxExpression) {
    return {
      value: idxExpression,
      type: 'expression'
    };
  }
  if (idxSubPart) {
    return {
      value: "".concat(columnName, "(").concat(idxSubPart, ")"),
      type: 'expression'
    };
  }
  if (columnName) {
    return {
      value: columnName,
      type: 'column'
    };
  }
  return null;
}
function generateTablesAndFields(_x2, _x3) {
  return _generateTablesAndFields.apply(this, arguments);
}
function _generateTablesAndFields() {
  _generateTablesAndFields = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(client, schemaName) {
    var query, queryResponse, _queryResponse, rows, tableMap, fieldMap;
    return _regeneratorRuntime().wrap(function _callee2$(_context2) {
      while (1) switch (_context2.prev = _context2.next) {
        case 0:
          query = "\n    select\n      t.table_name as tableName,\n      t.table_comment as tableComment,\n      c.column_name as columnName,\n      c.column_default as columnDefault,\n      case\n        when c.column_default is null then 'boolean'\n        when c.data_type = 'enum' then 'string'\n        when c.column_default regexp ? then 'number'\n        when c.extra like '%DEFAULT_GENERATED%' then 'expression'\n        else 'string'\n      end as defaultValueType,\n      c.is_nullable as columnIsNullable,\n      c.data_type as columnDataType,\n      c.column_type as columnType,\n      c.extra as columnExtra,\n      c.column_comment as columnComment,\n      c.generation_expression as generationExpression\n    from information_schema.tables t\n    join information_schema.columns c\n    on t.table_schema = c.table_schema and t.table_name = c.table_name\n    where\n      t.table_schema = ? and t.table_type = 'BASE TABLE'\n    order by\n      t.table_name, c.ordinal_position;\n  ";
          _context2.next = 3;
          return client.query(query, [NUMBER_REGEX, schemaName]);
        case 3:
          queryResponse = _context2.sent;
          _queryResponse = _slicedToArray(queryResponse, 1), rows = _queryResponse[0];
          tableMap = {};
          fieldMap = {};
          rows.forEach(function (row) {
            var tableName = row.tableName,
              tableComment = row.tableComment;
            var key = tableName;
            if (!tableMap[key]) {
              tableMap[key] = {
                name: tableName,
                note: {
                  value: tableComment || ''
                }
              };
            }
            if (!fieldMap[key]) {
              fieldMap[key] = [];
            }
            var field = generateField(row);
            fieldMap[key].push(field);
          });
          return _context2.abrupt("return", {
            tableList: Object.values(tableMap),
            fieldMap: fieldMap
          });
        case 9:
        case "end":
          return _context2.stop();
      }
    }, _callee2);
  }));
  return _generateTablesAndFields.apply(this, arguments);
}
function generateEnums(_x4, _x5) {
  return _generateEnums.apply(this, arguments);
}
/**
 * Mysql is automatically create index for primary keys, foreign keys, unique constraint. -> Ignore
 */
function _generateEnums() {
  _generateEnums = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(client, schemaName) {
    var query, queryResponse, _queryResponse2, rows, enumList;
    return _regeneratorRuntime().wrap(function _callee3$(_context3) {
      while (1) switch (_context3.prev = _context3.next) {
        case 0:
          query = "\n    select\n      t.table_name as tableName,\n      c.column_name as columnName,\n      TRIM(LEADING 'enum' FROM c.column_type) AS rawValues\n    from information_schema.tables t\n    join information_schema.columns c\n    on t.table_schema = c.table_schema and t.table_name = c.table_name\n    where\n      t.table_schema = ? and t.table_type = 'BASE TABLE' and c.data_type = 'enum'\n    order by\n      t.table_name, c.ordinal_position;\n  ";
          _context3.next = 3;
          return client.query(query, [schemaName]);
        case 3:
          queryResponse = _context3.sent;
          _queryResponse2 = _slicedToArray(queryResponse, 1), rows = _queryResponse2[0];
          enumList = rows.map(function (row) {
            var tableName = row.tableName,
              columnName = row.columnName,
              rawValues = row.rawValues;

            // i.e. ('value1','value2')
            var valueList = rawValues.slice(1, -1).split(',').map(function (value) {
              return {
                name: value.slice(1, -1)
              };
            });
            var enumName = getEnumName(tableName, columnName);
            return {
              name: enumName,
              values: valueList
            };
          });
          return _context3.abrupt("return", enumList);
        case 7:
        case "end":
          return _context3.stop();
      }
    }, _callee3);
  }));
  return _generateEnums.apply(this, arguments);
}
function generateIndexes(_x6, _x7) {
  return _generateIndexes.apply(this, arguments);
}
function _generateIndexes() {
  _generateIndexes = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4(client, schemaName) {
    var query, queryResponse, _queryResponse3, rows, tableIndexMap, indexMap;
    return _regeneratorRuntime().wrap(function _callee4$(_context4) {
      while (1) switch (_context4.prev = _context4.next) {
        case 0:
          query = "\n    with\n      pk_fk_uniques as (\n        select constraint_name, table_name\n        from information_schema.table_constraints\n        where table_schema = ?\n      )\n    select\n      st.table_name as tableName,\n      case\n        when st.non_unique = 0 then true\n        else false\n        end as isIdxUnique,\n      st.index_name as idxName,\n      st.column_name as columnName,\n      st.sub_part as idxSubPart,\n      st.index_type as idxType,\n      replace(st.expression, '`', '') as idxExpression\n    from information_schema.statistics st\n    where\n      st.table_schema = ?\n      and st.index_name not in (\n        select constraint_name\n        from pk_fk_uniques pfu\n        where pfu.table_name = st.table_name\n      )\n      and st.index_type in ('BTREE', 'HASH')\n    group by\n      st.table_name, st.non_unique, st.index_name, st.column_name, st.sub_part, st.index_type, st.expression\n    order by st.index_name;\n  ";
          _context4.next = 3;
          return client.query(query, [schemaName, schemaName]);
        case 3:
          queryResponse = _context4.sent;
          _queryResponse3 = _slicedToArray(queryResponse, 1), rows = _queryResponse3[0];
          tableIndexMap = rows.reduce(function (acc, row) {
            var tableName = row.tableName,
              idxName = row.idxName,
              idxType = row.idxType,
              isIdxUnique = row.isIdxUnique,
              columnName = row.columnName,
              idxExpression = row.idxExpression,
              idxSubPart = row.idxSubPart;
            var key = tableName;
            if (!acc[key]) {
              acc[key] = {};
            }
            if (!acc[key][idxName]) {
              // init first index
              acc[key][idxName] = {
                name: idxName,
                type: idxType,
                columns: [],
                unique: !!isIdxUnique
              };
            }
            var currentIndex = acc[key][idxName];
            var column = getIndexColumn(columnName, idxExpression, idxSubPart);
            if (column) {
              currentIndex.columns.push(column);
            }
            return acc;
          }, {});
          indexMap = {};
          Object.keys(tableIndexMap).forEach(function (tableName) {
            indexMap[tableName] = (0, _lodash.flatten)(Object.values(tableIndexMap[tableName]));
          });
          return _context4.abrupt("return", indexMap);
        case 9:
        case "end":
          return _context4.stop();
      }
    }, _callee4);
  }));
  return _generateIndexes.apply(this, arguments);
}
function generatePrimaryAndUniqueConstraint(_x8, _x9) {
  return _generatePrimaryAndUniqueConstraint.apply(this, arguments);
}
function _generatePrimaryAndUniqueConstraint() {
  _generatePrimaryAndUniqueConstraint = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5(client, schemaName) {
    var query, queryResponse, _queryResponse4, rows, inlineConstraintList, outOfConstraintList, compositeTableConstraintMap, compositeConstraintMap, constraintMap;
    return _regeneratorRuntime().wrap(function _callee5$(_context5) {
      while (1) switch (_context5.prev = _context5.next) {
        case 0:
          query = "\n    select\n      tc.table_name as tableName,\n      tc.constraint_name as constraintName,\n      group_concat(kcu.column_name order by kcu.ordinal_position separator ',') as columnNames,\n      count(kcu.column_name) as columnCount,\n      tc.constraint_type as constraintType\n    from information_schema.table_constraints tc\n    join information_schema.key_column_usage kcu\n    on\n      tc.constraint_schema = kcu.constraint_schema\n      and tc.constraint_name = kcu.constraint_name\n      and tc.table_name = kcu.table_name\n    where\n      (tc.constraint_type = 'PRIMARY KEY' or tc.constraint_type = 'UNIQUE')\n      and tc.table_schema = ?\n    group by tc.table_name, tc.constraint_name, tc.constraint_type\n    order by tc.table_name, tc.constraint_name;\n  ";
          _context5.next = 3;
          return client.query(query, [schemaName]);
        case 3:
          queryResponse = _context5.sent;
          _queryResponse4 = _slicedToArray(queryResponse, 1), rows = _queryResponse4[0];
          inlineConstraintList = rows.filter(function (constraint) {
            return constraint.columnCount === 1;
          });
          outOfConstraintList = rows.filter(function (constraint) {
            return constraint.columnCount > 1;
          });
          compositeTableConstraintMap = outOfConstraintList.reduce(function (acc, row) {
            var tableName = row.tableName,
              constraintName = row.constraintName,
              columnNames = row.columnNames,
              constraintType = row.constraintType;
            var key = tableName;
            if (!acc[key]) {
              acc[key] = {};
            }
            if (!acc[key][constraintName]) {
              acc[key][constraintName] = {
                name: constraintName
              };
            }
            if (constraintType === 'PRIMARY KEY') {
              acc[key][constraintName].primary = true;
            }
            if (constraintType === 'UNIQUE') {
              acc[key][constraintName].unique = true;
            }
            var columnList = columnNames.split(',').map(function (col) {
              return {
                type: 'column',
                value: col
              };
            });
            acc[key][constraintName].columns = columnList;
            return acc;
          }, {});
          compositeConstraintMap = {};
          Object.keys(compositeTableConstraintMap).forEach(function (tableName) {
            compositeConstraintMap[tableName] = (0, _lodash.flatten)(Object.values(compositeTableConstraintMap[tableName]));
          });
          constraintMap = inlineConstraintList.reduce(function (acc, row) {
            var tableName = row.tableName,
              columnNames = row.columnNames,
              constraintType = row.constraintType;
            var key = tableName;
            if (!acc[key]) {
              acc[key] = {};
            }
            var columnList = columnNames.split(',');
            columnList.forEach(function (columnName) {
              if (!acc[key][columnName]) {
                acc[key][columnName] = {};
              }
              if (constraintType === 'PRIMARY KEY') {
                acc[key][columnName].pk = true;
              }
              if (constraintType === 'UNIQUE' && !acc[key][columnName].pk) {
                acc[key][columnName].unique = true;
              }
            });
            return acc;
          }, {});
          return _context5.abrupt("return", {
            compositeConstraintMap: compositeConstraintMap,
            constraintMap: constraintMap
          });
        case 12:
        case "end":
          return _context5.stop();
      }
    }, _callee5);
  }));
  return _generatePrimaryAndUniqueConstraint.apply(this, arguments);
}
function generateForeignKeys(_x10, _x11) {
  return _generateForeignKeys.apply(this, arguments);
}
function _generateForeignKeys() {
  _generateForeignKeys = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee6(client, schemaName) {
    var query, queryResponse, _queryResponse5, rows, foreignKeyList;
    return _regeneratorRuntime().wrap(function _callee6$(_context6) {
      while (1) switch (_context6.prev = _context6.next) {
        case 0:
          query = "\n    select\n      rc.constraint_name as constraintName,\n      rc.table_name as foreignTableName,\n      group_concat(kcu.column_name order by kcu.ordinal_position separator ',') as foreignColumnNames,\n      kcu.referenced_table_name as refTableName,\n      group_concat(kcu.referenced_column_name order by kcu.ordinal_position separator ',') as refColumnNames,\n      rc.update_rule as onUpdate,\n      rc.delete_rule as onDelete\n    from information_schema.referential_constraints rc\n    join information_schema.key_column_usage kcu\n    on\n      rc.constraint_name = kcu.constraint_name\n      and rc.table_name = kcu.table_name\n      and rc.constraint_schema = kcu.table_schema\n    where rc.constraint_schema = ?\n    group by\n      rc.constraint_name,\n      rc.table_name,\n      kcu.referenced_table_name,\n      rc.update_rule,\n      rc.delete_rule\n    order by\n      rc.table_name;\n  ";
          _context6.next = 3;
          return client.query(query, [schemaName]);
        case 3:
          queryResponse = _context6.sent;
          _queryResponse5 = _slicedToArray(queryResponse, 1), rows = _queryResponse5[0];
          foreignKeyList = rows.map(function (row) {
            var constraintName = row.constraintName,
              onDelete = row.onDelete,
              onUpdate = row.onUpdate,
              foreignTableName = row.foreignTableName,
              foreignColumnNames = row.foreignColumnNames,
              refTableName = row.refTableName,
              refColumnNames = row.refColumnNames;
            var endpoint1 = {
              tableName: foreignTableName,
              fieldNames: foreignColumnNames.split(','),
              relation: '*'
            };
            var endpoint2 = {
              tableName: refTableName,
              fieldNames: refColumnNames.split(','),
              relation: '1'
            };
            return {
              name: constraintName,
              endpoints: [endpoint1, endpoint2],
              onDelete: onDelete === 'NO ACTION' ? null : onDelete,
              onUpdate: onUpdate === 'NO ACTION' ? null : onUpdate
            };
          });
          return _context6.abrupt("return", foreignKeyList);
        case 7:
        case "end":
          return _context6.stop();
      }
    }, _callee6);
  }));
  return _generateForeignKeys.apply(this, arguments);
}
function combineIndexAndCompositeConstraint(userDefinedIndexMap, compositeConstraintMap) {
  var indexMap = Object.assign(userDefinedIndexMap, {});
  Object.keys(compositeConstraintMap).forEach(function (tableName) {
    var _indexMap$tableName;
    var compositeConstraint = compositeConstraintMap[tableName];
    if (!indexMap[tableName]) {
      indexMap[tableName] = compositeConstraint;
      return;
    }
    (_indexMap$tableName = indexMap[tableName]).push.apply(_indexMap$tableName, _toConsumableArray(compositeConstraint));
  });
  return indexMap;
}
function fetchSchemaJson(_x12) {
  return _fetchSchemaJson.apply(this, arguments);
}
function _fetchSchemaJson() {
  _fetchSchemaJson = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee7(connection) {
    var client, schemaName, result, _result, _result$, tableList, fieldMap, enumList, rawIndexMap, _result$2, constraintMap, compositeConstraintMap, foreignKeyList, indexMap;
    return _regeneratorRuntime().wrap(function _callee7$(_context7) {
      while (1) switch (_context7.prev = _context7.next) {
        case 0:
          _context7.next = 2;
          return connectMySQL(connection);
        case 2:
          client = _context7.sent;
          // In MySQL, a schema is equal database
          schemaName = client.config.database;
          _context7.next = 6;
          return Promise.all([generateTablesAndFields(client, schemaName), generateEnums(client, schemaName), generateIndexes(client, schemaName), generatePrimaryAndUniqueConstraint(client, schemaName), generateForeignKeys(client, schemaName)]);
        case 6:
          result = _context7.sent;
          client.end();
          _result = _slicedToArray(result, 5), _result$ = _result[0], tableList = _result$.tableList, fieldMap = _result$.fieldMap, enumList = _result[1], rawIndexMap = _result[2], _result$2 = _result[3], constraintMap = _result$2.constraintMap, compositeConstraintMap = _result$2.compositeConstraintMap, foreignKeyList = _result[4]; // combine normal index and composite key
          indexMap = combineIndexAndCompositeConstraint(rawIndexMap, compositeConstraintMap);
          return _context7.abrupt("return", {
            tables: tableList,
            fields: fieldMap,
            refs: foreignKeyList,
            enums: enumList,
            indexes: indexMap,
            tableConstraints: constraintMap
          });
        case 11:
        case "end":
          return _context7.stop();
      }
    }, _callee7);
  }));
  return _fetchSchemaJson.apply(this, arguments);
}