GUI/StartupManager.cs
author moel.mich
Thu, 06 May 2010 19:20:38 +0000
changeset 109 70d0c3102424
parent 101 f8589379e32c
child 120 41dc766e28b8
permissions -rw-r--r--
Added an Identifier class for IHardware, ISensor and IParameter Identifier properties.
moel@82
     1
/*
moel@82
     2
  
moel@82
     3
  Version: MPL 1.1/GPL 2.0/LGPL 2.1
moel@82
     4
moel@82
     5
  The contents of this file are subject to the Mozilla Public License Version
moel@82
     6
  1.1 (the "License"); you may not use this file except in compliance with
moel@82
     7
  the License. You may obtain a copy of the License at
moel@82
     8
 
moel@82
     9
  http://www.mozilla.org/MPL/
moel@82
    10
moel@82
    11
  Software distributed under the License is distributed on an "AS IS" basis,
moel@82
    12
  WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
moel@82
    13
  for the specific language governing rights and limitations under the License.
moel@82
    14
moel@82
    15
  The Original Code is the Open Hardware Monitor code.
moel@82
    16
moel@82
    17
  The Initial Developer of the Original Code is 
moel@82
    18
  Michael Möller <m.moeller@gmx.ch>.
moel@82
    19
  Portions created by the Initial Developer are Copyright (C) 2009-2010
moel@82
    20
  the Initial Developer. All Rights Reserved.
moel@82
    21
moel@82
    22
  Contributor(s):
moel@82
    23
moel@82
    24
  Alternatively, the contents of this file may be used under the terms of
moel@82
    25
  either the GNU General Public License Version 2 or later (the "GPL"), or
moel@82
    26
  the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
moel@82
    27
  in which case the provisions of the GPL or the LGPL are applicable instead
moel@82
    28
  of those above. If you wish to allow use of your version of this file only
moel@82
    29
  under the terms of either the GPL or the LGPL, and not to allow others to
moel@82
    30
  use your version of this file under the terms of the MPL, indicate your
moel@82
    31
  decision by deleting the provisions above and replace them with the notice
moel@82
    32
  and other provisions required by the GPL or the LGPL. If you do not delete
moel@82
    33
  the provisions above, a recipient may use your version of this file under
moel@82
    34
  the terms of any one of the MPL, the GPL or the LGPL.
moel@82
    35
 
moel@82
    36
*/
moel@82
    37
moel@82
    38
using System;
moel@82
    39
using System.Collections.Generic;
moel@82
    40
using System.IO;
moel@82
    41
using System.Windows.Forms;
moel@82
    42
using Microsoft.Win32;
moel@82
    43
using OpenHardwareMonitor.TaskScheduler;
moel@82
    44
moel@82
    45
