<?php

namespace App\Http\Controllers;

use App\Components\Helpers;
use App\Http\Models\Device;
use App\Http\Models\SsGroup;
use App\Http\Models\SsNode;
use App\Http\Models\User;
use App\Http\Models\UserLabel;
use App\Http\Models\UserSubscribe;
use App\Http\Models\UserSubscribeLog;
use Illuminate\Http\Request;
use Redirect;
use Response;

/**
 * 订阅控制器
 *
 * Class SubscribeController
 *
 * @package App\Http\Controllers
 */
class SubscribeController extends Controller
{
	protected static $systemConfig;

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

	// 订阅码列表
	public function subscribeList(Request $request)
	{
		$user_id = $request->input('user_id');
		$username = $request->input('username');
		$status = $request->input('status');

		$query = UserSubscribe::with(['user:id,username']);

		if(isset($user_id)){
			$query->where('user_id', $user_id);
		}

		if(isset($username)){
			$query->whereHas('user', function($q) use ($username){
				$q->where('username', 'like', '%'.$username.'%');
			});
		}

		if(isset($status)){
			$query->where('status', $status);
		}

		$view['subscribeList'] = $query->orderBy('id', 'desc')->paginate(20)->appends($request->except('page'));

		return Response::view('subscribe.subscribeList', $view);
	}

	//订阅记录
	public function subscribeLog(Request $request)
	{
		$id = $request->input('id');
		$query = UserSubscribeLog::with('user:username');

		if(isset($id)){
			$query->where('sid', $id);
		}

		$view['subscribeLog'] = $query->orderBy('id', 'desc')->paginate(20)->appends($request->except('page'));

		return Response::view('subscribe.subscribeLog', $view);
	}

	// 订阅设备列表
	public function deviceList(Request $request)
	{
		$type = $request->input('type');
		$platform = $request->input('platform');
		$name = $request->input('name');
		$status = $request->input('status');

		$query = Device::query();

		if(isset($type)){
			$query->where('type', $type);
		}

		if(isset($platform)){
			$query->where('platform', $platform);
		}

		if(isset($name)){
			$query->where('name', 'like', '%'.$name.'%');
		}

		if(isset($status)){
			$query->where('status', $status);
		}

		$view['deviceList'] = $query->paginate(20)->appends($request->except('page'));

		return Response::view('subscribe.deviceList', $view);
	}

	// 设置用户的订阅的状态
	public function setSubscribeStatus(Request $request)
	{
		$id = $request->input('id');
		$status = $request->input('status', 0);

		if(empty($id)){
			return Response::json(['status' => 'fail', 'data' => '', 'message' => '操作异常']);
		}

		if($status){
			UserSubscribe::query()->where('id', $id)->update(['status' => 1, 'ban_time' => 0, 'ban_desc' => '']);
		}else{
			UserSubscribe::query()->where('id', $id)->update(['status' => 0, 'ban_time' => time(), 'ban_desc' => '后台手动封禁']);
		}

		return Response::json(['status' => 'success', 'data' => '', 'message' => '操作成功']);
	}

	// 设置设备是否允许订阅的状态
	public function setDeviceStatus(Request $request)
	{
		$id = $request->input('id');
		$status = $request->input('status', 0);

		if(empty($id)){
			return Response::json(['status' => 'fail', 'data' => '', 'message' => '操作异常']);
		}

		Device::query()->where('id', $id)->update(['status' => $status]);

		return Response::json(['status' => 'success', 'data' => '', 'message' => '操作成功']);
	}

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

		// 校验合法性
		$subscribe = UserSubscribe::query()->with('user')->where('status', 1)->where('code', $code)->first();
		if(!$subscribe){
			exit($this->noneNode());
		}

		$user = User::query()->whereIn('status', [0, 1])->where('enable', 1)->where('id', $subscribe->user_id)->first();
		if(!$user){
			exit($this->noneNode());
		}

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

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

		// 获取这个账号可用节点
		$userLabelIds = UserLabel::query()->where('user_id', $user->id)->pluck('label_id');
		if(empty($userLabelIds)){
			exit($this->noneNode());
		}

		$query = SsNode::query()->selectRaw('ss_node.*')->leftjoin("ss_node_label", "ss_node.id", "=", "ss_node_label.node_id");

		// 启用混合订阅时，加入V2Ray节点，未启用时仅下发SSR节点信息
		if(!self::$systemConfig['mix_subscribe']){
			$query->where('ss_node.type', 1);
		}

