using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Text;
using System.Windows.Forms;
using System.Reflection;

namespace ProcessHacker
{
    public partial class ProcessGroups : Form
    {
        private int _phandle;

        public ProcessGroups(int PID)
        {
            InitializeComponent();

            Misc.SetDoubleBuffered(listGroups, typeof(ListView), true);

            listGroups.ContextMenu = ListViewMenu.GetMenu(listGroups);

            _phandle = Win32.OpenProcess(Win32.PROCESS_RIGHTS.PROCESS_QUERY_INFORMATION, 0, PID);

            if (_phandle == 0)
            {
                MessageBox.Show("Could not open process handle:\n\n" + Win32.GetLastErrorMessage(),
                    "Process Hacker", MessageBoxButtons.OK,
                    MessageBoxIcon.Error);

                this.Close();
                return;
            }

            Win32.TOKEN_GROUPS groups = Win32.ReadTokenGroups(_phandle);

            if (groups.GroupCount == 0)
            {
                Win32.CloseHandle(_phandle);

                MessageBox.Show("Could not read process groups:\n\n" + Win32.GetLastErrorMessage(),
                "Process Hacker", MessageBoxButtons.OK,
                MessageBoxIcon.Error);

                this.Close();
                return;
            }

            for (int i = 0; i < groups.GroupCount; i++)
            {
                string name = Win32.GetAccountName(groups.Groups[i].SID,
                    Properties.Settings.Default.ShowAccountDomains);

                if (name == "")
                    continue;

                ListViewItem item = listGroups.Items.Add(name.ToLower(), name, 0);

                item.BackColor = GetAttributeColor(groups.Groups[i].Attributes);
                item.SubItems.Add(new ListViewItem.ListViewSubItem(item,
                    GetAttributeString(groups.Groups[i].Attributes)));
            }
        }

        private void ProcessGroups_Load(object sender, EventArgs e)
        {
            this.Size = Properties.Settings.Default.GroupWindowSize;
            ColumnSettings.LoadSettings(Properties.Settings.Default.GroupListColumns, listGroups);
        }

        private void ProcessGroups_FormClosing(object sender, FormClosingEventArgs e)
        {
            Properties.Settings.Default.GroupWindowSize = this.Size;
            Properties.Settings.Default.GroupListColumns = ColumnSettings.SaveSettings(listGroups);

            Win32.CloseHandle(_phandle);
        }

        private void buttonClose_Click(object sender, EventArgs e)
        {
            this.Close();
        }

        private string GetAttributeString(Win32.SID_ATTRIBUTES Attributes)
        {
            string text = "";

            if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_INTEGRITY) != 0)
            {
                if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_INTEGRITY_ENABLED) != 0)
                    return "Integrity";
                else
                    return "Integrity (Disabled)";
            }
            else if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_LOGON_ID) != 0)
                text = "Logon ID";
            else if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_MANDATORY) != 0)
                text = "Mandatory";
            else if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_OWNER) != 0)
                text = "Owner";
            else if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_RESOURCE) != 0)
                text = "Resource";
            else if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_USE_FOR_DENY_ONLY) != 0)
                text = "Use for Deny Only";

            if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_ENABLED_BY_DEFAULT) != 0)
                return text + " (Default Enabled)";
            else if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_ENABLED) != 0)
                return text;
            else
                return text + " (Disabled)";
        }

        private Color GetAttributeColor(Win32.SID_ATTRIBUTES Attributes)
        {
            if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_INTEGRITY) != 0)
            {
                if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_INTEGRITY_ENABLED) == 0)
                    return Color.FromArgb(0xe0e0e0);
                else
                    return Color.White;
            }

            if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_ENABLED_BY_DEFAULT) != 0)
                return Color.FromArgb(0xe0f0e0);
            else if ((Attributes & Win32.SID_ATTRIBUTES.SE_GROUP_ENABLED) != 0) 
                return Color.White;
            else
                return Color.FromArgb(0xf0e0e0);
        }
    }
}