feat(block): enable block

Author: sanemat

lib/env_pull_request.rb

@@ -2,7 +2,7 @@
 require 'env_pull_request/version'
 
 module EnvPullRequest
-  def self.new
-    Base.new
+  def self.new(&block)
+    Base.new(&block)
   end
 end

lib/env_pull_request/base.rb

@@ -5,8 +5,13 @@ class Base
     include Utils
     attr_reader :pull_request_id
 
-    def initialize
-      @pull_request_id = fetch_pull_request_id
+    def initialize(&block)
+      @pull_request_id =
+        if block_given?
+          block.call || fetch_pull_request_id
+        else
+          fetch_pull_request_id
+        end
     end
 
     def fetch_pull_request_id