ClientAdd.cs

﻿using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using System.Data.SQLite;
using Win2012WMIiSCSIVhdxManager;

namespace NoDiskSystem
{
    public delegate void DelegateUpdateTreeClient(TreeNode node, string tag, Client client);
    public partial class ClientAdd : Form
    {

        public event DelegateUpdateTreeClient updateTree;

        int client_enable = 1;
        string client_name = "";
        string client_mac = "";
        string client_group_id = "";
        string client_work_path = "";
        string client_disk_list_id = "";
        string client_name_prefix = "";

        string iscsiURL;
        string username;
        string password;

        List<VdiskTemplate> vdiskTempletList = new List<VdiskTemplate>();
        List<VdiskTemplate> tovdiskTempletList = new List<VdiskTemplate>();
        public ClientAdd()
        {
            InitializeComponent();
            this.磁盘fromdataGridView.AutoGenerateColumns = false;
            this.磁盘todataGridView.AutoGenerateColumns = false;
        }

        private void button3_Click(object sender, EventArgs e)
        {
            this.Close();
        }

        private void ClientAdd_Load(object sender, EventArgs e)
        {

            IniFile ini = new IniFile(Environment.CurrentDirectory + "\\setting.ini");
            iscsiURL = ini.ReadString("IscsiServer", "URL", "");
            username = ini.ReadString("IscsiServer", "USERNAME", "");
            password = ini.ReadString("IscsiServer", "PASSWORD", "");


            using (SQLiteConnection conn = new SQLiteConnection("data source=nodisk.db"))
            {
                using (SQLiteCommand cmd = new SQLiteCommand())
                {
                    cmd.Connection = conn;
                    conn.Open();

                    SQLiteHelper sh = new SQLiteHelper(cmd);

                    DataTable dt = sh.Select("select * from SYSTEM_SETTING;");



                    foreach (DataRow row in dt.Rows)
                    {
                        
                        client_name_prefix = row["client_name_prefix"].ToString();
                        client_work_path = row["client_disk_path"].ToString();
                    }

                    
                    string sqlString = "select count(*) from CLIENT;";
                    int count = int.Parse(sh.ExecuteScalar(sqlString).ToString());
                    client_name = client_name_prefix + "-" + ++count;
                    名称textBox.Text = client_name;
                    conn.Close();


                }
            }

            using (SQLiteConnection conn = new SQLiteConnection("data source=nodisk.db"))
            {
                using (SQLiteCommand cmd = new SQLiteCommand())
                {
                    DataTable dts = new DataTable();
                    dts.Columns.Add("Text", Type.GetType("System.String"));
                    dts.Columns.Add("Value", Type.GetType("System.String"));
                    dts.Rows.Add("未分组", "0");
                    

                    分组comboBox.DataSource = dts;
                    分组comboBox.DisplayMember = "Text";
                    分组comboBox.ValueMember = "Value";

                    cmd.Connection = conn;
                    conn.Open();

                    SQLiteHelper sh = new SQLiteHelper(cmd);

                    DataTable dt = sh.Select("select * from CLIENT_GROUP;");


                    int count = 1;
                    foreach (DataRow row in dt.Rows)
                    {
                        dts.Rows.Add(row["client_group_name"], count++);
                    }


                    
                    conn.Close();


                }
            }

            工作目录textBox.Text = client_work_path;


            using (SQLiteConnection conn = new SQLiteConnection("data source=nodisk.db"))
            {
                using (SQLiteCommand cmd = new SQLiteCommand())
                {
                    cmd.Connection = conn;
                    conn.Open();

                    SQLiteHelper sh = new SQLiteHelper(cmd);

                    DataTable dt = sh.Select("select * from VDISK_TEMPLET;");

                    VdiskTemplate vdiskTemp;

                    foreach (DataRow row in dt.Rows)
                    {

                        vdiskTemp = new VdiskTemplate();
                        vdiskTemp.DiskId = row["disk_id"].ToString();
                        vdiskTemp.DiskName = row["disk_name"].ToString();
                        vdiskTemp.DiskPath = row["disk_path"].ToString();
                        vdiskTempletList.Add(vdiskTemp);

                    }

                    this.磁盘fromdataGridView.DataSource = vdiskTempletList;

                    conn.Close();
                }
            }


        }

