sails-inverse-model/scanners/scanner_mysql.js

Sails Inverse Model helps you build models, controllers and views JS Sails from any database. In addition, you can quickly and individually generate each model, view, controller or all three at the same time.

var gencode = require('gencode');
var data = require('./data.json');

exports.generate = function (pathsql, cb) {
  gencode
    .utils
    .toArray(pathsql, 'utf8', '\n')
    .then((value) => { //Too: \n, \t, -, etc.
      var allData = "";
      var tables = [],
        dataValid = [];
      value.map((item) => {
        item = item
          .toString()
          .trim()
          .toLowerCase();

        if (!isComment(item)) {
          allData += item;
        }
      });
      var splits = allData.split(";");
      for (var i = 0; i < splits.length; i++) {
        if (isValid(splits[i])) {
          dataValid.push(splits[i]);
        }
      }

      for (var i = 0; i < dataValid.length; i++) {
        if (dataValid[i].toString().toLowerCase().trim().startsWith('create table')) {
          tables.push(verifyContains(dataValid[i].toString()));
        }
      }
      getTablesJSON(tables, function (err, result) {
        //console.log(":>", JSON.stringify(result, null, 2));
        cb(err, result);
      });

    }, (error) => {
      cb(error, null);
      //console.log("ERROR=>", error);
    });
};

//exports.generate("../test/script.sql");

function verifyContains(item) {
  var result = "";
  var state = true;
  var start,
    end = 0;
  var add;
  for (var i = 0; i < data.contains.length; i++) {
    result = "";
    state = true;
    while (state) {
      start = item.indexOf(data.contains[i]);
      if (start != -1) {
        end = item
          .substring(start, item.length)
          .indexOf(")") + start;
        result += item.substring(0, start);
        add = item.substring(start, end);
        while (add.indexOf(",") != -1) {
          add = add.replace(",", "-");
        }
        result += add;
        item = item.substring(end, item.length);
      } else {
        state = false;
      }
    }
    result += item;
    item = result;
  }
  return result;
}

function cleanItem(item) {
  var line = item;
  for (var i = 0; i < data.start.length; i++) {
    if (item.startsWith(data.start[i])) {
      line = item.substring(data.start[i].length + 1, item.length);
      break;
    }
  }
  return line;
}

function getTableName(item) {
  var firstLine = item.indexOf('(');
  var name = item
    .substring(0, firstLine)
    .trim()
    .replace('`', '')
    .replace('`', '');
  return name;
}

function isAtribute(line) {
  for (var j = 0; j < data.attr.length; j++) {
    if (line.toString().trim().startsWith(data.attr[j])) {
      return false;
    }
  }
  return true;
}

function getAtributes(item) {
  var attrLines = item.indexOf('(');
  item = item.substring(attrLines + 1, item.length);
  var atributes = [];
  var lines = item.split(',');
  var split;
  for (var i = 0; i < lines.length; i++) {
    lines[i] = lines[i].replace("not null", "not_null");
    if (isAtribute(lines[i])) {

      split = lines[i] != ""
        ? lines[i].split(" ")
        : "";
      atributes.push({
        Name: getValue(0, split),
        Type: getValue(1, split),
        Size: getValue(2, split),
        NotNull: split.length > 2,
        AI: split.length > 3
      });
    }
  }
  return atributes;
}

function getValue(pos, values) {
  var result = "";

  switch (pos) {
    case 0:
      result = values[pos]
        .replace('`', '')
        .replace('`', '');
      break;
    case 1:
      var val = values[pos];
      if (!val.startsWith("enum")) {
        var start = val.indexOf("(");
        result = start == -1
          ? val
          : val.substring(0, start);
      } else {
        result = val;
      }
      break;
    case 2:
      var val = values[1];
      if (!val.startsWith("enum")) {
        var start = val.indexOf("(");
        result = start == -1
          ? ""
          : val.substring(start + 1, val.indexOf(")"));
      } else {
        result = "";
      }
      break;
    default:
  }
  return result;
}

function getTablesJSON(tables, cb) {
  var tablesJSON = [];
  var item,
    line;
  for (var i = 0; i < tables.length; i++) {
    item = tables[i]
      .toString()
      .toLowerCase();
    tables[i] = cleanItem(item);

    tablesJSON.push({
      table_name: getTableName(tables[i]),
      atr: getAtributes(tables[i])
    });
  }
  console.log("d: " + tablesJSON.length);
  //console.log(JSON.stringify(tablesJSON, null, 4));
  cb(null, tablesJSON);
}

function isComment(line) {
  return line
    .trim()
    .startsWith('--');
}

function isValid(line) {
  if (line !== "") {
    for (var i = 0; i < data.ignore.length; i++) {
      if (line.startsWith(data.ignore[i])) {
        return false;
      }
    }
  } else {
    return false;
  }
  return true;
}