owo

core/LightTube/Database/DatabaseManager.cs

@@ -0,0 +1,154 @@
+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;
+			}
+		}
+	}
+}