From 494f11202f58066699762c88b56db776d18d95f9 Mon Sep 17 00:00:00 2001
From: Daniel Supernault <danielsupernault@gmail.com>
Date: Wed, 15 Jun 2022 00:46:20 -0600
Subject: [PATCH] Update LiveStream model

---
 app/Http/Controllers/LiveStreamController.php | 429 +++++++++++-------
 app/Models/LiveStream.php                     |  58 ++-
 2 files changed, 280 insertions(+), 207 deletions(-)

diff --git a/app/Http/Controllers/LiveStreamController.php b/app/Http/Controllers/LiveStreamController.php
index 51cc160f9..ecc999770 100644
--- a/app/Http/Controllers/LiveStreamController.php
+++ b/app/Http/Controllers/LiveStreamController.php
@@ -12,225 +12,304 @@ use App\Services\LiveStreamService;
 
 class LiveStreamController extends Controller
 {
-    public function createStream(Request $request)
-    {
-    	abort_if(!config('livestreaming.enabled'), 400);
-    	abort_if(!$request->user(), 403);
+	public function createStream(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->user(), 403);
 
-    	if(config('livestreaming.broadcast.limits.enabled')) {
-    		if($request->user()->is_admin) {
+		if(config('livestreaming.broadcast.limits.enabled')) {
+			if($request->user()->is_admin) {
 
-    		} else {
-    			$limits = config('livestreaming.broadcast.limits');
-    			$user = $request->user();
-    			abort_if($limits['admins_only'] && $user->is_admin == false, 401, 'LSE:003');
-    			if($limits['min_account_age']) {
-    				abort_if($user->created_at->gt(now()->subDays($limits['min_account_age'])), 403, 'LSE:005');
-    			}
+			} else {
+				$limits = config('livestreaming.broadcast.limits');
+				$user = $request->user();
+				abort_if($limits['admins_only'] && $user->is_admin == false, 401, 'LSE:003');
+				if($limits['min_account_age']) {
+					abort_if($user->created_at->gt(now()->subDays($limits['min_account_age'])), 403, 'LSE:005');
+				}
 
-    			if($limits['min_follower_count']) {
-    				$account = AccountService::get($user->profile_id);
-    				abort_if($account['followers_count'] < $limits['min_follower_count'], 403, 'LSE:008');
-    			}
-    		}
-    	}
+				if($limits['min_follower_count']) {
+					$account = AccountService::get($user->profile_id);
+					abort_if($account['followers_count'] < $limits['min_follower_count'], 403, 'LSE:008');
+				}
+			}
+		}
 
-    	$this->validate($request, [
-    		'name' => 'nullable|string|max:80',
-    		'description' => 'nullable|string|max:240',
-    		'visibility' => 'required|in:public,private'
-    	]);
+		$this->validate($request, [
+			'name' => 'nullable|string|max:80',
+			'description' => 'nullable|string|max:240',
+			'visibility' => 'required|in:public,private'
+		]);
 
-    	$stream = new LiveStream;
-    	$stream->name = $request->input('name');
-    	$stream->description = $request->input('description');
-    	$stream->visibility = $request->input('visibility');
-    	$stream->profile_id = $request->user()->profile_id;
-    	$stream->stream_id = Str::random(40);
-    	$stream->stream_key = Str::random(64);
-    	$stream->save();
+		$stream = new LiveStream;
+		$stream->name = $request->input('name');
+		$stream->description = $request->input('description');
+		$stream->visibility = $request->input('visibility');
+		$stream->profile_id = $request->user()->profile_id;
+		$stream->stream_id = Str::random(40) . '_' . $stream->profile_id;
+		$stream->stream_key = 'streamkey-' . Str::random(64);
+		$stream->save();
 
-    	return [
-    		'url' => $stream->getStreamKeyUrl(),
-    		'id' => $stream->stream_id
-    	];
-    }
+		return [
+			'host' => $stream->getStreamServer(),
+			'key' => $stream->stream_key,
+			'url' => $stream->getStreamKeyUrl(),
+			'id' => $stream->stream_id
+		];
+	}
 
