abiraj 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d96e91447aac465771ed31896d6f99e959f61cca70f4bce689a48b15410feef6
-  data.tar.gz: c3bbd736acae1d901bbe6a7185f57aabca7a454a57686a3f08607edf72a05fae
+  metadata.gz: 07dd79f7c92f15bf9aa690977178a66a52dfc9d7bb050721d3beee183b8b2de3
+  data.tar.gz: 64d625886da84d225c3ccdc8cf2e38af4835cb7fd55c7c7e02699fc54b7c2a0f
 SHA512:
-  metadata.gz: c46a230e1a113a33e2103ca84a1bcb12782934b98f5968dc2ac826e8036ef020aaca4b7afc57f1cbf17463e7223ba81f3593dfaf4ec61a2407f7d80006e8562d
-  data.tar.gz: 23be80da0ad17c335482ea27fdb147f83fe19d3a3d4dd8815fffa830e618d67488df0088a4233b727c8f60954c592acda65e958da4ce987d3cff7272650b5d28
+  metadata.gz: f75573ba26f415d651d21538fb32e1341d3c9ad417d3bd8e0c7fa63b2603866dd4c00817592e54811256e0f49f800392dc13dbdce9aa264588ead0366517350e
+  data.tar.gz: d8824ca8b7af97b590b80cff3d353373e11d013d9109f904f5f1b52f3a6353b6fd3556c59c1e3f8520abf6397eb9cdea319fc10c611da8f1bde431e146ac4ac2

data/lib/abiraj.rb

