<?php

namespace App\Http\Controllers\User;

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

class SubscribeController extends Controller {
	protected static $systemConfig;
	private $subType;

	public function __construct() {
		self::$systemConfig = Helpers::systemConfig();
	}

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

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

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

		// 检查用户是否有效
		$user = User::find($subscribe->user_id);
		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 > 0){
				exit($this->infoGenerator('您的账号处于封禁状态，请在'.date('Y-m-d H:i:s', $user->ban_time).'之后再更新！'));
			}

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

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

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

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

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

		// 获取这个账号可用节点
		$query = Node::query()
		             ->whereStatus(1)
		             ->whereIsSubscribe(1)
		             ->groupNodePermit($user->group_id)
		             ->where('level', '<=', $user->level);

		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(self::$systemConfig['rand_subscribe']){
			$nodeList = Arr::shuffle($nodeList);
		}

		$scheme = null;

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

		// 控制客户端最多获取节点数
		foreach($nodeList as $key => $node){
			// 控制显示的节点数
			if(self::$systemConfig['subscribe_max'] && $key >= self::$systemConfig['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(self::$systemConfig['is_custom_subscribe']){
			$headers['Subscription-Userinfo'] = 'upload='.$user->u.'; download='.$user->d.'; total='.$user->transfer_enable.'; expire='.strtotime($user->expire_time);
		}

		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:
			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(self::$systemConfig['website_name']).'&udpport=0&uot=0');
				break;

		}
		return $result.PHP_EOL;
	}

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