Set correct Content-Type

Author: pelf

Gemfile.lock

@@ -2,13 +2,20 @@ PATH
   remote: .
   specs:
     qrtrx (0.1.0)
+      mimemagic (~> 0.3)
       rqrcode (~> 0.10)
 
 GEM
   remote: https://rubygems.org/
   specs:
     chunky_png (1.3.10)
+    coderay (1.1.2)
     diff-lcs (1.3)
+    method_source (0.9.0)
+    mimemagic (0.3.2)
+    pry (0.11.3)
+      coderay (~> 1.1.0)
+      method_source (~> 0.9.0)
     rake (10.5.0)
     rqrcode (0.10.1)
       chunky_png (~> 1.0)
@@ -31,6 +38,7 @@ PLATFORMS
 
 DEPENDENCIES
   bundler (~> 1.16)
+  pry
   qrtrx!
   rake (~> 10.0)
   rspec (~> 3.0)

lib/qrtrx/server.rb

@@ -1,6 +1,7 @@
 require 'rqrcode'
 require 'socket'
 require 'securerandom'
+require 'mimemagic'
 
 module Qrtrx
   class Server
@@ -17,7 +18,6 @@ def start
       socket = server.accept
 
       file_path = File.join(Dir.pwd, file_name)
-
       File.open(file_path, "rb") do |file|
         socket.print http_header(file)
         IO.copy_stream(file, socket)
@@ -50,14 +50,13 @@ def ip_address
 
     def http_header(file)
       "HTTP/1.1 200 OK\r\n" +
-      "Content-Type: #{content_type(file)}\r\n" +
+      "Content-Type: #{content_type}\r\n" +
       "Content-Length: #{file.size}\r\n" +
       "Connection: close\r\n\r\n"
     end
 
-    def content_type(file)
-      # TODO
-      'text/plain'
+    def content_type
+      MimeMagic.by_path(file_name).type
     end
 
   end

qrtrx.gemspec

@@ -33,6 +33,8 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "bundler", "~> 1.16"
   spec.add_development_dependency "rake", "~> 10.0"
   spec.add_development_dependency "rspec", "~> 3.0"
+  # spec.add_development_dependency "pry"
 
   spec.add_dependency "rqrcode", "~> 0.10"
+  spec.add_dependency "mimemagic", "~> 0.3"
 end