majuna/app/models/onions.py

import base64
import hashlib

from app.brm.brn import BRN
from app.extensions import db
from app.models import AbstractConfiguration, AbstractResource


class Onion(AbstractConfiguration):
    @property
    def brn(self) -> BRN:
        return BRN(
            group_id=self.group_id,
            product="eotk",
            provider="*",
            resource_type="onion",
            resource_id=self.onion_name
        )

    group_id = db.Column(db.Integer(), db.ForeignKey("group.id"), nullable=False)
    domain_name = db.Column(db.String(255), nullable=False)
    onion_name = db.Column(db.String(56), nullable=False, unique=True)

    onion_public_key = db.Column(db.LargeBinary, nullable=False)
    onion_private_key = db.Column(db.LargeBinary, nullable=False)

    tls_public_key = db.Column(db.LargeBinary, nullable=False)
    tls_private_key = db.Column(db.LargeBinary, nullable=False)

    group = db.relationship("Group", back_populates="onions")

    @property
    def calculated_onion_name(self):
        p = self.onion_public_key[32:]

        h = hashlib.sha3_256()
        h.update(b".onion checksum")
        h.update(p)
        h.update(b"\x03")
        checksum = h.digest()

        result = bytearray(p)
        result.extend(checksum[0:2])
        result.append(0x03)

        onion = base64.b32encode(result).decode("utf-8").strip("=")
        return onion.lower()


class Eotk(AbstractResource):
    group_id = db.Column(db.Integer(), db.ForeignKey("group.id"), nullable=False)
    instance_id = db.Column(db.String(100), nullable=True)
    provider = db.Column(db.String(20), nullable=False)
    region = db.Column(db.String(20), nullable=False)

    group = db.relationship("Group", back_populates="eotks")

    @property
    def brn(self) -> BRN:
        return BRN(
            group_id=self.group_id,
            provider=self.provider,
            product="eotk",
            resource_type="instance",
            resource_id=self.region
        )