poke/core/LightTube/Database/DatabaseManager.cs

using System;
using System.Collections.Generic;
using System.Linq;
using InnerTube;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Primitives;
using MongoDB.Driver;
using Newtonsoft.Json;

namespace LightTube.Database
{
	public static class DatabaseManager
	{
		public static readonly string ApiUaRegex = "LightTubeApiClient\\/([0-9.]*) ([\\S]+?)\\/([0-9.]*) \\(([\\s\\S]+?)\\)";

		private static IMongoCollection<LTUser> _userCollection;
		private static IMongoCollection<LTLogin> _tokenCollection;
		private static IMongoCollection<LTChannel> _channelCacheCollection;
		private static IMongoCollection<LTPlaylist> _playlistCollection;
		private static IMongoCollection<LTVideo> _videoCacheCollection;
		public static LoginManager Logins { get; private set; }
		public static ChannelManager Channels { get; private set; }
		public static PlaylistManager Playlists { get; private set; }

		public static void Init(string connstr, Youtube youtube)
		{
			MongoClient client = new(connstr);
			IMongoDatabase database = client.GetDatabase("lighttube");
			_userCollection = database.GetCollection<LTUser>("users");
			_tokenCollection = database.GetCollection<LTLogin>("tokens");
			_playlistCollection = database.GetCollection<LTPlaylist>("playlists");
			_channelCacheCollection = database.GetCollection<LTChannel>("channelCache");
			_videoCacheCollection = database.GetCollection<LTVideo>("videoCache");
			Logins = new LoginManager(_userCollection, _tokenCollection);
			Channels = new ChannelManager(_channelCacheCollection);
			Playlists = new PlaylistManager(_userCollection, _playlistCollection, _videoCacheCollection, youtube);
		}

		public static void CreateLocalAccount(this HttpContext context)
		{
			bool accountExists = false;

			// Check local account
			if (context.Request.Cookies.TryGetValue("account_data", out string accountJson))
			{
				try
				{
					if (accountJson != null)
					{
						LTUser tempUser = JsonConvert.DeserializeObject<LTUser>(accountJson) ?? new LTUser();
						if (tempUser.UserID == "Local Account" && tempUser.PasswordHash == "local_account")
							accountExists = true;
					}
				}
				catch { }
			}

			// Account already exists, just leave it there
			if (accountExists) return;

			LTUser user = new()
			{
				UserID = "Local Account",
				PasswordHash = "local_account",
				SubscribedChannels = new List<string>()
			};

			context.Response.Cookies.Append("account_data", JsonConvert.SerializeObject(user), new CookieOptions
			{
				Expires = DateTimeOffset.MaxValue 
			});
		}

		public static bool TryGetUser(this HttpContext context, out LTUser user, string requiredScope)
		{
			// Check local account
			if (context.Request.Cookies.TryGetValue("account_data", out string accountJson))
			{
				try
				{
					if (accountJson != null)
					{
						LTUser tempUser = JsonConvert.DeserializeObject<LTUser>(accountJson) ?? new LTUser();
						if (tempUser.UserID == "Local Account" && tempUser.PasswordHash == "local_account")
						{
							user = tempUser;
							return true;
						}
					}
				}
				catch
				{
					user = null;
					return false;
				}
			}
			
			// Check cloud account
			if (!context.Request.Cookies.TryGetValue("token", out string token))
				if (context.Request.Headers.TryGetValue("Authorization", out StringValues tokens))
					token = tokens.ToString();
				else
				{
					user = null;
					return false;
				}

			try
			{
				if (token != null)
				{
					user = Logins.GetUserFromToken(token).Result;
					LTLogin login = Logins.GetLoginFromToken(token).Result;
					if (login.Scopes.Contains(requiredScope))
					{
#pragma warning disable 4014
						login.UpdateLastAccess(DateTimeOffset.Now);
#pragma warning restore 4014
						return true;
					}
					return false;
				}
			}
			catch
			{
				user = null;
				return false;
			}

			user = null;
			return false;
		}

		public static bool TryGetRssUser(string token, out LTUser user)
		{
			if (token is null)
			{
				user = null;
				return false;
			}

			try
			{
				user = Logins.GetUserFromRssToken(token).Result;
				return true;
			}
			catch
			{
				user = null;	
				return false;
			}
		}
	}
}