<?php

namespace App\Http\Controllers\User;

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::to('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, 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();
    }

}