chassis 0.1.0

data/lib/chassis/form.rb

@@ -0,0 +1,56 @@
+module Chassis
+  UnknownFormFieldError = Chassis.error do |field|
+     "#{field} given but not allowed."
+  end
+
+  class FormModule < Module
+    module InstanceMethods
+      def initialize(hash = {})
+        assert_valid_keys! HashUtils.symbolize(hash)
+        super
+        yield self if block_given?
+      end
+
+      def values
+        dirty_attributes
+      end
+
+      private
+      def assert_valid_keys!(hash)
+        return if accepted_keys.empty?
+
+        hash.keys.each do |key|
+          raise UnknownFormFieldError, key unless accepted_keys.include? key
+        end
+      end
+
+      def accepted_keys
+        self.class.accepted_keys
+      end
+    end
+
+    module ClassMethods
+      def attribute(name, type, options = {})
+        accepted_keys << name.to_sym
+        super(name, type, options)
+      end
+
+      def accepted_keys
+        @accepted_keys ||= []
+      end
+    end
+
+    def included(base)
+      base.include Virtus.model
+      base.extend ClassMethods
+      base.include Virtus::DirtyAttribute
+      base.include InstanceMethods
+    end
+  end
+
+  class << self
+    def form
+      FormModule.new
+    end
+  end
+end

data/lib/chassis/hash_utils.rb

@@ -0,0 +1,16 @@
+module Chassis
+  module HashUtils
+    def symbolize(hash)
+      hash.inject({}) do |memo, pair|
+        key, value = pair
+
+        if value.is_a? Hash
+          memo.merge! key.to_sym => symbolize(value)
+        else
+          memo.merge! key.to_sym => value
+        end
+      end
+    end
+    module_function :symbolize
+  end
+end

data/lib/chassis/heroku.rb

@@ -0,0 +1,5 @@
+# Whatever random things need to be done to make everything
+# behave correctly on heroku.
+
+$stdout.sync = true
+$stderr.sync = true

data/lib/chassis/initializable.rb

@@ -0,0 +1,11 @@
+module Chassis
+  module Initializable
+    def initialize(values = {})
+      super()
+      values.each_pair do |key, value|
+        send "#{key}=", value
+      end
+      yield self if block_given?
+    end
+  end
+end

data/lib/chassis/logger.rb

@@ -0,0 +1,8 @@
+module Chassis
+  class Logger < ::Logger
+    def initialize(logdev = Chassis.stream, shift_age = 0, shift_size = 1048576)
+      super
+      self.level = ENV['LOG_LEVEL'].to_sym if ENV['LOG_LEVEL']
+    end
+  end
+end

data/lib/chassis/observable.rb

@@ -0,0 +1,19 @@
+module Chassis
+  module Observable
+    def add_observer(observer)
+      @observers ||= []
+      @observers << observer
+    end
+
+    private
+    def notify_observers(event, *args)
+      return unless defined? @observers
+
+      @observers.each do |observer|
+        if observer.respond_to? event
+          observer.send event, *[self, args].flatten
+        end
+      end
+    end
+  end
+end

data/lib/chassis/persistence.rb

@@ -0,0 +1,49 @@
+module Chassis
+  module Persistence
+    module ClassMethods
+      def create(*args, &block)
+        record = new(*args, &block)
+        record.save
+        record
+      end
+
+      def repo
+        begin
+          @repo ||= StringUtils.constantize "#{name}Repo"
+        rescue NameError
+          fail "#{name}Repo not defined! Define this method to specify a different repo"
+        end
+      end
+    end
+
+    class << self
+      def included(base)
+        base.class_eval do
+          include Initializable
+          include Equalizer.new(:id)
+
+          attr_accessor :id
+        end
+
+        base.extend ClassMethods
+      end
+    end
+
+    def save
+      yield self if block_given?
+      repo.save self
+    end
+
+    def delete
+      repo.delete self
+    end
+
+    def new_record?
+      id.nil?
+    end
+
+    def repo
+      self.class.repo
+    end
+  end
+end

data/lib/chassis/rack/bouncer.rb

