<?php

namespace app;

use extend\ba\Random;

class InvitationCodeUtil
{
    /**
     * 自定义进制字符集（30进制）
     * 避免 0、1、O、I 混淆字符
     */
    private static array $r = ['M', 'J', 'U', 'D', 'Z', 'X', '9', 'C', '7', 'P', 'E', '8', '6', 'B', 'G', 'H', 'S', '2', '5', 'F', 'R', '4', 'Q', 'W', 'K', '3', 'V', 'Y', 'T', 'N'];

    /**
     * 补位字符
     */
    private static string $b = 'A';

    /**
     * 进制长度
     */
    private static int $binLen = 30;

    /**
     * 最小长度
     */
    private static int $s = 6;

    /**
     * 补位字符串 - 生成注册码用
     */
    private static string $re = 'REGISTER';

    /**
     * 补位字符串 - 生成序列码用
     */
    private static string $e = 'KSLFXFR';

    /**
     * 根据ID生成六位注册邀请码
     *
     * @param int $id
     * @return string
     */
    public static function toRegisteredCode(int $id): string
    {
        $str = self::encodeBase($id);
        if (strlen($str) < self::$s) {
            $pad = substr(self::$re, 0, self::$s - strlen($str));
            $str .= $pad;
        }
        return $str;
    }

    /**
     * 根据ID生成六位序列码
     *
     * @param int $id
     * @return string
     */
    public static function toSerialCode(int $id): string
    {
        $str = self::encodeBase($id);
        if (strlen($str) < self::$s) {
            $pad = substr(self::$e, 0, self::$s - strlen($str));
            $str .= $pad;
        }
        return $str;
    }

    /**
     * 将邀请码还原为 ID
     *
     * @param string $code
     * @return int
     */
    public static function codeToId(string $code): int
    {
        $chars = str_split($code);
        $res = 0;

        foreach ($chars as $i => $ch) {
            if ($ch === self::$b) {
                break;
            }

            $index = array_search($ch, self::$r, true);
            if ($index === false) {
                continue;
            }

            $res = $i === 0 ? $index : $res * self::$binLen + $index;
        }

        return $res;
    }

    /**
     * 雪花 ID（需引入 PHP 雪花 ID 工具）
     * 推荐使用: https://github.com/Gerardojbaez/laravel-snowflake 或你自己的封装
     *
     * @return int
     */
    public static function getSnowFlakeId(): int
    {
        // 简化版本，实际应使用类库如 Godruoyi\Snowflake
        if (!class_exists('Snowflake')) {
            throw new \RuntimeException("Snowflake class not found.");
        }
        return Random::generateRandomPrefixedId();
    }

    /**
     * 进制转换编码主逻辑
     *
     * @param int $id
     * @return string
     */
    private static function encodeBase(int $id): string
    {
        $buf = [];
        do {
            $index = $id % self::$binLen;
            array_unshift($buf, self::$r[$index]);
            $id = intdiv($id, self::$binLen);
        } while ($id > 0);

        return implode('', $buf);
    }
}