HomeControl.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Diagnostics;
using System.Drawing;
using System.Globalization;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using System.Windows.Forms;
using System.Windows.Forms.DataVisualization.Charting;

namespace ScreenTime
{
    public partial class HomeControl : UserControl
    {
        [DllImport("user32.dll")]
        private static extern IntPtr GetForegroundWindow();

        [DllImport("user32.dll", SetLastError = true, CharSet = CharSet.Auto)]
        private static extern int GetWindowText(IntPtr hWnd, StringBuilder lpString, int nMaxCount);

        [DllImport("user32.dll")]
        private static extern uint GetWindowThreadProcessId(IntPtr hWnd, out uint lpdwProcessId);

        string finString;
        long start;
        long end;
        string currentApp;
        int today = DateTimeOffset.UtcNow.Day;
        private CancellationTokenSource cancellationTokenSource = new CancellationTokenSource();
        string status = "Offline";


        public HomeControl()
        {
            InitializeComponent();
            InitializeTimer();
            //TodaysUsage();
            UpdateStatus("Online");
            statusLbl.Text = status;

        }

        private void InitializeTimer()
        {
            System.Windows.Forms.Timer timer = new System.Windows.Forms.Timer();
            timer.Interval = 1000; // 1 second
            if (!DesignMode)
            {
                timer.Tick += Timer_Tick;
            }
            timer.Start();
        }



        private void UpdateStatus(string newStatus)
        {
            status = newStatus;
            statusLbl.Text = status;
            Debug.WriteLine($"Sent status: {status}");
        }


        private void Timer_Tick(object sender, EventArgs e)
        {
            System.Windows.Forms.Timer timer = (System.Windows.Forms.Timer)sender;
            timer.Stop();

            int currentDay = DateTimeOffset.UtcNow.Day;
            try
            {
                IntPtr handle = GetForegroundWindow();
                StringBuilder windowText = new StringBuilder(256);
                if (GetWindowText(handle, windowText, 256) > 0)
                {
                    string activeApp = windowText.ToString();

                    // Extract the part after the last hyphen
                    string simpleAppName = GetSimplifiedAppName(activeApp);

                    // Get the process ID of the active window
                    GetWindowThreadProcessId(handle, out uint processId);
                    string processName = GetProcessNameById((int)processId);
                    if (processName.Contains("chrome.exe") && (Constants.BRAINROT_TITLES.Contains(simpleAppName)))
                    {
                        processName = simpleAppName;
                        //UpdateStatus("Brain rotting");
                    }





                    // Check if the process name or window title is in the exclusion lists
                    if (Constants.SYSTEM_PROCESS_NAMES.Contains(processName) || Constants.SYSTEM_WINDOW_TITLES.Contains(simpleAppName))
                    {
                        // If the current app is being tracked, stop tracking it
                        if (currentApp != null)
                        {
                            end = DateTimeOffset.UtcNow.ToUnixTimeSeconds();
                            Debug.WriteLine($"{currentApp}, Start-{start}, End-{end}");
                            /*
                            if (!DesignMode)
                            {
                                await Task.Run(() => DatabaseHelper.InsertDataAsync(currentApp, start, end));
                            }
                            */
                            DatabaseHelper.InsertData(currentApp, start, end);

                            currentApp = null; // Reset currentApp
                        }
                         appString.Text = "Tracking paused for system app";
                         return;
                    }

                    if (currentDay != today)
                    {
                        if (currentApp != null)
                        {
                            end = DateTimeOffset.UtcNow.ToUnixTimeSeconds() - 1;
                            Debug.WriteLine($"{currentApp}, Start-{start}, End-{end}");
                            /*
                            if (!DesignMode)
                            {
                                await Task.Run(() => DatabaseHelper.InsertDataAsync(currentApp, start, end));
                            }
                            */
                            DatabaseHelper.InsertData(currentApp, start, end);

                        }

                        today = currentDay;
                    }


                    if (currentApp != processName)
                    {
                        if (currentApp != null)
                        {
                            end = DateTimeOffset.UtcNow.ToUnixTimeSeconds();
                            Debug.WriteLine($"{currentApp}, Start-{start}, End-{end}");

                            /*
                            if (!DesignMode)
                            {
                                await Task.Run(() => DatabaseHelper.InsertDataAsync(currentApp, start, end));
                            }
                            */
                            DatabaseHelper.InsertData(currentApp, start, end);
                        }

                        currentApp = processName;
                        start = DateTimeOffset.UtcNow.ToUnixTimeSeconds();
                        if (processName.Contains("chrome.exe") && (Constants.BRAINROT_TITLES.Contains(simpleAppName)))
                        {
                            processName = simpleAppName;
                            UpdateStatus("Brain rotting");
                        }
                        else if (Constants.GAMING_TITLES.Contains(processName))
                        {
                            UpdateStatus("Gaming");
                        }
                        else if (Constants.CODING_TITLES.Contains(processName))
                        {
                            UpdateStatus("Coding");
                        }
                        else {
                            UpdateStatus("Online");
                        }
                        finString = processName;
                    }

                    appString.Text = $"{finString}";
                }
                else
                {
                     //TODO: rn doesnt stop the db recoprding process
                                                       // If the current app is being tracked, stop tracking it
                    if (currentApp != null)
                    {
                        end = DateTimeOffset.UtcNow.ToUnixTimeSeconds();
                        Debug.WriteLine($"{currentApp}, Start-{start}, End-{end}");
                        /*
                        if (!DesignMode)
                        {
                            await Task.Run(() => DatabaseHelper.InsertDataAsync(currentApp, start, end));
                        }
                        */
                        DatabaseHelper.InsertData(currentApp, start, end);

                        currentApp = null; // Reset currentApp
                    }
                    appString.Text = $"No active app";
                    UpdateStatus("Online");
                    return;
                }
            }
            finally
            {
                timer.Start();
            }
        }

