CMLeonOS/Gui/ShellComponents/Lock.cs

// The CMLeonOS Project (https://github.com/Leonmmcoset/CMLeonOS)
// Copyright (C) 2025-present LeonOS 2 Developer Team
// 
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
// 
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

using Cosmos.System.Graphics;
using CMLeonOS;
using CMLeonOS.Gui.UILib;
using CMLeonOS.Logger;

using System;
using System.Collections.Generic;
using System.Drawing;

namespace CMLeonOS.Gui.ShellComponents
{
    internal class Lock : Process
    {
        internal Lock() : base("Lock", ProcessType.Application) { }

        AppWindow window;

        Table userTable;
        
        TextBox passwordBox;
        Button logOnButton;

        WindowManager wm = ProcessManager.GetProcess<WindowManager>();

        Sound.SoundService soundService = ProcessManager.GetProcess<Sound.SoundService>();

        private static class Images
        {
            [IL2CPU.API.Attribs.ManifestResourceStream(ResourceName = "CMLeonOS.Gui.Resources.Lock.Key.bmp")]
            private static byte[] _iconBytes_Key;
            internal static Bitmap Icon_Key = new Bitmap(_iconBytes_Key);

            [IL2CPU.API.Attribs.ManifestResourceStream(ResourceName = "CMLeonOS.Gui.Resources.Lock.User.bmp")]
            private static byte[] _iconBytes_User;
            internal static Bitmap Icon_User = new Bitmap(_iconBytes_User);

            [IL2CPU.API.Attribs.ManifestResourceStream(ResourceName = "CMLeonOS.Gui.Resources.Lock.UserArrow.bmp")]
            private static byte[] _iconBytes_UserArrow;
            internal static Bitmap Icon_UserArrow = new Bitmap(_iconBytes_UserArrow);
        }

        private const int width = 700;
        private const int height = 420;
        private const int leftPanelWidth = 240;
        private const int rightContentPadding = 28;
        private const int rightContentBottomPadding = 30;
        private const int accountsLabelY = 102;
        private const int tableY = 124;
        private const int tableHeight = 140;
        private const int passwordLabelY = tableY + tableHeight + 18;
        private const int passwordY = passwordLabelY + 22;
        private const int passwordHeight = 30;
        private const int logOnButtonY = passwordY + 44;
        private const int logOnButtonHeight = 38;

        private readonly Color windowBackground = Color.FromArgb(232, 238, 246);
        private readonly Color outerBorder = Color.FromArgb(169, 181, 198);
        private readonly Color leftPanel = Color.FromArgb(34, 53, 84);
        private readonly Color leftPanelAccent = Color.FromArgb(74, 124, 201);
        private readonly Color leftPanelSoft = Color.FromArgb(53, 77, 116);
        private readonly Color rightPanel = Color.FromArgb(247, 250, 253);
        private readonly Color titleColor = Color.FromArgb(28, 39, 56);
        private readonly Color bodyColor = Color.FromArgb(92, 103, 119);
        private readonly Color hintColor = Color.FromArgb(132, 142, 156);
        private readonly Color inputBackground = Color.White;
        private readonly Color inputForeground = Color.FromArgb(31, 42, 55);
        private readonly Color selectionBackground = Color.FromArgb(223, 236, 255);
        private readonly Color selectionBorder = Color.FromArgb(91, 140, 223);
        private readonly Color primaryButton = Color.FromArgb(53, 111, 214);
        private readonly Color primaryButtonBorder = Color.FromArgb(33, 83, 171);
        private readonly Color primaryButtonText = Color.White;

        private double shakiness = 0;

        private List<User> users = new List<User>();