-    public function getUserStream(Request $request)
-    {
-    	abort_if(!config('livestreaming.enabled'), 400);
-    	abort_if(!$request->user(), 403);
+	public function getUserStream(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->user(), 403);
 
-    	$stream = LiveStream::whereProfileId($request->input('profile_id'))->first();
+		$stream = LiveStream::whereProfileId($request->input('profile_id'))->first();
 
-    	if(!$stream) {
-    		return [];
-    	}
+		if(!$stream) {
+			return [];
+		}
 
-    	$res = [];
-    	$owner = $stream->profile_id == $request->user()->profile_id;
+		$res = [];
+		$owner = $stream->profile_id == $request->user()->profile_id;
 
-    	if($stream->visibility === 'private') {
-    		abort_if(!$owner && !FollowerService::follows($request->user()->profile_id, $stream->profile_id), 403, 'LSE:011');
-    	}
+		if($stream->visibility === 'private') {
+			abort_if(!$owner && !FollowerService::follows($request->user()->profile_id, $stream->profile_id), 403, 'LSE:011');
+		}
 
-    	if($owner) {
-    		$res['stream_key'] = $stream->stream_key;
-    		$res['stream_id'] = $stream->stream_id;
-    		$res['stream_url'] = $stream->getStreamKeyUrl();
-    	}
+		if($owner) {
+			$res['stream_key'] = $stream->stream_key;
+			$res['stream_id'] = $stream->stream_id;
+			$res['stream_url'] = $stream->getStreamKeyUrl();
+		}
 
-    	if($stream->live_at == null) {
-    		$res['hls_url'] = null;
-	    	$res['name'] = $stream->name;
-	    	$res['description'] = $stream->description;
-	    	return $res;
-    	}
+		if($stream->live_at == null) {
+			$res['hls_url'] = null;
+			$res['name'] = $stream->name;
+			$res['description'] = $stream->description;
+			return $res;
+		}
 
-    	$res = [
-    		'hls_url' => $stream->getHlsUrl(),
-    		'name' => $stream->name,
-    		'description' => $stream->description
-    	];
+		$res = [
+			'hls_url' => $stream->getHlsUrl(),
+			'name' => $stream->name,
+			'description' => $stream->description
+		];
 
-    	return response()->json($res, 200, [], JSON_UNESCAPED_SLASHES);
-    }
+		return response()->json($res, 200, [], JSON_UNESCAPED_SLASHES);
+	}
 
-    public function deleteStream(Request $request)
-    {
-    	abort_if(!config('livestreaming.enabled'), 400);
-    	abort_if(!$request->user(), 403);
+	public function deleteStream(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->user(), 403);
 
-    	LiveStream::whereProfileId($request->user()->profile_id)
-    		->get()
-    		->each(function($stream) {
-    			Storage::deleteDirectory("public/live-hls/{$stream->stream_id}");
-    			$stream->delete();
-    		});
+		LiveStream::whereProfileId($request->user()->profile_id)
+			->get()
+			->each(function($stream) {
+				Storage::deleteDirectory("public/live-hls/{$stream->stream_id}");
+				$stream->delete();
+			});
 
-    	return [200];
-    }
+		return [200];
+	}
 
-    public function getActiveStreams(Request $request)
-    {
-    	abort_if(!config('livestreaming.enabled'), 400);
-    	abort_if(!$request->user(), 403);
+	public function getActiveStreams(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->user(), 403);
 
-    	return LiveStream::whereVisibility('local')->whereNotNull('live_at')->get()->map(function($stream) {
-    		return [
-    			'account' => AccountService::get($stream->profile_id),
-    			'stream_id' => $stream->stream_id
-    		];
-    	});
-    }
+		return LiveStream::whereVisibility('local')->whereNotNull('live_at')->get()->map(function($stream) {
+			return [
+				'account' => AccountService::get($stream->profile_id),
+				'stream_id' => $stream->stream_id
+			];
+		});
+	}
 
-    public function getLatestChat(Request $request)
-    {
-    	abort_if(!config('livestreaming.enabled'), 400);
-    	abort_if(!$request->user(), 403);
+	public function getLatestChat(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->user(), 403);
 
-    	$stream = LiveStream::whereProfileId($request->input('profile_id'))
-    		->whereNotNull('live_at')
-    		->first();
+		$stream = LiveStream::whereProfileId($request->input('profile_id'))
+			->whereNotNull('live_at')
+			->first();
 
-    	if(!$stream) {
-    		return [];
-    	}
+		if(!$stream) {
+			return [];
+		}
 
-    	$owner = $stream->profile_id == $request->user()->profile_id;
-    	if($stream->visibility === 'private') {
-    		abort_if(!$owner && !FollowerService::follows($request->user()->profile_id, $stream->profile_id), 403, 'LSE:021');
-    	}
+		$owner = $stream->profile_id == $request->user()->profile_id;
+		if($stream->visibility === 'private') {
+			abort_if(!$owner && !FollowerService::follows($request->user()->profile_id, $stream->profile_id), 403, 'LSE:021');
+		}
 
-    	$res = collect(LiveStreamService::getComments($stream->profile_id))
-    		->map(function($r) {
-    			return json_decode($r);
-    		});
+		$res = collect(LiveStreamService::getComments($stream->profile_id))
+			->map(function($res) {
+				return json_decode($res);
+			});
 
-    	return $res;
-    }
+		return $res;
+	}
 
