From 13f00edefd80139655235f5a1ffc0fecc02a3004 Mon Sep 17 00:00:00 2001
From: qianlifeng <qianlf2008@163.com>
Date: Mon, 6 Jan 2014 22:21:08 +0800
Subject: [PATCH] fix shutdown didn't work issue

---
 WinAlfred.Plugin.System/Sys.cs |  3 ++-
 WinAlfred/App.xaml             |  2 +-
 WinAlfred/App.xaml.cs          | 39 ++++++++++++++++++++++++----------
 WinAlfred/MainWindow.xaml.cs   | 23 +++++++++++++++++++-
 WinAlfred/WinAlfred.csproj     | 11 ++++++++++
 5 files changed, 64 insertions(+), 14 deletions(-)

diff --git a/WinAlfred.Plugin.System/Sys.cs b/WinAlfred.Plugin.System/Sys.cs
index 9919ca6c3..a2badb7ce 100644
--- a/WinAlfred.Plugin.System/Sys.cs
+++ b/WinAlfred.Plugin.System/Sys.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Diagnostics;
 using System.Linq;
 using System.Runtime.InteropServices;
 using System.Text;
@@ -44,7 +45,7 @@ namespace WinAlfred.Plugin.System
                 SubTitle = "Shutdown Computer",
                 Score = 100,
                 IcoPath = "Images\\exit.png",
-                Action = () => ExitWindowsEx(EWX_SHUTDOWN,0)
+                Action = () => Process.Start("shutdown","/s /t 0")
             });
             availableResults.Add(new Result
             {
diff --git a/WinAlfred/App.xaml b/WinAlfred/App.xaml
index 9acfef20e..1a43596b7 100644
--- a/WinAlfred/App.xaml
+++ b/WinAlfred/App.xaml
@@ -1,7 +1,7 @@
 <Application x:Class="WinAlfred.App"
              xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
              xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
-             StartupUri="MainWindow.xaml">
+            >
     <Application.Resources>
         <ResourceDictionary>
             <Style x:Key="defaultQueryBoxStyle" BasedOn="{x:Null}" TargetType="{x:Type TextBox}">
diff --git a/WinAlfred/App.xaml.cs b/WinAlfred/App.xaml.cs
index cc2295037..dfb3747a1 100644
--- a/WinAlfred/App.xaml.cs
+++ b/WinAlfred/App.xaml.cs
@@ -1,11 +1,28 @@
-using System.Windows;
-
-namespace WinAlfred
-{
-    /// <summary>
-    /// App.xaml 的交互逻辑
-    /// </summary>
-    public partial class App : Application
-    {
-    }
-}
+using System;
+using System.Threading;
+using System.Windows;
+
+namespace WinAlfred
+{
+    /// <summary>
+    /// App.xaml 的交互逻辑
+    /// </summary>
+    public partial class App : Application
+    {
+        protected override void OnStartup(StartupEventArgs e)
+        {
+            base.OnStartup(e);
+            bool startupFlag;
+            Mutex mutex = new Mutex(true, "WinAlfred", out startupFlag);
+            if (!startupFlag)
+            {
+                Environment.Exit(0);
+            }
+            else
+            {
+                MainWindow mainWindow = new MainWindow();
+                mainWindow.Show();
+            }
+        }
+    }
+}
diff --git a/WinAlfred/MainWindow.xaml.cs b/WinAlfred/MainWindow.xaml.cs
index d231c8c9e..5027b88ce 100644
--- a/WinAlfred/MainWindow.xaml.cs
+++ b/WinAlfred/MainWindow.xaml.cs
@@ -6,6 +6,8 @@ using System.Windows;
 using System.Windows.Controls;
 using System.Windows.Forms;
 using System.Windows.Input;
+using IWshRuntimeLibrary;
+using Microsoft.Win32;
 using WinAlfred.Commands;
 using WinAlfred.Helper;
 using WinAlfred.Plugin;
@@ -101,12 +103,31 @@ namespace WinAlfred
             }
         }
 
+        public void SetAutoStart(bool IsAtuoRun)
+        {
+            string LnkPath = Environment.GetFolderPath(Environment.SpecialFolder.Startup) + "//WinAlfred.lnk";
+            if (IsAtuoRun)
+            {
+                WshShell shell = new WshShell();
+                IWshShortcut shortcut = (IWshShortcut)shell.CreateShortcut(LnkPath);
+                shortcut.TargetPath = System.Reflection.Assembly.GetExecutingAssembly().Location;
+                shortcut.WorkingDirectory = Environment.CurrentDirectory;
+                shortcut.WindowStyle = 1; //normal window
+                shortcut.Description = "WinAlfred";
+                shortcut.Save();
+            }
+            else
+            {
+                System.IO.File.Delete(LnkPath);
+            }
+        }
+
         private void MainWindow_OnLoaded(object sender, RoutedEventArgs e)
         {
             Plugins.Init(this);
             cmdDispatcher = new Command(this);
             InitialTray();
-
+            SetAutoStart(true);
             //var engine = new Jurassic.ScriptEngine();
             //MessageBox.Show(engine.Evaluate("5 * 10 + 2").ToString());
         }
diff --git a/WinAlfred/WinAlfred.csproj b/WinAlfred/WinAlfred.csproj
index c695a139c..1dbb749da 100644
--- a/WinAlfred/WinAlfred.csproj
+++ b/WinAlfred/WinAlfred.csproj
@@ -189,6 +189,17 @@
   <ItemGroup>
     <Resource Include="app.ico" />
   </ItemGroup>
+  <ItemGroup>
+    <COMReference Include="IWshRuntimeLibrary">
+      <Guid>{F935DC20-1CF0-11D0-ADB9-00C04FD58A0B}</Guid>
+      <VersionMajor>1</VersionMajor>
+      <VersionMinor>0</VersionMinor>
+      <Lcid>0</Lcid>
+      <WrapperTool>tlbimp</WrapperTool>
+      <Isolated>False</Isolated>
+      <EmbedInteropTypes>True</EmbedInteropTypes>
+    </COMReference>
+  </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <Import Project="$(SolutionDir)\.nuget\NuGet.targets" Condition="Exists('$(SolutionDir)\.nuget\NuGet.targets')" />
   <PropertyGroup>