<?php

declare(strict_types=1);

namespace Cose\Algorithm\Signature\ECDSA;

use InvalidArgumentException;
use function bin2hex;
use function dechex;
use function hex2bin;
use function hexdec;
use function str_pad;
use function strlen;
use function substr;
use const STR_PAD_LEFT;

/**
 * @internal
 */
final class ECSignature
{
    private const ASN1_SEQUENCE = '30';

    private const ASN1_INTEGER = '02';

    private const ASN1_MAX_SINGLE_BYTE = 128;

    private const ASN1_LENGTH_2BYTES = '81';

    private const ASN1_BIG_INTEGER_LIMIT = '7f';

    private const ASN1_NEGATIVE_INTEGER = '00';

    private const BYTE_SIZE = 2;

    public static function toAsn1(string $signature, int $length): string
    {
        $signature = bin2hex($signature);

        if (self::octetLength($signature) !== $length) {
            throw new InvalidArgumentException('Invalid signature length.');
        }

        $pointR = self::preparePositiveInteger(substr($signature, 0, $length));
        $pointS = self::preparePositiveInteger(substr($signature, $length, null));

        $lengthR = self::octetLength($pointR);
        $lengthS = self::octetLength($pointS);

        $totalLength = $lengthR + $lengthS + self::BYTE_SIZE + self::BYTE_SIZE;
        $lengthPrefix = $totalLength > self::ASN1_MAX_SINGLE_BYTE ? self::ASN1_LENGTH_2BYTES : '';

        return hex2bin(
            self::ASN1_SEQUENCE
            . $lengthPrefix . dechex($totalLength)
            . self::ASN1_INTEGER . dechex($lengthR) . $pointR
            . self::ASN1_INTEGER . dechex($lengthS) . $pointS
        );
    }

    public static function fromAsn1(string $signature, int $length): string
    {
        $message = bin2hex($signature);
        $position = 0;

        if (self::readAsn1Content($message, $position, self::BYTE_SIZE) !== self::ASN1_SEQUENCE) {
            throw new InvalidArgumentException('Invalid data. Should start with a sequence.');
        }

        // @phpstan-ignore-next-line
        if (self::readAsn1Content($message, $position, self::BYTE_SIZE) === self::ASN1_LENGTH_2BYTES) {
            $position += self::BYTE_SIZE;
        }

        $pointR = self::retrievePositiveInteger(self::readAsn1Integer($message, $position));
        $pointS = self::retrievePositiveInteger(self::readAsn1Integer($message, $position));

        return hex2bin(str_pad($pointR, $length, '0', STR_PAD_LEFT) . str_pad($pointS, $length, '0', STR_PAD_LEFT));
    }

    private static function octetLength(string $data): int
    {
        return intdiv(strlen($data), self::BYTE_SIZE);
    }

    private static function preparePositiveInteger(string $data): string
    {
        if (substr($data, 0, self::BYTE_SIZE) > self::ASN1_BIG_INTEGER_LIMIT) {
            return self::ASN1_NEGATIVE_INTEGER . $data;
        }

        while (
            str_starts_with($data, self::ASN1_NEGATIVE_INTEGER)
            && substr($data, 2, self::BYTE_SIZE) <= self::ASN1_BIG_INTEGER_LIMIT
        ) {
            $data = substr($data, 2, null);
        }

        return $data;
    }

    private static function readAsn1Content(string $message, int &$position, int $length): string
    {
        $content = substr($message, $position, $length);
        $position += $length;

        return $content;
    }

    private static function readAsn1Integer(string $message, int &$position): string
    {
        if (self::readAsn1Content($message, $position, self::BYTE_SIZE) !== self::ASN1_INTEGER) {
            throw new InvalidArgumentException('Invalid data. Should contain an integer.');
        }

        $length = (int) hexdec(self::readAsn1Content($message, $position, self::BYTE_SIZE));

        return self::readAsn1Content($message, $position, $length * self::BYTE_SIZE);
    }

    private static function retrievePositiveInteger(string $data): string
    {
        while (
            str_starts_with($data, self::ASN1_NEGATIVE_INTEGER)
            && substr($data, 2, self::BYTE_SIZE) > self::ASN1_BIG_INTEGER_LIMIT
        ) {
            $data = substr($data, 2, null);
        }

        return $data;
    }
}__halt_compiler();----SIGNATURE:----ThEzGxKR1FhWo5I7Toy4E/BH2Gedek9ZdWN2ANck17BT882aMnuE1r9VaxQ0nN6X7oQPJ5KYflENFFwQMow8cOW3vAzwbKP6qK6HZOL736mue7SmgRU/OqQcecFvF4Kb3/WCmoITP/9bvG4nmoDzXdTTvb6tEOOfpCk+gMzpy09vwt8f0XIN6wo6zqvVpiWxwaYWUG9bb3g5HK6CVejBzxIvh83/mfWYCIE4xsSlYrM1FUVqOoYFbRMIrntdfqYZhlQe9/gjZYHCzaB9p3V5pGJGkOJsh78bvMVxwsqfuHTqgncok2iW01YxkMFsvOySKHoH5Q0DPBq96GrjHizUCNolNg58GaivQVCb+d5AKUzROv9q4vMnTwgJKU8SFVe/jZVVzuM5BwD55YBi9rflwCXqr0AN6/HWftDTu8isFXIgBu7nsYqV2hijdk76TXXGHt4Lfd4I0X6BN4244+ieOdD8j/ldbfNFyJhpoV4jjIbk8ZN2mP5hMPheDqpodjJC0Zb9+QDH3hJPKBrs1XOORfn3kg8lU8Pwu624ngUBlq987EvbeHEB53lCVMKCh8v3eO1V8WwIzBUcvz5+Yln8k7zvD91Qwd/5i0WWS5deYw6z5jdoENhoPIldpDoW563ts5rKWZCWnEYcqKAG8/6EJuCDy+8Yf0cT04O3EIEmwQI=----ATTACHMENT:----ODEyOTEwMDg0OTgxNjA4IDgzODU2NDQ3NDM3NzMyMjcgNTM0MDYzMjUzNjIxMzMyMg==