vpn, proxy and tor detection working

lib/usual_suspect/configuration.rb

@@ -1,10 +1,10 @@
 # lib/usual_suspect/configuration.rb
 module UsualSuspect
   class Configuration
-    attr_accessor :block_criteria
+    attr_accessor :vpn_api_key
 
     def initialize
-      @block_criteria = []
+      @vpn_api_key = nil
     end
   end
 

lib/usual_suspect/tasks/templates/create_usual_suspect_events.rb.erb

@@ -6,7 +6,9 @@ class CreateUsualSuspectEvents < ActiveRecord::Migration[6.0]
       t.datetime :sign_in_at
       t.boolean :password_change_after_login
       t.boolean :geovelocity_failed
-      t.boolean :vpn_usage
+      t.boolean :using_vpn
+      t.boolean :using_proxy
+      t.boolean :using_tor
       t.string :sign_in_ip
       t.string :city
       t.string :country

lib/usual_suspect/user_extension.rb

@@ -4,14 +4,18 @@ module UserExtension
 
     def update_login(ip, location, session_token)
       event = UsualSuspectEvent.new(user: self, session_token: session_token)
+      vpn_tor_proxy_usage = UsualSuspect::VpnChecker.check_vpn(ip)
 
       event.assign_attributes(
         sign_in_at: Time.current,
         sign_in_ip: ip,
         city: location.city,
         country: location.country,
         latitude: location.latitude,
-        longitude: location.longitude
+        longitude: location.longitude,
+        using_vpn: vpn_tor_proxy_usage['security']['vpn'],
+        using_proxy: vpn_tor_proxy_usage['security']['proxy'],
+        using_tor: vpn_tor_proxy_usage['security']['tor'],
       )
 
       event.save

lib/usual_suspect/vpn_checker.rb

@@ -0,0 +1,20 @@
+module UsualSuspect
+  module VpnChecker
+    def self.check_vpn(ip_address)
+	  api_key = UsualSuspect.configuration.vpn_api_key
+	  return false unless api_key
+
+	  # Call IP2Location API
+	  # Example URL (adjust according to IP2Location's API documentation)
+	  url = "https://vpnapi.io/api/#{ip_address}?key=#{api_key}"
+
+	  response = Net::HTTP.get(URI(url))
+	  result = JSON.parse(response)
+
+	  result
+    rescue => e
+      # Handle errors (e.g., network issues, invalid response)
+      false
+    end
+  end
+end