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