sitehub 0.4.1

data/lib/sitehub/collection.rb

@@ -0,0 +1,35 @@
+class SiteHub
+  class Collection < Hash
+    class DuplicateVersionException < Exception;
+    end
+
+    def valid?
+      raise "implement me"
+    end
+
+    def resolve
+      raise "implement me"
+    end
+
+    class << self
+      def inherited clazz
+
+        def clazz.overrides
+          @overrides ||=[]
+        end
+
+        def clazz.method_added(name)
+          if name == :add && !overrides.include?(name)
+            overrides << name
+            alias_method :add_backup, :add
+
+            self.send(:define_method, :add) do |id, value, *args|
+              raise DuplicateVersionException, 'supply unique labels' if self[id]
+              add_backup id, value, *args
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/sitehub/collection/route_collection.rb

@@ -0,0 +1,28 @@
+require_relative '../collection'
+class SiteHub
+  class Collection < Hash
+    class RouteCollection < Collection
+
+      def add id, route, *opts
+        self[id] = route
+      end
+
+      def valid?
+        !self.empty?
+      end
+
+      def resolve(env: nil)
+        return self.values.first unless self.values.find { |route| route.rule }
+        result = self.values.find { |route| route.applies?(env) }
+        result && result.resolve(env: env)
+      end
+
+      def transform &block
+        each  do |id, value|
+          self[id] = block.call(value)
+        end
+      end
+
+    end
+  end
+end

data/lib/sitehub/collection/split_route_collection.rb

@@ -0,0 +1,50 @@
+require_relative '../collection'
+require_relative 'split_route_collection/split'
+class SiteHub
+  class Collection < Hash
+    class SplitRouteCollection < Collection
+      class InvalidSplitException < Exception;
+      end
+
+      def initialize(hash={})
+        hash.each do |value, percentage|
+          add(value.id, value, percentage)
+        end
+      end
+
+      def add id, value, percentage
+        raise InvalidSplitException, 'splits must be a Fixnum' unless percentage.is_a?(Fixnum)
+        lower = values.last ? values.last.upper : 0
+        upper = lower + percentage
+
+        raise InvalidSplitException, 'total split percentages can not be greater than 100%' if upper > 100
+        self[id] = Split.new(lower, upper, value)
+      end
+
+      def resolve(*args)
+        random = rand(100)
+        result = values.find { |split| random >= split.lower && random < split.upper }
+        result ? result.value : nil
+      end
+
+      def transform &block
+        values.each do |split|
+          split.value = block.call(split.value)
+        end
+      end
+
+      def [] key
+        result = super
+        result && result.value
+      end
+
+      def valid?
+        last = values.last
+        return true if last && last.upper == 100
+
+        warn('splits do not add up to 100% and no default has been specified')
+        false
+      end
+    end
+  end
+end

data/lib/sitehub/collection/split_route_collection/split.rb

@@ -0,0 +1,18 @@
+class SiteHub
+  class Collection
+    class SplitRouteCollection < Collection
+      class Split
+        attr_reader :upper, :lower
+        attr_accessor :value
+
+        def initialize lower, upper, value
+          @upper, @lower, @value = upper, lower, value
+        end
+
+        def == other
+          other.is_a?(Split) && other.lower == self.lower && other.upper == self.upper
+        end
+      end
+    end
+  end
+end

data/lib/sitehub/constants.rb

