mini_mime 0.1.0

data/lib/mini_mime.rb

@@ -0,0 +1,173 @@
+require "mini_mime/version"
+require "thread"
+
+module MiniMime
+  BINARY_ENCODINGS = %w(base64 8bit)
+
+  # return true if this filename is known to have binary encoding
+  #
+  # puts MiniMime.binary?("file.gif") => true
+  # puts MiniMime.binary?("file.txt") => false
+  def self.binary?(filename)
+    info = Db.lookup_by_filename(filename)
+    !!(info && BINARY_ENCODINGS.include?(info.encoding))
+  end
+
+  # return true if this content type is known to have binary encoding
+  #
+  # puts MiniMime.binary_content_type?("text/plain") => false
+  # puts MiniMime.binary?("application/x-compress") => true
+  def self.binary_content_type?(mime)
+    info = Db.lookup_by_content_type(mime)
+    !!(info && BINARY_ENCODINGS.include?(info.encoding))
+  end
+
+  # return first matching content type for a file
+  #
+  # puts MiniMime.content_type("test.xml") => "application/xml"
+  # puts MiniMime.content_type("test.gif") => "image/gif"
+  def self.content_type(filename)
+    info = Db.lookup_by_filename(filename)
+    info && info.content_type
+  end
+
+  class Info
+    attr_accessor :extension, :content_type, :encoding
+    def initialize(buffer)
+      @extension,@content_type,@encoding = buffer.split(/\s+/).map!(&:freeze)
+    end
+
+    def [](idx)
+      if idx == 0
+        @extension
+      elsif idx == 1
+        @content_type
+      elsif idx == 2
+        @encoding
+      end
+    end
+  end
+
+  class Db
+    LOCK = Mutex.new
+
+    def self.lookup_by_filename(filename)
+      extension = File.extname(filename)
+      if extension
+        extension.sub!(".", "")
+        extension.downcase!
+        if extension.length > 0
+          LOCK.synchronize do
+            @db ||= new
+            @db.lookup_by_extension(extension)
+          end
+        else
+          nil
+        end
+      end
+    end
+
+    def self.lookup_by_content_type(content_type)
+      LOCK.synchronize do
+        @db ||= new
+        @db.lookup_by_content_type(content_type)
+      end
+    end
+
+
+    class Cache
+      def initialize(size)
+        @size = size
+        @hash = {}
+      end
+
+      def []=(key,val)
+        rval = @hash[key] = val
+        if @hash.length > @size
+          @hash.shift
+        end
+        rval
+      end
+
+      def fetch(key, &blk)
+        @hash.fetch(key, &blk)
+      end
+    end
+
+    class RandomAccessDb
+
+      MAX_CACHED = 100
+
+      def initialize(name, sort_order)
+        @path = File.expand_path("../db/#{name}", __FILE__)
+        @file = File.open(@path)
+
+        @row_length = @file.readline.length
+        @file_length = @file.size
+        @rows = @file_length / @row_length
+
+        @hit_cache = Cache.new(MAX_CACHED)
+        @miss_cache = Cache.new(MAX_CACHED)
+
+        @sort_order = sort_order
+      end
+
+      def lookup(val)
+        @hit_cache.fetch(val) do
+          @miss_cache.fetch(val) do
+            data = lookup_uncached(val)
+            if data
+              @hit_cache[val] = data
+            else
+              @miss_cache[val] = nil
+            end
+
+            data
+          end
+        end
+      end
+
+      #lifted from marcandre/backports
+      def lookup_uncached(val)
+        from = 0
+        to = @rows - 1
+        result = nil
+
+        while from <= to do
+          midpoint = from + (to-from).div(2)
+          current = resolve(midpoint)
+          data = current[@sort_order]
+          if data > val
+            to = midpoint - 1
+          elsif data < val
+            from = midpoint + 1
+          else
+            result = current
+            break
+          end
+        end
+        result
+      end
+
+      def resolve(row)
+        @file.seek(row*@row_length)
+        Info.new(@file.readline)
+      end
+
+    end
+
+    def initialize
+      @ext_db = RandomAccessDb.new("ext_mime.db", 0)
+      @content_type_db = RandomAccessDb.new("content_type_mime.db", 1)
+    end
+
+    def lookup_by_extension(extension)
+      @ext_db.lookup(extension)
+    end
+
+    def lookup_by_content_type(content_type)
+      @content_type_db.lookup(content_type)
+    end
+
+  end
+end

data/lib/mini_mime/version.rb

@@ -0,0 +1,3 @@
+module MiniMime
+  VERSION = "0.1.0"
+end

data/mini_mime.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+# frozen_string_literal: true
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'mini_mime/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "mini_mime"
+  spec.version       = MiniMime::VERSION
+  spec.authors       = ["Sam Saffron"]
+  spec.email         = ["sam.saffron@gmail.com"]
+
+  spec.summary       = %q{A lightweight mime type lookup toy}
+  spec.description   = %q{A lightweight mime type lookup toy}
+  spec.homepage      = "https://github.com/discourse/mini_mime"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.13"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "minitest", "~> 5.0"
+  spec.add_development_dependency "mime-types", "~> 3"
+end

metadata

@@ -0,0 +1,115 @@
+--- !ruby/object:Gem::Specification
+name: mini_mime
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Sam Saffron
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2016-12-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: mime-types
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+description: A lightweight mime type lookup toy
+email:
+- sam.saffron@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- CHANGELOG
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- lib/db/content_type_mime.db
+- lib/db/ext_mime.db
+- lib/mini_mime.rb
+- lib/mini_mime/version.rb
+- mini_mime.gemspec
+homepage: https://github.com/discourse/mini_mime
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: A lightweight mime type lookup toy
+test_files: []