From 362d767e97445a8d2e8e5a3574770f0dd43c4bca Mon Sep 17 00:00:00 2001
From: BarsTiger <kit.barsic@gmail.com>
Date: Mon, 17 Jul 2023 14:58:02 +0300
Subject: [PATCH] Added identity here

---
 dragonion_core/proto/encryption/__init__.py |  1 +
 dragonion_core/proto/encryption/identity.py | 28 +++++++++++++++++++++
 2 files changed, 29 insertions(+)
 create mode 100644 dragonion_core/proto/encryption/__init__.py
 create mode 100644 dragonion_core/proto/encryption/identity.py

diff --git a/dragonion_core/proto/encryption/__init__.py b/dragonion_core/proto/encryption/__init__.py
new file mode 100644
index 0000000..2ae2839
--- /dev/null
+++ b/dragonion_core/proto/encryption/__init__.py
@@ -0,0 +1 @@
+pass
diff --git a/dragonion_core/proto/encryption/identity.py b/dragonion_core/proto/encryption/identity.py
new file mode 100644
index 0000000..9aad43b
--- /dev/null
+++ b/dragonion_core/proto/encryption/identity.py
@@ -0,0 +1,28 @@
+from attrs import define
+from cryptography.hazmat.primitives.asymmetric import rsa
+from cryptography.hazmat.primitives import serialization
+import base64
+
+
+@define
+class Identity:
+    """
+    Valid identity has fields
+    username - in most cases same as filename, represents identity's username
+    private_key - key to decrypt incoming messages and get public key,
+                  shouldn't be sent anywhere, generated via generate function
+    """
+    username: str
+    private_key: rsa.RSAPrivateKey = None
+
+    def generate(self):
+        self.private_key = rsa.generate_private_key(
+            public_exponent=65537,
+            key_size=4096
+        )
+
+    def public_key(self):
+        return base64.urlsafe_b64encode(self.private_key.public_key().public_bytes(
+            encoding=serialization.Encoding.DER,
+            format=serialization.PublicFormat.PKCS1
+        ))