后台

2025-08-19 17:30:51 +08:00
parent fd44965f8a
commit 353c72c602
74 changed files with 10575 additions and 44 deletions
--- a/app/InvitationCodeUtil.php
+++ b/app/InvitationCodeUtil.php
@@ -0,0 +1,131 @@
+<?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);
+    }
+}