        private void RenderBackground()
        {
            window.Clear(windowBackground);
            window.DrawRectangle(0, 0, width, height, outerBorder);

            int leftWidth = leftPanelWidth;
            int rightX = leftWidth + 1;
            int rightWidth = width - rightX;

            window.DrawFilledRectangle(0, 0, leftWidth, height, leftPanel);
            window.DrawFilledRectangle(leftWidth - 6, 0, 6, height, leftPanelAccent);
            window.DrawFilledRectangle(rightX, 0, rightWidth, height, rightPanel);

            window.DrawFilledRectangle(20, 22, 56, 56, leftPanelSoft);
            window.DrawRectangle(20, 22, 56, 56, Color.FromArgb(109, 149, 214));
            window.DrawImageAlpha(Images.Icon_Key, 32, 34);

            window.DrawString("Welcome Back", Color.White, 20, 96);
            window.DrawString("CMLeonOS Desktop", Color.FromArgb(190, 208, 233), 20, 120);

            string selectedUserName = GetSelectedUser()?.Username ?? "Guest";
            window.DrawString("Selected account", Color.FromArgb(171, 190, 217), 20, 168);
            window.DrawString(selectedUserName, Color.White, 20, 190);
            window.DrawString(GetSelectedUserSubtitle(), Color.FromArgb(176, 197, 224), 20, 214);

            window.DrawFilledRectangle(20, height - 72, leftWidth - 40, 44, leftPanelSoft);
            window.DrawRectangle(20, height - 72, leftWidth - 40, 44, Color.FromArgb(88, 118, 171));
            window.DrawString("Tip: type password.", Color.White, 32, height - 62);
            window.DrawString("Press Enter to sign in.", Color.FromArgb(190, 208, 233), 32, height - 44);

            int rightContentX = leftWidth + rightContentPadding;
            window.DrawString("Sign in", titleColor, rightContentX, 28);
            window.DrawString("Choose an account, then enter your password.", bodyColor, rightContentX, 54);
            window.DrawString("Continue to the desktop when ready.", bodyColor, rightContentX, 72);

            window.DrawString("Accounts", hintColor, rightContentX, accountsLabelY);
            window.DrawString("Password", hintColor, rightContentX, passwordLabelY);
            window.DrawString("Secure local session", Color.FromArgb(118, 128, 141), rightContentX, height - rightContentBottomPadding);

        }

        private User GetSelectedUser()
        {
            if (userTable == null || users == null || users.Count == 0)
            {
                return null;
            }

            if (userTable.SelectedCellIndex < 0 || userTable.SelectedCellIndex >= users.Count)
            {
                return null;
            }

            return users[userTable.SelectedCellIndex];
        }

        private string GetSelectedUserSubtitle()
        {
            User selectedUser = GetSelectedUser();
            if (selectedUser == null)
            {
                return "No local users available";
            }

            return selectedUser.Admin ? "Administrator account" : "Standard local account";
        }

        private void RefreshLayout()
        {
            RenderBackground();
            wm.Update(window);
        }

        private void ShowError(string text)
        {
            MessageBox messageBox = new MessageBox(this, "Logon Failed", text);
            messageBox.Show();
        }

        private void Shake()
        {
            shakiness = 24;
        }

        private void LogOn()
        {
            if (userTable.SelectedCellIndex < 0 || passwordBox.Text.Trim() == string.Empty)
            {
                return;
            }

            if (userTable.SelectedCellIndex >= users.Count)
            {
                return;
            }

            User selectedUser = users[userTable.SelectedCellIndex];
            string password = passwordBox.Text.Trim();

            Logger.Logger.Instance.Info("Lock", $"Attempting login for user: {selectedUser.Username}");
            Logger.Logger.Instance.Info("Lock", $"UserSystem.GetUsers() returned: {(users == null ? "null" : users.Count.ToString())}");

            if (users == null || users.Count == 0)
            {
                Logger.Logger.Instance.Error("Lock", "User list is null or empty!");
                ShowError("User system error. Please restart.");
                Shake();
                return;
            }

            Logger.Logger.Instance.Info("Lock", $"User found: {selectedUser.Username}, Admin: {selectedUser.Admin}");
            string hashedInputPassword = UserSystem.HashPasswordSha256(password);

            if (selectedUser.Password != hashedInputPassword)
            {
                Logger.Logger.Instance.Warning("Lock", $"Authentication failed for user: {selectedUser.Username}");
                passwordBox.Text = string.Empty;

                if (selectedUser.LockedOut)
                {
                    TimeSpan remaining = selectedUser.LockoutEnd - DateTime.Now;
                    if (remaining.Minutes > 0)
                    {
                        ShowError($"Try again in {remaining.Minutes}m, {remaining.Seconds}s.");
                    }
                    else
                    {
                        ShowError($"Try again in {remaining.Seconds}s.");
                    }
                }

                wm.Update(window);
                soundService.PlaySystemSound(Sound.SystemSound.Alert);
                Shake();
                return;
            }

            Logger.Logger.Instance.Info("Lock", $"Authentication successful for user: {selectedUser.Username}");
            TryStop();
            UserSystem.SetCurrentLoggedInUser(selectedUser);
            ProcessManager.AddProcess(wm, new ShellComponents.Taskbar()).Start();
            ProcessManager.AddProcess(wm, new ShellComponents.Dock.Dock()).Start();
            soundService.PlaySystemSound(Sound.SystemSound.Login);

            Logger.Logger.Instance.Info("Lock", $"{selectedUser.Username} logged on to GUI.");
        }