@@ -0,0 +1,23 @@
+require 'sitehub/constants/http_header_keys'
+require 'sitehub/constants/rack_http_header_keys'
+
+class SiteHub
+  REQUEST_MAPPING = 'sitehub.request_mapping'.freeze
+  RESPONSE = 'sitehub.response'.freeze
+  ASYNC_CALLBACK = 'async.callback'.freeze
+  RECORDED_ROUTES_COOKIE = 'sitehub.recorded_route'.freeze
+  ERRORS = 'sitehub.errors'.freeze
+  TIME_STAMP_FORMAT = "%d/%b/%Y:%H:%M:%S %z".freeze
+  EMPTY_STRING = ''.freeze
+  UNDERSCORE = '_'.freeze
+  SEMICOLON = ';'.freeze
+  SPACE = ' '.freeze
+  SEMICOLON_WITH_SPACE = "#{SEMICOLON}#{SPACE}".freeze
+  COMMA_WITH_SPACE = ", "
+
+  HYPHEN = '-'.freeze
+  QUESTION_MARK = '?'.freeze
+  EQUALS_SIGN = '='.freeze
+  FULL_STOP = '.'.freeze
+  NEW_LINE = "\n".freeze
+end

data/lib/sitehub/constants/http_header_keys.rb

@@ -0,0 +1,25 @@
+class SiteHub
+  module Constants
+    module HttpHeaderKeys
+
+      ShouldNotTransfer = %w(proxy-connection).freeze
+
+      LOCATION_HEADER = 'location'.freeze
+      HOST_HEADER = 'host'.freeze
+      X_FORWARDED_HOST_HEADER = 'x-forwarded-host'.freeze
+      X_FORWARDED_FOR_HEADER = 'x-forwarded-for'.freeze
+      CONNECTION_HEADER = 'connection'.freeze
+      KEEP_ALIVE = 'keep-alive'.freeze
+      PROXY_AUTHENTICATE = 'proxy-authenticate'.freeze
+      PROXY_AUTHORIZATION = 'proxy-authorization'.freeze
+      TE = 'te'.freeze
+      TRAILERS = 'trailers'.freeze
+      TRANSFER_ENCODING = 'transfer-encoding'.freeze
+      CONTENT_ENCODING = 'content-encoding'.freeze
+      SET_COOKIE = 'Set-Cookie'.freeze
+      CONTENT_LENGTH = 'Content-Length'.freeze
+
+      HopByHop = [CONNECTION_HEADER, KEEP_ALIVE, PROXY_AUTHENTICATE, PROXY_AUTHORIZATION, TE, TRAILERS, TRANSFER_ENCODING, CONTENT_ENCODING].freeze
+    end
+  end
+end

data/lib/sitehub/constants/rack_http_header_keys.rb

@@ -0,0 +1,17 @@
+class SiteHub
+  module Constants
+    module RackHttpHeaderKeys
+      PATH_INFO      = 'PATH_INFO'.freeze
+      REQUEST_METHOD = 'REQUEST_METHOD'.freeze
+      SCRIPT_NAME    = 'SCRIPT_NAME'.freeze
+      QUERY_STRING   = 'QUERY_STRING'.freeze
+      X_FORWARDED_FOR = 'HTTP_X_FORWARDED_FOR'.freeze
+      REMOTE_ADDR = 'REMOTE_ADDR'.freeze
+      REMOTE_USER = 'REMOTE_USER'.freeze
+      HTTP_VERSION = 'HTTP_VERSION'.freeze
+      REMOTE_ADDRESS_ENV_KEY = "REMOTE_ADDR".freeze
+      HTTP_HEADER_FILTER_EXCEPTIONS = %w(CONTENT_TYPE CONTENT_LENGTH).freeze
+      TRANSACTION_ID = 'HTTP_SITEHUB_TRANSACTION_ID'.freeze
+    end
+  end
+end

data/lib/sitehub/cookie.rb

