update transform.go

DarkNeos/transform.go

@@ -2,12 +2,14 @@ package darkneos
 
 import (
 	"errors"
+	"unicode/utf16"
 
-	"github.com/sktt1ryze/ygopro-proxy/DarkNeos/ygoprobuffer"
 	"github.com/sktt1ryze/ygopro-proxy/DarkNeos/ygopropb"
+	"google.golang.org/protobuf/proto"
 )
 
 const FILLING_TOKEN uint16 = 0xcccc;
+const UTF16_BUFFER_MAX_LEN int = 20
 
 const (
   ProtobufToRawBuf = 1
@@ -16,25 +18,61 @@ const (
 
 func Transform(src []byte, tranformType int) ([]byte, error) {
   if tranformType == ProtobufToRawBuf {
-    return make([]byte, 0), nil
+    message := &ygopropb.YgoCtosMsg{}
+    err := proto.Unmarshal(src, message)
+    if err != nil {
+      return nil, err
+    }
+    
+    switch message.Msg.(type) {
+      case *(ygopropb.YgoCtosMsg_CtosPlayerInfo):
+        return TransformPlayerInfo(message.GetCtosPlayerInfo()), nil
+      default:
+        return nil, errors.New("Unhandled YgoCtosMsg type")
+    }
   } else if tranformType == RawBufToProtobuf {
-    // todo
-    return make([]byte, 0), nil
+    return nil, errors.New("Unhandled tranformType")
   } else {
     return nil, errors.New("Unknown tranformType")
   }
 }
 
 
-func TransformPlayerInfo(pb ygopropb.CtosPlayerInfo) ygoprobuffer.CtosPlayerInfo {
-  name := [ygoprobuffer.PLAYER_NAME_MAX_LEN]uint16 {}
-  for i := range name {
-    name[i] = FILLING_TOKEN
+func TransformPlayerInfo(pb *ygopropb.CtosPlayerInfo) []byte {
+  info := strToUtf16Buffer(pb.Name)
+
+  return uint16BufToByteBuf(info)
+}
+
+
+func strToUtf16Buffer(s string) []uint16 {
+  b := make([]uint16, UTF16_BUFFER_MAX_LEN, UTF16_BUFFER_MAX_LEN)
+  for i := range b {
+    b[i] = FILLING_TOKEN
   }
 
-  info_buf := ygoprobuffer.CtosPlayerInfo {
-    Name: name,
+  s_utf16 := utf16.Encode([]rune(s))
+
+  // todo: optimize
+  for i, v := range s_utf16 {
+    if i < UTF16_BUFFER_MAX_LEN {
+      b[i] = v
+      
+      if i == len(s_utf16) - 1 && i < len(b) - 1 {
+        b[i + 1] = 0
+      }
+    } else { break }
+  }
+
+  return b
+}
+
+func uint16BufToByteBuf(u16_b []uint16) []byte {
+  b := make([]byte, 0, len(u16_b) * 2)
+
+  for _, v := range u16_b {
+    b = append(b, byte(v >> 8), byte(v))
   }
 
-  return info_buf
+  return b
 }