feat: First utils version

.github/workflows/pypi.yaml

@@ -0,0 +1,29 @@
+name: Publish release to PyPi
+
+on:
+  release:
+    types: [ created ]
+
+jobs:
+  pypi-publish:
+    name: Publish release to PyPI
+    runs-on: ubuntu-latest
+    environment:
+      name: pypi
+      url: https://pypi.org/project/libertai-utils
+    permissions:
+      id-token: write
+    steps:
+      - uses: actions/checkout@v4
+      - name: Install poetry
+        run: pipx install poetry
+      - uses: actions/setup-python@v5
+        with:
+          python-version-file: 'pyproject.toml'
+          cache: 'poetry'
+      - name: Install dependencies
+        run: poetry install
+      - name: Build package
+        run: poetry build
+      - name: Publish package distributions to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

README.md

@@ -1,2 +1,3 @@
 # libertai-utils
-Libertai python Utilities
+
+LibertAI internal types and functions

libertai/crypto/ethereum.py → libertai_utils/blockchains/ethereum.py

@@ -3,14 +3,15 @@
 from web3 import Web3
 
 
-def get_address_from_signature(message: str, signature: str) -> str:
-    """Recover the Ethereum address used to sign the message"""
+def is_eth_signature_valid(message: str, signature: str, address: str) -> bool:
+    """Check if a message signature with an Ethereum wallet is valid"""
     w3 = Web3(Web3.HTTPProvider(""))
     encoded_message = encode_defunct(text=message)
-    return w3.eth.account.recover_message(
+    recovered_address = w3.eth.account.recover_message(
         encoded_message,
         signature=HexBytes(signature),
     )
+    return format_eth_address(address) == format_eth_address(recovered_address)
 
 
 def format_eth_address(address: str) -> str:

libertai_utils/blockchains/index.py

@@ -0,0 +1,20 @@
+from libertai_utils.blockchains.ethereum import is_eth_signature_valid, format_eth_address
+from libertai_utils.blockchains.solana import is_solana_signature_valid
+from libertai_utils.interfaces.subscription import SubscriptionChain
+
+
+def is_signature_valid(chain: SubscriptionChain, message: str, signature: str, address: str) -> bool:
+    valid = False
+
+    if chain == SubscriptionChain.base:
+        valid = is_eth_signature_valid(message, signature, address)
+    elif chain == SubscriptionChain.solana:
+        valid = is_solana_signature_valid(message, signature, address)
+
+    return valid
+
+
+def format_address(address: str, chain: SubscriptionChain) -> str:
+    if chain == SubscriptionChain.base:
+        return format_eth_address(address)
+    return address

libertai_utils/blockchains/solana.py

@@ -0,0 +1,26 @@
+import base64
+
+from nacl.exceptions import BadSignatureError
+from nacl.signing import VerifyKey
+from solders.pubkey import Pubkey
+
+
+def is_solana_signature_valid(message: str, signature: str, address: str) -> bool:
+    """Check if a message signature with a Solana wallet is valid
+
+    Args:
+        message: Message that was signed
+        signature: base64-encoded message signature
+        address: base58-encoded public key
+    """
+    public_key = Pubkey.from_string(address)
+    bytes_message = bytes(message, encoding="utf-8")
+    bytes_signature = bytes(signature, "utf-8")
+
+    try:
+        retrieved_message = VerifyKey(public_key.__bytes__()).verify(
+            smessage=bytes_message, signature=base64.b64decode(bytes_signature)
+        )
+        return bytes_message == retrieved_message
+    except BadSignatureError:
+        return False

libertai/interfaces/subscription.py → libertai_utils/interfaces/subscription.py

@@ -17,6 +17,7 @@ class SubscriptionType(str, Enum):
 
 class SubscriptionChain(str, Enum):
     base = "base"
+    solana = "solana"
 
 
 class SubscriptionAccount(BaseModel):