From 3b2f81ea1f5619b99b6aafcfe565cf9814adc09d Mon Sep 17 00:00:00 2001
From: Tnze <cjd001113@outlook.com>
Date: Wed, 26 May 2021 02:15:45 +0800
Subject: [PATCH] Quoted string

---
 nbt/snbt_decode.go      | 42 ++++++++++++++++++++++++++++++++---------
 nbt/snbt_decode_test.go |  2 +-
 2 files changed, 34 insertions(+), 10 deletions(-)

diff --git a/nbt/snbt_decode.go b/nbt/snbt_decode.go
index 9495812..1229557 100644
--- a/nbt/snbt_decode.go
+++ b/nbt/snbt_decode.go
@@ -1,5 +1,9 @@
 package nbt
 
+import (
+	"strings"
+)
+
 type decodeState struct {
 	data   []byte
 	off    int // next read offset in data
@@ -33,16 +37,12 @@ func writeLiteral(e *Encoder, d *decodeState, tagName string) error {
 	start := d.readIndex()
 	d.scanWhile(scanContinue)
 	literal := d.data[start:d.readIndex()]
-
-	switch literal[0] {
-	case '"', '\'': // TAG_String
-		str := literal // TODO: Parse string
-		e.writeTag(TagString, "")
+	switch v := parseLiteral(literal); v.(type) {
+	case string:
+		str := v.(string)
+		e.writeTag(TagString, tagName)
 		e.writeInt16(int16(len(str)))
-		e.w.Write(str)
-
-	default:
-		e.w.Write(literal) // TODO: Parse other literal
+		e.w.Write([]byte(str))
 	}
 	return nil
 }
@@ -119,3 +119,27 @@ func (d *decodeState) scanWhile(op int) {
 	d.off = len(data) + 1 // mark processed EOF with len+1
 	d.opcode = d.scan.eof()
 }
+
+// parseLiteral parse an SNBT literal, might be
+// TAG_String, TAG_Int, TAG_Float, ... etc.
+// so returned value is one of string, int32, float32 ...
+func parseLiteral(literal []byte) interface{} {
+	switch literal[0] {
+	case '"', '\'': // Quoted String
+		var sb strings.Builder
+		for i := 1; ; i++ {
+			c := literal[i]
+			switch c {
+			case literal[0]:
+				return sb.String()
+			case '\\':
+				i++
+				c = literal[i]
+			}
+			sb.WriteByte(c)
+		}
+	default:
+
+	}
+	panic(phasePanicMsg)
+}
diff --git a/nbt/snbt_decode_test.go b/nbt/snbt_decode_test.go
index f6a43ec..023bbea 100644
--- a/nbt/snbt_decode_test.go
+++ b/nbt/snbt_decode_test.go
@@ -8,7 +8,7 @@ import (
 func TestEncoder_WriteSNBT(t *testing.T) {
 	var buf bytes.Buffer
 	e := NewEncoder(&buf)
-	if err := e.WriteSNBT(`{ abc: a123}`); err != nil {
+	if err := e.WriteSNBT(`{ "abc": "a123"}`); err != nil {
 		t.Fatal(err)
 	}
 	t.Log(buf.Bytes())