fix: multiple array index when sqlify back

taozhi8833998 · taozhi8833998 · commit 9d6eb3f899d8 · 2025-08-19T09:32:02.000+08:00
diff --git a/src/column.js b/src/column.js
@@ -184,7 +184,7 @@ function columnToSQL(column, isDual) {
     return result.filter(hasVal).join(' ')
   }
   if (expr.parentheses && Reflect.has(expr, 'array_index') && expr.type !== 'cast') str = `(${str})`
-  if (expr.array_index && expr.type !== 'column_ref') {
+  if (expr.array_index && expr.type !== 'column_ref' && expr.type !== 'function') {
     str = `${str}${arrayIndexToSQL(expr.array_index)}`
   }
   return [str, asToSQL(column.as)].filter(hasVal).join(' ')
diff --git a/test/snowflake.spec.js b/test/snowflake.spec.js
@@ -539,6 +539,13 @@ describe('snowflake', () => {
         'SELECT "my ""column"""'
       ]
     },
+    {
+      title: 'array index',
+      sql: [
+        "SELECT split('12-34', '-')[0] FROM DUAL",
+        "SELECT split('12-34', '-')[0] FROM DUAL"
+      ]
+    },
   ]
   SQL_LIST.forEach(sqlInfo => {
     const { title, sql } = sqlInfo

Original file line number	Diff line number	Diff line change
`@@ -184,7 +184,7 @@ function columnToSQL(column, isDual) {`
`184`	`184`	`return result.filter(hasVal).join(' ')`
`185`	`185`	`}`
`186`	`186`	if (expr.parentheses && Reflect.has(expr, 'array_index') && expr.type !== 'cast') str = `(${str})`
`187`		`- if (expr.array_index && expr.type !== 'column_ref') {`
	`187`	`+ if (expr.array_index && expr.type !== 'column_ref' && expr.type !== 'function') {`
`188`	`188`	str = `${str}${arrayIndexToSQL(expr.array_index)}`
`189`	`189`	`}`
`190`	`190`	`return [str, asToSQL(column.as)].filter(hasVal).join(' ')`