-    public function addChatComment(Request $request)
-    {
-    	abort_if(!config('livestreaming.enabled'), 400);
-    	abort_if(!$request->user(), 403);
+	public function addChatComment(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->user(), 403);
 
-    	$this->validate($request, [
-    		'profile_id' => 'required|exists:profiles,id',
-    		'message' => 'required|max:140'
-    	]);
+		$this->validate($request, [
+			'profile_id' => 'required|exists:profiles,id',
+			'message' => 'required|max:140'
+		]);
 
-    	$stream = LiveStream::whereProfileId($request->input('profile_id'))->firstOrFail();
+		$stream = LiveStream::whereProfileId($request->input('profile_id'))->firstOrFail();
 
-    	$owner = $stream->profile_id == $request->user()->profile_id;
-    	if($stream->visibility === 'private') {
-    		abort_if(!$owner && !FollowerService::follows($request->user()->profile_id, $stream->profile_id), 403, 'LSE:022');
-    	}
+		$owner = $stream->profile_id == $request->user()->profile_id;
+		if($stream->visibility === 'private') {
+			abort_if(!$owner && !FollowerService::follows($request->user()->profile_id, $stream->profile_id), 403, 'LSE:022');
+		}
 
-    	$res = [
-    		'pid' => (string) $request->user()->profile_id,
-    		'username' => $request->user()->username,
-    		'text' => $request->input('message'),
-    		'ts' => now()->timestamp
-    	];
+		$res = [
+			'pid' => (string) $request->user()->profile_id,
+			'username' => $request->user()->username,
+			'text' => $request->input('message'),
+			'ts' => now()->timestamp
+		];
 
-    	LiveStreamService::addComment($stream->profile_id, json_encode($res, JSON_UNESCAPED_SLASHES));
+		LiveStreamService::addComment($stream->profile_id, json_encode($res, JSON_UNESCAPED_SLASHES));
 
-    	return $res;
-    }
+		return $res;
+	}
 
-    public function editStream(Request $request)
-    {
-    	abort_if(!config('livestreaming.enabled'), 400);
-    	abort_if(!$request->user(), 403);
+	public function editStream(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->user(), 403);
 
-    	$this->validate($request, [
-    		'name' => 'nullable|string|max:80',
-    		'description' => 'nullable|string|max:240'
-    	]);
+		$this->validate($request, [
+			'name' => 'nullable|string|max:80',
+			'description' => 'nullable|string|max:240'
+		]);
 
-    	$stream = LiveStream::whereProfileId($request->user()->profile_id)->firstOrFail();
-    	$stream->name = $request->input('name');
-    	$stream->description = $request->input('description');
-    	$stream->save();
+		$stream = LiveStream::whereProfileId($request->user()->profile_id)->firstOrFail();
+		$stream->name = $request->input('name');
+		$stream->description = $request->input('description');
+		$stream->save();
 
-    	return;
-    }
+		return;
+	}
 
-    public function deleteChatComment(Request $request)
-    {
-    	abort_if(!config('livestreaming.enabled'), 400);
-    	abort_if(!$request->user(), 403);
+	public function deleteChatComment(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->user(), 403);
 
-    	$this->validate($request, [
-    		'profile_id' => 'required|exists:profiles,id',
-    		'message' => 'required'
-    	]);
+		$this->validate($request, [
+			'profile_id' => 'required|exists:profiles,id',
+			'message' => 'required'
+		]);
 
-    	abort_if($request->user()->profile_id != $request->input('profile_id'), 403);
+		abort_if($request->user()->profile_id != $request->input('profile_id'), 403);
 
-    	$stream = LiveStream::whereProfileId($request->user()->profile_id)->firstOrFail();
+		$stream = LiveStream::whereProfileId($request->user()->profile_id)->firstOrFail();
 
-    	$payload = $request->input('message');
-    	$payload = json_encode($payload, JSON_UNESCAPED_SLASHES);
-    	LiveStreamService::deleteComment($stream->profile_id, $payload);
+		$payload = $request->input('message');
+		$payload = json_encode($payload, JSON_UNESCAPED_SLASHES);
+		LiveStreamService::deleteComment($stream->profile_id, $payload);
 
-    	return;
-    }
+		return;
+	}
 