        private void LogOnClick(int x, int y)
        {
            LogOn();
        }

        #region Process
        public override void Start()
        {
            base.Start();
            
            users = UserSystem.GetUsers() ?? new List<User>();
            Logger.Logger.Instance.Info("Lock", $"Lock started. Total users: {users?.Count ?? 0}");
            if (users != null)
            {
                foreach (var user in users)
                {
                    Logger.Logger.Instance.Info("Lock", $"  - User: {user.Username}, Admin: {user.Admin}");
                }
            }
            
            window = new AppWindow(this, (int)(wm.ScreenWidth / 2 - width / 2), (int)(wm.ScreenHeight / 2 - height / 2), width, height);
            window.Title = "CMLeonOS Login";
            window.Icon = Images.Icon_Key;
            window.CanMove = false;
            window.CanClose = false;
            wm.AddWindow(window);

            RenderBackground();

            int contentX = leftPanelWidth + rightContentPadding;
            int contentWidth = width - contentX - rightContentPadding;
            
            userTable = new Table(window, contentX, tableY, contentWidth, tableHeight);
            userTable.CellHeight = 40;
            userTable.Background = inputBackground;
            userTable.Foreground = inputForeground;
            userTable.Border = Color.FromArgb(203, 212, 224);
            userTable.SelectedBackground = selectionBackground;
            userTable.SelectedForeground = inputForeground;
            userTable.SelectedBorder = selectionBorder;
            userTable.AllowSelection = true;
            userTable.AllowDeselection = false;
            userTable.ScrollbarThickness = 18;
            userTable.TableCellSelected = _ => RefreshLayout();
            
            foreach (var user in users)
            {
                var cell = new TableCell(user.Username);
                userTable.Cells.Add(cell);
            }
            
            if (userTable.Cells.Count > 0)
            {
                userTable.SelectedCellIndex = 0;
            }
            
            wm.AddWindow(userTable);

            passwordBox = new TextBox(window, contentX, passwordY, contentWidth, passwordHeight);
            passwordBox.Shield = true;
            passwordBox.PlaceholderText = "Enter password";
            passwordBox.Background = inputBackground;
            passwordBox.Foreground = inputForeground;
            passwordBox.PlaceholderForeground = hintColor;
            passwordBox.Changed = RefreshLayout;
            passwordBox.Submitted = LogOn;
            wm.AddWindow(passwordBox);

            logOnButton = new Button(window, contentX, logOnButtonY, contentWidth, logOnButtonHeight);
            logOnButton.Text = "Log On";
            logOnButton.Background = primaryButton;
            logOnButton.Border = primaryButtonBorder;
            logOnButton.Foreground = primaryButtonText;
            logOnButton.Image = Images.Icon_UserArrow;
            logOnButton.ImageLocation = Button.ButtonImageLocation.Left;
            logOnButton.OnClick = LogOnClick;
            wm.AddWindow(logOnButton);

            RefreshLayout();
        }

        public override void Run()
        {
            int oldX = window.X;
            int newX = (int)((wm.ScreenWidth / 2) - (width / 2) + (Math.Sin(shakiness) * 8));
            if (oldX != newX)
            {
                window.Move(newX, window.Y);
            }
            shakiness /= 1.1;
        }

        public override void Stop()
        {
            base.Stop();
            wm.RemoveWindow(window);
        }
        #endregion
    }
}