waitress-core 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d788390d6fd47ac3ea0120241146d3b940217b97
-  data.tar.gz: 162ee7fc128c28cf58dbaf959f8b2a7ea084b3c4
+  metadata.gz: a5fa1729728798da3d7a4ec0949db1a3fe2deae3
+  data.tar.gz: 667594fb143c6d05c2ce4b158f3b35eb388aadf2
 SHA512:
-  metadata.gz: 2af638e5e7084801949f532ea4702c1b58a94660c6ca39fe1030dd400a5430c056ddeb67d43cb190e5c7d9121cc42a722890a304b910e39f3717d2d206f830c3
-  data.tar.gz: 236b011d0e408bc37eef94173fc921b0d687ec053c02b9d835145aedecab5628c28b112dad7888e8318d63a6ae94e41c0432dc5eea8795e1493055a9f71afc40
+  metadata.gz: 83b1ecfba256a594a6e99b7610656257dfdb83445f889ab907e952c3757c7745e447da388e8be16335f62d483f1c88c013d2c306b23be2f1335b0b762654ed3a
+  data.tar.gz: 2fe5debcbbc9770416491d0a876d5190e7d6ab802858ed10654dac67b9f033d327d24fe32be8612e2954433a88704dc9cfee05e09e03acbb3c3259d136a9f52c

data/lib/waitress.rb

@@ -16,6 +16,7 @@ require 'waitress/handlers/handler404'
 require 'waitress/handlers/libhandler'
 require 'waitress/chef'
 require 'waitress/evalbind'
+require 'waitress/restful/restbuilder'
 
 require 'waitress_http11'
 

data/lib/waitress/restful/restbuilder.rb

@@ -0,0 +1,79 @@
+module Waitress
+  class REST < Handler
+
+    require 'json'
+    require 'scon'
+    require 'yaml'
+
+    attr_accessor :priority
+
+    SUPPORTED_FORMATS = [:json, :scon, :yaml, :yml]
+
+    def self.build! schema, &call
+      pattern = /:([^\/:\?\[]+)(\?)?(\[[a-zA-Z0-9\-_\s,]+\])?\//
+
+      a = schema.gsub(pattern) do |match|
+        capture_name = $1
+        optional = ($2 == "?")
+        enumerations = nil
+        enumerations = $3.gsub(/\[|\]/, "").split(/,\s?/) unless $3.nil?
+        buildRe = "(?<#{capture_name}>"
+        if enumerations.nil?
+          buildRe << "[^\\/]+"
+        else
+          buildRe << enumerations.join("|")
+        end
+        buildRe << ")"
+        buildRe << "?" if optional
+        buildRe << "/"
+        buildRe << "?" if optional
+
+        buildRe
+      end
+
+      Waitress::REST.new(Regexp.new(a), &call)
+    end
+
+    def initialize regex, &call
+      @regex = regex
+      @call = call
+      @priority = 200
+    end
+
+    def respond? request, vhost
+      (request.uri =~ @regex) != nil
+    end
+
+    def encode_format content_hash, request, response, type
+      ret = ""
+      if type == :json
+        if request.get_query.include? "pretty"
+          ret = JSON.pretty_generate(content_hash)
+        else
+          ret = JSON.generate(content_hash)
+        end
+      elsif type == :scon
+        ret = SCON.generate!(content_hash)
+      elsif type == :yaml || type == :yml
+        ret = content_hash.to_yaml
+      end
+      ret
+    end
+
+    def serve request, response, client, vhost
+      match = request.uri.match @regex
+
+      form = :json
+      if (request.get_query.include? "format")
+        val = request.get_query["format"].downcase.to_sym
+        form = val if SUPPORTED_FORMATS.include?(val)
+      end
+
+      response.mime ".#{form.to_s}"
+
+      content_hash = @call.call(match, request, response, vhost)
+      response.body encode_format(content_hash, request, response, form)
+    end
+
+  end
+end

data/lib/waitress/version.rb

@@ -1,3 +1,3 @@
 module Waitress
-  VERSION = "0.3.0"
+  VERSION = "0.4.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: waitress-core
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Jaci Brunning
@@ -226,6 +226,7 @@ files:
 - lib/waitress/resources/http/img/404.png
 - lib/waitress/resources/http/index.html
 - lib/waitress/response.rb
+- lib/waitress/restful/restbuilder.rb
 - lib/waitress/server.rb
 - lib/waitress/util/less_watcher.rb
 - lib/waitress/util/util.rb