App_win_iot_tmp/ConUpdate/ViewModel/ProgressViewModel.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.IO.Compression;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows;

namespace ConUpdate.ViewModel
{
    class ProgressViewModel : ViewModelBase
    {
        private double currentProgress = 0;
        public double CurrentProgress
        {
            get { return currentProgress; }
            set
            {
                currentProgress = value;
                Val = Math.Round(((value / _maxLength) * 100), 2);
                Message = value + "/" + _maxLength;
            }
        }
        private LogHelper log = new LogHelper();
        private double _maxLength;
        private string containerName = "IOTContainer.exe";
        #region 组件

        private double val = 0.00;
        public double Val
        {
            get { return val; }
            set
            {
                val = value;
                OnPropertyChanged("Val");
            }
        }

        private string title = string.Empty;
        public string Title
        {
            get { return title; }
            set
            {
                title = value;
                OnPropertyChanged("Title");
            }
        }

        private int maxValue = 100;
        public int MaxValue
        {
            get { return maxValue; }
            set
            {
                maxValue = value;
                OnPropertyChanged("MaxValue");
            }
        }

        private string message = string.Empty;
        public string Message
        {
            get { return message; }
            set
            {
                message = value;
                OnPropertyChanged("Message");
            }
        }
        #endregion


        public ProgressViewModel(string filePath)
        {
            Task.Run(() =>
            {
                UpdateContainer(filePath);
            });
        }
        #region 方法
        /// <summary>
        /// 更新容器
        /// </summary>
        /// <param name="filePath"></param>
        public void UpdateContainer(string filePath)
        {
            try
            {
                log.WriteInfoFile($"关闭容器{containerName}中...");
                CloseContainer();
                log.WriteInfoFile($"加载压缩文件{filePath}...");
                Title = "加载新版文件...请稍等...";
                var basePath = AppDomain.CurrentDomain.BaseDirectory;
                var tempDir = Path.Combine(basePath, "updateTempFolder");
                if (Directory.Exists(tempDir))
                {
                    Directory.Delete(tempDir, true);
                }
                ZipFile.ExtractToDirectory(filePath, tempDir);
                log.WriteInfoFile($"解压文件{filePath}...");
                Title = "正在更新版本...";
                GetFilesCount(tempDir, ref _maxLength);
                CopyDireToDire(tempDir, basePath);
                Directory.Delete(tempDir, true);
                //File.Delete(filePath);
                Title = "更新已完成...";
                log.WriteInfoFile($"更新容器成功！...");
                OpenContainer();
                Environment.Exit(0);
            }
            catch (Exception ex)
            {
                Title = "更新失败...";
                log.WriteInfoFile("容器更新失败：" + ex.Message);
                OpenContainer();
                Environment.Exit(0);
            }
        }
        /// <summary>
        /// 将一个文件夹下的所有东西复制到另一个文件夹 (可备份文件夹)
        /// </summary>
        /// <param name="sourceDire">源文件夹全名</param>
        /// <param name="destDire">目标文件夹全名</param>
        /// <param name="backupsDire">备份文件夹全名</param>
        public void CopyDireToDire(string sourceDire, string destDire, string backupsDire = null)
        {
            if (Directory.Exists(sourceDire) && Directory.Exists(destDire))
            {
                DirectoryInfo sourceDireInfo = new DirectoryInfo(sourceDire);
                FileInfo[] fileInfos = sourceDireInfo.GetFiles();
                foreach (FileInfo fInfo in fileInfos)
                {
                    CurrentProgress += 1;
                    string sourceFile = fInfo.FullName;
                    string destFile = sourceFile.Replace(sourceDire, destDire);
                    if (backupsDire != null && File.Exists(destFile))
                    {
                        Directory.CreateDirectory(backupsDire);
                        string backFile = destFile.Replace(destDire, backupsDire);
                        File.Copy(destFile, backFile, true);
                    }
                    File.Copy(sourceFile, destFile, true);
                }
                DirectoryInfo[] direInfos = sourceDireInfo.GetDirectories();
                foreach (DirectoryInfo dInfo in direInfos)
                {
                    string sourceDire2 = dInfo.FullName;
                    string destDire2 = sourceDire2.Replace(sourceDire, destDire);
                    string backupsDire2 = null;
                    if (backupsDire != null)
                    {
                        backupsDire2 = sourceDire2.Replace(sourceDire, backupsDire);
                    }
                    Directory.CreateDirectory(destDire2);
                    CopyDireToDire(sourceDire2, destDire2, backupsDire2);
                }
            }
        }
        /// <summary>
        /// 获取文件总数
        /// </summary>
        /// <param name="dir"></param>
        /// <param name="count"></param>
        public void GetFilesCount(string dir, ref double count)
        {
            count += new DirectoryInfo(dir).GetFiles().Length;
            var dirs = new DirectoryInfo(dir).GetDirectories();
            foreach (var item in dirs)
            {
                GetFilesCount(item.FullName, ref count);
            }
        }
        /// <summary>
        /// 开启容器
        /// </summary>
        public void OpenContainer()
        {
            App app = ((App)Application.Current);
            //if (killexe)
            {
                System.Security.Principal.WindowsIdentity identity = System.Security.Principal.WindowsIdentity.GetCurrent();
                //创建Windows用户主题
                System.Windows.Forms.Application.EnableVisualStyles();
                System.Security.Principal.WindowsPrincipal principal = new System.Security.Principal.WindowsPrincipal(identity);
                //判断当前登录用户是否为管理员
                if (!principal.IsInRole(System.Security.Principal.WindowsBuiltInRole.Administrator))
                {
                    //创建启动对象
                    System.Diagnostics.ProcessStartInfo startInfo = new System.Diagnostics.ProcessStartInfo();
                    //设置运行文件
                    startInfo.WorkingDirectory = AppDomain.CurrentDomain.BaseDirectory.Replace("\\", "/");//设置应用程序在其所在目录
                    startInfo.FileName = startInfo.WorkingDirectory + "/"+ containerName;                //设置启动参数
                    startInfo.Arguments = app.info + "&true";
                    log.WriteInfoFile("发送" + startInfo.FileName);
                    //设置启动动作,确保以管理员身份运行
                    startInfo.Verb = "runas";
                    //如果不是管理员，则启动UAC
                    Process.Start(startInfo);
                    //退出
                    System.Windows.Forms.Application.Exit();
                }
                else
                {
                    System.Diagnostics.Process p = new System.Diagnostics.Process();
                    p.StartInfo.WorkingDirectory = AppDomain.CurrentDomain.BaseDirectory.Replace("\\", "/");//设置应用程序在其所在目录
                    p.StartInfo.FileName = p.StartInfo.WorkingDirectory + "/"+ containerName;
                    p.StartInfo.Arguments = app.info + "&true";
                    log.WriteInfoFile("管理员发送" + p.StartInfo.FileName);
                    p.StartInfo.UseShellExecute = false;    //是否使用操作系统shell启动
                    p.StartInfo.RedirectStandardInput = true;//接受来自调用程序的输入信息
                    p.StartInfo.RedirectStandardOutput = true;//由调用程序获取输出信息
                    p.StartInfo.RedirectStandardError = true;//重定向标准错误输出
                    p.StartInfo.CreateNoWindow = false;//不显示程序窗口
                    p.Start();//启动程序
                }
            }
        }
        /// <summary>
        /// 关闭容器
        /// </summary>
        public void CloseContainer()
        {
            try
            {
                Process[] pro = Process.GetProcesses();
                var process = pro.FirstOrDefault(p => string.Equals(p.ProcessName, containerName.Substring(0, containerName.IndexOf(".")), StringComparison.OrdinalIgnoreCase));
                if(process != null)
                {
                    process.Kill();
                    log.WriteInfoFile("关闭容器...");
                }
            }
            catch (Exception ex)
            {
                log.WriteInfoFile("关闭容器失败：" + ex.ToString());
            }
        }
        #endregion

    }
}