Add context, http client, models and utils

Author: Inbal Tako

requirements.txt

@@ -1,2 +1,4 @@
 requests~=2.23.0
-pycrypto~=2.6.1
+pycrypto~=2.6.1
+aiohttp~=3.6.2
+crypto~=1.4.1

securenative/config.py

@@ -1,11 +1,11 @@
-from securenative.config.securenative_options import SecurenativeOptions
+from securenative.config.securenative_options import SecureNativeOptions
 from securenative.enums.failover_strategy import FailOverStrategy
 
 
 class ConfigurationBuilder(object):
 
     def __init__(self):
-        self.api_key = ""
+        self.api_key = None
         self.api_url = "https://api.securenative.com/collector/api/v1"
         self.interval = 1000
         self.max_events = 1000
@@ -57,5 +57,4 @@ def with_fail_over_strategy(self, fail_over_strategy):
 
     @staticmethod
     def get_default_securenative_options():
-        return SecurenativeOptions("", "https://api.securenative.com/collector/api/v1", 1000,
-                                   1000, 1500, True, False, "CRITICAL", FailOverStrategy.FAIL_OPEN)
+        return SecureNativeOptions()

securenative/config/configuration_builder.py

@@ -9,7 +9,7 @@ class ConfigurationManager(object):
     CUSTOM_CONFIG_FILE_ENV_NAME = "SECURENATIVE_COMFIG_FILE"
     config = ConfigParser.ConfigParser()
 
-    @staticmethod
+    @classmethod
     def read_resource_file(cls, resource_path):
         cls.config.read(resource_path)
         sections = cls.config.sections()
@@ -25,7 +25,7 @@ def read_resource_file(cls, resource_path):
 
         return properties
 
-    @staticmethod
+    @classmethod
     def _get_resource_path(cls, env_name):
         env_value = os.environ.get(env_name)
 
@@ -38,7 +38,13 @@ def _get_resource_path(cls, env_name):
     def config_builder():
         return ConfigurationBuilder.default_config_builder()
 
-    @staticmethod
+    @classmethod
+    def _get_env_or_default(cls, properties, key, default):
+        if properties[key]:
+            return properties[key]
+        return default
+
+    @classmethod
     def load_config(cls):
         builder = ConfigurationBuilder()
         options = builder.get_default_securenative_options()

securenative/config/configuration_manager.py

@@ -1,7 +1,10 @@
-class SecurenativeOptions(object):
+from securenative.enums.failover_strategy import FailOverStrategy
 
-    def __init__(self, api_key, api_url, interval, max_events, timeout, auto_send,
-                 disable, log_level, fail_over_strategy):
+
+class SecureNativeOptions(object):
+
+    def __init__(self, api_key=None, api_url=None, interval=1000, max_events=1000, timeout=1500, auto_send=True,
+                 disable=False, log_level="CRITICAL", fail_over_strategy=FailOverStrategy.FAIL_OPEN):
         self.api_key = api_key
         self.api_url = api_url
         self.interval = interval

securenative/config/securenative_options.py

@@ -0,0 +1,45 @@
+from securenative.context.securenative_context import SecureNativeContext
+
+
+class ContextBuilder(object):
+
+    def __init__(self):
+        self.context = SecureNativeContext()
+
+    def with_client_token(self, client_token):
+        self.context.client_token = client_token
+        return self
+
+    def with_ip(self, ip):
+        self.context.ip = ip
+        return self
+
+    def with_remote_ip(self, remote_ip):
+        self.context.remote_ip = remote_ip
+        return self
+
+    def with_headers(self, headers):
+        self.context.headers = headers
+        return self
+
+    def with_url(self, url):
+        self.context.url = url
+        return self
+
+    def with_method(self, method):
+        self.context.method = method
+        return self
+
+    def with_body(self, body):
+        self.context.body = body
+        return self
+
+    @staticmethod
+    def default_context_builder():
+        return ContextBuilder()
+
+    def from_http_request(self, request):  # TODO!
+        pass
+
+    def build(self):
+        return self.context

securenative/context/context_builder.py

@@ -0,0 +1,10 @@
+class SecureNativeContext(object):
+
+    def __init__(self, client_token=None, ip=None, remote_ip=None, headers=None, url=None, method=None, body=None):
+        self.client_token = client_token
+        self.ip = ip
+        self.remote_ip = remote_ip
+        self.headers = headers
+        self.url = url
+        self.method = method
+        self.body = body

securenative/context/securenative_context.py

@@ -0,0 +1,6 @@
+from enum import Enum
+
+
+class ApiRoute(Enum):
+    TRACK = "track"
+    VERIFY = "verify"

securenative/enums/api_route.py

@@ -0,0 +1,23 @@
+from enum import Enum
+
+
+class EventTypes(Enum):
+    LOG_IN = "sn.user.login"
+    LOG_IN_CHALLENGE = "sn.user.login.challenge"
+    LOG_IN_FAILURE = "sn.user.login.failure"
+    LOG_OUT = "sn.user.logout"
+    SIGN_UP = "sn.user.signup"
+    AUTH_CHALLENGE = "sn.user.auth.challenge"
+    AUTH_CHALLENGE_SUCCESS = "sn.user.auth.challenge.success"
+    AUTH_CHALLENGE_FAILURE = "sn.user.auth.challenge.failure"
+    TWO_FACTOR_DISABLE = "sn.user.2fa.disable"
+    EMAIL_UPDATE = "sn.user.email.update"
+    PASSWORD_REST = "sn.user.password.reset"
+    PASSWORD_REST_SUCCESS = "sn.user.password.reset.success"
+    PASSWORD_UPDATE = "sn.user.password.update"
+    PASSWORD_REST_FAILURE = "sn.user.password.reset.failure"
+    USER_INVITE = "sn.user.invite"
+    ROLE_UPDATE = "sn.user.role.update"
+    PROFILE_UPDATE = "sn.user.profile.update"
+    PAGE_VIEW = "sn.user.page.view"
+    VERIFY = "sn.verify"

securenative/enums/event_types.py

@@ -0,0 +1,7 @@
+from enum import Enum
+
+
+class RiskLevel(Enum):
+    LOW = "low"
+    MEDIUM = "medium"
+    HIGH = "high"