        private void button1_Click(object sender, EventArgs e)
        {
            FolderBrowserDialog path = new FolderBrowserDialog();
            path.ShowDialog();
            工作目录textBox.Text = path.SelectedPath;
        }

        private void 复原button_Click(object sender, EventArgs e)
        {
            using (SQLiteConnection conn = new SQLiteConnection("data source=nodisk.db"))
            {
                using (SQLiteCommand cmd = new SQLiteCommand())
                {
                    cmd.Connection = conn;
                    conn.Open();

                    SQLiteHelper sh = new SQLiteHelper(cmd);

                    DataTable dt = sh.Select("select * from VDISK_TEMPLET;");

                    vdiskTempletList.Clear();

                    VdiskTemplate vdiskTemp;

                    foreach (DataRow row in dt.Rows)
                    {

                        vdiskTemp = new VdiskTemplate();
                        vdiskTemp.DiskId = row["disk_id"].ToString();
                        vdiskTemp.DiskName = row["disk_name"].ToString();
                        vdiskTemp.DiskPath = row["disk_path"].ToString();
                        vdiskTempletList.Add(vdiskTemp);

                    }

                    this.磁盘fromdataGridView.DataSource = null;
                    this.磁盘fromdataGridView.DataSource = vdiskTempletList;

                    this.磁盘todataGridView.DataSource = null;
                    tovdiskTempletList.Clear();

                    conn.Close();
                }
            }

            磁盘todataGridView.DataSource = new List<VdiskTemplate>();
        }

        private void button添加_Click(object sender, EventArgs e)
        {
            string disk_id = "";
            string disk_name = "";
            string disk_path = "";

            if (磁盘fromdataGridView.Rows.Count != 0)
            {
                int a = 磁盘fromdataGridView.CurrentRow.Index;
                disk_id = 磁盘fromdataGridView.Rows[a].Cells["diskId"].Value.ToString();
                disk_name = 磁盘fromdataGridView.Rows[a].Cells["diskName"].Value.ToString();
                disk_path = 磁盘fromdataGridView.Rows[a].Cells["diskPath"].Value.ToString();

                foreach (VdiskTemplate vdisk in vdiskTempletList)
                {
                    if (vdisk.DiskId == disk_id)
                    {
                        vdiskTempletList.Remove(vdisk);
                        break;
                    }
                }
                磁盘fromdataGridView.DataSource = null;
                磁盘fromdataGridView.DataSource = vdiskTempletList;

                //int addindex = this.磁盘todataGridView.Rows.Add();
                //this.磁盘todataGridView.Rows[addindex].Cells["toDiskId"].Value = disk_id;
                //this.磁盘todataGridView.Rows[addindex].Cells["toDiskName"].Value = disk_name;
                磁盘todataGridView.DataSource = null;
                VdiskTemplate tovdisk = new VdiskTemplate();
                tovdisk.DiskId = disk_id;
                tovdisk.DiskName = disk_name;
                tovdisk.DiskPath = disk_path;
                tovdiskTempletList.Add(tovdisk);
                磁盘todataGridView.DataSource = tovdiskTempletList;
            }

            
        }