		$nodeList = $query->where('ss_node.status', 1)->where('ss_node.is_subscribe', 1)->whereIn('ss_node_label.label_id', $userLabelIds)->groupBy('ss_node.id')->orderBy('ss_node.sort', 'desc')->orderBy('ss_node.id', 'asc')->get()->toArray();
		if(empty($nodeList)){
			exit($this->noneNode());
		}

		// 打乱数组
		if(self::$systemConfig['rand_subscribe']){
			shuffle($nodeList);
		}

		$scheme = '';

		// 展示到期时间和剩余流量
		if(self::$systemConfig['is_custom_subscribe']){
			$scheme .= $this->expireDate($user).$this->lastTraffic($user);
		}

		// 控制客户端最多获取节点数
		foreach($nodeList as $key => $node){
			// 控制显示的节点数
			if(self::$systemConfig['subscribe_max'] && $key >= self::$systemConfig['subscribe_max']){
				break;
			}
			// 获取分组名称
			$host = $node['server']? : $node['ip'];
			if($node['type'] == 1){
				$group = SsGroup::query()->where('id', $node['group_id'])->first();
				$group = empty($group)? Helpers::systemConfig()['website_name'] : $group->name;
				$obfs_param = $user->obfs_param? : $node['obfs_param'];

				if($node['single']){
					$port = $node['port'];
					$protocol = $node['protocol'];
					$method = $node['method'];
					$obfs = $node['obfs'];
					$passwd = $node['passwd'];
					$protocol_param = $user->port.':'.$user->passwd;
				}else{
					$port = $user->port;
					$protocol = $user->protocol;
					$method = $user->method;
					$obfs = $user->obfs;
					$passwd = $user->passwd;
					$protocol_param = $user->protocol_param;
				}

				// 生成ssr scheme
				$scheme .= 'ssr://'.base64url_encode($host.':'.$port.':'.$protocol.':'.$method.':'.$obfs.':'.base64url_encode($passwd).'/?obfsparam='.base64url_encode($obfs_param).'&protoparam='.base64url_encode($protocol_param).'&remarks='.base64url_encode($node['name']).'&group='.base64url_encode($group).'&udpport=0&uot=0')."\n";
			}else{
				// 生成v2ray scheme
				$v2_json = [
					"v"    => "2",
					"ps"   => $node['name'],
					"add"  => $host,
					"port" => $node['v2_port'],
					"id"   => $user->vmess_id,
					"aid"  => $node['v2_alter_id'],
					"net"  => $node['v2_net'],
					"type" => $node['v2_type'],
					"host" => $node['v2_host'],
					"path" => $node['v2_path'],
					"tls"  => $node['v2_tls']? "tls" : ""
				];

				$scheme .= 'vmess://'.base64url_encode(json_encode($v2_json, JSON_PRETTY_PRINT))."\n";
			}
		}

		// 适配Quantumult的自定义订阅头
		if(self::$systemConfig['is_custom_subscribe']){
			$headers = [
				'Content-type'          => 'application/octet-stream; charset=utf-8',
				'Cache-Control'         => 'no-store, no-cache, must-revalidate',
				'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);
		}else{
			return Response::make(base64url_encode($scheme));
		}
	}

	// 写入订阅访问日志
	private function log($subscribeId, $ip, $headers)
	{
		$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();
	}

	// 抛出无可用的节点信息，用于兼容防止客户端订阅失败
	private function noneNode()
	{
		return base64url_encode('ssr://'.base64url_encode('0.0.0.0:1:origin:none:plain:'.base64url_encode('0000').'/?obfsparam=&protoparam=&remarks='.base64url_encode('无可用节点或账号被封禁或订阅被封禁').'&group='.base64url_encode('错误').'&udpport=0&uot=0')."\n");
	}

	/**
	 * 过期时间
	 *
	 * @param object $user
	 *
	 * @return string
	 */
	private function expireDate($user)
	{
		$text = '到期时间: '.$user->expire_time;

		return 'ssr://'.base64url_encode('0.0.0.1:1:origin:none:plain:'.base64url_encode('0000').'/?obfsparam=&protoparam=&remarks='.base64url_encode($text).'&group='.base64url_encode(Helpers::systemConfig()['website_name']).'&udpport=0&uot=0')."\n";
	}

	/**
	 * 剩余流量
	 *
	 * @param object $user
	 *
	 * @return string
	 */
	private function lastTraffic($user)
	{
		$text = '剩余流量: '.flowAutoShow($user->transfer_enable-$user->u-$user->d);

		return 'ssr://'.base64url_encode('0.0.0.2:1:origin:none:plain:'.base64url_encode('0000').'/?obfsparam=&protoparam=&remarks='.base64url_encode($text).'&group='.base64url_encode(Helpers::systemConfig()['website_name']).'&udpport=0&uot=0')."\n";
	}
}