na-get-lib/NaGet.SubCommands/NaGetInstall.cs

   1 using System;\r
   2 using NaGet.Packages.Install;\r
   3 using NaGet.Packages;\r
   4 using NaGet.Net;\r
   5 \r
   6 namespace NaGet.SubCommands\r
   7 {\r
   8         public class NaGetInstall : NaGetTaskSet\r
   9         {\r
  10                 private bool done = false;\r
  11                 \r
  12                 private int currentTaskSetIndex = -1;\r
  13                 \r
  14                 private PackageListsManager pkgListMan;\r
  15                 \r
  16                 /// <summary>\r
  17                 /// ダウンロードに使うダウンローダオブジェクト\r
  18                 /// </summary>\r
  19                 public Downloader Downloader {\r
  20                         get {\r
  21                                 if (downloader == null) {\r
  22                                         downloader = new Downloader();\r
  23                                 }\r
  24                                 return downloader;\r
  25                         }\r
  26                 }\r
  27                 \r
  28                 private Downloader downloader;\r
  29                 \r
  30                 private bool packageInstallerDownloaded = false;\r
  31                 \r
  32                 public override bool Cancelable {\r
  33                         get { return ! done; }\r
  34                 }\r
  35                 \r
  36                 /// <summary>\r
  37                 /// インストールするパッケージ\r
  38                 /// </summary>\r
  39                 public Installation[] Installations;\r
  40                 \r
  41                 /// <summary>\r
  42                 /// コンストラクタ\r
  43                 /// </summary>\r
  44                 /// <param name="pkgs">インストールするパッケージ</param>\r
  45                 public NaGetInstall(PackageListsManager pkgListMan, Package[] pkgs)\r
  46                         : this(pkgListMan, Installation.ConvertInstallations(pkgs))\r
  47                 {\r
  48                 }\r
  49                 \r
  50                 /// <summary>\r
  51                 /// コンストラクタ\r
  52                 /// </summary>\r
  53                 /// <param name="installations">インストール処理の配列</param>\r
  54                 public NaGetInstall(PackageListsManager pkgMan, Installation[] installations)\r
  55                 {\r
  56                         pkgListMan = pkgMan;\r
  57                         \r
  58                         Installations = installations;\r
  59                         initializeMainTaskSetNames();\r
  60                 }\r
  61                 \r
  62                 private void initializeMainTaskSetNames()\r
  63                 {\r
  64                         System.Collections.Generic.List<string> taskSetNames = new System.Collections.Generic.List<string>();\r
  65                         \r
  66                         for (int i =0; i < Installations.Length; i++) {\r
  67                                 taskSetNames.Add(string.Format("取得: {0}", Installations[i].ToString()));\r
  68                                 taskSetNames.Add(string.Format("ウイルススキャン: {0}", Installations[i].ToString()));\r
  69                         }\r
  70                         taskSetNames.Add("インストーラの検証");\r
  71                         for (int i =0; i < Installations.Length; i++) {\r
  72                                 taskSetNames.Add(string.Format("インストール: {0}", Installations[i].ToString()));\r
  73                         }\r
  74                         taskSetNames.Add(string.Format("リスト更新: {0}", NaGet.Env.ArchiveInstalledPackageListFile));\r
  75                         taskSetNames.Add(string.Format("リスト更新: {0}", NaGet.Env.SystemInstalledPackageListFile));\r
  76                         \r
  77                         TaskSetNames = taskSetNames.ToArray();\r
  78                 }\r
  79                                 \r
  80                 public override void Run()\r
  81                 {\r
  82                         currentTaskSetIndex = 0;\r
  83                         RaiseTaskSetEvent(NaGetTaskSetEventType.STARTED, "インストール処理開始");\r
  84                         \r
  85                         {\r
  86                                 // ハッシュ非適合なインストーラの表\r
  87                                 System.Collections.Generic.List<Installation> invalidInstallers = null;\r
  88                                 \r
  89                                 while (invalidInstallers == null || invalidInstallers.Count > 0) {\r
  90                                         currentTaskSetIndex = 0;\r
  91                                         packageInstallerDownloaded = false;\r
  92                                         \r
  93                                         runDownloadAndVirusCheckInstallers();\r
  94                                         if (done) return; // もしrunDownloadInstallers()内でエラー終了していたなら終了\r
  95                                         \r
  96                                         packageInstallerDownloaded = true;\r
  97                                         \r
  98                                         // ハッシュの壊れているインストーラを取得\r
  99                                         invalidInstallers = runCheckHashForInstaller();\r
 100                                         \r
 101                                         // ハッシュが壊れているときの対策\r
 102                                         if (invalidInstallers.Count > 0) {\r
 103                                                 string msg = string.Format("{0}個のパッケージでファイルが壊れている可能性があります\n強制的にインストールを続行しますか?",\r
 104                                                                            invalidInstallers.Count);\r
 105                                                 NaGetTaskQueryResult result = RaiseTaskSetQueryEvent(msg, NaGetTaskQueryResult.CONTINUE\r
 106                                                                                                      | NaGetTaskQueryResult.RETRY\r
 107                                                                                                      | NaGetTaskQueryResult.CANCEL);\r
 108                                                 \r
 109                                                 switch (result) {\r
 110                                                         case NaGetTaskQueryResult.CONTINUE:\r
 111                                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.WARNING, "ハッシュの非整合を無視してインストールを継続");\r
 112                                                                 invalidInstallers.Clear(); // ハッシュ非適合パッケージを強制的に抹消\r
 113                                                                 break;\r
 114                                                         case NaGetTaskQueryResult.RETRY:\r
 115                                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.INFO, "ダウンロード処理を再試行");\r
 116                                                                 \r
 117                                                                 foreach (Installation invalidInst in invalidInstallers) {\r
 118                                                                         invalidInst.RemoveDownloadedFile();\r
 119                                                                 }\r
 120                                                                 \r
 121                                                                 break;\r
 122                                                         //case NaGetTaskQueryResult.CANCEL:\r
 123                                                         default:\r
 124                                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.CANCELED, "パッケージのインストール処理がキャンセルされました");\r
 125                                                                 done = true;\r
 126                                                                 return;\r
 127                                                 }\r
 128                                         }\r
 129                                         currentTaskSetIndex ++;\r
 130                                 }\r
 131                         }\r
 132                         \r
 133                         foreach (Installation inst in Installations) {\r
 134                                 string installTaskMsg = TaskSetNames[currentTaskSetIndex];\r
 135                                 if (inst.Silent && (!inst.SupportsSilentOnly)) {\r
 136                                         installTaskMsg += " (サイレントインストール)";\r
 137                                 }\r
 138                                 \r
 139                                 RaiseTaskSetEvent(NaGetTaskSetEventType.STARTED_TASKSET, installTaskMsg);\r
 140                                 \r
 141                                 try {\r
 142                                         \r
 143                                         inst.ErrorDataReceived += this.ReceivedErrorData;\r
 144                                         inst.OutputDataReceived += this.ReceivedOutputData;\r
 145                                         int exitCode = inst.Install();\r
 146                                         if (exitCode != 0) {\r
 147                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.WARNING, "インストールが正常に終えていない可能性があります。インストーラの終了コード:"+exitCode);\r
 148                                         }\r
 149                                         \r
 150                                         pkgListMan.WriteInstallationLog(inst);\r
 151                                 } catch (Exception e) {\r
 152                                         RaiseTaskSetEvent(NaGetTaskSetEventType.ERROR, e.Message);\r
 153                                         done = true;\r
 154                                         return;\r
 155                                 }\r
 156                                 currentTaskSetIndex ++;\r
 157                                 \r
 158                                 RaiseTaskSetEvent(NaGetTaskSetEventType.COMPLETED_TASKSET, installTaskMsg);\r
 159                                 \r
 160                                 if (cancelCalled) {\r
 161                                         RaiseTaskSetEvent(NaGetTaskSetEventType.CANCELED, "パッケージのインストール処理がキャンセルされました");\r
 162                                         done = true;\r
 163                                         return;\r
 164                                 }\r
 165                         }\r
 166                         pkgListMan.SaveSystemInstalledLogList(); // ログのコミット\r
 167                         \r
 168                         runLocalUpdate();\r
 169                         \r
 170                         done = true;\r
 171                         \r
 172                         RaiseTaskSetEvent(NaGetTaskSetEventType.COMPLETED, "終了", 100);\r
 173                 }\r
 174                 \r
 175                 \r
 176                 /// <summary>\r
 177                 /// 処理内容のダウンロード・ウイルススキャン部分のサブルーチン\r
 178                 /// </summary>\r
 179                 private void runDownloadAndVirusCheckInstallers()\r
 180                 {\r
 181                         using (DownloadScanner scanner = new DownloadScanner()) {\r
 182                                 scanner.Init();\r
 183                                 foreach (Installation inst in Installations) {\r
 184                                         if (! inst.IsInstallablePackage()) {\r
 185                                                 string msg = string.Format("{0}はインストールすることができません", inst.ToString());\r
 186                                                 \r
 187                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.ERROR, msg);\r
 188                                                 done = true;\r
 189                                                 return;\r
 190                                         }\r
 191                                         \r
 192                                         RaiseTaskSetEvent(NaGetTaskSetEventType.STARTED_TASKSET, TaskSetNames[currentTaskSetIndex]);\r
 193                                         \r
 194                                         if (! inst.Downloaded) {\r
 195                                                 try {\r
 196                                                         inst.Download(Downloader);\r
 197                                                 } catch (NaGetTaskCanceledException) {\r
 198                                                         RaiseTaskSetEvent(NaGetTaskSetEventType.CANCELED, "インストーラのダウンロード処理がキャンセルされました");\r
 199                                                         done = true;\r
 200                                                         return;\r
 201                                                 } catch (System.Net.WebException e) {\r
 202                                                         RaiseTaskSetEvent(NaGetTaskSetEventType.WARNING, e.Message);\r
 203                                                         if (System.Net.NetworkInformation.NetworkInterface.GetIsNetworkAvailable()) {\r
 204                                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.ERROR, "ネットワークに接続されていません。");\r
 205                                                         } else {\r
 206                                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.ERROR, "ネットワークに接続できませんでした。ネットワークが切断されているか、ファイアウォールによって遮断された可能性があります。");\r
 207                                                         }\r
 208                                                         done = true;\r
 209                                                         return;\r
 210                                                 } catch (Exception e) {\r
 211                                                         RaiseTaskSetEvent(NaGetTaskSetEventType.ERROR, e.Message);\r
 212                                                         done = true;\r
 213                                                         return;\r
 214                                                 }\r
 215                                         }\r
 216                                         currentTaskSetIndex ++;\r
 217                                         \r
 218                                         if (inst.Downloaded) { // 正常終了\r
 219                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.COMPLETED_TASKSET, TaskSetNames[currentTaskSetIndex]);\r
 220                                         } else { // インストールが完了せずに終わった=失敗=エラー\r
 221                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.ERROR, string.Format("{0}のインストーラを正常にダウンロードできませんでした", inst.ToString()));\r
 222                                                 done = true;\r
 223                                                 return;\r
 224                                         }\r
 225                                         \r
 226                                         RaiseTaskSetEvent(NaGetTaskSetEventType.STARTED_TASKSET, TaskSetNames[currentTaskSetIndex]);\r
 227                                         if (scanner.HasScanner) {\r
 228                                                 try {\r
 229                                                         inst.ScanInstallerFile(scanner);\r
 230                                                 } catch (System.Runtime.InteropServices.COMException ex) {\r
 231                                                         RaiseTaskSetEvent(NaGetTaskSetEventType.WARNING,\r
 232                                                                           string.Format("{0} (E{1})", ex.Message, ex.ErrorCode));\r
 233                                                 } catch (System.IO.FileNotFoundException ex) {\r
 234                                                         if (ex.InnerException is System.Runtime.InteropServices.COMException) {\r
 235                                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.WARNING,\r
 236                                                                                   string.Format("{0} (E{1})", ex.InnerException.Message, ((System.Runtime.InteropServices.COMException) ex.InnerException).ErrorCode));\r
 237                                                         }\r
 238                                                         RaiseTaskSetEvent(NaGetTaskSetEventType.ERROR, "インストーラがウイルススキャナによって削除されました。");\r
 239                                                         done = true;\r
 240                                                         return;\r
 241                                                 }\r
 242                                         } else {\r
 243                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.INFO, string.Format("ダウンロードしたファイルはウイルススキャンされませんでした（ウイルススキャンソフトが検出できませんでした）"));\r
 244                                         }\r
 245                                         RaiseTaskSetEvent(NaGetTaskSetEventType.COMPLETED_TASKSET, TaskSetNames[currentTaskSetIndex]);\r
 246                                 }\r
 247                         }\r
 248                 }\r
 249                 \r
 250                 /// <summary>\r
 251                 /// ダウンロードしたパッケージが整合したか否かハッシュでチェック\r
 252                 /// </summary>\r
 253                 /// <returns>整合しなかったインストーラのリスト</returns>\r
 254                 private System.Collections.Generic.List<Installation> runCheckHashForInstaller()\r
 255                 {\r
 256                         System.Collections.Generic.List<Installation> invalidInstallers = new System.Collections.Generic.List<Installation>();\r
 257                         \r
 258                         int i = 0;\r
 259                         foreach (Installation inst in Installations) {\r
 260                                 float percent = (CurrentTaskSetIndex+((float)i / Installations.Length))*100f/TaskSetNames.Length;\r
 261                                 \r
 262                                 if (inst.GetRegisteredHashCount() > 0) {\r
 263                                         RaiseTaskSetEvent(NaGetTaskSetEventType.INFO, "検証: "+inst.ToString(), percent);\r
 264                                         \r
 265                                         if (inst.IsInstallablePackage() && inst.VerifyHashValues() == false) {\r
 266                                                 invalidInstallers.Add(inst);\r
 267                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.WARNING, "検証: "+inst.ToString() + " 非整合", percent);\r
 268                                         } else {\r
 269                                                 RaiseTaskSetEvent(NaGetTaskSetEventType.INFO, "検証: "+inst.ToString() + " OK", percent);\r
 270                                         }\r
 271                                 }\r
 272                                 i++;\r
 273                         }\r
 274                         \r
 275                         return invalidInstallers;\r
 276                 }\r
 277                 \r
 278                 private void runLocalUpdate()\r
 279                 {\r
 280                         // インストールトリストの更新\r
 281                         RaiseTaskSetEvent(NaGetTaskSetEventType.STARTED_TASKSET, TaskSetNames[currentTaskSetIndex]);\r
 282                         pkgListMan.DetectInstalledPkgs();\r
 283                         pkgListMan.SaveInstalledPackageList();\r
 284                         currentTaskSetIndex++;\r
 285                         RaiseTaskSetEvent(NaGetTaskSetEventType.COMPLETED_TASKSET, TaskSetNames[currentTaskSetIndex-1]);\r
 286                 \r
 287                         // システムにインストールされているリストの更新\r
 288                         RaiseTaskSetEvent(NaGetTaskSetEventType.STARTED_TASKSET, TaskSetNames[currentTaskSetIndex]);\r
 289                         pkgListMan.DetectSystemInstalledPkgs();\r
 290                         pkgListMan.SaveSystemInstalledPackageList();\r
 291                         currentTaskSetIndex++;\r
 292                         RaiseTaskSetEvent(NaGetTaskSetEventType.COMPLETED_TASKSET, TaskSetNames[currentTaskSetIndex-1]);\r
 293                 }\r
 294                 \r
 295                 public override bool Done {\r
 296                         get { return done; }\r
 297                 }\r
 298                 \r
 299                 public override int CurrentTaskSetIndex {\r
 300                         get { return currentTaskSetIndex; }\r
 301                 }\r
 302                 \r
 303                 private bool cancelCalled = false;\r
 304                 \r
 305                 public override bool Cancel()\r
 306                 {\r
 307                         cancelCalled = true;\r
 308                         if (! packageInstallerDownloaded) {\r
 309                                 return Downloader.Cancel();\r
 310                         } else return true;\r
 311                 }\r
 312         }\r
 313 }\r