net-knocker 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7d734c93c4e362b454bace35ef5f0e46efd227139cdafdb3c19c030a98961f85
-  data.tar.gz: 53de1160612bbd364785b5e97ea84c343eeb27080621fbb0702bc13b0f3329fc
+  metadata.gz: a9789879bc518f157cbecee0b2275e9740cb424bc0e4337b57183c299b8a6c82
+  data.tar.gz: 2817492fa9cadde9138836aa266dfae4120d26c6b45aa4a4a14fd99a84cd8ed3
 SHA512:
-  metadata.gz: c611fc7c1d7eaba3e86db70a4c386e98f8d394d1909d0ebee5f1972149384cf723e94470a20eea622ecb1083fabe588a3fed093ab841136569ec5d06d672d0db
-  data.tar.gz: f8b6973f49d8d1152a4adee0d71c42e5c70c40adfb1fac8ab745695439c1bbd6c9cc047346a47530a811c77bae9c9dfbd859a9212d0a06a9ea53e552bb156a69
+  metadata.gz: f254f92c4758ac0b595244d4bb7aa9556a18fa0ffdedfa80273fe055b466cca7f862f2916601f9d8ac6d36d9576623abccb558147c590f339437b459b4189990
+  data.tar.gz: eff3af079d67b23c36b7d534d78c82c70863124804bb27582b2c913072a77d879b0cfc4ffa84ad766cb05b38d65965526d5ec62d9e379a852356229b148a50c1

data/.pryrc

@@ -1 +1,3 @@
+# frozen_string_literal: true
+
 require_relative 'lib/net/knocker'

data/.rubocop.yml

@@ -1,5 +1,5 @@
 AllCops:
-  TargetRubyVersion: 2.4
+  TargetRubyVersion: 2.5
   NewCops: enable
 
 Style/StringLiterals:
@@ -13,3 +13,6 @@ Style/Documentation:
 
 Layout/LineLength:
   Max: 120
+
+Metrics/MethodLength:
+  Max: 20

data/lib/net/knocker/env.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module Net
+  class Knocker
+    module Env
+      def env_secret
+        @env_secret ||= ENV.fetch('NET_KNOCKER_SECRET')
+      end
+
+      def http_client
+        @http_client ||= Net::HTTP.new(url.host, url.port).tap do |http|
+          http.use_ssl = true
+          http.cert = ssl_certificate
+          http.key = ssl_key
+          http.set_debug_output $stderr if ENV.key? 'NET_KNOCKER_HTTP_DEBUG'
+        end
+      end
+
+      def should_omit?
+        @should_omit ||= (%w[NET_KNOCKER_URL NET_KNOCKER_SECRET] - ENV.keys).any?
+      end
+
+      def ssl_key
+        @ssl_key ||= begin
+          key_class = case ssl_certificate.signature_algorithm
+                      when /^ecdsa/
+                        OpenSSL::PKey::EC
+                      when /^rsa/
+                        OpenSSL::PKey::RSA
+                      when /^dsa/
+                        OpenSSL::PKey::DSA
+                      else
+                        raise "Unknown key type: #{ssl_certificate.signature_algorithm}"
+                      end
+          key_class.new(env_secret)
+        end
+      end
+
+      def ssl_certificate
+        @ssl_certificate ||= OpenSSL::X509::Certificate.new(env_secret)
+      end
+
+      def url
+        @url ||= URI.parse(ENV.fetch('NET_KNOCKER_URL')) unless should_omit?
+      end
+
+      def log
+        @log ||= if defined? logger
+                   logger
+                 elsif defined? Rails.logger
+                   Rails.logger
+                 else
+                   ::Logger.new($stderr)
+                 end
+      end
+    end
+  end
+end

