From 053ba4c3ed971f1a710f8e25658df71b7088cf39 Mon Sep 17 00:00:00 2001
From: reid <reid@watcha.com>
Date: Thu, 20 Feb 2020 18:57:59 +0900
Subject: [PATCH] Add cue indentifier

---
 lib/captions/formats/vtt.rb | 10 +++++++++-
 spec/sample/test.vtt        |  4 ++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/lib/captions/formats/vtt.rb b/lib/captions/formats/vtt.rb
index 8ae8943..a8be4d6 100644
--- a/lib/captions/formats/vtt.rb
+++ b/lib/captions/formats/vtt.rb
@@ -7,6 +7,9 @@ class VTT < Base
     # VTT file comments/style section
     VTT_METADATA = /^NOTE|^STYLE/
 
+    # The identifier is a name that identifies the cue.
+    CUE_IDENTIFIER = /^\d+((?!-->).)*$/
+
     # Auto Keyword used in Alignment
     AUTO_KEYWORD = "auto"
 
@@ -23,7 +26,6 @@ class VTT < Base
     # Parse VTT file and update CueList
     def parse
       base_parser do
-        count = 1
         cue_count = 0
         meta_data_section = false
         cue = nil
@@ -32,6 +34,8 @@ def parse
           line = line.strip
           if line.empty?
             meta_data_section = false
+          elsif !meta_data_section and is_cue_identifier?(line)
+            next
           elsif is_meta_data?(line)
             meta_data_section = true
           elsif is_time?(line)
@@ -85,6 +89,10 @@ def is_text?(text)
       !text.empty? and text.is_a?(String) and text != VTT_HEADER
     end
 
+    def is_cue_identifier?(text)
+      text.match?(CUE_IDENTIFIER)
+    end
+
     def set_properties(cue, properties)
       properties.each do |prop|
         prop, value = prop.split(":")
diff --git a/spec/sample/test.vtt b/spec/sample/test.vtt
index 343d942..8411484 100644
--- a/spec/sample/test.vtt
+++ b/spec/sample/test.vtt
@@ -9,6 +9,7 @@ not be 100% accurate
 - I didn’t pay too close attention to
 when the cues should start or end.
 
+1
 00:00:04.080 --> 00:06.320 align:start position:21% line:90%,end
 I'm going to flip through this deck.
 
@@ -16,11 +17,14 @@ STYLE
 ::cue(#\31) { color: lime; }
 ::cue(#crédit\ de\ transcription) { color: red; }
 
+2
 00:06.400 --> 00:07.880 align:middle position:10% line:-1
 And I want you to see one card.
+32
 
 NOTE Comments
 
+3
 00:00:08.300 --> 00:00:09.400 align:middle line:90%,end
 Not this one. That's too obvious.