namespace OpenHardwareMonitor.GUI {
moel@82
    46
  public class StartupManager {
moel@82
    47
moel@82
    48
    private TaskSchedulerClass scheduler;
moel@82
    49
    private bool startup;
moel@82
    50
moel@82
    51
    private const string REGISTRY_RUN =
moel@82
    52
      @"Software\Microsoft\Windows\CurrentVersion\Run";
moel@82
    53
moel@82
    54
    public StartupManager() {
moel@82
    55
      try {
moel@82
    56
        scheduler = new TaskSchedulerClass();
moel@82
    57
        scheduler.Connect(null, null, null, null);
moel@104
    58
      } catch {
moel@82
    59
        scheduler = null;
moel@82
    60
      }
moel@82
    61
moel@82
    62
      if (scheduler != null) {
moel@82
    63
        try {
moel@82
    64
          ITaskFolder folder = scheduler.GetFolder("\\Open Hardware Monitor");
moel@82
    65
          IRegisteredTask task = folder.GetTask("Startup");
moel@82
    66
          startup = task != null;
moel@101
    67
        } catch (IOException) {
moel@82
    68
          startup = false;
moel@82
    69
        }
moel@82
    70
      } else {
moel@82
    71
        RegistryKey key = Registry.CurrentUser.OpenSubKey(REGISTRY_RUN);
moel@82
    72
        startup = false;
moel@82
    73
        if (key != null) {
moel@82
    74
          string value = (string)key.GetValue("OpenHardwareMonitor");
moel@82
    75
          if (value != null)
moel@82
    76
            startup = value == Application.ExecutablePath;
moel@82
    77
        }
moel@82
    78
      }
moel@82
    79
    }
moel@82
    80
moel@82
    81
    private void CreateSchedulerTask() {
moel@82
    82
      ITaskDefinition definition = scheduler.NewTask(0);
moel@82
    83
      definition.RegistrationInfo.Description =
moel@82
    84
        "This task starts the Open Hardware Monitor on Windows startup.";
moel@82
    85
      definition.Principal.RunLevel =
moel@82
    86
        TASK_RUNLEVEL.TASK_RUNLEVEL_HIGHEST;
moel@82
    87
      definition.Settings.DisallowStartIfOnBatteries = false;
moel@82
    88
      definition.Settings.StopIfGoingOnBatteries = false;
moel@82
    89
      definition.Settings.ExecutionTimeLimit = "PT0S";
moel@82
    90
moel@82
    91
      ILogonTrigger trigger = (ILogonTrigger)definition.Triggers.Create(
moel@82
    92
        TASK_TRIGGER_TYPE2.TASK_TRIGGER_LOGON);
moel@82
    93
moel@82
    94
      IExecAction action = (IExecAction)definition.Actions.Create(
moel@82
    95
        TASK_ACTION_TYPE.TASK_ACTION_EXEC);
moel@82
    96
      action.Path = Application.ExecutablePath;
moel@82
    97
      action.WorkingDirectory =
moel@82
    98
        Path.GetDirectoryName(Application.ExecutablePath);
moel@82
    99
moel@82
   100
      ITaskFolder root = scheduler.GetFolder("\\");
moel@82
   101
      ITaskFolder folder;
moel@82
   102
      try {
moel@82
   103
        folder = root.GetFolder("Open Hardware Monitor");
moel@104
   104
      } catch (IOException) {
moel@82
   105
        folder = root.CreateFolder("Open Hardware Monitor", "");
moel@82
   106
      }
moel@82
   107
      folder.RegisterTaskDefinition("Startup", definition,
moel@82
   108
        (int)TASK_CREATION.TASK_CREATE_OR_UPDATE, null, null,
moel@82
   109
        TASK_LOGON_TYPE.TASK_LOGON_INTERACTIVE_TOKEN, "");
moel@82
   110
    }
moel@82
   111
moel@82
   112
    private void DeleteSchedulerTask() {
moel@82
   113
      ITaskFolder root = scheduler.GetFolder("\\");
moel@82
   114
      try {
moel@82
   115
        ITaskFolder folder = root.GetFolder("Open Hardware Monitor");
moel@82
   116
        folder.DeleteTask("Startup", 0);
moel@104
   117
      } catch (IOException) { }
moel@82
   118
      try {
moel@82
   119
        root.DeleteFolder("Open Hardware Monitor", 0);
moel@104
   120
      } catch (IOException) { }
moel@82
   121
    }
moel@82
   122
moel@82
   123
    private void CreateRegistryRun() {
moel@82
   124
      RegistryKey key = Registry.CurrentUser.CreateSubKey(REGISTRY_RUN);
moel@82
   125
      key.SetValue("OpenHardwareMonitor", Application.ExecutablePath);
moel@82
   126
    }
moel@82
   127
moel@82
   128
    private void DeleteRegistryRun() {
moel@82
   129
      RegistryKey key = Registry.CurrentUser.CreateSubKey(REGISTRY_RUN);
moel@82
   130
      key.DeleteValue("OpenHardwareMonitor");
moel@82
   131
    }
moel@82
   132
moel@82
   133
    public bool Startup {
moel@82
   134
      get {
moel@82
   135
        return startup;
moel@82
   136
      }
moel@82
   137
      set {
moel@82
   138
        if (startup != value) {
moel@82
   139
          startup = value;
moel@82
   140
          if (scheduler != null) {
moel@82
   141
            if (startup)
moel@82
   142
              CreateSchedulerTask();
moel@82
   143
            else
moel@82
   144
              DeleteSchedulerTask();
moel@82
   145
          } else {
moel@82
   146
            if (startup)
moel@82
   147
              CreateRegistryRun();
moel@82
   148
            else
moel@82
   149
              DeleteRegistryRun();
moel@82
   150
          }
moel@82
   151
        }
moel@82
   152
      }
moel@82
   153
    }
moel@82
   154
  }
moel@82
   155
moel@82
   156
 
moel@82
   157
moel@82
   158
 
moel@82
   159
moel@82
   160
}