annotate smart_toolShed/SMART/Java/Installer/SmartInstallerTask.java @ 4:1fc014126d55

Uploaded

1 import java.util.*;
2 import java.awt.event.ActionEvent;
3 import java.awt.event.ActionListener;
4 import java.io.*;
5 import javax.swing.*;
6 import javax.swing.filechooser.*;
7 import javax.swing.border.*;
8 import javax.swing.SwingUtilities;
9 import java.net.*;
10 import java.util.Stack;
11 import java.util.zip.ZipEntry;
12 import java.util.zip.ZipInputStream;
13 
14 public class SmartInstallerTask extends SwingWorker<Boolean, String> {
15 
16   int BUFFER = 1024;
17 
18   int       architecture         = 0;
19   String    installDirectoryName = null;
20   JTextArea logArea              = null;
21   boolean[] selectedPrograms     = null;
22 
23   // program chooser buttons
24   String    programChoosers[] = {"R", "R Color Brewer Package", "R HMisc Package", "Python 2.6", "S-MART"};
25 
26   // Web addresses for the tools
27   String packageAddresses[][] = {
28     {"http://cran.cict.fr/bin/windows/base/R-2.11.0-win32.exe", "http://cran.cict.fr/bin/windows64/base/R-2.11.0-win64.exe"},
29     {"", ""},
30     {"", ""},
31     {"http://www.python.org/ftp/python/2.6.5/python-2.6.5.msi", "http://www.python.org/ftp/python/2.6.5/python-2.6.5.amd64.msi"},
32     {"http://urgi.versailles.inra.fr/content/download/1929/17848/file/s-mart-1.0.15.zip", "http://urgi.versailles.inra.fr/content/download/1929/17848/file/s-mart-1.0.15.zip"}
33   };
34 
35   // Packages to install
36   String rPackages[] = {"RColorBrewer", "Hmisc"};
37 
38   // Script lines
39   String scriptLines[][] = {
40     {"\"<INSTALLDIR>\\R-2.11.0-win32.exe\"", "\"<INSTALLDIR>\\R-2.11.0-win64.exe\""},
41     {"\"<RFILE>\" CMD BATCH \"<INSTALLDIR>\\installRColorBrewer.R\"", "\"<RFILE>\" CMD BATCH \"<INSTALLDIR>\\installRColorBrewer.R\""},
42     {"\"<RFILE>\" CMD BATCH \"<INSTALLDIR>\\installHmisc.R\"", "\"<RFILE>\" CMD BATCH \"<INSTALLDIR>\\installHmisc.R\""},
43     {"msiexec /i \"<INSTALLDIR>\\python-2.6.5.msi\"", "msiexec /i \"<INSTALLDIR>\\python-2.6.5.amd64.msi\""},
44     {"", ""}
45   };
46 
47   // Files to uncompress
48   String compressedFiles[][] = {
49     {"", ""},
50     {"", ""},
51     {"", ""},
52     {"", ""},
53     {"<INSTALLDIR>\\s-mart-1.0.15.zip", "<INSTALLDIR>\\s-mart-1.0.15.zip"}
54   };
55 
56 
57   public SmartInstallerTask(JTextArea ta, boolean[] b, String s, int a) {
58     logArea              = ta;
59     selectedPrograms     = b;
60     installDirectoryName = s;
61     architecture         = a;
62   }
63 
64 
65   @Override
66   public Boolean doInBackground() {
67     boolean installOk;
68     publish("Starting install\n");
69     writeFiles();
70     for (int i = 0; i < selectedPrograms.length; i++) {
71       if (selectedPrograms[i]) {
72         if (! install(i)) {
73           return Boolean.FALSE;
74         }
75       }
76     }
77     removeFiles();
78     setEnvironmentVariables();
79     publish("Ending install\n");
80     return Boolean.TRUE;
81   }
82 
83 
84   @Override
85   protected void process(List<String> chunks) {
86     for (String chunk: chunks) {
87       logArea.append(chunk);
88     }
89   }
90 
91 
92   private boolean launch(String command) {
93     return realLaunch(new ProcessBuilder(command), command);
94   }
95 
96   private boolean launch(String[] command) {
97     return realLaunch(new ProcessBuilder(command), Arrays.toString(command));
98   }
99 
00   private boolean realLaunch(ProcessBuilder pb, String command) {
01     BufferedReader outputReader;
02     pb                          = pb.redirectErrorStream(true);
03     Process        process      = null;
04     publish("      Starting command '" + command + "'\n");
05     try {
06       process = pb.start();
07       BufferedInputStream outputStream = new BufferedInputStream(process.getInputStream());
08       InputStream is        = process.getInputStream();
09       InputStreamReader isr = new InputStreamReader(is);
10       outputReader          = new BufferedReader(isr);
11     }
12     catch (Exception exception) {
13       publish("      !Process cannot be started (command is '" + command + "')!\n");
14       exception.printStackTrace();
15       return false;
16     }
17     if (outputReader == null) {
18       publish("      !Problem in the output of the command!\n");
19       return false;
20     }
21     else {
22       publish("      Output is:\n");
23       try {
24         publish("        ---\n");
25         String line;
26         while ((line = outputReader.readLine()) != null) {
27           publish("        " + line + "\r\n");
28         }
29         publish("        ---\n");
30       }
31       catch (IOException e) {
32         publish("      !Cannot get the output of the command!\n");
33         return false;
34       }
35     }
36     int exitValue = process.exitValue();
37     if (exitValue != 0) {
38       publish("      !Problem during the execution of the command '" + command + "'!\n");
39       return false;
40     }
41     publish("      Ending command '" + command + "'\n");
42     return true;
43   }
44 
45 
46   private File lookForFile(String fileName, String[] putativePlaces) {
47     publish("      Looking for file " + fileName + "\n");
48     for (String place: putativePlaces) {
49       File file = new File(place, fileName);
50       publish("      Look at " + file.getAbsolutePath() + "\n");
51       if (file.exists()) {
52         publish("      Found it in expected place " + file.getAbsolutePath() + "\n");
53         return file;
54       }
55     }
56     Stack<File> files = new Stack<File>();
57     files.push(new File("\\"));
58     while (! files.empty()) {
59       File file = files.pop();
60       for (File childFile: file.listFiles()) {
61         if (childFile.isDirectory()) {
62           files.push(childFile);
63         }
64         else {
65           if (fileName.compareToIgnoreCase(childFile.getName()) == 0) {
66             publish("      Found it in unexpected place " + childFile.getAbsolutePath() + "\n");
67             return childFile;
68           }
69         }
70       }
71     }
72     publish("      !Cannot file file '" + fileName + "'!\n");
73     return null;
74   }
75 
76 
77   private boolean writeFile(String fileName, String content) {
78     try {
79       FileWriter     fw = new FileWriter(fileName);
80       BufferedWriter bw = new BufferedWriter(fw);
81       bw.write(content);
82       bw.close();
83       fw.close();
84     }
85     catch (Exception e) {
86       publish("      !Cannot write file '" + fileName + "'!\n");
87       return false;
88     }
89     return true;
90   }
91 
92 
93   private boolean removeFile(String fileName) {
94     File file = new File(fileName);
95     if (file.exists()) {
96       if (! file.delete()) {
97         publish("      !Cannot delete file '" + file.getAbsolutePath() + "'!\n");
98         return false;
99       }
00     }
01     return true;
02   }
03 
04 
05   private boolean writeFiles() {
06     for (String rPackage: rPackages) {
07       String fileName = installDirectoryName + File.separator + "install" + rPackage + ".R";
08       String content  = "install.packages(\"" + rPackage + "\", repos = \"http://cran.cict.fr\", dependencies = TRUE)\n";
09       if (! writeFile(fileName, content)) {
10         publish("    !Cannot write file for R package '" + rPackage + "'!\n");
11         return false;
12       }
13     }
14     return true;
15   }
16 
17   private boolean removeFiles() {
18     for (String rPackage: rPackages) {
19       File file = new File(installDirectoryName + File.separator + "install" + rPackage + ".R");
20       if (! file.delete()) {
21         publish("!Cannot delete R install file for " + rPackage + "!\n");
22         return false;
23       }
24     }
25     File file = new File(installDirectoryName + File.separator + "createUser.sql");
26     if (! file.delete()) {
27       publish("!Cannot delete mySQL configuration file!\n");
28       return false;
29     }
30     return true;
31   }
32 
33   private boolean install(int element) {
34     publish("  Starting install of " + programChoosers[element] + "\n");
35     downloadPackage(element);
36     executeInstall(element);
37     uncompressPackage(element);
38     removePackage(element);
39     postProcess(element);
40     publish("  Ending install of " + programChoosers[element] + "\n");
41     return true;
42   }
43 
44 
45   private String getLocalName(String remoteName) {
46     String localName = installDirectoryName + File.separator + (new File(remoteName)).getName();
47     int    position  = localName.indexOf("?");
48     if (position >= 0) {
49       localName = localName.substring(0, position);
50     }
51     return localName;
52   }
53 
54 
55   private boolean downloadPackage(int element) {
56     String fileName  = packageAddresses[element][architecture];
57     if (! "".equals(fileName)) {
58       publish("    Starting download of " + programChoosers[element] + "\n");
59       try {
60         BufferedInputStream  bis = new BufferedInputStream(new URL(fileName).openStream());
61         FileOutputStream     fos = new FileOutputStream(getLocalName(fileName));
62         BufferedOutputStream bos = new BufferedOutputStream(fos, BUFFER);
63         byte[] data = new byte[BUFFER];
64         int x       = 0;
65         while((x = bis.read(data, 0, BUFFER)) >= 0) {
66           bos.write(data, 0, x);
67         }
68         bos.close();
69         fos.close();
70         bis.close();
71       }
72       catch (IOException e) {
73         publish("    !Cannot download file '" + fileName + "'!\n");
74         return false;
75       }
76       publish("    Ending download of " + programChoosers[element] + "\n");
77     }
78     return true;
79   }
80 
81 
82   private String replaceSubstring(String line) {
83     if (line.contains("<INSTALLDIR>")) {
84       String protectedDirectory = installDirectoryName.replaceAll("\\\\", "\\\\\\\\");
85       line = line.replaceAll("<INSTALLDIR>", protectedDirectory);
86     }
87     if (line.contains("<RFILE>")) {
88       String   userName = System.getenv().get("USERNAME");
89       String[] possibleRDirectories = {"C:\\Program Files\\R-2.11.0", "C:\\Documents and Settings\\" + userName + "\\Mes documents\\R\\R-2.11.0\\bin", "C:\\Documents and Settings\\" + userName + "\\My documents\\R\\R-2.11.0\\bin"};
90       String rDirectory = lookForFile("'.exe", possibleRDirectories).getAbsolutePath();
91       rDirectory = rDirectory.replaceAll("\\\\", "\\\\\\\\");
92       line = line.replaceAll("<RFILE>", rDirectory);
93     }
94     return line;
95   }
96 
97 
98   private boolean executeInstall(int element) {
99     String commands           = scriptLines[element][architecture];
00     if (! "".equals(commands)) {
01       for (String command: commands.split(";")) {
02         command = replaceSubstring(command);
03         publish("    Starting command '" + command + "'\n");
04         Process process = null;
05         try {
06           process = Runtime.getRuntime().exec(command);
07         }
08         catch (IOException e) {
09           publish("    !Cannot execute command '" + command + "'!\n");
10           return false;
11         }
12         try {
13           process.waitFor();
14         }
15         catch (InterruptedException e) {
16           publish("    !Cannot wait for the end of the command '" + command + "'!\n");
17           return false;
18         }
19         int exitValue = process.exitValue();
20         if (exitValue != 0) {
21           publish("    !Problem during the execution of the command '" + command + "'!\n");
22           return false;
23         }
24         publish("    Ending command '" + command + "'\n");
25       }
26     }
27     return true;
28   }
29 
30 
31   private boolean uncompressPackage(int element) {
32     String file = compressedFiles[element][architecture];
33     if (! "".equals(file)) {
34       file = replaceSubstring(file);
35       publish("    Starting uncompressing file '" + file + "'\n");
36       try {
37         FileInputStream     fis = new FileInputStream(file);
38         BufferedInputStream bis = new BufferedInputStream(fis);
39         ZipInputStream      zis = new ZipInputStream(bis);
40         ZipEntry            entry;
41         while ((entry = zis.getNextEntry()) != null) {
42           if (! entry.isDirectory()) {
43             File newFile = new File(installDirectoryName + File.separator + entry.getName());
44             // create parent directories
45             File upDirectory = newFile.getParentFile();
46             while (upDirectory != null){
47               if (! upDirectory.exists()) {
48                 upDirectory.mkdir();
49                 publish("      Creating directory '" + upDirectory.getAbsolutePath() + "'\n");
50               }
51               upDirectory = upDirectory.getParentFile();
52             }
53             // write the files to the disk
54             publish("      Extracting '" + entry.getName() + "' to '" + newFile.getAbsolutePath() + "'\n");
55             int  count;
56             byte data[] = new byte[BUFFER];
57             FileOutputStream     fos = new FileOutputStream(newFile);
58             BufferedOutputStream bos = new BufferedOutputStream(fos, BUFFER);
59             while ((count = zis.read(data, 0, BUFFER)) != -1){
60               bos.write(data, 0, count);
61             }
62             bos.flush();
63             bos.close();
64             fos.close();
65           }
66         }
67         zis.close();
68         bis.close();
69         fis.close();
70       }
71       catch(FileNotFoundException e) {
72         publish("    !Cannot find file '" + file + "'!\n");
73         return false;
74       }
75       catch(Exception e){
76         publish("    !Cannot uncompress file '" + file + "'!\n");
77         return false;
78       }
79       publish("    Ending uncompressing file '" + file + "'\n");
80     }
81     return true;
82   }
83 
84 
85   private boolean removePackage(int element) {
86     String packageName = packageAddresses[element][architecture];
87     if ("".equals(packageName)) {
88       return true;
89     }
90     String fileName = getLocalName(packageAddresses[element][architecture]);
91     return removeFile(fileName);
92   }
93 
94 
95   private boolean postProcess(int element) {
96     switch (element) {
97       case 4:
98         // Move S-MART files to parent directory
99         File installDirectory = new File(installDirectoryName + File.separator + "S-Mart");
00         for (File file: installDirectory.listFiles()) {
01           File destinationFile = new File(file.getParentFile().getParentFile(), file.getName());
02           if (! file.renameTo(destinationFile)) {
03             publish("     !Cannot move '" + file.getAbsolutePath() + "' to '" + destinationFile.getAbsolutePath() + "'!\n");
04           }
05         }
06         if (! installDirectory.delete()) {
07           publish("     !Cannot remove installation S-MART directory '" + installDirectory.getAbsolutePath() + "'!\n");
08         }
09     }
10     return true;
11   }
12 
13 
14   private boolean setEnvironmentVariables() {
15     String[] command = {"REG", "ADD", "HKCU\\Environment", "/v", "PYTHONPATH", "/t", "REG_SZ", "/d", "\"" + installDirectoryName + "\\Python\"", "/f"};
16     return launch(command);
17   }
18 }
19