fix(magic.go): use offset according to each filetype (#3)

Co-authored-by: Liam Galvin <liam@liam-galvin.co.uk>
liamg · Jul 28, 2023 · 3fda65c · 3fda65c
1 parent e2b75d5
commit 3fda65c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/magic.go b/magic.go
@@ -108,7 +108,7 @@ func worker(ctx context.Context, work chan job) {
 			select {
 			case <-ctx.Done():
 				return
-			case job.resultChan <- job.reference.check(job.input, 0):
+			case job.resultChan <- job.reference.check(job.input, job.reference.Offset):
 			}
 		}
 	}

diff --git a/magic_test.go b/magic_test.go
@@ -66,5 +66,11 @@ func TestNestedLookupSync(t *testing.T) {
 	require.Nil(t, err)
 
 	assert.Equal(t, fileType.Extension, "wav")
+}
+
+func TestLookupWithOffset(t *testing.T) {
+	fileType, err := Lookup([]byte{0x00, 0x00, 0x00, 0x20, 0x66, 0x74, 0x79, 0x70, 0x33, 0x67, 0x97})
+	require.Nil(t, err)
 
+	assert.Equal(t, fileType.Extension, "3gp")
 }