Add X-TIMESTAMP-MAP header support

philipgiuliani · Nov 23, 2023 · 2d4b16f · 2d4b16f
1 parent 6eeb754
commit 2d4b16f
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 4 deletions.
diff --git a/src/glubs/webvtt.gleam b/src/glubs/webvtt.gleam
@@ -72,7 +72,12 @@ fn metadata_to_string(metadata: List(#(String, String))) -> StringBuilder {
     False ->
       metadata
       |> list.map(fn(item) {
-        string_builder.from_strings([item.0, ": ", item.1])
+        let separator = case item.0 == "X-TIMESTAMP-MAP" {
+          True -> "="
+          False -> ": "
+        }
+
+        string_builder.from_strings([item.0, separator, item.1])
       })
       |> string_builder.join("\n")
   }
@@ -146,9 +151,14 @@ fn parse_metadata(
 ) -> Result(List(#(String, String)), String) {
   metadata
   |> list.try_map(fn(meta) {
-    case string.split_once(meta, ": ") {
-      Ok(entry) -> Ok(entry)
-      Error(Nil) -> Error("Invalid metadata item")
+    case meta {
+      "X-TIMESTAMP-MAP=" <> header -> Ok(#("X-TIMESTAMP-MAP", header))
+      _other -> {
+        case string.split_once(meta, ": ") {
+          Ok(entry) -> Ok(entry)
+          Error(Nil) -> Error("Invalid metadata item")
+        }
+      }
     }
   })
 }

diff --git a/test/glubs/webvtt_test.gleam b/test/glubs/webvtt_test.gleam
@@ -46,6 +46,24 @@ pub fn parse_header_with_metadata_test() {
   |> should.equal(text)
 }
 
+pub fn parse_header_with_timestamp_map_test() {
+  let text = "WEBVTT\nX-TIMESTAMP-MAP=LOCAL:00:00:00.000,MPEGTS:284722\n"
+  let structured =
+    WebVTT(
+      comment: None,
+      metadata: [#("X-TIMESTAMP-MAP", "LOCAL:00:00:00.000,MPEGTS:284722")],
+      items: [],
+    )
+
+  text
+  |> webvtt.parse()
+  |> should.equal(Ok(structured))
+
+  structured
+  |> webvtt.to_string()
+  |> should.equal(text)
+}
+
 pub fn parse_cue_test() {
   "WEBVTT\n\n1\n00:00.123 --> 00:00.456\nTest"
   |> webvtt.parse()