@@ -0,0 +1,33 @@
+module Chassis
+  module Rack
+    class Bouncer
+      BOUNCERS = [
+        lambda { |req| ; req.user_agent =~ /masscan/ }
+      ]
+
+      def initialize(app, &bouncer)
+        @app, @bouncer = app, bouncer
+      end
+
+      def call(env)
+        if bounce?(env)
+          [403, { }, [ ]]
+        else
+          @app.call env
+        end
+      end
+
+      def bounce?(env)
+        request = ::Rack::Request.new env
+
+        bouncers.any? do |bouncer|
+          bouncer.call request
+        end
+      end
+
+      def bouncers
+        [BOUNCERS, @bouncer].flatten.compact
+      end
+    end
+  end
+end

data/lib/chassis/rack/builder_shim_patch.rb

@@ -0,0 +1,7 @@
+module Rack
+  class Builder
+    def shim(middleware, *args, &block)
+      @use.unshift proc { |app| middleware.new(app, *args, &block) }
+    end
+  end
+end

data/lib/chassis/rack/health_check.rb

@@ -0,0 +1,45 @@
+module Chassis
+  HealthCheckError = Class.new RuntimeError
+
+  module Rack
+    class Ping
+      def initialize(app, &block)
+        @app, @block = app, block
+      end
+
+      def call(env)
+        if '/ping' == env.fetch('PATH_INFO')
+          if @block
+            begin
+              result = @block.call(::Rack::Request.new(env))
+              raise "health check did not return correctly" unless result
+            rescue => boom
+              raise HealthCheckError, boom.to_s
+            end
+          end
+
+          [200, {'Content-Type' => 'text/plain'}, ['pong']]
+        else
+          @app.call env
+        end
+      end
+    end
+
+    class HealthCheck
+      def initialize(app)
+        @app = app
+      end
+
+      def call(env)
+        begin
+          @app.call env
+        rescue HealthCheckError => ex
+          env['rack.errors'].write ex.to_s
+          env['rack.errors'].write ex.backtrace.join("\n")
+          env['rack.errors'].flush
+          exit!
+        end
+      end
+    end
+  end
+end

data/lib/chassis/rack/instrumentation.rb

@@ -0,0 +1,20 @@
+require 'rack/runtime'
+
+module Chassis
+  module Rack
+    class Instrumentation
+      def initialize(app, namespace = nil)
+        stack = ::Rack::Builder.new
+        stack.use ::Rack::Runtime
+        stack.use ::Harness::RackInstrumenter, namespace
+        stack.run app
+
+        @app = stack.to_app
+      end
+
+      def call(env)
+        @app.call env
+      end
+    end
+  end
+end

data/lib/chassis/rack/json_body_parser.rb

@@ -0,0 +1,20 @@
+module Chassis
+  module Rack
+    class JsonBodyParser < ::Rack::PostBodyContentTypeParser
+      def call(env)
+        body = env[POST_BODY].read
+
+        if body.strip.empty?
+          @app.call env
+        else
+          begin
+            env[POST_BODY].rewind
+            super
+          rescue JSON::ParserError => ex
+            [400, { 'Content-Type' => 'text/plain' }, [ex.to_s]]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/chassis/rack/no_robots.rb

@@ -0,0 +1,24 @@
+module Chassis
+  module Rack
+    class NoRobots
+      def initialize(app)
+        @app = app
+      end
+
+      def call(env)
+        if env['PATH_INFO'] == '/robots.txt'
+          [200, {'Content-Type' => 'text/plain'}, [robots_txt]]
+        else
+          @app.call env
+        end
+      end
+
+      def robots_txt
+<<-txt
+User Agent: *
+Disallow: /
+txt
+      end
+    end
+  end
+end

data/lib/chassis/registry.rb

@@ -0,0 +1,30 @@
+module Chassis
+  UnregisteredError = Chassis.error do |key|
+    "#{key.inspect} is not registered!"
+  end
+
+  class Registry
+    def initialize
+      @map = { }
+    end
+
+    def []=(key, value)
+      map[key] = value
+    end
+
+    def fetch(key)
+      map.fetch key do
+        fail UnregisteredError, key
+      end
+    end
+
+    def clear
+      map.clear
+    end
+
+    private
+    def map
+      @map
+    end
+  end
+end