-    public function getConfig(Request $request)
-    {
-    	abort_if(!config('livestreaming.enabled'), 400);
-    	abort_if(!$request->user(), 403);
+	public function getConfig(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->user(), 403);
 
-    	$res = [
-    		'enabled' => config('livestreaming.enabled'),
-    		'broadcast' => [
-    			'sources' => config('livestreaming.broadcast.sources'),
-    			'limits' => config('livestreaming.broadcast.limits')
-    		],
-    	];
+		$res = [
+			'enabled' => config('livestreaming.enabled'),
+			'broadcast' => [
+				'sources' => config('livestreaming.broadcast.sources'),
+				'limits' => config('livestreaming.broadcast.limits')
+			],
+		];
 
-    	return response()->json($res, 200, [], JSON_UNESCAPED_SLASHES);
-    }
+		return response()->json($res, 200, [], JSON_UNESCAPED_SLASHES);
+	}
+
+	public function clientBroadcastPublish(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		$key = $request->input('name');
+		$name = $request->input('name');
+
+		abort_if(!$name, 400);
+
+		if(empty($key)) {
+			abort_if(!$request->filled('tcurl'), 400);
+			$url = $this->parseStreamUrl($request->input('tcurl'));
+			$key = $request->filled('name') ? $request->input('name') : $url['name'];
+		}
+
+		$token = substr($name, 0, 10) === 'streamkey-';
+
+		if($token) {
+			$stream = LiveStream::whereStreamKey($key)->firstOrFail();
+			return redirect($stream->getStreamRtmpUrl(), 301);
+		} else {
+			$stream = LiveStream::whereStreamId($key)->firstOrFail();
+		}
+
+		if($request->filled('name') && $token == false) {
+			$stream->live_at = now();
+			$stream->save();
+			return [];
+		} else {
+			abort(400);
+		}
+
+		abort(400);
+	}
+
+	public function clientBroadcastFinish(Request $request)
+	{
+		abort_if(!config('livestreaming.enabled'), 400);
+		abort_if(!$request->filled('tcurl'), 400);
+		$url = $this->parseStreamUrl($request->input('tcurl'));
+		$name = $url['name'] ?? $request->input('name');
+
+		$stream = LiveStream::whereStreamId($name)->whereStreamKey($url['key'])->firstOrFail();
+
+		if(config('livestreaming.broadcast.delete_token_after_finished')) {
+			$stream->delete();
+		} else {
+			$stream->live_at = null;
+			$stream->save();
+		}
+
+		return [];
+	}
+
+	protected function parseStreamUrl($url)
+	{
+		$name = null;
+		$key = null;
+		$query = parse_url($url, PHP_URL_QUERY);
+		$parts = explode('&', $query);
+		foreach($parts as $part) {
+			if (!strlen(trim($part))) {
+				continue;
+			}
+			$s = explode('=', $part);
+			if(in_array($s[0], ['name', 'key'])) {
+				if($s[0] === 'name') {
+					$name = $s[1];
+				}
+				if($s[0] === 'key') {
+					$key = $s[1];
+				}
+			}
+		}
+
+		return ['name' => $name, 'key' => $key];
+	}
 }
diff --git a/app/Models/LiveStream.php b/app/Models/LiveStream.php
index 197bd70ad..98b8d8211 100644
--- a/app/Models/LiveStream.php
+++ b/app/Models/LiveStream.php
@@ -8,40 +8,34 @@ use Storage;
 
 class LiveStream extends Model
 {
-    use HasFactory;
+	use HasFactory;
 
-    public function getHlsUrl()
-    {
-    	$path = Storage::url("live-hls/{$this->stream_id}/index.m3u8");
-    	return url($path);
-    }
+	public function getHlsUrl()
+	{
+		$path = Storage::url("live-hls/{$this->stream_id}/index.m3u8");
+		return url($path);
+	}
 
-    public function getStreamKeyUrl()
-    {
-    	$proto = 'rtmp://';
-    	$host = config('livestreaming.server.host');
-    	$port = ':' . config('livestreaming.server.port');
-    	$path = '/' . config('livestreaming.server.path') . '?';
-    	$query = http_build_query([
-    		'name' => $this->stream_id,
-    		'key' => $this->stream_key,
-    		'ts' => time()
-    	]);
+	public function getStreamServer()
+	{
+		$proto = 'rtmp://';
+		$host = config('livestreaming.server.host');
+		$port = ':' . config('livestreaming.server.port');
+		$path = '/' . config('livestreaming.server.path');
+		return $proto . $host . $port . $path;
+	}
 
-    	return $proto . $host . $port . $path . $query;
-    }
+	public function getStreamKeyUrl()
+	{
+		$path = $this->getStreamServer() . '?';
+		$query = http_build_query([
+			'name' => $this->stream_key,
+		]);
+		return $path . $query;
+	}
 
-    public function getStreamRtmpUrl()
-    {
-    	$proto = 'rtmp://';
-    	$host = config('livestreaming.server.host');
-    	$port = ':' . config('livestreaming.server.port');
-    	$path = '/' . config('livestreaming.server.path') . '/'. $this->stream_id . '?';
-    	$query = http_build_query([
-    		'key' => $this->stream_key,
-    		'ts' => time()
-    	]);
-
-    	return $proto . $host . $port . $path . $query;
-    }
+	public function getStreamRtmpUrl()
+	{
+		return $this->getStreamServer() . '/' . $this->stream_id;
+	}
 }