MBTI_wang/api/app/controller/admin/Invite.php

<?php
namespace app\controller\admin;

use app\BaseController;
use app\common\service\WechatService;
use think\facade\Db;

/**
 * 管理端 - 小程序邀请二维码（企业版 + 个人版）
 */
class Invite extends BaseController
{
    /**
     * 同时生成：企业测评码（e_企业ID → enterprise 页）与个人首页码（scene=p → index）
     * GET /api/v1/admin/invite/qrcode
     *
     * 兼容旧前端：仍返回顶层 qrcode / scene / page 指向企业版。
     */
    public function qrcode()
    {
        $admin = $this->request->user ?? null;
        $role  = $admin['role'] ?? '';
        if (!$admin || !in_array($role, ['admin', 'enterprise_admin', 'superadmin'], true)) {
            return error('无权限访问', 403);
        }

        $enterpriseId = null;
        if ($role === 'enterprise_admin') {
            $row = Db::name('users')->where('id', (int) ($admin['userId'] ?? 0))->find();
            $enterpriseId = isset($row['enterpriseId']) ? (int) $row['enterpriseId'] : null;
        } elseif ($role === 'superadmin') {
            $enterpriseId = (int) $this->request->param('enterpriseId', 0);
            if ($enterpriseId <= 0) {
                $enterpriseId = self::resolveDefaultEnterpriseIdForPlatform();
            }
            if ($enterpriseId <= 0) {
                return error('请传 enterpriseId，或在系统设置中配置默认企业', 400);
            }
        } else {
            $enterpriseId = (int) $this->request->param('enterpriseId', 0);
            if ($enterpriseId <= 0) {
                return error('请指定企业（企业管理员无需传参，使用所属企业）', 400);
            }
        }
        if ($enterpriseId <= 0) {
            return error('无法确定企业，仅企业管理员或指定 enterpriseId 可生成邀请码', 400);
        }

        $sceneEnterprise = 'e_' . $enterpriseId;
        $pageEnterprise  = 'pages/enterprise/index';

        $resultEnt = WechatService::getWxacodeUnlimited($sceneEnterprise, $pageEnterprise, 430);
        if (isset($resultEnt['errcode'])) {
            return error('企业版小程序码失败：' . ($resultEnt['errmsg'] ?? ''), 500);
        }
        $binEnt = $resultEnt['binary'] ?? '';
        if ($binEnt === '') {
            return error('企业版小程序码生成失败', 500);
        }
        $b64Ent = 'data:image/png;base64,' . base64_encode($binEnt);

        $scenePersonal = 'p';
        $pagePersonal  = 'pages/index/index';
        $resultPer     = WechatService::getWxacodeUnlimited($scenePersonal, $pagePersonal, 430);
        if (isset($resultPer['errcode'])) {
            return error('个人版小程序码失败：' . ($resultPer['errmsg'] ?? ''), 500);
        }
        $binPer = $resultPer['binary'] ?? '';
        if ($binPer === '') {
            return error('个人版小程序码生成失败', 500);
        }
        $b64Per = 'data:image/png;base64,' . base64_encode($binPer);

        return success([
            'qrcode'     => $b64Ent,
            'scene'      => $sceneEnterprise,
            'page'       => $pageEnterprise,
            'enterprise' => [
                'qrcode' => $b64Ent,
                'scene'  => $sceneEnterprise,
                'page'   => $pageEnterprise,
                'label'  => '企业版',
            ],
            'personal' => [
                'qrcode' => $b64Per,
                'scene'  => $scenePersonal,
                'page'   => $pagePersonal,
                'label'  => '个人版',
            ],
        ]);
    }

    /**
     * 超管生成企业版太阳码时：请求未带 enterpriseId 则按系统 defaultEnterpriseId → 首家运营企业
     */
    private static function resolveDefaultEnterpriseIdForPlatform(): int
    {
        try {
            $systemRow = Db::name('system_config')->where('key', 'system')->where('enterprise_id', 0)->find();
            if ($systemRow && !empty($systemRow['value'])) {
                $sysVal = is_string($systemRow['value']) ? json_decode($systemRow['value'], true) : $systemRow['value'];
                if (is_array($sysVal) && !empty($sysVal['defaultEnterpriseId'])) {
                    $de = (int) $sysVal['defaultEnterpriseId'];
                    if ($de > 0) {
                        return $de;
                    }
                }
            }
        } catch (\Throwable $e) {
            // ignore
        }
        try {
            $first = Db::name('enterprises')
                ->whereNull('deletedAt')
                ->where('status', 'operating')
                ->order('id', 'asc')
                ->value('id');
            if ($first) {
                return (int) $first;
            }
        } catch (\Throwable $e) {
            // ignore
        }

        try {
            $any = Db::name('enterprises')->whereNull('deletedAt')->order('id', 'asc')->value('id');

            return $any ? (int) $any : 0;
        } catch (\Throwable $e) {
            return 0;
        }
    }
}