        private void button删除_Click(object sender, EventArgs e)
        {
            string disk_id = "";
            string disk_name = "";
            string disk_path = "";


            if (磁盘todataGridView.Rows.Count != 0) 
            {
                int a = 磁盘todataGridView.CurrentRow.Index;
                disk_id = 磁盘todataGridView.Rows[a].Cells["toDiskId"].Value.ToString();
                disk_name = 磁盘todataGridView.Rows[a].Cells["toDiskName"].Value.ToString();
                disk_path = 磁盘todataGridView.Rows[a].Cells["toDiskPath"].Value.ToString();

                foreach (VdiskTemplate vdisk in tovdiskTempletList)
                {
                    if (vdisk.DiskId == disk_id)
                    {
                        tovdiskTempletList.Remove(vdisk);
                        break;
                    }
                }
                磁盘todataGridView.DataSource = null;
                磁盘todataGridView.DataSource = tovdiskTempletList;

                //int addindex = this.磁盘todataGridView.Rows.Add();
                //this.磁盘todataGridView.Rows[addindex].Cells["toDiskId"].Value = disk_id;
                //this.磁盘todataGridView.Rows[addindex].Cells["toDiskName"].Value = disk_name;
                磁盘fromdataGridView.DataSource = null;
                VdiskTemplate fromvdisk = new VdiskTemplate();
                fromvdisk.DiskId = disk_id;
                fromvdisk.DiskName = disk_name;
                fromvdisk.DiskPath = disk_path;
                vdiskTempletList.Add(fromvdisk);
                磁盘fromdataGridView.DataSource = vdiskTempletList;
            }

            
        }

        private void 排序向上button_Click(object sender, EventArgs e)
        {
            string disk_id = "";
            string disk_name = "";
            string disk_path = "";

            if (磁盘todataGridView.Rows.Count != 0)
            {
                int rowIndex = 磁盘todataGridView.CurrentRow.Index;
                disk_id = 磁盘todataGridView.Rows[rowIndex].Cells["toDiskId"].Value.ToString();
                disk_name = 磁盘todataGridView.Rows[rowIndex].Cells["toDiskName"].Value.ToString();
                disk_path = 磁盘todataGridView.Rows[rowIndex].Cells["toDiskPath"].Value.ToString();

                if (磁盘todataGridView.Rows.Count == 1)
                {
                    return;
                }
                else {
                    if (rowIndex == 0)
                    {
                        return;
                    }
                    else {
                        for (int i = 0; i < tovdiskTempletList.Count; i++)
                        {
                            if (((VdiskTemplate)tovdiskTempletList[i]).DiskId == disk_id)
                            {

                                ((VdiskTemplate)tovdiskTempletList[i]).DiskId = ((VdiskTemplate)tovdiskTempletList[i - 1]).DiskId;
                                ((VdiskTemplate)tovdiskTempletList[i]).DiskName = ((VdiskTemplate)tovdiskTempletList[i - 1]).DiskName;

                                ((VdiskTemplate)tovdiskTempletList[i - 1]).DiskId = disk_id;
                                ((VdiskTemplate)tovdiskTempletList[i - 1]).DiskName = disk_name;
                                ((VdiskTemplate)tovdiskTempletList[i - 1]).DiskPath = disk_path;

                                磁盘todataGridView.DataSource = null;
                                磁盘todataGridView.DataSource = tovdiskTempletList;

                                磁盘todataGridView.Rows[rowIndex - 1].Selected = true;
                                磁盘todataGridView.Rows[rowIndex].Selected = false;

                                磁盘todataGridView.CurrentCell = 磁盘todataGridView.Rows[rowIndex - 1].Cells[0];

                                break;
                            }
                        }
                    }
                }
            }
        }