data/lib/net/knocker/peer.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Net
+  class Knocker
+    class Peer
+      def self.any?
+        return unless ENV
+
+        Dir['/proc/*/environ']
+          .select { |path| File.readable_real? path }
+          .grep_v(self_patterns)
+          .select do |path|
+          File.read(path)
+              .then do |content|
+            (env_var_patterns.any? { |pattern| content.include? pattern })
+          end
+        end
+          .any?
+      end
+
+      def self.env_var_patterns
+        @env_var_patterns ||= %W[NET_KNOCKER_URL NET_KNOCKER_SECRET #{ENV['NET_KNOCKER_URL']}].freeze
+      end
+
+      def self.self_patterns
+        @self_patterns ||= %r{proc/((?:thread-)?self|#{Process.pid})/environ}.freeze
+      end
+    end
+  end
+end

data/lib/net/knocker/version.rb

@@ -2,6 +2,6 @@
 
 module Net
   class Knocker
-    VERSION = '0.0.1'
+    VERSION = '0.0.2'
   end
 end

data/lib/net/knocker.rb

@@ -1,89 +1,43 @@
 # frozen_string_literal: true
+
 require_relative './knocker/version'
+require_relative './knocker/env'
+require_relative './knocker/peer'
 require 'logger'
 require 'net/https'
 require 'uri'
 
 module Net
   class Knocker
+    include Knocker::Env
+
     def self.check
       new
     end
 
-    def self.skip_checkin?
-      new.skip_checkin?
-    end
-
-    def checkin_uri
-      @checkin_uri ||= URI.parse(ENV.fetch('NET_KNOCKER_URL'))
-    end
-
-    def log
-      @log ||= if defined? logger
-                  logger
-                elsif defined? Rails.logger
-                Rails.logger
-                else
-                  ::Logger.new($stderr)
-                end
+    def self.should_omit?
+      new.should_omit?
     end
 
     def in
-      return false if skip_checkin?
-
-      check_connection 'PUT'
+      send_request 'PUT'
     end
 
     def out
-      return false if skip_checkin?
+      return log.info("[#{self.class}:#{Process.pid}] omitting send request") if Peer.any?
 
-      check_connection 'DELETE'
-    end
-
-    def skip_checkin?
-      @skip_checkin ||= %w[NET_KNOCKER_URL NET_KNOCKER_SECRET].all? { |k| !(ENV.key? k) }
+      send_request 'DELETE'
     end
 
     private
 
-    def check_connection(http_method)
+    def send_request(http_method)
+      return log.info("[#{self.class}:#{Process.pid}] omitting send request") if should_omit?
+
       http_client.send_request(http_method, '/') do |response|
         log.debug("[#{self.class}][#{__method__}] Public IP: #{response.body} ")
       end
     end
-
-    def env_secret
-      @env_secret ||= ENV.fetch('NET_KNOCKER_SECRET')
-    end
-
-    def http_client
-      @http_client ||= Net::HTTP.new(checkin_uri.host, checkin_uri.port).tap do |http|
-        http.use_ssl = true
-        http.cert = ssl_certificate
-        http.key = ssl_key
-        http.set_debug_output $stderr if ENV.key? 'NET_KNOCKER_HTTP_DEBUG'
-      end
-    end
-
-    def ssl_key
-      @ssl_key ||= begin
-        key_class = case ssl_certificate.signature_algorithm
-                    when /^ecdsa/
-                      OpenSSL::PKey::EC
-                    when /^rsa/
-                      OpenSSL::PKey::RSA
-                    when /^dsa/
-                      OpenSSL::PKey::DSA
-                    else
-                      raise "Unknown key type: #{ssl_certificate.signature_algorithm}"
-                    end
-        key_class.new(env_secret)
-      end
-    end
-
-    def ssl_certificate
-      @ssl_certificate ||= OpenSSL::X509::Certificate.new(env_secret)
-    end
   end
 end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: net-knocker
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Tio Teath
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-04-10 00:00:00.000000000 Z
+date: 2022-04-12 00:00:00.000000000 Z
 dependencies: []
 description: 
 email:
@@ -31,6 +31,8 @@ files:
 - bin/console
 - bin/setup
 - lib/net/knocker.rb
+- lib/net/knocker/env.rb
+- lib/net/knocker/peer.rb
 - lib/net/knocker/version.rb
 - net-knocker.gemspec
 homepage: https://github.com/todosmodos/net-knocker