<?php

namespace App\Http\Controllers\User;

use App\Components\IP;
use App\Http\Controllers\Controller;
use App\Models\UserSubscribe;
use App\Models\UserSubscribeLog;
use Arr;
use Illuminate\Http\Request;
use Redirect;
use Response;

class SubscribeController extends Controller
{
    private $subType;

    // 通过订阅码获取订阅信息
    public function getSubscribeByCode(Request $request, $code)
    {
        if (empty($code)) {
            return Redirect::route('login');
        }
        $this->subType = $request->input('type');

        // 检查订阅码是否有效
        $subscribe = UserSubscribe::whereCode($code)->first();
        if (!$subscribe) {
            exit($this->infoGenerator('使用的订阅链接错误！请重新从官网获取！'));
        }

        if ($subscribe->status !== 1) {
            exit($this->infoGenerator('您的订阅链接已被封禁，请前往官网查询原因！'));
        }

        // 检查用户是否有效
        $user = $subscribe->user;
        if (!$user) {
            exit($this->infoGenerator('错误订阅链接，账号不存在！请前往官网重新获取订阅链接'));
        }

        if ($user->status === -1) {
            exit($this->infoGenerator('您的账号已经被禁止使用，请重新注册'));
        }

        if ($user->enable !== 1) {
            $unusedTransfer = $user->transfer_enable - $user->u - $user->d;
            if ($user->ban_time) {
                exit($this->infoGenerator('您的账号处于封禁状态，请在'.date('Y-m-d H:i:s', $user->ban_time).'之后再更新！'));
            }

            if ($unusedTransfer <= 0) {
                exit($this->infoGenerator('账号流量耗尽！请前往官网购买或重置流量！'));
            }

            if ($user->expired_at < date('Y-m-d')) {
                exit($this->infoGenerator('账号过期！请前往官网购买！'));
            }

            exit($this->infoGenerator('账号存在问题，请前往官网查询！'));
        }

        // 更新访问次数
        $subscribe->increment('times', 1);

        // 记录每次请求
        $this->subscribeLog($subscribe->id, IP::getClientIp(), $request->headers);

        // 获取这个账号可用节点
        $query = $user->whereIsSubscribe(1)->userAccessNodes();

        if ($this->subType === 1) {
            $query = $query->whereIn('type', [1, 4]);
        } elseif ($this->subType) {
            $query = $query->whereType($this->subType);
        }

        $nodeList = $query->orderByDesc('sort')->orderBy('id')->get()->toArray();
        if (empty($nodeList)) {
            exit($this->infoGenerator('无可用节点'));
        }

        // 打乱数组
        if (sysConfig('rand_subscribe')) {
            $nodeList = Arr::shuffle($nodeList);
        }

        $scheme = null;

        // 展示到期时间和剩余流量
        if (sysConfig('is_custom_subscribe')) {
            $scheme .= $this->infoGenerator('到期时间: '.($user->expired_at < date('Y-m-d') ? '过期' : $user->expired_at)).$this->infoGenerator('剩余流量: '.flowAutoShow($user->transfer_enable - $user->u - $user->d));
        }

        // 控制客户端最多获取节点数
        foreach ($nodeList as $key => $node) {
            // 控制显示的节点数
            if (sysConfig('subscribe_max') && $key >= sysConfig('subscribe_max')) {
                break;
            }

            $scheme .= $this->getUserNodeInfo($user->id, $node['id'], 0).PHP_EOL;
        }

        $headers = [
            'Content-type'  => 'application/octet-stream; charset=utf-8',
            'Cache-Control' => 'no-store, no-cache, must-revalidate',
            //'Content-Disposition' => 'attachment; filename='.$filename
        ];

        // 适配Quantumult的自定义订阅头
        if (sysConfig('is_custom_subscribe')) {
            $headers['Subscription-Userinfo'] = 'upload='.$user->u.'; download='.$user->d.'; total='.$user->transfer_enable.'; expire='.strtotime($user->expired_at);
        }

        return Response::make(base64url_encode($scheme), 200, $headers);
    }

    // 抛出错误的节点信息，用于兼容防止客户端订阅失败
    private function infoGenerator($text): string
    {
        $result = null;
        switch ($this->subType) {
            case 2:
                $result = 'vmess://'.base64url_encode(json_encode([
                        "v"   => "2", "ps" => $text, "add" => "0.0.0.0", "port" => 0, "id" => 0,
                        "aid" => 0, "net" => "tcp", "type" => "none", "host" => "", "path" => "/",
                        'tls' => "tls",
                    ], JSON_PRETTY_PRINT));
                break;
            case 3:
                $result = 'trojan://0@0.0.0.0:0?peer=0.0.0.0#'.rawurlencode($text);
                break;
            case 1:
            case 4:
            default:
                $result = 'ssr://'.base64url_encode('0.0.0.0:0:origin:none:plain:'.base64url_encode('0000').'/?obfsparam=&protoparam=&remarks='.base64url_encode($text).'&group='.base64url_encode(sysConfig('website_name')).'&udpport=0&uot=0');
                break;
        }

        return $result.PHP_EOL;
    }

    // 写入订阅访问日志
    private function subscribeLog($subscribeId, $ip, $headers): void
    {
        $log = new UserSubscribeLog();
        $log->user_subscribe_id = $subscribeId;
        $log->request_ip = $ip;
        $log->request_time = date('Y-m-d H:i:s');
        $log->request_header = $headers;
        $log->save();
    }
}