From 9323a5f4ef0895404ffabb71278e728a1d5816cf Mon Sep 17 00:00:00 2001
From: Michael Go <michael.go@shopify.com>
Date: Thu, 24 Oct 2024 19:57:02 -0300
Subject: [PATCH] passing tokenizer unit test

---
 lib/liquid/tokenizer.rb | 60 +++++++++++++++++------------------------
 1 file changed, 24 insertions(+), 36 deletions(-)

diff --git a/lib/liquid/tokenizer.rb b/lib/liquid/tokenizer.rb
index 5c539cb0..c2d9f454 100644
--- a/lib/liquid/tokenizer.rb
+++ b/lib/liquid/tokenizer.rb
@@ -8,24 +8,15 @@ module Liquid
       @source         = source
       @line_number    = line_number || (line_numbers ? 1 : nil)
       @for_liquid_tag = for_liquid_tag
-      @offset         = 0
+      @offset = 0
       @state = :outside
       @ss = StringScanner.new(source)
-      @n = 0
     end
 
     def shift
-      @n += 1
-
-      puts @ss.string if @n > 10000
-      puts @ss.pos if @n > 10000
       return nil if @ss.eos?
 
-      token = +""
-
-      if (t = take_new_next)
-        token << t
-      end
+      token = take_new_next
 
       until @state == :outside || @ss.eos?
         token << if @state == :inside_variable || @state == :inside_tag
@@ -41,41 +32,37 @@ module Liquid
     private
 
     def take_rest
-      puts "take_rest: #{@state}"
-      xtake_rest
-    end
-
-    def take_new_next
-      puts "take_new_next: #{@state}"
-      xtake_new_next
-    end
-
-    def take_inside
-      puts "take_inside: #{@state}"
-      xtake_inside
-    end
-
-    def xtake_rest
       token = @ss.rest
       @ss.terminate
       @state = :outside
       token
     end
 
-    "{% foo %}"
-    def xtake_new_next
+    def take_new_next
       start_pos = @ss.pos
 
       if (n = @ss.skip_until(VariableStart))
-        @state = :inside_variable unless n == 2
-        end_pos = @ss.pos
+        if n == 2
+          @state = :inside_variable
+          end_pos = @ss.pos
         token = @source.byteslice(start_pos, end_pos - start_pos)
+        else
+          @ss.pos -= 2
+          token = @source.byteslice(start_pos, n - 2)
+        end
+
         @line_number += @for_liquid_tag ? 1 : token.count("\n") if @line_number
         token
       elsif (n = @ss.skip_until(TagStart))
-        @state = :inside_tag unless n == 2
-        end_pos = @ss.pos
-        token = @source.byteslice(start_pos, end_pos - start_pos)
+        if n == 2
+          @state = :inside_tag
+          end_pos = @ss.pos
+          token = @source.byteslice(start_pos, end_pos - start_pos)
+        else
+          @ss.pos -= 2
+          token = @source.byteslice(start_pos, n - 2)
+        end
+
         @line_number += @for_liquid_tag ? 1 : token.count("\n") if @line_number
         token
       else
@@ -83,19 +70,20 @@ module Liquid
       end
     end
 
-    def xtake_inside
+    def take_inside
       start_pos = @ss.pos
+
       if @state == :inside_variable
         if (n = @ss.skip_until(VariableEnd))
           @state = :outside
-          token = @source.byteslice(start_pos, start_pos + n)
+          token = @source.byteslice(start_pos, n)
           @line_number += @for_liquid_tag ? 1 : token.count("\n") if @line_number
           token
         end
       elsif @state == :inside_tag
         if (n = @ss.skip_until(TagEnd))
           @state = :outside
-          token = @source.byteslice(start_pos, start_pos + n)
+          token = @source.byteslice(start_pos, n)
           @line_number += @for_liquid_tag ? 1 : token.count("\n") if @line_number
           token
         end