Add ::text[] support

BemiHQ · Dec 9, 2024 · 8c42c30 · 8c42c30
1 parent 2a012f3
commit 8c42c30
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 2 deletions.
diff --git a/scripts/install.sh b/scripts/install.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-VERSION="0.19.3"
+VERSION="0.19.4"
 
 # Detect OS and architecture
 OS=$(uname -s | tr '[:upper:]' '[:lower:]')

diff --git a/src/main.go b/src/main.go
@@ -6,7 +6,7 @@ import (
 	"time"
 )
 
-const VERSION = "0.19.3"
+const VERSION = "0.19.4"
 
 func main() {
 	config := LoadConfig()

diff --git a/src/query_handler_test.go b/src/query_handler_test.go
@@ -371,6 +371,10 @@ func TestHandleQuery(t *testing.T) {
 			"description": {"objoid"},
 			"values":      {},
 		},
+		"SELECT word FROM (VALUES ('abort', 'U', 't', 'unreserved', 'can be bare label')) t(word, catcode, barelabel, catdesc, baredesc) WHERE word <> ALL('{a,abs,absolute,action}'::text[])": {
+			"description": {"word"},
+			"values":      {"abort"},
+		},
 		// SHOW
 		"SHOW search_path": {
 			"description": {"search_path"},

diff --git a/src/select_remapper.go b/src/select_remapper.go
@@ -444,6 +444,29 @@ func (selectRemapper *SelectRemapper) remapTypecast(node *pgQuery.Node) *pgQuery
 			if typeName == "regclass" {
 				return typeCast.Arg
 			}
+
+			if typeName == "text" {
+				arrayStr := typeCast.Arg.GetAConst().GetSval().Sval
+				arrayStr = strings.Trim(arrayStr, "{}")
+				elements := strings.Split(arrayStr, ",")
+
+				funcCall := &pgQuery.FuncCall{
+					Funcname: []*pgQuery.Node{
+						pgQuery.MakeStrNode("list_value"),
+					},
+				}
+
+				for _, elem := range elements {
+					funcCall.Args = append(funcCall.Args,
+						pgQuery.MakeAConstStrNode(elem, 0))
+				}
+
+				return &pgQuery.Node{
+					Node: &pgQuery.Node_FuncCall{
+						FuncCall: funcCall,
+					},
+				}
+			}
 		}
 	}
 	return node