mtcute/packages/core/src/utils/crypto/abstract.ts

import { MaybeAsync } from '../../types/index.js'
import { AesModeOfOperationIge } from './common.js'
import { factorizePQSync } from './factorization.js'

export interface IEncryptionScheme {
    encrypt(data: Uint8Array): MaybeAsync<Uint8Array>

    decrypt(data: Uint8Array): MaybeAsync<Uint8Array>
}

export interface ICryptoProvider {
    initialize?(): MaybeAsync<void>

    sha1(data: Uint8Array): MaybeAsync<Uint8Array>

    sha256(data: Uint8Array): MaybeAsync<Uint8Array>

    pbkdf2(
        password: Uint8Array,
        salt: Uint8Array,
        iterations: number,
        keylen?: number, // = 64
        algo?: string, // sha1 or sha512 (default sha512)
    ): MaybeAsync<Uint8Array>

    hmacSha256(data: Uint8Array, key: Uint8Array): MaybeAsync<Uint8Array>

    // in telegram, iv is always either used only once, or is the same for all calls for the key
    createAesCtr(key: Uint8Array, iv: Uint8Array, encrypt: boolean): MaybeAsync<IEncryptionScheme>

    createAesIge(key: Uint8Array, iv: Uint8Array): MaybeAsync<IEncryptionScheme>

    createAesEcb(key: Uint8Array): MaybeAsync<IEncryptionScheme>

    factorizePQ(pq: Uint8Array): MaybeAsync<[Uint8Array, Uint8Array]>
}

export abstract class BaseCryptoProvider {
    createAesIge(key: Uint8Array, iv: Uint8Array): MaybeAsync<IEncryptionScheme> {
        const ecb = this.createAesEcb(key)

        if ('then' in ecb) {
            return ecb.then((ecb) => new AesModeOfOperationIge(key, iv, ecb))
        }

        return new AesModeOfOperationIge(key, iv, ecb)
    }

    factorizePQ(pq: Uint8Array) {
        return factorizePQSync(pq)
    }

    abstract createAesEcb(key: Uint8Array): MaybeAsync<IEncryptionScheme>
}

export type CryptoProviderFactory = () => ICryptoProvider