Use a singelton pattern

Author: Inbal Tako

lib/securenative.rb

@@ -1,62 +1,39 @@
-require_relative 'securenative/event_manager'
-require_relative 'securenative/config'
 require_relative 'securenative/sn_exception'
-require_relative 'securenative/utils'
-require 'json'
+require_relative 'securenative/secure_native_sdk'
 
-class SecureNative
-  def initialize(api_key, options: SecureNativeOptions.new)
-    if api_key == nil
-      raise SecureNativeSDKException.new
-    end
-
-    @api_key = api_key
-    @options = options
-    @event_manager = EventManager.new(@api_key, options: @options)
-  end
-
-  def api_key
-    @api_key
-  end
+$securenative = nil
 
-  def version
-    Config::SDK_VERSION
-  end
-
-  def track(event)
-    validate_event(event)
-    @event_manager.send_async(event, Config::TRACK_EVENT)
-  end
-
-  def verify(event)
-    validate_event(event)
-    res = @event_manager.send_sync(event, Config::VERIFY_EVENT)
-    if res.status_code == 200
-      return JSON.parse(res.body)
+module SecureNative
+  def self.init(api_key, options: SecureNativeOptions.new)
+    if $securenative == nil
+      $securenative = SecureNativeSDK.new(api_key, options: options)
     end
-    nil
   end
 
-  def flow(event) # Note: For future purposes
-    validate_event(event)
-    @event_manager.send_async(event, Config::FLOW_EVENT)
+  def self.track(event)
+    sdk = _get_or_throw
+    sdk.track(event)
   end
 
-  def verify_webhook(hmac_header, body)
-    Utils.verify_signature(@api_key, body, hmac_header)
+  def self.verify(event)
+    sdk = _get_or_throw
+    sdk.verify(event)
   end
 
-  def flush
-    @event_manager.flush
+  def self.verify_webhook(hmac_header, body)
+    sdk = _get_or_throw
+    sdk.verify_webhook(hmac_header = hmac_header, body = body)
   end
 
-  private
+  def self.flush
+    sdk = _get_or_throw
+    sdk.flush
+  end
 
-  def validate_event(event)
-    unless event.params.nil?
-      if event.params.length > Config::MAX_ALLOWED_PARAMS
-        event.params = event.params[0, Config::MAX_ALLOWED_PARAMS]
-      end
+  def self._get_or_throw
+    if $securenative == nil
+      raise SecureNativeSDKException.new
     end
+    $securenative
   end
 end

lib/securenative/secure_native_sdk.rb

@@ -0,0 +1,62 @@
+require_relative 'event_manager'
+require_relative 'config'
+require_relative 'sn_exception'
+require_relative 'utils'
+require 'json'
+
+class SecureNativeSDK
+  def initialize(api_key, options: SecureNativeOptions.new)
+    if api_key == nil
+      raise SecureNativeSDKException.new
+    end
+
+    @api_key = api_key
+    @options = options
+    @event_manager = EventManager.new(@api_key, options: @options)
+  end
+
+  def api_key
+    @api_key
+  end
+
+  def version
+    Config::SDK_VERSION
+  end
+
+  def track(event)
+    validate_event(event)
+    @event_manager.send_async(event, Config::TRACK_EVENT)
+  end
+
+  def verify(event)
+    validate_event(event)
+    res = @event_manager.send_sync(event, Config::VERIFY_EVENT)
+    if res.status_code == 200
+      return JSON.parse(res.body)
+    end
+    nil
+  end
+
+  def flow(event) # Note: For future purposes
+    validate_event(event)
+    @event_manager.send_async(event, Config::FLOW_EVENT)
+  end
+
+  def verify_webhook(hmac_header, body)
+    Utils.verify_signature(@api_key, body, hmac_header)
+  end
+
+  def flush
+    @event_manager.flush
+  end
+
+  private
+
+  def validate_event(event)
+    unless event.params.nil?
+      if event.params.length > Config::MAX_ALLOWED_PARAMS
+        event.params = event.params[0, Config::MAX_ALLOWED_PARAMS]
+      end
+    end
+  end
+end

test/test_sdk.rb

@@ -13,26 +13,27 @@ def build_event(type = EventType::LOG_IN, id)
   )
 end
 
-describe SecureNative do
+describe SecureNativeSDK do
   let(:sn_options) {SecureNativeOptions.new}
-  let(:sn_client) {SecureNative.new(ENV["SN_API_KEY"], options: sn_options)}
 
   it "use sdk without api key" do
     api_key = nil
-    expect {SecureNative.new(api_key, options: sn_options)}.to raise_error(SecureNativeSDKException)
+    expect {SecureNative.init(api_key, options: sn_options)}.to raise_error(SecureNativeSDKException)
   end
 
   it "track an event" do
     user_id = SecureRandom.uuid
     event = build_event(EventType::LOG_IN, user_id)
-    sn_client.track(event)
-    expect(sn_client.flush).not_to be_empty
+    SecureNative.init(ENV["SN_API_KEY"], options: sn_options)
+    SecureNative.track(event)
+    expect(SecureNative.flush).not_to be_empty
   end
 
   it "verify an event" do
     user_id = SecureRandom.uuid
     event = build_event(EventType::LOG_OUT, user_id)
-    res = sn_client.verify(event)
+    SecureNative.init(ENV["SN_API_KEY"], options: sn_options)
+    res = SecureNative.verify(event)
 
     expect(res).not_to be_empty
     expect(res['triggers']).not_to be_empty