archive-inst/Program.cs

   1 using System;\r
   2 using System.Text;\r
   3 using System.IO;\r
   4 using Microsoft.Build.BuildEngine;\r
   5 using NaGet.Packages;\r
   6 using NaGet.Packages.Install;\r
   7 \r
   8 namespace ArchiveInstall\r
   9 {\r
  10         class Program\r
  11         {\r
  12                 public const string InstalledFileListName = ".applistation.installedfiles.xml";\r
  13                 \r
  14                 public const string InstalledPackageFileName = ".applistation.package.xml";\r
  15                 \r
  16                 private static void extract(string arcFile, string extractDestDir)\r
  17                 {\r
  18                         StringBuilder output = new StringBuilder(1024);\r
  19                         int res = NaGet.InteropServices.CommonArchiverExtracter.ExtractArchive(arcFile, extractDestDir, output, IntPtr.Zero);\r
  20                         if (res != 0) {\r
  21                                 Environment.Exit(res);\r
  22                         }\r
  23                 }\r
  24                 \r
  25                 private static void install(string fakeTargetDir, string targetDir)\r
  26                 {\r
  27                         string installSrc = fakeTargetDir;\r
  28                         \r
  29                         if (Directory.GetFiles(installSrc).Length == 0 &&\r
  30                             Directory.GetDirectories(installSrc).Length == 1) {\r
  31                                 installSrc = Directory.GetDirectories(installSrc)[0];\r
  32                         }\r
  33                         \r
  34                         // ハッシュ比較\r
  35                         if (Directory.Exists(targetDir)) {\r
  36                                 InstalledFileList list = null;\r
  37                                 try {\r
  38                                         list = NaGet.Utils.GetDeserializedObject<InstalledFileList>(Path.Combine(targetDir, InstalledFileListName));\r
  39                                 } catch {\r
  40                                 }\r
  41                                 if (list != null) {\r
  42                                         // 変更されたファイル（設定ファイル）をキープする処理\r
  43                                         foreach (InstalledFile changedFile in list.Verify(targetDir)) {\r
  44                                                 string changedFilePath = Path.Combine(targetDir, changedFile.Path);\r
  45                                                 string toBeChangedFilePath = Path.Combine(installSrc, changedFile.Path);\r
  46                                                 \r
  47                                                 // 新規のファイルは退避させる\r
  48                                                 if (File.Exists(toBeChangedFilePath)) {\r
  49                                                         File.Move(toBeChangedFilePath, toBeChangedFilePath + ".newfile");\r
  50                                                 }\r
  51                                                 \r
  52                                                 // 変更済みファイルをinstallSrcの方に反映させる\r
  53                                                 if (File.Exists(changedFilePath)) {\r
  54                                                         if (! Directory.Exists(Path.GetDirectoryName(toBeChangedFilePath))) {\r
  55                                                                 // フォルダーがない場合は作る\r
  56                                                                 Directory.CreateDirectory(Path.GetDirectoryName(toBeChangedFilePath));\r
  57                                                         }\r
  58                                                         File.Copy(changedFilePath, toBeChangedFilePath);\r
  59                                                 }\r
  60                                         }\r
  61                                 }\r
  62                         } else {\r
  63                                 Directory.CreateDirectory(targetDir);\r
  64                         }\r
  65                         \r
  66                         // まずはフォルダーツリーを作成\r
  67                         foreach (string dir in Directory.GetDirectories(installSrc, "*", SearchOption.AllDirectories)) {\r
  68                                 string dirPath = NaGet.Utils.GetRelativePath(installSrc, dir);\r
  69                                 string targetDirPath = Path.Combine(targetDir, dirPath);\r
  70                                 if (! Directory.Exists(targetDirPath)) {\r
  71                                         Directory.CreateDirectory(targetDirPath);\r
  72                                 }\r
  73                         }\r
  74                         \r
  75                         // ファイルをインストール(高速化のため移動)\r
  76                         foreach (string file in Directory.GetFiles(installSrc, "*", SearchOption.AllDirectories)) {\r
  77                                 string filePath = NaGet.Utils.GetRelativePath(installSrc, file);\r
  78                                 string targetFilePath = Path.Combine(targetDir, filePath);\r
  79                                 if (File.Exists(targetFilePath)) {\r
  80                                         File.Delete(targetFilePath);\r
  81                                 }\r
  82                                 File.Move(file, targetFilePath);\r
  83                         }\r
  84                 }\r
  85                 \r
  86                 private static void postInstall(string targetDir, Package package)\r
  87                 {\r
  88                         // SYSTEM32へのコピーの実行\r
  89                         if (!string.IsNullOrEmpty(package.System32CopyFiles)) {\r
  90                                 if (! Directory.Exists(NaGet.Env.ArchiveSystem32)) {\r
  91                                         Directory.CreateDirectory(NaGet.Env.ArchiveSystem32);\r
  92                                 }\r
  93                                 \r
  94                                 string logfile = Path.Combine(NaGet.Env.ArchiveSystem32, ".applistation.install."+package.Name+".log");\r
  95                                 if (File.Exists(logfile)) {\r
  96                                         File.SetAttributes(logfile, FileAttributes.Normal);\r
  97                                 }\r
  98                                 using (FileStream fs = new FileStream(logfile, FileMode.Create))\r
  99                                 using (StreamWriter sw = new StreamWriter(fs)){\r
 100                                         foreach (string pathpat in package.System32CopyFiles.Split(';')) {\r
 101                                                 foreach (string path in NaGet.Utils.ExtendWildcardFile(targetDir, pathpat)) {\r
 102                                                         if ((File.GetAttributes(path) & FileAttributes.Directory) == 0) { // もしファイルならば\r
 103                                                                 string destPath = Path.Combine(NaGet.Env.ArchiveSystem32, Path.GetFileName(path));\r
 104                                                                 File.Copy(path, destPath, true);\r
 105                                                                 \r
 106                                                                 sw.WriteLine(Path.GetFileName(destPath));\r
 107                                                         }\r
 108                                                 }\r
 109                                         }\r
 110                                 }\r
 111                                 File.SetAttributes(logfile, FileAttributes.Hidden);\r
 112                         }\r
 113                         \r
 114                         // インストールスクリプトの実行\r
 115                         if (! string.IsNullOrEmpty(package.InstallScript) ) {\r
 116                                 Engine engine = MSBuild.Engine;\r
 117                                 Project proj = new Project(engine);\r
 118                                 try {\r
 119                                         proj.LoadXml(package.InstallScript);\r
 120                                 } catch (InvalidProjectFileException e) {\r
 121                                         throw new ApplicationException("InstallScript is invalid", e);\r
 122                                 }\r
 123                                 \r
 124                                 engine.BuildProject(proj, "Install");\r
 125                         }\r
 126                         \r
 127                         // 直下のファイルで*.exeがGUIプログラムならプログラムフォルダーに登録\r
 128                         foreach (string exeFile in Directory.GetFiles(targetDir, "*.exe")) {\r
 129                                 if (NaGet.InteropServices.PEFileInfoUtils.GetPEFileType(exeFile) == NaGet.InteropServices.PEFileType.WinGUI) {\r
 130                                         string progGrpPath = Path.Combine(NaGet.Env.ArchiveProgramGroup, package.Name);\r
 131                                         if (! Directory.Exists(progGrpPath)) Directory.CreateDirectory(progGrpPath);\r
 132                                         \r
 133                                         using (NaGet.InteropServices.ShellLink lnk = new NaGet.InteropServices.ShellLink() ) {\r
 134                                                 //string path = NaGet.Utils.GetRelativePath(progGrpPath, exeFile);// lnkファイルに相対パス指定不能\r
 135                                                 string path = exeFile;\r
 136                                                 \r
 137                                                 lnk.Path = path;\r
 138                                                 //lnk.SetIconLocation(path, 0);\r
 139                                                 \r
 140                                                 // .lnk ファイル名\r
 141                                                 string lnkFilePath = Path.Combine(progGrpPath, lnk.GetSuitableShellLinkNameFor() + ".lnk");\r
 142                                                 if (File.Exists(lnkFilePath)) { // ファイル名がかぶってしまったとき\r
 143                                                         lnkFilePath = Path.Combine(progGrpPath, Path.GetFileNameWithoutExtension(exeFile) + ".lnk");\r
 144                                                 }\r
 145                                                 \r
 146                                                 // 保存\r
 147                                                 lnk.ToPersistFile().Save(lnkFilePath, true);\r
 148                                         }\r
 149                                 }\r
 150                         }\r
 151                 }\r
 152                 \r
 153                 private static void storeInstalledFileList(string targetDir)\r
 154                 {\r
 155                         string installedFileListPath = Path.Combine(targetDir, InstalledFileListName);\r
 156                         if (File.Exists(installedFileListPath)) {\r
 157                                 File.SetAttributes(installedFileListPath, FileAttributes.Normal);\r
 158                         }\r
 159                         InstalledFileList installedFileList = InstalledFileList.CreateFromFiles(targetDir);\r
 160                         NaGet.Utils.PutSerializeObject(installedFileListPath, installedFileList);\r
 161                         \r
 162                         File.SetAttributes(installedFileListPath, FileAttributes.Hidden | FileAttributes.ReadOnly);\r
 163                 }\r
 164                 \r
 165                 private static void storePackageXml(Package package, string destDir)\r
 166                 {\r
 167                         if (package == null) {\r
 168                                 return;\r
 169                         }\r
 170                         \r
 171                         InstalledPackage pkg = InstalledPackage.PackageConverter(package);\r
 172                         UninstallInformation uninfo = pkg.UninstallInfo;\r
 173                         uninfo.InstallLocation = destDir;\r
 174                         uninfo.UninstallString = string.Format("archive-inst -x \"{0}\"", package.Name);\r
 175                         uninfo.EstimatedSize = (int) (NaGet.Utils.GetFileSize(destDir) >> 10);\r
 176                         uninfo.InstallDateString = DateTime.Now.ToString("yyyyMMdd");\r
 177                         pkg.UninstallInfo = uninfo;\r
 178                         \r
 179                         string packageXmlFilePath = Path.Combine(destDir, InstalledPackageFileName);\r
 180                         if (File.Exists(packageXmlFilePath)) {\r
 181                                 File.SetAttributes(packageXmlFilePath, FileAttributes.Normal);\r
 182                         }\r
 183                         NaGet.Utils.PutSerializeObject(packageXmlFilePath, pkg);\r
 184                         File.SetAttributes(packageXmlFilePath, FileAttributes.Hidden | FileAttributes.ReadOnly);\r
 185                 }\r
 186                 \r
 187                 private static void removePackage(InstalledPackage package, string targetDir)\r
 188                 {\r
 189                         // アンインストールスクリプトの実行\r
 190                         if (! string.IsNullOrEmpty(package.InstallScript)) {\r
 191                                 Engine engine = MSBuild.Engine;\r
 192                                 Project proj = new Project(engine);\r
 193                                 try {\r
 194                                         proj.LoadXml(package.InstallScript);\r
 195                                 } catch (InvalidProjectFileException e) {\r
 196                                         throw new ApplicationException("InstallScript is invalid", e);\r
 197                                 }\r
 198                                 \r
 199                                 engine.BuildProject(proj, "Uninstall");\r
 200                         }\r
 201                         \r
 202                         // GUIプログラムでプログラムフォルダーに登録しているのを解除\r
 203                         string progGrpPath = Path.Combine(NaGet.Env.ArchiveProgramGroup, package.Name);\r
 204                         if (Directory.Exists(progGrpPath)) {\r
 205                                 NaGet.Utils.SetAttributeRecursive(progGrpPath, FileAttributes.Normal);\r
 206                                 Directory.Delete(progGrpPath, true);\r
 207                         }\r
 208                         \r
 209                         // SYSTEM32からの削除の実行\r
 210                         if (! string.IsNullOrEmpty(package.System32CopyFiles) ) {\r
 211                                 string logfile = Path.Combine(NaGet.Env.ArchiveSystem32, ".applistation.install."+package.Name+".log");\r
 212                                 \r
 213                                 if (File.Exists(logfile)) {\r
 214                                         using (FileStream fs = new FileStream(logfile, FileMode.Open))\r
 215                                         using (StreamReader sr = new StreamReader(fs)){\r
 216                                                 string fileName = sr.ReadLine().Trim();\r
 217                                                 string filePath = Path.Combine(NaGet.Env.ArchiveSystem32, fileName);\r
 218                                                 \r
 219                                                 if (File.Exists(filePath)) {\r
 220                                                         File.SetAttributes(filePath, FileAttributes.Normal);\r
 221                                                         File.Delete(filePath);\r
 222                                                 }\r
 223                                         }\r
 224                                         File.SetAttributes(logfile, FileAttributes.Normal);\r
 225                                         File.Delete(logfile);\r
 226                                 }\r
 227                         }\r
 228                         \r
 229                         NaGet.Utils.SetAttributeRecursive(targetDir, FileAttributes.Normal);\r
 230                         Directory.Delete(targetDir, true);\r
 231                 }\r
 232                 \r
 233                 private static void parseMainArguments(string[] args, out string arcFile, out string targetDir, out Package package)\r
 234                 {\r
 235                         if (args.Length < 1) {\r
 236                                 throw new ArgumentException();\r
 237                         }\r
 238                         \r
 239                         switch (args[0].ToLower()) {\r
 240                                 case "-t":\r
 241                                         if (args.Length != 3) {\r
 242                                                 throw new ArgumentException();\r
 243                                         }\r
 244                                         \r
 245                                         arcFile = args[1];\r
 246                                         targetDir = args[2];\r
 247                                         package = null;\r
 248                                         break;\r
 249                                 case "-i":\r
 250                                         if (args.Length != 3) {\r
 251                                                 throw new ArgumentException();\r
 252                                         }\r
 253                                         \r
 254                                         arcFile = args[1];\r
 255                                         PackageList<Package> pkgList = NaGet.Utils.GetDeserializedObject<PackageList<Package>>(NaGet.Env.PackageListFile);\r
 256                                         package = pkgList.GetPackageForName(args[2]);\r
 257                                         targetDir = Path.Combine(NaGet.Env.ArchiveProgramFiles, package.Name);\r
 258                                         break;\r
 259                                 case "-x":\r
 260                                         if (args.Length != 2) {\r
 261                                                 throw new ArgumentException();\r
 262                                         }\r
 263                                         \r
 264                                         arcFile = null;\r
 265                                         targetDir = Path.Combine(NaGet.Env.ArchiveProgramFiles, args[1]);\r
 266                                         package = null;\r
 267                                         \r
 268                                         string filepath = Path.Combine(targetDir, InstalledPackageFileName);\r
 269                                         if (File.Exists(filepath)) {\r
 270                                                 package = NaGet.Utils.GetDeserializedObject<InstalledPackage>(filepath);\r
 271                                         } else {\r
 272                                                 Console.Error.WriteLine("Not found or already removed package : {0}", args[1]);\r
 273                                                 Environment.Exit(100);\r
 274                                         }\r
 275                                         break;\r
 276                                 default:\r
 277                                         arcFile = null;\r
 278                                         targetDir = null;\r
 279                                         package = null;\r
 280                                         Console.Error.WriteLine("Unreconized command \"{0}\".", args[0]);\r
 281                                         Environment.Exit(100);\r
 282                                         break;\r
 283                         }\r
 284                 }\r
 285                 \r
 286                 [STAThread]\r
 287                 public static void Main(string[] args)\r
 288                 {\r
 289                         // アーカイブSYSTEM32をパスに足す\r
 290                         NaGet.Utils.AddDirectoryToPath(NaGet.Env.ArchiveSystem32);\r
 291                         \r
 292                         string arcFile = null;\r
 293                         string targetDir = null;\r
 294                         Package package = null;\r
 295                         \r
 296                         // 引数パースおよびヘルプ表示\r
 297                         try {\r
 298                                 parseMainArguments(args, out arcFile, out targetDir, out package);\r
 299                         } catch (ArgumentException) {\r
 300                                 string executeFileName = System.AppDomain.CurrentDomain.FriendlyName;\r
 301                                 Console.Write("Usage:");\r
 302                                 Console.WriteLine("\t{0} -t archive.zip target_dir\tExtraction", executeFileName);\r
 303                                 Console.WriteLine("\t{0} -i archive.zip PackageName\tInstall", executeFileName);\r
 304                                 Console.WriteLine("\t{0} -x PackageName\t\tUninstall", executeFileName);\r
 305                                 Console.WriteLine();\r
 306                         }\r
 307 \r
 308                         // インストールおよび展開処理\r
 309                         if (arcFile != null) {\r
 310                                 string tempExtractDir = targetDir + "___temp___"; // HACK\r
 311                                 Directory.CreateDirectory(tempExtractDir);\r
 312                                 \r
 313                                 try {\r
 314                                         // STEP1. 書庫の展開\r
 315                                         if (package != null && package.Type == InstallerType.ITSELF) {\r
 316                                                 // 書庫でない場合展開せずにそのままコピーする\r
 317                                                 string destFile = Path.Combine(tempExtractDir, Path.GetFileName(arcFile));\r
 318                                                 File.Copy(arcFile, destFile);\r
 319                                         } else {\r
 320                                                 extract(arcFile, tempExtractDir);\r
 321                                         }\r
 322                                         \r
 323                                         // STEP2. インストール\r
 324                                         install(tempExtractDir, targetDir);\r
 325                                         \r
 326                                         if (package != null) {\r
 327                                                 // STEP3. カスタマイズ可能な後処理\r
 328                                                 postInstall(targetDir, package);\r
 329                                                 \r
 330                                                 // STEP4. パッケージ情報をインストール先(targetDir)に置く\r
 331                                                 storeInstalledFileList(targetDir);\r
 332                                                 storePackageXml(package, targetDir);\r
 333                                         }\r
 334                                 } catch (DllNotFoundException) {\r
 335                                         Console.Error.WriteLine("E: Does not exist archive dll for {0}", arcFile); // TODO\r
 336                                         Environment.Exit(10);\r
 337                                 } finally {\r
 338                                         Directory.Delete(tempExtractDir, true);\r
 339                                 }\r
 340                         } else if (package is InstalledPackage) {\r
 341                                 removePackage((InstalledPackage) package, targetDir);\r
 342                         }\r
 343                 }\r
 344         }\r
 345 }\r