data/lib/chassis/repo.rb

@@ -0,0 +1,73 @@
+module Chassis
+  RecordNotFoundError = Chassis.error do |klass, id|
+    "Could not locate #{klass} with id #{id}"
+  end
+
+  QueryNotImplementedError = Chassis.error do |selector|
+    "Adapter does not support #{selector.class}!"
+  end
+
+  GraphQueryNotImplementedError = Chassis.error do |selector|
+    "Adapter does not know how to graph #{selector.class}!"
+  end
+
+  NoQueryResultError = Chassis.error do |selector|
+    "Query #{selector.class} must return results!"
+  end
+
+  class Repo
+    include Interchange.new(*[
+      :all, :find, :create, :update, :delete,
+      :first, :last, :query, :graph_query,
+      :sample, :empty?, :count, :clear,
+      :initialize_storage
+    ])
+
+    class << self
+      def default
+        @default ||= new
+      end
+    end
+
+    def find(klass, id)
+      raise ArgumentError, "id cannot be nil!" if id.nil?
+      super
+    end
+
+    def save(record)
+      if record.id
+        update record
+      else
+        create record
+      end
+    end
+
+    def query!(klass, selector)
+      result = query klass, selector
+      no_results = result.respond_to?(:empty?) ? result.empty? : result.nil?
+
+      if no_results && block_given?
+        yield klass, selector
+      elsif no_results
+        fail NoQueryResultError, selector
+      end
+
+      result
+    end
+  end
+
+  class << self
+    def repo
+      Repo.default
+    end
+  end
+end
+
+require_relative 'repo/delegation'
+require_relative 'repo/lazy_association'
+require_relative 'repo/record_map'
+require_relative 'repo/base_repo'
+require_relative 'repo/null_repo'
+require_relative 'repo/memory_repo'
+require_relative 'repo/redis_repo'
+require_relative 'repo/pstore_repo'

data/lib/chassis/repo/base_repo.rb

@@ -0,0 +1,99 @@
+module Chassis
+  class BaseRepo
+    def create(record)
+      record.id ||= next_id
+      map.set record
+    end
+
+    def update(record)
+      map.set record
+    end
+
+    def delete(record)
+      map.delete record
+    end
+
+    def clear
+      map.clear
+    end
+
+    def all(klass)
+      map.all klass
+    end
+
+    def find(klass, id)
+      map.get klass, id
+    end
+
+    def first(klass)
+      all(klass).first
+    end
+
+    def last(klass)
+      all(klass).last
+    end
+
+    def sample(klass)
+      all(klass).sample
+    end
+
+    def empty?(klass)
+      all(klass).empty?
+    end
+
+    def count(klass)
+      all(klass).count
+    end
+
+    def query(klass, selector)
+      if query_implemented? klass, selector
+        send query_method(klass, selector), klass, selector
+      else
+        raise QueryNotImplementedError, selector
+      end
+    end
+
+    def graph_query(klass, selector)
+      if graph_query_implemented? klass, selector
+        send graph_query_method(klass, selector), klass, selector
+      else
+        raise GraphQueryNotImplementedError, selector
+      end
+    end
+
+    def initialize_storage
+
+    end
+
+    private
+    def map
+      @map
+    end
+
+    def next_id
+      @counter ||= 0
+      @counter = @counter + 1
+      @counter
+    end
+
+    def query_method(klass, selector)
+      "query_#{selector_key(selector)}"
+    end
+
+    def query_implemented?(klass, selector)
+      respond_to? query_method(klass, selector)
+    end
+
+    def graph_query_method(klass, selector)
+      "graph_query_#{selector_key(selector)}"
+    end
+
+    def graph_query_implemented?(klass, selector)
+      respond_to? graph_query_method(klass, selector)
+    end
+
+    def selector_key(selector)
+      StringUtils.underscore(StringUtils.demodulize(selector.class.name))
+    end
+  end
+end