@@ -1,122 +1,286 @@
-require "net/http"
-require "json"
-require 'openssl'
-
-module Abiraj
-  class Error < StandardError; end
-  
-  def self.logger(message:, level: 'info')
-    level = 'warn' if level == 'error'
-    prefix = "[ABIRAJ] "
-    if defined?(Chef) && Chef.const_defined?(:Log)
-      case level.downcase
-      when "error"
-        Chef::Log.error(prefix + message)
-      when "debug"
-        Chef::Log.debug(prefix + message)
-      when "warn"
-        Chef::Log.warn(prefix + message)
-      else
-        Chef::Log.info(prefix + message)
-      end
-    end
-  end
-  
-  class Account
-    def initialize(server_url:, authtoken:, org: nil, certificate: nil)
-      @init = false
-      
-      # Validate inputs
-      if server_url.nil? || authtoken.nil? || server_url.strip.empty? || authtoken.strip.empty?
-        Abiraj.logger(message: "Server URL and authtoken are required to initialize Abiraj.", level: "error")
-        return
-      end
-
-      unless server_url.start_with?('http://', 'https://')
-        Abiraj.logger(message: "Invalid server URL. It must start with http:// or https://", level: "error")
-        return
-      end
-
-      # Load certificate if provided
-      # if server_url.start_with?('https://') && certificate
-      #   begin
-      #     cert_content = File.read(certificate)
-      #     Abiraj.logger(message: "Loaded certificate content successfully.")
-      #     @certificate = OpenSSL::X509::Certificate.new(cert_content)
-      #   rescue StandardError => e
-      #     Abiraj.logger(message: "Failed to read or parse certificate: #{e.message}", level: "error")
-      #     return
-      #   end
-      # end
-      
-      @server_url = server_url.strip
-      @authtoken = authtoken.strip
-      @org = org&.strip
-      @certificate = certificate if certificate
-      @init = true
-      Abiraj.logger(message: "Abiraj initialized successfully.")
-    end
-
-    def get(account_id: nil, account_name: nil, account_title: nil)
-      return log_and_return_nil("Abiraj is not initialized.") unless @init
-
-      Abiraj.logger(message: "Fetching account data.")
-      
-      # Prepare request parameters
-      params = {}
-      params['account_id'] = account_id.to_s if account_id
-      params['account_name'] = account_name unless account_name.to_s.strip.empty?
-      params['account_title'] = account_title unless account_title.to_s.strip.empty?
-
-      # Send GET request
-      account_data = get_request(params, "/api/get_account_details_dict")
-      if account_data
-        Abiraj.logger(message: "Account data fetched successfully.")
-        return account_data
-      else
-        log_and_return_nil("Unable to fetch account data.")
-      end
-    end
-
-    private
-
-    def log_and_return_nil(message)
-      Abiraj.logger(message: message, level: "error")
-      nil
-    end
-
-    def get_request(params, endpoint)
-      Abiraj.logger(message: "Raising API request to Abiraj server.")
-      
-      uri = URI(@server_url + endpoint)
-      uri.query = URI.encode_www_form(params) unless params.empty?
-
-      http = Net::HTTP.new(uri.host, uri.port)
-      http.use_ssl = uri.scheme == 'https'
-      http.verify_mode = OpenSSL::SSL::VERIFY_PEER
-
-      if http.use_ssl? && @certificate
-        http.ca_file = @certificate
-      end
-
-      request = Net::HTTP::Get.new(uri)
-      request['authtoken'] = @authtoken
-      request['org'] = @org if @org
-
-      begin
-        response = http.request(request)
-        Abiraj.logger(message: "HTTP response code: #{response.code}", level: "debug")
-        
-        if response.code.to_i == 200
-          JSON.parse(response.body)
-        else
-          log_and_return_nil("Unexpected response code #{response.code}: #{response.body}")
-        end
-      rescue JSON::ParserError => e
-        log_and_return_nil("Failed to parse JSON response: #{e.message}")
-      rescue StandardError => e
-        log_and_return_nil("HTTP request failed: #{e.message}")
-      end
-    end
-  end
-end
+require "net/http"
+require "json"
+require "openssl"
+
+module Abiraj
+  class Error < StandardError; end
+
+  class << self
+    attr_accessor :server_url, :authtoken, :org, :certificate
+    def log(message:, level: "info")
+      prefix = "[Abiraj] "
+      message = message.to_s
+      case level.downcase
+      when "error"
+        Chef::Log.error(prefix + message)
+        nil
+      when "debug"
+        Chef::Log.debug(prefix + message)
+      when "warn"
+        Chef::Log.warn(prefix + message)
+      when "info"
+        Chef::Log.info(prefix + message)
+      end
+    end
+  end
+
+  class Init
+    def initialize(server_url:, authtoken:, org: nil, certificate: nil)
+      if server_url.nil? || authtoken.nil? || server_url.strip.empty? || authtoken.strip.empty?
+        Abiraj.log(message: "Server URL and authtoken are required to initialize Abiraj", level: "error")
+        return nil
+      end
+
+      unless server_url.start_with?("http://", "https://")
+        Abiraj.log(message: "Invalid server URL. It must start with http:// or https://", level: "error")
+        return nil
+      end
+
+      Abiraj.server_url = server_url.strip
+      Abiraj.authtoken = authtoken.strip
+      Abiraj.org = org unless org.to_s.strip.empty? || org.nil?
+      Abiraj.certificate = certificate
+      Abiraj.log(message: "Abiraj initialized successfully")
+    end
+  end
+
+  class Account
+    def self.get(account_id: nil, account_name: nil, account_title: nil)
+      begin
+        unless Abiraj.server_url && Abiraj.authtoken
+          Abiraj.log(message: "Abiraj is not initialized. Please initialize before using.", level: "error")
+          return nil
+        end
+
+        Abiraj.log(message: "Fetching account data")
+        params = {}
+        params["account_id"] = account_id.to_s if account_id
+        params["account_name"] = account_name unless account_name.to_s.strip.empty?
+        params["account_title"] = account_title unless account_title.to_s.strip.empty?
+        account = self.new.send(:get_request, params, "/api/get_account_details_dict")
+        if account
+          Abiraj.log(message: "Account data fetched successfully")
+          return account
+        else
+          Abiraj.log(message: "Unable to fetch account data.", level: "error")
+          return nil
+        end
+
+      rescue StandardError => e
+        Abiraj.log(message: "Failed to fetch account data: #{e.message}", level: "error")
+        return nil
+      end
+    end
+
+    protected
+
+    def get_request(params, endpoint)
+      Abiraj.log(message: "Raising API request to Abiraj server.")
+      uri = URI(Abiraj.server_url + endpoint)
+      uri.query = URI.encode_www_form(params) unless params.empty?
+      http = set_http(uri)
+      request = Net::HTTP::Get.new(uri)
+      request["authtoken"] = Abiraj.authtoken
+      if Abiraj.org && !Abiraj.org.strip.empty?
+        request["org"] = Abiraj.org
+      end
+      begin
+        response = http.request(request)
+        response = JSON.parse(response.body)
+        if response['status_code'] == 200
+          return response
+        else
+          Abiraj.log(message: "#{response['status_code']}: #{response['message']}", level: "error")
+        end
+      rescue JSON::ParserError => e
+        Abiraj.log(message: "Failed to parse JSON response: #{e.message}", level: "error")
+        return nil
+      rescue StandardError => e
+        Abiraj.log(message: "HTTP request failed: #{e.message}", level: "error")
+        return nil
+      end
+    end
+
+    def set_http(uri)
+      begin
+        Abiraj.log(message: "Setting up HTTP connection.")
+        
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = uri.scheme == "https"
+
+        if uri.scheme == "https"
+          http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+          configure_ssl(http, uri)
+        end
+        http
+      rescue StandardError => e
+        Abiraj.log(message: "Failed to set HTTP connection: #{e.message}", level: "error")
+        return nil
+      end
+    end
+
+    def configure_ssl(http, uri)
+      begin
+        if Abiraj.server_url.start_with?("https://")
+          if Abiraj.certificate
+            handle_ssl_certificate(http)
+          else
+            Abiraj.log(message: "No SSL certificate provided. Attempting to fetch from server.")
+            fetch_and_set_server_certificate(http, uri)
+          end
+        end
+      rescue StandardError => e
+        Abiraj.log(message: "Failed to configure SSL: #{e.message}", level: "error")
+        return nil
+      end
+    end
+
+    def handle_ssl_certificate(http)
+      begin
+        Abiraj.log(message: "Adding SSL certificate.")
+        if Abiraj.certificate.start_with?("-----BEGIN CERTIFICATE-----")
+          cert_object = OpenSSL::X509::Certificate.new(Abiraj.certificate)
+          http.cert_store = OpenSSL::X509::Store.new
+          http.cert_store.add_cert(cert_object)
+        elsif File.exist?(Abiraj.certificate)
+          http.ca_file = Abiraj.certificate
+        else
+          Abiraj.log(message: "Invalid certificate value or file path", level: "error")
+          return nil
+        end
+        http
+      rescue OpenSSL::X509::CertificateError => e
+        Abiraj.log(message: "Invalid certificate format: #{e.message}", level: "error")
+        return nil
+      end
+    end
+
+    def fetch_and_set_server_certificate(http, uri)
+      begin
+        tcp_socket = TCPSocket.new(uri.host, uri.port || 443)
+        ssl_socket = OpenSSL::SSL::SSLSocket.new(tcp_socket)
+        ssl_socket.connect
+        server_cert = ssl_socket.peer_cert
+        ssl_socket.close
+        tcp_socket.close
+        http.cert_store = OpenSSL::X509::Store.new
+        http.cert_store.add_cert(server_cert)
+        Abiraj.log(message: "Certificate fetched and added to HTTP connection.")
+      rescue StandardError => e
+        Abiraj.log(message: "Failed to fetch SSL certificate: #{e.message}. Disabling SSL verification.", level: "warn")
+        http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+      end
+    end
+  end
+
+  class Accounts
+    def self.get(account_ids: [])
+      if Abiraj.server_url.nil? || Abiraj.authtoken.nil? || Abiraj.server_url.strip.empty? || Abiraj.authtoken.strip.empty?
+        Abiraj.log(message: "Abiraj is not initialized. Please initialize before using.", level: "error")
+        return nil
+      end
+      securden_accounts = self.new.send(:post_request, account_ids, "/api/get_multiple_accounts_details")
+    end
+
+    protected
+
+    def post_request(data, request_path)
+      Abiraj.log(message: "Raising API request to Abiraj server.")
+      payload = { "account_ids": data }
+      uri = URI(Abiraj.server_url + request_path)
+      http = set_http(uri)
+      request = Net::HTTP::Post.new(uri)
+      request["authtoken"] = Abiraj.authtoken
+      request["Content-Type"] = "application/json" # Set the content type for JSON data
+      request["org"] = Abiraj.org if Abiraj.org && !Abiraj.org.strip.empty?
+      request.body = payload.to_json unless payload.nil? || payload.empty?
+
+      begin
+        response = http.request(request)
+        response = JSON.parse(response.body)
+        if response['status_code'] == 200
+          return response
+        else
+          Abiraj.log(message: "#{response['status_code']}: #{response['message']}", level: "error")
+        end
+      rescue JSON::ParserError => e
+        Abiraj.log(message: "Failed to parse JSON response: #{e.message}", level: "error")
+        return nil
+      rescue StandardError => e
+        Abiraj.log(message: "HTTP request failed: #{e.message}", level: "error")
+        return nil
+      end
+    end
+
+    def set_http(uri)
+      begin
+        Abiraj.log(message: "Setting up HTTP connection.")
+        
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = uri.scheme == "https"
+
+        if uri.scheme == "https"
+          http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+          configure_ssl(http, uri)
+        end
+        http
+      rescue StandardError => e
+        Abiraj.log(message: "Failed to set HTTP connection: #{e.message}", level: "error")
+        return nil
+      end
+    end
+
+    def configure_ssl(http, uri)
+      begin
+        if Abiraj.server_url.start_with?("https://")
+          if Abiraj.certificate
+            handle_ssl_certificate(http)
+          else
+            Abiraj.log(message: "No SSL certificate provided. Attempting to fetch from server.")
+            fetch_and_set_server_certificate(http, uri)
+          end
+        end
+      rescue StandardError => e
+        Abiraj.log(message: "Failed to configure SSL: #{e.message}", level: "error")
+        return nil
+      end
+    end
+
+    def handle_ssl_certificate(http)
+      begin
+        Abiraj.log(message: "Adding SSL certificate.")
+        if Abiraj.certificate.start_with?("-----BEGIN CERTIFICATE-----")
+          cert_object = OpenSSL::X509::Certificate.new(Abiraj.certificate)
+          http.cert_store = OpenSSL::X509::Store.new
+          http.cert_store.add_cert(cert_object)
+        elsif File.exist?(Abiraj.certificate)
+          http.ca_file = Abiraj.certificate
+        else
+          Abiraj.log(message: "Invalid certificate value or file path", level: "error")
+          return nil
+        end
+        http
+      rescue OpenSSL::X509::CertificateError => e
+        Abiraj.log(message: "Invalid certificate format: #{e.message}", level: "error")
+        return nil
+      end
+    end
+
+    def fetch_and_set_server_certificate(http, uri)
+      begin
+        tcp_socket = TCPSocket.new(uri.host, uri.port || 443)
+        ssl_socket = OpenSSL::SSL::SSLSocket.new(tcp_socket)
+        ssl_socket.connect
+        server_cert = ssl_socket.peer_cert
+        ssl_socket.close
+        tcp_socket.close
+        http.cert_store = OpenSSL::X509::Store.new
+        http.cert_store.add_cert(server_cert)
+        Abiraj.log(message: "Certificate fetched and added to HTTP connection.")
+      rescue StandardError => e
+        Abiraj.log(message: "Failed to fetch SSL certificate: #{e.message}. Disabling SSL verification.", level: "warn")
+        http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+      end
+    end
+  end
+end

metadata

@@ -1,25 +1,25 @@
 --- !ruby/object:Gem::Specification
 name: abiraj
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Abiraj
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-12-27 00:00:00.000000000 Z
+date: 2025-01-01 00:00:00.000000000 Z
 dependencies: []
 description: This plugin sends a GET request to an API to fetch account data (passwords,
   etc.) based on the account ID, name, and title.
 email:
-- contact@abiraj.com
+- devops-support@abiraj.com
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
 - lib/abiraj.rb
-- lib/securden/version.rb
+- lib/abiraj/version.rb
 homepage: 
 licenses: []
 metadata: {}