abiraj 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b5894ed4fd9f1f42f60427ef95452ad54a5933cd32c3454ea11a4674300946ea
-  data.tar.gz: e1db6072dc4c4ce90613e11502b3f138ae36f061bbc3e05fea49bdbcd20ef7de
+  metadata.gz: 07dd79f7c92f15bf9aa690977178a66a52dfc9d7bb050721d3beee183b8b2de3
+  data.tar.gz: 64d625886da84d225c3ccdc8cf2e38af4835cb7fd55c7c7e02699fc54b7c2a0f
 SHA512:
-  metadata.gz: fc1147d7ee3a4dec5bb05076b055d1460a30301d851c6f5d35bb8415bf28776d7405b3da36da58e45806b2e3cbd63fcf019a266b551b67c621caf354496f3fad
-  data.tar.gz: a6e2dacfe433ad8c92bffe69645fa59af42f599abf0ac1d4ca240a065d2e982562af47568709381c2fc0df2ca28138991dff219161daee0d6f780f5532e150f7
+  metadata.gz: f75573ba26f415d651d21538fb32e1341d3c9ad417d3bd8e0c7fa63b2603866dd4c00817592e54811256e0f49f800392dc13dbdce9aa264588ead0366517350e
+  data.tar.gz: d8824ca8b7af97b590b80cff3d353373e11d013d9109f904f5f1b52f3a6353b6fd3556c59c1e3f8520abf6397eb9cdea319fc10c611da8f1bde431e146ac4ac2

data/lib/abiraj/version.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
 module Abiraj
-    VERSION = "0.1.1"
+    VERSION = "0.1.0"
   end
   

data/lib/abiraj.rb

@@ -4,147 +4,283 @@ require "openssl"
 
 module Abiraj
   class Error < StandardError; end
-  
-  def self.logger(message:, level: "info")
-    prefix = "[Abiraj] "
-    message = message.to_s
-    puts prefix + message
-    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)
+
+  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 Account
+
+  class Init
     def initialize(server_url:, authtoken:, org: nil, certificate: nil)
-      @init = false
       if server_url.nil? || authtoken.nil? || server_url.strip.empty? || authtoken.strip.empty?
-        logger("Server URL and authtoken are required to initialize Abiraj.", "error")
+        Abiraj.log(message: "Server URL and authtoken are required to initialize Abiraj", level: "error")
+        return nil
       end
 
       unless server_url.start_with?("http://", "https://")
-        logger("Invalid server URL. It must start with http:// or https://", "error")
-        return
-      end
-      @server_url = server_url.strip
-      @authtoken = authtoken.strip
-      @org = org&.strip
-      if certificate
-        @certificate = certificate
+        Abiraj.log(message: "Invalid server URL. It must start with http:// or https://", level: "error")
+        return nil
       end
-      @init = true
-      logger("Abiraj initialized successfully.")
-    end
 
-    def get(account_id: nil, account_name: nil, account_title: nil)
-      if !@init
-        logger("Abiraj is not initialized.", "error")
-        return
-      end
-      logger("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_data = get_request(params, "/api/get_account_details_dict")
-      if account_data
-        logger("Account data fetched successfully.")
-        return account_data
-      else
-        logger("Unable to fetch account data.", "error")
-      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
 
-    private
+  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
 
-    def logger(message, level = "info")
-      Abiraj.logger(message: message, level: level)
-      if level == "error"
+      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)
-      logger("Raising API request to Abiraj server.")
-      
-      uri = URI(@server_url + 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"] = @authtoken
-      if @org && !@org.strip.empty?
-        request["org"] = @org
+      request["authtoken"] = Abiraj.authtoken
+      if Abiraj.org && !Abiraj.org.strip.empty?
+        request["org"] = Abiraj.org
       end
       begin
         response = http.request(request)
-        if response.code.to_i == 200
-          JSON.parse(response.body)
+        response = JSON.parse(response.body)
+        if response['status_code'] == 200
+          return response
         else
-          logger("Unexpected response code #{response.code}: #{response.body}", "error")
+          Abiraj.log(message: "#{response['status_code']}: #{response['message']}", level: "error")
         end
       rescue JSON::ParserError => e
-        logger("Failed to parse JSON response: #{e.message}", "error")
+        Abiraj.log(message: "Failed to parse JSON response: #{e.message}", level: "error")
+        return nil
       rescue StandardError => e
-        logger("HTTP request failed: #{e.message}", "error")
+        Abiraj.log(message: "HTTP request failed: #{e.message}", level: "error")
+        return nil
       end
     end
 
     def set_http(uri)
       begin
-        logger("Setting up HTTP connection.")
+        Abiraj.log(message: "Setting up HTTP connection.")
+        
         http = Net::HTTP.new(uri.host, uri.port)
         http.use_ssl = uri.scheme == "https"
-        http.verify_mode = OpenSSL::SSL::VERIFY_PEER
-        http = set_ssl(http)
-        return http
+
+        if uri.scheme == "https"
+          http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+          configure_ssl(http, uri)
+        end
+        http
       rescue StandardError => e
-        logger("Failed to set HTTP connection: #{e.message}", "error")
+        Abiraj.log(message: "Failed to set HTTP connection: #{e.message}", level: "error")
         return nil
       end
     end
 
-    def set_ssl(http)
+    def configure_ssl(http, uri)
       begin
-        if @server_url.start_with?("https://") && @certificate
-          logger("Setting SSL connection with certificate.")
-          
-          if @certificate.start_with?("-----BEGIN CERTIFICATE-----")
-            logger("Binding certificate value.")
-            
-            cert_object = OpenSSL::X509::Certificate.new(@certificate)
-            http.cert_store = OpenSSL::X509::Store.new
-            http.cert_store.add_cert(cert_object)
-            
-          elsif File.exist?(@certificate)
-            logger("Binding certificate file.")
-            http.ca_file = @certificate
-            
+        if Abiraj.server_url.start_with?("https://")
+          if Abiraj.certificate
+            handle_ssl_certificate(http)
           else
-            logger("Invalid certificate content or file path.", "error")
-            return nil
+            Abiraj.log(message: "No SSL certificate provided. Attempting to fetch from server.")
+            fetch_and_set_server_certificate(http, uri)
           end
-          
-        elsif @server_url.start_with?("https://") && !@certificate
-          logger("Disabling SSL verification. Certificate required for SSL secure connection", "warn")
-          http.verify_mode = OpenSSL::SSL::VERIFY_NONE
         end
-    
-        return http
-        
+      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
-        logger("Invalid certificate format: #{e.message}", "error")
+        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
-        logger("Failed to set SSL connection: #{e.message}", "error")
+        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
-    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,29 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: abiraj
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Abiraj
-autorequire:
+autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-12-30 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/abiraj/version.rb
-homepage:
+homepage: 
 licenses: []
 metadata: {}
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -38,8 +38,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.23
-signing_key:
+rubygems_version: 3.0.3
+signing_key: 
 specification_version: 4
 summary: Abiraj Chef plugin tool to extend and securely manage secrets and credential.
 test_files: []