        private string GetSimplifiedAppName(string fullAppName)
        {
            string[] websiteKeywords = new string[] { "YouTube", "/ X", "Reddit", "4chan", "on X:" };

            // Check if the fullAppName contains any of the website keywords
            foreach (var keyword in websiteKeywords)
            {
                if (fullAppName.Contains(keyword))
                {
                    // Return the keyword with the browser name

                    return $"{keyword} (chrome.exe)";
                }
            }
            int lastHyphenIndex = fullAppName.LastIndexOf('-');
            if (lastHyphenIndex >= 0 && lastHyphenIndex < fullAppName.Length - 1)
            {
                return fullAppName.Substring(lastHyphenIndex + 1).Trim();
            }
            return fullAppName; // Return the full name if no hyphen is found
        }

        private string GetProcessNameById(int processId)
        {
            try
            {
                var process = System.Diagnostics.Process.GetProcessById(processId);
                return process.ProcessName + ".exe";
            }
            catch (Exception)
            {
                return "Unknown";
            }
        }

        
        public void HandleFormClosing()
        {
            cancellationTokenSource.Cancel();
            // End the current session with the current timestamp
            if (!string.IsNullOrEmpty(currentApp))
            {
                UpdateStatus("Offline");
                end = DateTimeOffset.UtcNow.ToUnixTimeSeconds();
                Debug.WriteLine($"{currentApp}, Start-{start}, End-{end}");
                //Task.Run(() => DatabaseHelper.InsertDataAsync(currentApp, start, end));
                DatabaseHelper.InsertData(currentApp, start, end);
            }
            cancellationTokenSource.Dispose();
            DatabaseHelper.Dispose();
        }

        public void TodaysUsage()
        {
            DateTime currentDate = DateTime.Now;
            
            string bruh = currentDate.ToString("dd/MM/yyyy");
            string updateTime = currentDate.ToString("dd/MM/yyyy HH:mm");
            CalendarHelper calendarHelper = new CalendarHelper();
            long startDay = calendarHelper.GetDayStart(bruh);
            long endDay = calendarHelper.GetDayEnd(bruh);
            Debug.WriteLine($"{startDay}, {endDay}");

            var analyzer = new UsageAnalyzer();

            var dataFromDay = DatabaseHelper.GetDataFromSpecificDay(startDay, endDay);

            var appUsageSeconds = analyzer.CalculateAppUsageSeconds(dataFromDay);

            string savedTimeZone = Properties.Settings.Default.UserTimeZone;

            string testing = $"Last update: {updateTime}\nTimezone: {savedTimeZone}\n";

            var sortedAppUsage = appUsageSeconds.OrderByDescending(app => app.Value);
            

            int totalUsage = 0;

            foreach (var app in sortedAppUsage)
            {
                Debug.WriteLine($"{app.Key}: {app.Value} seconds");


                if (app.Value > 3599)
                {
                    TimeSpan timeSpan = TimeSpan.FromSeconds(app.Value);
                    totalUsage += app.Value;
                    int minutes = timeSpan.Minutes;
                    int hours = timeSpan.Hours;
                    testing += $"{app.Key} for {hours} hours and {minutes} minutes.\n";

                }
                else if (app.Value > 59)
                {
                    TimeSpan timeSpan = TimeSpan.FromSeconds(app.Value);
                    totalUsage += app.Value;
                    int minutes = timeSpan.Minutes;
                    testing += $"{app.Key} for {minutes} minutes.\n";
                }
                else
                {
                    totalUsage += app.Value;
                    testing += $"{app.Key} for {app.Value} seconds.\n";
                }
                
            }
            TimeSpan timeSpan1 = TimeSpan.FromSeconds(totalUsage);
            int totalMinutes = timeSpan1.Minutes;
            int totalHours = timeSpan1.Hours;


            testing += $"\nTotal of {totalHours} hours and {totalMinutes} minutes.";

            todayUsageLbl.Text = testing;
        }
        

        private void appString_Click(object sender, EventArgs e)
        {
           
        }

        private void tableLayoutPanel1_Paint(object sender, PaintEventArgs e)
        {

        }

        private void todayUsageLbl_Click(object sender, EventArgs e)
        {

        }

        private void lblColumn_Click(object sender, EventArgs e)
        {

        }

        private void lblColumn2_Click(object sender, EventArgs e)
        {

        }

        private void button1_Click(object sender, EventArgs e)
        {
            TodaysUsage();
        }

        private void statusLbl_Click(object sender, EventArgs e)
        {

        }
    }
}