        private void 排序向下button_Click(object sender, EventArgs e)
        {
            string disk_id = "";
            string disk_name = "";
            string disk_path = "";

            if (磁盘todataGridView.Rows.Count != 0)
            {
                int rowIndex = 磁盘todataGridView.CurrentRow.Index;
                disk_id = 磁盘todataGridView.Rows[rowIndex].Cells["toDiskId"].Value.ToString();
                disk_name = 磁盘todataGridView.Rows[rowIndex].Cells["toDiskName"].Value.ToString();
                disk_path = 磁盘todataGridView.Rows[rowIndex].Cells["toDiskPath"].Value.ToString();

                if (磁盘todataGridView.Rows.Count == 1)
                {
                    return;
                }
                else
                {
                    if (rowIndex == (磁盘todataGridView.Rows.Count - 1))
                    {
                        return;
                    }
                    else
                    {
                        for (int i = 0; i < tovdiskTempletList.Count; i++)
                        {
                            if (((VdiskTemplate)tovdiskTempletList[i]).DiskId == disk_id)
                            {

                                ((VdiskTemplate)tovdiskTempletList[i]).DiskId = ((VdiskTemplate)tovdiskTempletList[i + 1]).DiskId;
                                ((VdiskTemplate)tovdiskTempletList[i]).DiskName = ((VdiskTemplate)tovdiskTempletList[i + 1]).DiskName;

                                ((VdiskTemplate)tovdiskTempletList[i + 1]).DiskId = disk_id;
                                ((VdiskTemplate)tovdiskTempletList[i + 1]).DiskName = disk_name;
                                ((VdiskTemplate)tovdiskTempletList[i + 1]).DiskPath = disk_path;

                                磁盘todataGridView.DataSource = null;
                                磁盘todataGridView.DataSource = tovdiskTempletList;

                                磁盘todataGridView.Rows[rowIndex + 1].Selected = true;
                                磁盘todataGridView.Rows[rowIndex].Selected = false;

                                磁盘todataGridView.CurrentCell = 磁盘todataGridView.Rows[rowIndex + 1].Cells[0];

                                break;
                            }
                        }
                    }
                }
            }
        }

