fix: support insert or action in sqlite

Author: taozhi8833998

pegjs/sqlite.pegjs

@@ -1848,7 +1848,7 @@ replace_insert_stmt
         tableList: Array.from(tableList),
         columnList: columnListTableAlias(columnList),
         ast: {
-          type: ri,
+          ...ri,
           table: [t],
           columns: c,
           values: v,
@@ -1877,7 +1877,7 @@ insert_no_columns_stmt
         tableList: Array.from(tableList),
         columnList: columnListTableAlias(columnList),
         ast: {
-          type: ri,
+          ...ri,
           table: [t],
           columns: null,
           values: v,
@@ -1905,7 +1905,7 @@ insert_into_set
         tableList: Array.from(tableList),
         columnList: columnListTableAlias(columnList),
         ast: {
-          type: ri,
+          ...ri,
           table: [t],
           columns: null,
           partition: p,
@@ -1924,8 +1924,28 @@ on_duplicate_update_stmt
   }
 
 replace_insert
-  = KW_INSERT   { return 'insert'; }
-  / KW_REPLACE  { return 'replace'; }
+  = KW_INSERT tail:(__ KW_OR __ ('ABORT'i / 'FAIL'i / 'IGNORE'i / 'REPLACE'i / 'ROLLBACK'i))? {
+    const result = {
+      type: 'insert',
+    }
+    if (!tail || tail.length === 0) {
+      return result;
+    }
+    result.or = [
+      {
+        type: 'origin',
+        value: 'or',
+      },
+      {
+        type: 'origin',
+        value: tail[3],
+      }
+    ]
+    return result
+  }
+  / KW_REPLACE  {
+    return { type: 'replace' }
+  }
 
 value_clause
   = KW_VALUES __ l:value_list  { return l; }

src/insert.js

@@ -63,6 +63,7 @@ function insertToSQL(stmt) {
   const {
     table,
     type,
+    or: orExpr = [],
     prefix = 'into',
     columns,
     conflict,
@@ -74,7 +75,7 @@ function insertToSQL(stmt) {
     set,
   } = stmt
   const { keyword, set: duplicateSet } = onDuplicateUpdate || {}
-  const clauses = [toUpper(type), toUpper(prefix), tablesToSQL(table), partitionToSQL(partition)]
+  const clauses = [toUpper(type), orExpr.map(literalToSQL).join(' '), toUpper(prefix), tablesToSQL(table), partitionToSQL(partition)]
   if (Array.isArray(columns)) clauses.push(`(${columns.map(literalToSQL).join(', ')})`)
   clauses.push(commonOptionConnector(Array.isArray(values) ? 'VALUES' : '', valuesToSQL, values))
   clauses.push(commonOptionConnector('ON CONFLICT', conflictToSQL, conflict))

test/sqlite.spec.js

@@ -238,5 +238,11 @@ describe('sqlite', () => {
     const sql = `DELETE FROM users WHERE last_login > ? RETURNING id, email as email_address`
     expect(getParsedSql(sql)).to.be.equal(`DELETE FROM "users" WHERE "last_login" > ? RETURNING "id", "email" AS "email_address"`)
   })
-
+  it('should support insert or replace', () => {
+    const keywords = ['ABORT', 'FAIL', 'IGNORE', 'REPLACE', 'ROLLBACK']
+    keywords.forEach(keyword => {
+      const sql = `INSERT OR ${keyword} INTO test (category) VALUES ('Infra, Layer1, DePIN')`
+      expect(getParsedSql(sql)).to.be.equal(`INSERT OR ${keyword} INTO "test" (category) VALUES ('Infra, Layer1, DePIN')`)
+    })
+  })
 })