import std/streams
import std/tables

import utils/twtstr

# extension -> type
type MimeTypes* = distinct Table[string, string]

template getOrDefault*(mimeTypes: MimeTypes; k, fallback: string): string =
  Table[string, string](mimeTypes).getOrDefault(k, fallback)

template contains*(mimeTypes: MimeTypes; k: string): bool =
  k in Table[string, string](mimeTypes)

template hasKeyOrPut*(mimeTypes: var MimeTypes; k, v: string): bool =
  Table[string, string](mimeTypes).hasKeyOrPut(k, v)

# Add mime types found in stream to mimeTypes.
# No error handling for now.
proc parseMimeTypes*(mimeTypes: var MimeTypes; stream: Stream) =
  var line: string
  while stream.readLine(line):
    if line.len == 0:
      continue
    if line[0] == '#':
      continue
    let t = line.untilLower(AsciiWhitespace)
    if t == "": continue
    var i = t.len
    while i < line.len:
      i = line.skipBlanks(i)
      let ext = line.untilLower(AsciiWhitespace, i)
      i += ext.len
      if ext != "":
        discard mimeTypes.hasKeyOrPut(ext, t)
  stream.close()

proc parseMimeTypes*(stream: Stream): MimeTypes =
  var mimeTypes: MimeTypes
  mimeTypes.parseMimeTypes(stream)
  return mimeTypes