# Copyright (c) 2026 Emanuele Bellocchia
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
"""Module for sr25519 keys."""
# Imports
from typing import Any
import sr25519
from typing_extensions import override
from bip_utils.ecc.common.ikeys import IPrivateKey, IPublicKey
from bip_utils.ecc.common.ipoint import IPoint
from bip_utils.ecc.curve.elliptic_curve_types import EllipticCurveTypes
from bip_utils.utils.misc import DataBytes
class Sr25519KeysConst:
"""Class container for sr25519 keys constants."""
# Public key length in bytes
PUB_KEY_BYTE_LEN: int = 32
# Private key length in bytes
PRIV_KEY_BYTE_LEN: int = 64
class Sr25519PublicKey(IPublicKey):
"""Sr25519 public key class."""
m_ver_key: bytes
@override
@classmethod
def FromBytes(cls,
key_bytes: bytes) -> IPublicKey:
"""
Construct class from key bytes.
Args:
key_bytes (bytes): Key bytes
Returns:
IPublicKey: IPublicKey object
Raises:
ValueError: If key bytes are not valid
"""
return cls(key_bytes)
@override
@classmethod
def FromPoint(cls,
key_point: IPoint) -> IPublicKey:
"""
Construct class from key point.
Args:
key_point (IPoint object): Key point
Returns:
IPublicKey: IPublicKey object
Raises:
ValueError: If key point is not valid
"""
raise RuntimeError("Constrution from point not supported")
def __init__(self,
key_bytes: bytes) -> None:
"""
Construct class from key object.
Args:
key_bytes (bytes): Key bytes
Raises:
ValueError: If key is not valid
"""
if len(key_bytes) != self.CompressedLength():
raise ValueError("Invalid public key")
self.m_ver_key = key_bytes
@override
@staticmethod
def CurveType() -> EllipticCurveTypes:
"""
Get the elliptic curve type.
Returns:
EllipticCurveTypes: Elliptic curve type
"""
return EllipticCurveTypes.SR25519
@override
@staticmethod
def CompressedLength() -> int:
"""
Get the compressed key length.
Returns:
int: Compressed key length
"""
return Sr25519KeysConst.PUB_KEY_BYTE_LEN
@override
@staticmethod
def UncompressedLength() -> int:
"""
Get the uncompressed key length.
Returns:
int: Uncompressed key length
"""
return Sr25519PublicKey.CompressedLength()
@override
def UnderlyingObject(self) -> Any:
"""
Get the underlying object.
Returns:
Any: Underlying object
"""
return self.m_ver_key
@override
def RawCompressed(self) -> DataBytes:
"""
Return raw compressed public key.
Returns:
DataBytes object: DataBytes object
"""
return DataBytes(self.m_ver_key)
@override
def RawUncompressed(self) -> DataBytes:
"""
Return raw uncompressed public key.
Returns:
DataBytes object: DataBytes object
"""
# Same as compressed
return self.RawCompressed()
@override
def Point(self) -> IPoint:
"""
Get public key point.
Returns:
IPoint object: IPoint object
"""
raise RuntimeError("Getting point from public key not supported")
class Sr25519PrivateKey(IPrivateKey):
"""Sr25519 private key class."""
m_sign_key: bytes
@override
@classmethod
def FromBytes(cls,
key_bytes: bytes) -> IPrivateKey:
"""
Construct class from key bytes.
Args:
key_bytes (bytes): Key bytes
Returns:
IPrivateKey: IPrivateKey object
Raises:
ValueError: If key bytes are not valid
"""
return cls(key_bytes)
def __init__(self,
key_bytes: bytes) -> None:
"""
Construct class from key object.
Args:
key_bytes (bytes): Key bytes
Raises:
ValueError: If key is not valid
"""
if len(key_bytes) != self.Length():
raise ValueError("Invalid private key")
self.m_sign_key = key_bytes
@override
@staticmethod
def CurveType() -> EllipticCurveTypes:
"""
Get the elliptic curve type.
Returns:
EllipticCurveTypes: Elliptic curve type
"""
return EllipticCurveTypes.SR25519
@override
@staticmethod
def Length() -> int:
"""
Get the key length.
Returns:
int: Key length
"""
return Sr25519KeysConst.PRIV_KEY_BYTE_LEN
@override
def UnderlyingObject(self) -> Any:
"""
Get the underlying object.
Returns:
Any: Underlying object
"""
return self.m_sign_key
@override
def Raw(self) -> DataBytes:
"""
Return raw private key.
Returns:
DataBytes object: DataBytes object
"""
return DataBytes(self.m_sign_key)
@override
def PublicKey(self) -> IPublicKey:
"""
Get the public key correspondent to the private one.
Returns:
IPublicKey object: IPublicKey object
"""
return Sr25519PublicKey(sr25519.public_from_secret_key(self.m_sign_key)) # pylint: disable=no-member