work on writing nbt files. we can now read and rewrite the simple tes…

…t nbt file perfectly

lib/nbt/tag.rb

@@ -3,14 +3,25 @@ module Tag
     attr_reader :name
     attr_reader :payload
 
-    def type
-      self.class.type
+    def type_id
+      self.class.type_id
+    end
+
+    def binary_type_id
+      # I hope i'm doing this wrong.
+      ::BinData::Int8be.new.read(type_id.chr).to_binary_s
     end
 
     def read_name(io)
       @name = NBT::TagName.new.read(io).data
     end
 
+    def name_to_nbt_string
+      nm = NBT::TagName.new
+      nm.data = @name
+      nm.to_binary_s
+    end
+
     def tag_type_to_class(tag_type)
       NBT::Tag.tag_type_to_class(tag_type)
     end

lib/nbt/tag/byte.rb

@@ -18,7 +18,9 @@ def to_s(indent = 0)
       end
 
       def to_nbt_string(named = true)
-
+        result = binary_type_id
+        result += name_to_nbt_string if named
+        result + ::BinData::Int8be.new.read(@payload).to_binary_s
       end
     end
   end

lib/nbt/tag/compound.rb

@@ -30,7 +30,14 @@ def to_s(indent = 0)
       end
 
       def to_nbt_string(named = true)
+        result = binary_type_id
+        result += name_to_nbt_string if named
 
+        result = @payload.inject(result) do |r, load|
+          r + load.to_nbt_string(true)
+        end
+
+        result + NBT::Tag::End.new(nil).to_nbt_string
       end
 
       def find_tag(name)

lib/nbt/tag/end.rb

@@ -16,7 +16,7 @@ def to_s(indent = 0)
       end
 
       def to_nbt_string(named = false)
-
+        binary_type_id
       end
     end
   end

lib/nbt/tag/string.rb

@@ -6,9 +6,8 @@ class String
       def initialize(io, named = true)
         read_name(io) if named
 
-        l = ::BinData::Int16be.new.read(io).value.to_i
-
-        @payload = io.read(l)
+        # Tag names are actually TAG_Strings, so yay.
+        @payload = NBT::TagName.new.read(io).data
       end
 
       def self.type_id
@@ -20,7 +19,11 @@ def to_s(indent = 0)
       end
 
       def to_nbt_string(named = true)
-
+        result = binary_type_id
+        result += name_to_nbt_string if named
+        tag = NBT::TagName.new
+        tag.data = @payload
+        result + tag.to_binary_s
       end
     end
   end

lib/nbt/tag_name.rb

@@ -2,7 +2,7 @@ module NBT
   class TagName < BinData::Record
     # Spec says this is a TAG_Short, which is signed. Staying strict
     # though this may really be unsigned in practice.
-    int16be :len, :value => lambda { data.length }
+    int16be :len, :value => Proc.new { data.length }
     string :data, :read_length => :len
   end
 end

script.rb

@@ -1,3 +1,5 @@
+# encoding: UTF-8
+
 require 'bundler'
 Bundler.require :default
 
@@ -18,7 +20,11 @@
 end
 
 puts @compound.to_s
-puts
+
+puts @compound.to_nbt_string
+#Zlib::GzipWriter.open('lolwut.nbt') do |gz|
+#  gz.write @compound.to_nbt_string
+#end
 
 Zlib::GzipReader.open('doc/bigtest.nbt') do |f|
   # ostensibly this will always be a single TAG_Compound, per the spec