hobby-rpc 0.0.0 → 0.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 34cd6d8707a326830ee359ffd413ca313a4893fbe5545cbb3423171a659f57a6
-  data.tar.gz: c7e6e3f1202778b76b956d46d087951c0fc5b53ed9779c7b775898825e3b014e
+  metadata.gz: 448a25904b66f7dcc5aa8dcdd8cd9f9c93a49b2b92cc1f6e0b10876e3ff226f0
+  data.tar.gz: c547bfeac7df6e8b9385e0fa22935c0e19363e09e2c2716da93cbd9683d13cd8
 SHA512:
-  metadata.gz: 89ac9149cff9c6c44608cd97ec8a2ac8ab83ce069fc27c65279903f25a9d422edb8def1a973ff9d73f20e9f649d5250f35a595e4255b7d5760eff26dffbe9304
-  data.tar.gz: ccb6c62344c3dbdc7c25628768eb9e38ec87f236bf0fef59de9fbeddea7372a8f49dd14a6fe39f447f5f78d9176b9d3a81e2985f7f9aff2b61fcb7405b2d5571
+  metadata.gz: c6f5a67aa5da6f11076c89f612c75632cd5545d264e6671f84cc2f2f6851c27d62e8d5b1431d07e0ad070bc0ca42969ea2a464deabeb0f78cc457d74415a019a
+  data.tar.gz: e39d939760b011b2fce2069abb3c0e712dd3186b1fcbe5ce057633ecb40fcb6d378162dbc34c18647015e77970908909bbaa4f342693eb330d14f13bdae425e4

data/lib/hobby/rpc.rb

@@ -1,23 +1,34 @@
 module Hobby
   class RPC
     class CORS
-      def initialize app
+      def initialize app, origins: nil
         @app = app
+        @origins = origins
       end
       
       def call env
+        origin = '*'
+      
+        if @origins
+          if @origins.include? env['HTTP_ORIGIN']
+            origin = env['HTTP_ORIGIN']
+          else
+            return [400, {}, '400']
+          end
+        end
+      
         case env['REQUEST_METHOD']
         when 'OPTIONS'
           response = Rack::Response.new
       
-          response.add_header 'Access-Control-Allow-Origin', '*'
+          response.add_header 'Access-Control-Allow-Origin', origin
           response.add_header 'Access-Control-Allow-Methods', 'POST, OPTIONS'
           response.add_header 'Access-Control-Allow-Headers', 'Authorization, Content-Type'
           response.add_header 'Access-Control-Max-Age', '86400'
       
           response.to_a
         else
-          status, headers, body = @app.call(env).to_a
+          status, headers, body = @app.call env
           headers['Access-Control-Allow-Origin'] = '*'
           [status, headers, body]
         end
@@ -54,8 +65,6 @@ module Hobby
     include Auth[User]
     include JSON::Keys
     
-    use CORS
-    
     key :fn, String
     optional {
       key :in
@@ -75,5 +84,13 @@ module Hobby
         response.status = 400
       end
     }
+    
+    def initialize hash = {}
+      if cors_origins = hash[:cors_origins]
+        use CORS, origins: cors_origins
+      else
+        use CORS
+      end
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hobby-rpc
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.0.1
 platform: ruby
 authors:
 - Anatoly Chernov