stackrender
diff --git a/‎lib/sql.js‎
Lines changed: 67 additions & 6 deletions b/‎lib/sql.js‎
Lines changed: 67 additions & 6 deletions
@@ -14,7 +14,7 @@ const escapeMap = {
   '\\'   : '\\\\',
 }
 
-const surportedTypes = ['select', 'delete']
+const surportedTypes = ['select', 'delete', 'update', 'insert']
 
 function escape(str) {
   const res = []
@@ -141,6 +141,7 @@ function funcToSQL(expr) {
  * @return {string}
  */
 function columnsToSQL(columns, tables) {
+  if (!columns) return
   const baseTable = Array.isArray(tables) && tables[0]
   let isDual = false
   if (baseTable && baseTable.type === 'dual') isDual = true
@@ -213,6 +214,40 @@ function withToSql(withExpr) {
   return `WITH ${isRecursive}${withExprStr}`
 }
 
+/**
+ * @param {Array} sets
+ * @return {string}
+ */
+function setToSQL(sets) {
+  const baseTable = sets[0]
+  const clauses = []
+  let str = ''
+  if (baseTable.column) str = `\`${baseTable.column}\``
+  str = baseTable.value && (str += ` = ${exprToSQL(baseTable.value)}`)
+
+  clauses.push(str)
+
+  for (let i = 1; i < sets.length; ++i) {
+    const joinExpr = sets[i]
+
+    if (joinExpr.column) str = `, \`${joinExpr.column}\``
+    str = joinExpr.value && (str += ` = ${exprToSQL(joinExpr.value)}`)
+
+    clauses.push(str)
+  }
+
+  return clauses.join('')
+}
+
+/**
+ * @param {Array} values
+ * @return {string}
+ */
+function valuesToSQL(values) {
+  const clauses = values.map(exprToSQL)
+  return `(${clauses.join('')})`
+}
+
 /**
  * @param {Object}      stmt
  * @param {?Array}      stmt.with
@@ -256,7 +291,9 @@ function selectToSQL(stmt) {
 
 function deleteToSQL(stmt) {
   const clauses = ['DELETE']
-  console.warn('delete', Array.isArray(stmt.tables))
+  if (stmt.columns === '*') clauses.push('*')
+  else columnsToSQL(stmt.columns, stmt.from) && clauses.push(columnsToSQL(stmt.columns, stmt.from))
+
   if (Array.isArray(stmt.tables)) clauses.push(tablesToSQL(stmt.tables))
   if (Array.isArray(stmt.from)) clauses.push('FROM', tablesToSQL(stmt.from))
 
@@ -265,6 +302,28 @@ function deleteToSQL(stmt) {
   return clauses.join(' ')
 }
 
+function updateToSQL(stmt) {
+  const clauses = ['UPDATE']
+  if (has(stmt, 'table') && stmt.table !== null) clauses.push(identifierToSql(stmt.table, false))
+  if (Array.isArray(stmt.tables)) clauses.push(tablesToSQL(stmt.tables))
+  if (Array.isArray(stmt.set)) clauses.push('SET', setToSQL(stmt.set))
+
+  if (has(stmt, 'where') && stmt.where !== null) clauses.push(`WHERE ${exprToSQL(stmt.where)}`)
+
+  return clauses.join(' ')
+}
+
+function insertToSQL(stmt) {
+  const clauses = ['INSERT INTO']
+  if (has(stmt, 'table') && stmt.table !== null) clauses.push(identifierToSql(stmt.table, false))
+  if (Array.isArray(stmt.tables)) clauses.push(tablesToSQL(stmt.tables))
+  if (Array.isArray(stmt.columns)) clauses.push(`(${stmt.columns.map(identifierToSql).join(', ')})`)
+  if (Array.isArray(stmt.values)) clauses.push('VALUES', valuesToSQL(stmt.values))
+  if (has(stmt, 'where') && stmt.where !== null) clauses.push(`WHERE ${exprToSQL(stmt.where)}`)
+
+  return clauses.join(' ')
+}
+
 function unaryToSQL(expr) {
   const str = `${expr.operator} ${exprToSQL(expr.expr)}`
   return expr.parentheses ? `(${str})` : str
@@ -314,18 +373,20 @@ exprToSQLConvertFn = {
 typeToSQLFn = {
   select : selectToSQL,
   delete : deleteToSQL,
+  update : updateToSQL,
+  insert : insertToSQL,
 }
 
-function onlySelectSupported(expr) {
+function checkSupported(expr) {
   const ast = expr && expr.ast ? expr.ast : expr
-  if (!surportedTypes.includes(ast.type)) throw new Error('Only SELECT statements supported at the moment')
+  if (!surportedTypes.includes(ast.type)) throw new Error(`${ast.type} statements not supported at the moment`)
 }
 
 module.exports = function toSQL(ast) {
   if (Array.isArray(ast)) {
-    ast.forEach(onlySelectSupported)
+    ast.forEach(checkSupported)
     return multipleToSQL(ast)
   }
-  onlySelectSupported(ast)
+  checkSupported(ast)
   return unionToSQL(ast)
 }