aboutsummaryrefslogtreecommitdiff
path: root/src/image.gleam
blob: 93fe3000a0043296c0efcd4dfbb46fdc687553a1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
pub type ImageInformation {
  ImageInformation(width: Int, height: Int, extension: String)
}

pub fn get_image_information(image) {
  case image {
    <<0x89, "PNG\r\n":utf8, 0x1A, "\n":utf8, _rest:bits>> ->
      parse_png_chunks(image, 8)
    <<
      "GIF":utf8,
      _version:bytes-3,
      width:little-16,
      height:little-16,
      _rest:bits,
    >> -> Ok(ImageInformation(width, height, "gif"))
    _ -> Error("Unsupported image format")
  }
}

fn parse_png_chunks(image, offset) {
  case image {
    <<
      _:unit(8)-size(offset),
      _length:32,
      "IHDR":utf8,
      width:32,
      height:32,
      _rest:bits,
    >> -> Ok(ImageInformation(width, height, "png"))
    <<_:size(offset), length:32, _:4, _:bits>> ->
      parse_png_chunks(image, offset + length + 12)
    _ -> Error("Invalid PNG chunk")
  }
}