        private void 应用button_Click(object sender, EventArgs e)
        {
            if((名称textBox.Text.Trim() == "") || (MACtextBox.Text.Trim() == "") || (工作目录textBox.Text.Trim() == ""))
            {
                MessageBox.Show("请填写完整的信息！");
                return;
            }
            string client_id = System.Guid.NewGuid().ToString("N");
            string client_name = 名称textBox.Text.Trim();
            string client_mac = MACtextBox.Text.Trim();
            string client_work_path = 工作目录textBox.Text.Trim();
            string client_description = 备注textBox.Text.Trim();
            string client_group_id = "";
            string client_group_tag = "";
            string client_disk_list_id = System.Guid.NewGuid().ToString("N");
            if (启用工作站checkBox.Checked == true)
            {
                client_enable = 1;
            }
            else {
                client_enable = 0;
            }

            using (SQLiteConnection conn = new SQLiteConnection("data source=nodisk.db"))
            {
                using (SQLiteCommand cmd = new SQLiteCommand())
                {
                    cmd.Connection = conn;
                    conn.Open();

                    SQLiteHelper sh = new SQLiteHelper(cmd);

                    string tempSelectedText = 分组comboBox.Text;

                    DataTable dt = sh.Select("select * from CLIENT_GROUP where client_group_name = '" + tempSelectedText + "';");

                    if (dt.Rows.Count == 0)
                    {
                        client_group_id = "未分组";
                        client_group_tag = "未分组";
                    }else{
                        foreach (DataRow row in dt.Rows)
                        {
                            client_group_id = row["client_group_id"].ToString();
                            client_group_tag = row["client_group_name"].ToString();
                        }
                    }
                    conn.Close();
                }
            }


            //判断是否有重复的MAC地址，如果有则不允许添加
            bool tempMacOnly = false;

            using (SQLiteConnection conn = new SQLiteConnection("data source=nodisk.db"))
            {
                using (SQLiteCommand cmd = new SQLiteCommand())
                {
                    cmd.Connection = conn;
                    conn.Open();

                    SQLiteHelper sh = new SQLiteHelper(cmd);

                    string tempMacString = "select count(*) from CLIENT where client_mac = '" + client_mac + "';";
                    int count = int.Parse(sh.ExecuteScalar(tempMacString).ToString());

                    if (count == 0)
                    {
                        tempMacOnly = true;
                    }
                    else
                    {
                        tempMacOnly = false;
                    }
                    conn.Close();
                }
            }


            if (tempMacOnly == true)
            {
                using (SQLiteConnection conn = new SQLiteConnection("data source=nodisk.db"))
                {
                    using (SQLiteCommand cmd = new SQLiteCommand())
                    {
                        cmd.Connection = conn;
                        conn.Open();

                        SQLiteHelper sh = new SQLiteHelper(cmd);

                        //更新主页datagrid的source
                        Client client = new Client();
                        client.ClientId = client_id;
                        client.ClientName = client_name;
                        client.ClientMac = client_mac;
                        client.clientWorkPath = client_work_path;
                        client.clientDescription = client_description;
                        client.ClientSuperEnable = "0";
                        client.ClientEnable = "1";
                        
                        

                        string addClientSql = "insert into CLIENT (client_id,client_name,client_mac,client_work_path,client_description,client_group_id,client_disk_list_id,client_enable,client_super_enable) values ('" + client_id + "','" + client_name + "','" + client_mac + "','" + client_work_path + "','" + client_description + "','" + client_group_id + "','" + client_disk_list_id + "','"  + client_enable + "','0');";

                        sh.ExecuteScalar(addClientSql);

                        int diskSort = 0;
                        string addClientDiskList = "";
                        string diskId = "";
                        foreach (VdiskTemplate v in tovdiskTempletList)
                        {
                            diskId = v.DiskId;
                           addClientDiskList = "insert into CLIENT_DISK_LIST (client_disk_list_id,disk_id,disk_sort) values ('" + client_disk_list_id + "','" + v.DiskId + "','" + diskSort++ + "');";

                           sh.ExecuteScalar(addClientDiskList);
                        }
                        

                        


                        


                        //创建客户端启动脚本


                        //创建客户端启动盘
                        foreach (VdiskTemplate v in tovdiskTempletList)
                        {
                            
                            //获取最新的还原点的磁盘路径

                            string parentPointPath = "";
                            string sort = "0";
                            DataTable rs = sh.Select("select * from VDISK_RESTORE_POINT where disk_id = '" + v.DiskId + "';");
                            int count = rs.Rows.Count;
                            if (count != 0)
                            {

                                foreach (DataRow rows in rs.Rows)
                                {
                                    if (float.Parse(sort) < float.Parse(rows["vdisk_restore_point_sort"].ToString()))
                                    {
                                        sort = rows["vdisk_restore_point_sort"].ToString();
                                        parentPointPath = rows["vdisk_restore_point_path"].ToString();
                                    }

                                }
                            }
                            else {
                                parentPointPath = v.DiskPath + v.DiskName + ".vhdx";
                            }


                            //创建磁盘文件
                            DiskManagementData ParamObj = new DiskManagementData(); // TODO: 初始化为适当的值

                            
                            ParamObj.VhdxType = 4; //动态盘
                            ParamObj.DevicePath = client_work_path + "\\" + v.DiskName + "-" + client_mac + ".vhdx";
                            ParamObj.TargetName = v.DiskName + "-" + client_mac;
                            ParamObj.DiskSize = ushort.Parse("10");
                            ParamObj.TargetIQN = "HstecsClient." + v.DiskName + "-" + client_mac;
                            //ParamObj.ParentPath = v.DiskPath + v.DiskName + ".vhdx";
                            ParamObj.ParentPath = parentPointPath;
                            ParamObj.serverURL = iscsiURL;
                            ParamObj.Username = username;
                            ParamObj.Password = password;

                            DiskManager.CreateVhdxDisk(ParamObj);

                        }

                        //更新tree及主页datagrid
                        TreeNode node = new TreeNode(client_name, IconIndexes.computer, IconIndexes.computer);
                        node.Tag = client_name;
                        node.Text = client_name;

                        conn.Close();
                        this.Close();

                        UpdateTree(node, client_group_tag, client);

                        MessageBox.Show("工作站添加成功！");
                        
                    }
                }
            }
            else {
                MessageBox.Show("已存在指定MAC地址的客户端！");
            }


        }

        private void UpdateTree(TreeNode node, string tag, Client client)
        {
            updateTree(node, tag, client);
        }
    }
}