From a321bfe7b2f6db5078de7b2e5ed5ddcccd65f319 Mon Sep 17 00:00:00 2001
From: NeonXP <i@neonxp.dev>
Date: Wed, 16 Nov 2022 05:11:19 +0300
Subject: initial commit

---
 model/query.go | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 model/query.go

(limited to 'model/query.go')

diff --git a/model/query.go b/model/query.go
new file mode 100644
index 0000000..75ee6b0
--- /dev/null
+++ b/model/query.go
@@ -0,0 +1,33 @@
+package model
+
+import (
+	"fmt"
+	"strconv"
+)
+
+// Query returns node by array query
+func (n *Node) Query(query []string) (*Node, error) {
+	if len(query) == 0 {
+		return n, nil
+	}
+	head, rest := query[0], query[1:]
+	switch n.Type {
+	case ArrayNode:
+		idx, err := strconv.Atoi(head)
+		if err != nil {
+			return nil, fmt.Errorf("index must be a number, got %s", head)
+		}
+		next, err := n.Index(idx)
+		if err != nil {
+			return nil, err
+		}
+		return next.Query(rest)
+	case ObjectNode:
+		next, err := n.Get(head)
+		if err != nil {
+			return nil, err
+		}
+		return next.Query(rest)
+	}
+	return nil, fmt.Errorf("can't get %s from node type %s", head, n.Type)
+}
-- 
cgit v1.2.3