@@ -0,0 +1,54 @@
+require 'sitehub/cookie/attribute'
+require 'sitehub/cookie/flag'
+require 'sitehub/constants'
+class SiteHub
+  class Cookie
+    attr_reader :attributes_and_flags, :name_attribute
+    include Constants
+
+    FIRST = 0
+
+    def initialize cookie_string
+      @attributes_and_flags =[]
+      pairs = cookie_string.split(SEMICOLON).map do |entry|
+        if entry.include?(EQUALS_SIGN)
+          Cookie::Attribute.new(*entry.split(EQUALS_SIGN))
+        else
+          Cookie::Flag.new(entry)
+        end
+      end
+
+      name_attribute = pairs.delete_at(FIRST)
+      attributes_and_flags.concat(pairs)
+      @name_attribute = Cookie::Attribute.new(name_attribute.name.to_s, name_attribute.value)
+
+    end
+
+    def name
+      name_attribute.name
+    end
+
+    def value
+      name_attribute.value
+    end
+
+    def find name
+      attributes_and_flags.find{|entry| entry.name == name}
+    end
+
+    def delete name
+      entry = find(name)
+      @attributes_and_flags = attributes_and_flags.delete_if{|e| e.name == name}
+      entry
+    end
+
+    def == other
+      other.is_a?(self.class) && self.attributes_and_flags.sort{|entry1, entry2| entry1.name <=> entry2.name} == other.attributes_and_flags.sort{|entry1, entry2| entry1.name <=> entry2.name}
+    end
+
+    def to_s
+      [name_attribute].concat(attributes_and_flags).join(SEMICOLON_WITH_SPACE)
+    end
+
+  end
+end

data/lib/sitehub/cookie/attribute.rb

@@ -0,0 +1,22 @@
+require 'sitehub/string_sanitiser'
+class SiteHub
+  class Cookie
+    class Attribute
+      include StringSanitiser
+      attr_accessor :name, :value
+
+      def initialize name, value
+        @name = sanitise(name).to_s.to_sym
+        @value = sanitise(value || EMPTY_STRING)
+      end
+
+      def to_s
+        "#{name}=#{value}"
+      end
+
+      def == other
+        other.is_a?(Attribute) && self.name == other.name && self.value == other.value
+      end
+    end
+  end
+end

data/lib/sitehub/cookie/flag.rb

@@ -0,0 +1,22 @@
+require 'sitehub/string_sanitiser'
+class SiteHub
+  class Cookie
+    class Flag
+      include StringSanitiser
+      attr_accessor :name
+
+      def initialize flag
+        @name = sanitise(flag).to_sym
+      end
+
+      def to_s
+        name.to_s
+      end
+
+      def == other
+        other.is_a?(Flag) && self.name == other.name
+      end
+    end
+
+  end
+end

data/lib/sitehub/cookie_rewriting.rb

@@ -0,0 +1,35 @@
+require 'sitehub/cookie'
+require 'sitehub/constants'
+class SiteHub
+
+  module CookieRewriting
+
+    ENDING_WITH_NEWLINE = /#{NEW_LINE}$/
+
+    def rewrite_cookies(headers, substitute_domain:)
+      cookies_hash = cookies_string_as_hash(headers[Constants::HttpHeaderKeys::SET_COOKIE])
+
+      cookies_hash.values.each do |cookie|
+        if domain_attribute = cookie.find(:domain)
+          value = domain_attribute.value
+          domain_attribute.value = substitute_domain.dup
+          domain_attribute.value.prepend(FULL_STOP) if value.start_with?(FULL_STOP)
+        end
+      end
+      headers[HttpHeaders::SET_COOKIE] = cookies_hash_to_string(cookies_hash)
+    end
+
+    def cookies_hash_to_string cookies_hash
+      cookies_hash.values.inject(EMPTY_STRING.dup) do |cookie_string, cookie|
+        cookie_string << "#{cookie.to_s}#{NEW_LINE}"
+      end.sub(ENDING_WITH_NEWLINE, EMPTY_STRING)
+    end
+
+    def cookies_string_as_hash cookie_string
+      cookie_string.lines.inject({}) do |cookies, cookie_line|
+        cookie = SiteHub::Cookie.new(cookie_line)
+        cookies[cookie.name] = cookie; cookies
+      end
+    end
+  end
+end

data/lib/sitehub/forward_proxies.rb

