require "uri"
require "../handler"
require "../mime-type"
require "../views/html/page"

class Guff::Handlers::GuffStuffHandler < Guff::Handler
  def initialize(models)
    super(models)
    @digests = {} of String => String
  end

  def call(context : HTTP::Server::Context)
    path = context.request.path.not_nil!

    if matching_request?(context.request.method, path)
      reply(context)
    else
      call_next(context)
    end
  end

  private def reply(context : HTTP::Server::Context)
    # get expanded path to file
    if path = expand_path(context.request.path.not_nil!)
      # get file digest
      file_digest = digest(path)

      # check for cache header
      if context.request.headers["if-none-match"]? == file_digest
        # cached, send 304 not modified
        context.response.status_code = 304

        # FIXME?
        # context.response.headers["connection"] = "close"
        # context.response.write("".to_slice)
      else
        # not cached, set code and send headers
        context.response.status_code = 200
        context.response.content_type = Guff::MimeType.mime_type(path)
        context.response.content_length = File.size(path)
        context.response.headers["etag"] = file_digest

        if context.request.method == "GET"
          # send body
          File.open(path) do |fh|
            IO.copy(fh, context.response)
          end
        end
      end
    else
      # file not found
      context.response.status_code = 404

      # FIXME?
      # context.response.write("".to_slice)
    end

    # FIXME?
    # context.response.close
  end

  VALID_METHODS = %w{GET HEAD}
  PATH_RE = %r{^/guff-stuff/}

  private def matching_request?(method, path)
    VALID_METHODS.includes?(method) && PATH_RE.match(path)
  end

  private def expand_path(req_path : String) : String?
    # unescape path, check for nil byte
    path = URI.unescape(req_path)
    return nil if path.includes?('\0')

    # build absolute path
    r = File.join(
      @models.config["stuff"],
      File.expand_path(path.gsub(PATH_RE, ""), "/")
    )

    # return path if file exists, or nil otherwise
    File.file?(r) ? r : nil
  end

  private def digest(path : String) : String
    @digests[path] ||= OpenSSL::Digest.new("SHA1").file(path).hexdigest
  end
end