@@ -0,0 +1,50 @@
+require 'sitehub/constants'
+require 'sitehub/forward_proxy'
+require 'rack/request'
+require 'rack/response'
+require 'rack/utils'
+require 'em-http'
+
+class SiteHub
+  class ForwardProxies
+
+    NOT_FOUND = Rack::Response.new(['page not found'], 404, {})
+    def call env
+      source_request = Rack::Request.new(env)
+
+      forward_proxy = mapped_route(path: source_request.path, request: source_request)
+
+      return forward_proxy.call(env) if forward_proxy
+      NOT_FOUND
+    end
+
+    def init
+      forward_proxies.values.each { |proxy| proxy.build }
+      self
+    end
+
+    def << route
+      forward_proxies[route.mapped_path] = route
+    end
+
+    def mapped_route(path:, request:)
+
+      key = forward_proxies.keys.find do |key|
+        case key
+          when Regexp
+            key.match(path)
+          else
+            path == key
+        end
+      end
+
+      forward_proxy_builder = forward_proxies[key]
+      forward_proxy_builder ? forward_proxy_builder.resolve(id: request.cookies[RECORDED_ROUTES_COOKIE], env: request.env) : nil
+    end
+
+    def forward_proxies
+      @forward_proxies ||= {}
+    end
+
+  end
+end

data/lib/sitehub/forward_proxy.rb

@@ -0,0 +1,67 @@
+require 'sitehub/http_headers'
+require 'sitehub/request_mapping'
+require 'sitehub/rules'
+require 'sitehub/resolver'
+require 'faraday'
+require 'sitehub/constants'
+class SiteHub
+  class ForwardProxy
+
+    ERROR_RESPONSE = Rack::Response.new(['error'], 500, {})
+
+    include HttpHeaders, Rules, Resolver, Constants
+
+    attr_reader :url, :id, :mapped_path, :http_client, :sitehub_cookie_path, :sitehub_cookie_name
+
+    def initialize(url:, id:, mapped_path: nil, rule: nil, sitehub_cookie_path: nil, sitehub_cookie_name:)
+      @id, @url, @rule, @mapped_path = id, url, rule, mapped_path
+      @sitehub_cookie_path = sitehub_cookie_path
+      @sitehub_cookie_name = sitehub_cookie_name
+      @http_client = Faraday.new(ssl: {verify:false}) do |con|
+        con.adapter :em_synchrony
+      end
+    end
+
+    def call env
+      source_request = Rack::Request.new(env)
+
+      request_mapping = request_mapping(source_request)
+
+      mapped_uri = URI(request_mapping.computed_uri)
+
+      request_headers = sanitise_headers(extract_http_headers(source_request.env))
+      request_headers[HOST_HEADER] = "#{mapped_uri.host}:#{mapped_uri.port}"
+      request_headers[X_FORWARDED_HOST_HEADER] = append_host(request_headers[X_FORWARDED_HOST_HEADER].to_s, source_request.url)
+      env[REQUEST_MAPPING] = request_mapping
+
+      begin
+        response = http_client.send(source_request.request_method.downcase, mapped_uri) do |request|
+          request.headers = request_headers
+          request.body = source_request.body.read
+          request.params = source_request.params
+        end
+
+        Rack::Response.new(response.body, response.status, sanitise_headers(response.headers)).tap do |r|
+          r.set_cookie(sitehub_cookie_name, path: (sitehub_cookie_path || source_request.path), value: id)
+        end
+      rescue Exception => e
+        env[ERRORS] << e.message
+        ERROR_RESPONSE.dup
+      end
+    end
+
+    def request_mapping(source_request)
+      RequestMapping.new(source_url: source_request.url, mapped_url: url, mapped_path: mapped_path)
+    end
+
+    def == other
+      other.is_a?(ForwardProxy) && url == other.url
+    end
+
+    private
+    def append_host(forwarded_host, destination_uri)
+      destination_uri = URI(destination_uri)
+      forwarded_host == EMPTY_STRING ? "#{destination_uri.host}:#{destination_uri.port}" : "#{forwarded_host},#{destination_uri.host}"
+    end
+  end
+end