Revision created by MOE tool push_codebase.
[gae.git] / java / src / main / com / google / appengine / tools / admin / Application.java
bloba01254f0ecf0a12a31f1e63ef91ee57e2cf65d93
1 // Copyright 2008 Google Inc. All Rights Reserved.
3 package com.google.appengine.tools.admin;
5 import com.google.appengine.tools.admin.AppAdminFactory.ApplicationProcessingOptions;
6 import com.google.appengine.tools.info.SdkImplInfo;
7 import com.google.appengine.tools.info.SdkInfo;
8 import com.google.appengine.tools.info.Version;
9 import com.google.appengine.tools.plugins.AppYamlProcessor;
10 import com.google.appengine.tools.plugins.SDKPluginManager;
11 import com.google.appengine.tools.plugins.SDKRuntimePlugin;
12 import com.google.appengine.tools.util.ApiVersionFinder;
13 import com.google.appengine.tools.util.FileIterator;
14 import com.google.appengine.tools.util.JarSplitter;
15 import com.google.appengine.tools.util.JarTool;
16 import com.google.apphosting.utils.config.AppEngineConfigException;
17 import com.google.apphosting.utils.config.AppEngineWebXml;
18 import com.google.apphosting.utils.config.AppEngineWebXmlReader;
19 import com.google.apphosting.utils.config.BackendsXml;
20 import com.google.apphosting.utils.config.BackendsXmlReader;
21 import com.google.apphosting.utils.config.BackendsYamlReader;
22 import com.google.apphosting.utils.config.CronXml;
23 import com.google.apphosting.utils.config.CronXmlReader;
24 import com.google.apphosting.utils.config.CronYamlReader;
25 import com.google.apphosting.utils.config.DispatchXml;
26 import com.google.apphosting.utils.config.DispatchXmlReader;
27 import com.google.apphosting.utils.config.DispatchYamlReader;
28 import com.google.apphosting.utils.config.DosXml;
29 import com.google.apphosting.utils.config.DosXmlReader;
30 import com.google.apphosting.utils.config.DosYamlReader;
31 import com.google.apphosting.utils.config.GenerationDirectory;
32 import com.google.apphosting.utils.config.IndexesXml;
33 import com.google.apphosting.utils.config.IndexesXmlReader;
34 import com.google.apphosting.utils.config.QueueXml;
35 import com.google.apphosting.utils.config.QueueXmlReader;
36 import com.google.apphosting.utils.config.QueueYamlReader;
37 import com.google.apphosting.utils.config.WebXml;
38 import com.google.apphosting.utils.config.WebXmlReader;
39 import com.google.common.collect.ImmutableSet;
40 import com.google.common.io.Files;
42 import org.mortbay.io.Buffer;
43 import org.mortbay.jetty.MimeTypes;
44 import org.xml.sax.SAXException;
46 import java.io.File;
47 import java.io.FileInputStream;
48 import java.io.FileNotFoundException;
49 import java.io.FileOutputStream;
50 import java.io.FileWriter;
51 import java.io.IOException;
52 import java.io.PrintWriter;
53 import java.net.URL;
54 import java.net.URLConnection;
55 import java.util.ArrayList;
56 import java.util.Arrays;
57 import java.util.HashSet;
58 import java.util.List;
59 import java.util.Set;
60 import java.util.logging.Level;
61 import java.util.logging.Logger;
62 import java.util.regex.Pattern;
64 import javax.activation.FileTypeMap;
65 import javax.tools.JavaCompiler;
66 import javax.tools.JavaFileObject;
67 import javax.tools.StandardJavaFileManager;
68 import javax.tools.ToolProvider;
69 import javax.xml.XMLConstants;
70 import javax.xml.transform.stream.StreamSource;
71 import javax.xml.validation.SchemaFactory;
73 /**
74 * An App Engine application. You can {@link #readApplication read} an
75 * {@code Application} from a path, and
76 * {@link com.google.appengine.tools.admin.AppAdminFactory#createAppAdmin create}
77 * an {@link com.google.appengine.tools.admin.AppAdmin} to upload, create
78 * indexes, or otherwise manage it.
81 public class Application implements GenericApplication {
83 private static final int MAX_COMPILED_JSP_JAR_SIZE = 1024 * 1024 * 5;
84 private static final String COMPILED_JSP_JAR_NAME_PREFIX = "_ah_compiled_jsps";
86 private static final int MAX_CLASSES_JAR_SIZE = 1024 * 1024 * 5;
87 private static final String CLASSES_JAR_NAME_PREFIX = "_ah_webinf_classes";
89 private static final String JAVA_7_RUNTIME_ID = "java7";
90 private static final ImmutableSet<String> ALLOWED_RUNTIME_IDS = ImmutableSet.of(
91 JAVA_7_RUNTIME_ID);
93 private static Pattern JSP_REGEX = Pattern.compile(".*\\.jspx?");
95 /** If available, this is set to a program to make symlinks, e.g. /bin/ln */
96 private static File ln = Utility.findLink();
97 private static File sdkDocsDir;
98 public static synchronized File getSdkDocsDir(){
99 if (null == sdkDocsDir){
100 sdkDocsDir = new File(SdkInfo.getSdkRoot(), "docs");
102 return sdkDocsDir;
105 private static Version sdkVersion;
106 public static synchronized Version getSdkVersion() {
107 if (null == sdkVersion) {
108 sdkVersion = SdkInfo.getLocalVersion();
110 return sdkVersion;
113 private static final String STAGEDIR_PREFIX = "appcfg";
115 private static final Logger logger = Logger.getLogger(Application.class.getName());
117 private static final MimeTypes mimeTypes = new MimeTypes();
119 private AppEngineWebXml appEngineWebXml;
120 private WebXml webXml;
121 private CronXml cronXml;
122 private DispatchXml dispatchXml;
123 private DosXml dosXml;
124 private String pagespeedYaml;
125 private QueueXml queueXml;
126 private IndexesXml indexesXml;
127 private BackendsXml backendsXml;
128 private File baseDir;
129 private File stageDir;
130 private File externalResourceDir;
131 private String apiVersion;
132 private String appYaml;
134 private UpdateListener listener;
135 private PrintWriter detailsWriter;
136 private int updateProgress = 0;
137 private int progressAmount = 0;
139 protected Application(){
143 * Builds a normalized path for the given directory in which
144 * forward slashes are used as the file separator on all platforms.
145 * @param dir A directory
146 * @return The normalized path
148 private static String buildNormalizedPath(File dir) {
149 String normalizedPath = dir.getPath();
150 if (File.separatorChar == '\\') {
151 normalizedPath = normalizedPath.replace('\\', '/');
153 return normalizedPath;
156 private Application(String explodedPath, String appId, String module, String appVersion) {
157 this.baseDir = new File(explodedPath);
158 explodedPath = buildNormalizedPath(baseDir);
159 File webinf = new File(baseDir, "WEB-INF");
160 if (!webinf.getName().equals("WEB-INF")) {
161 throw new AppEngineConfigException("WEB-INF directory must be capitalized.");
164 String webinfPath = webinf.getPath();
165 AppEngineWebXmlReader aewebReader = new AppEngineWebXmlReader(explodedPath);
166 WebXmlReader webXmlReader = new WebXmlReader(explodedPath);
167 AppYamlProcessor.convert(webinf, aewebReader.getFilename(), webXmlReader.getFilename());
169 validateXml(aewebReader.getFilename(), new File(getSdkDocsDir(), "appengine-web.xsd"));
170 appEngineWebXml = aewebReader.readAppEngineWebXml();
171 appEngineWebXml.setSourcePrefix(explodedPath);
172 if (appId == null) {
173 if (appEngineWebXml.getAppId() == null) {
174 throw new AppEngineConfigException(
175 "No app id supplied and XML files have no <application> element");
177 } else {
178 appEngineWebXml.setAppId(appId);
180 if (module != null) {
181 appEngineWebXml.setModule(module);
183 if (appVersion != null) {
184 appEngineWebXml.setMajorVersionId(appVersion);
187 webXml = webXmlReader.readWebXml();
188 webXml.validate();
190 CronXmlReader cronReader = new CronXmlReader(explodedPath);
191 validateXml(cronReader.getFilename(), new File(getSdkDocsDir(), "cron.xsd"));
192 cronXml = cronReader.readCronXml();
193 if (cronXml == null) {
194 CronYamlReader cronYaml = new CronYamlReader(webinfPath);
195 cronXml = cronYaml.parse();
198 QueueXmlReader queueReader = new QueueXmlReader(explodedPath);
199 validateXml(queueReader.getFilename(), new File(getSdkDocsDir(), "queue.xsd"));
200 queueXml = queueReader.readQueueXml();
201 if (queueXml == null) {
202 QueueYamlReader queueYaml = new QueueYamlReader(webinfPath);
203 queueXml = queueYaml.parse();
206 DispatchXmlReader dispatchXmlReader = new DispatchXmlReader(explodedPath,
207 DispatchXmlReader.DEFAULT_RELATIVE_FILENAME);
208 validateXml(dispatchXmlReader.getFilename(), new File(getSdkDocsDir(), "dispatch.xsd"));
209 dispatchXml = dispatchXmlReader.readDispatchXml();
210 if (dispatchXml == null) {
211 DispatchYamlReader dispatchYamlReader = new DispatchYamlReader(webinfPath);
212 dispatchXml = dispatchYamlReader.parse();
215 DosXmlReader dosReader = new DosXmlReader(explodedPath);
216 validateXml(dosReader.getFilename(), new File(getSdkDocsDir(), "dos.xsd"));
217 dosXml = dosReader.readDosXml();
218 if (dosXml == null) {
219 DosYamlReader dosYaml = new DosYamlReader(webinfPath);
220 dosXml = dosYaml.parse();
223 if (getAppEngineWebXml().getPagespeed() != null) {
224 StringBuilder pagespeedYamlBuilder = new StringBuilder();
225 AppYamlTranslator.appendPagespeed(
226 getAppEngineWebXml().getPagespeed(), pagespeedYamlBuilder, 0);
227 pagespeedYaml = pagespeedYamlBuilder.toString();
230 IndexesXmlReader indexReader = new IndexesXmlReader(explodedPath);
231 File datastoreSchema = new File(getSdkDocsDir(), "datastore-indexes.xsd");
232 validateXml(indexReader.getFilename(), datastoreSchema);
233 indexesXml = indexReader.readIndexesXml();
235 BackendsXmlReader backendsReader = new BackendsXmlReader(explodedPath);
236 validateXml(backendsReader.getFilename(), new File(getSdkDocsDir(), "backends.xsd"));
237 backendsXml = backendsReader.readBackendsXml();
238 if (backendsXml == null) {
239 BackendsYamlReader backendsYaml = new BackendsYamlReader(webinfPath);
240 backendsXml = backendsYaml.parse();
245 * Reads the App Engine application from {@code path}. The path may either
246 * be a WAR file or the root of an exploded WAR directory.
248 * @param path a not {@code null} path.
250 * @throws IOException if an error occurs while trying to read the
251 * {@code Application}.
252 * @throws com.google.apphosting.utils.config.AppEngineConfigException if the
253 * {@code Application's} appengine-web.xml file is malformed.
255 public static Application readApplication(String path)
256 throws IOException {
257 return new Application(path, null, null, null);
261 * Sets the external resource directory. Call this method before invoking
262 * {@link #createStagingDirectory(ApplicationProcessingOptions, ResourceLimits)}.
263 * <p>
264 * The external resource directory is a directory outside of the war directory where additional
265 * files live. These files will be copied into the staging directory during an upload, after the
266 * war directory is copied there. Consequently if there are any name collisions the files in the
267 * external resource directory will win.
269 * @param path a not {@code null} path to an existing directory.
271 * @throws IllegalArgumentException If {@code path} does not refer to an existing
272 * directory.
274 public void setExternalResourceDir(String path) {
275 if (path == null) {
276 throw new NullPointerException("path is null");
278 if (stageDir != null) {
279 throw new IllegalStateException(
280 "This method must be invoked prior to createStagingDirectory()");
282 File dir = new File(path);
283 if (!dir.exists()) {
284 throw new IllegalArgumentException("path does not exist: " + path);
286 if (!dir.isDirectory()) {
287 throw new IllegalArgumentException(path + " is not a directory.");
289 this.externalResourceDir = dir;
293 * Reads the App Engine application from {@code path}. The path may either
294 * be a WAR file or the root of an exploded WAR directory.
296 * @param path a not {@code null} path.
297 * @param appId if non-null, use this as an application id override.
298 * @param module if non-null, use this as a module id override.
299 * @param appVersion if non-null, use this as an application version override.
301 * @throws IOException if an error occurs while trying to read the
302 * {@code Application}.
303 * @throws com.google.apphosting.utils.config.AppEngineConfigException if the
304 * {@code Application's} appengine-web.xml file is malformed.
306 public static Application readApplication(String path,
307 String appId,
308 String module,
309 String appVersion) throws IOException {
310 return new Application(path, appId, module, appVersion);
314 * Returns the application identifier, from the AppEngineWebXml config
315 * @return application identifier
317 @Override
318 public String getAppId() {
319 return appEngineWebXml.getAppId();
323 * Returns the application version, from the AppEngineWebXml config
324 * @return application version
326 @Override
327 public String getVersion() {
328 return appEngineWebXml.getMajorVersionId();
331 @Override
332 public String getSourceLanguage() {
333 return appEngineWebXml.getSourceLanguage();
336 @Override
337 public String getModule() {
338 return appEngineWebXml.getModule();
341 @Override
342 public String getInstanceClass() {
343 return appEngineWebXml.getInstanceClass();
346 @Override
347 public boolean isPrecompilationEnabled() {
348 return appEngineWebXml.getPrecompilationEnabled();
351 @Override
352 public List<ErrorHandler> getErrorHandlers() {
353 class ErrorHandlerImpl implements ErrorHandler {
354 private final AppEngineWebXml.ErrorHandler errorHandler;
355 public ErrorHandlerImpl(AppEngineWebXml.ErrorHandler errorHandler) {
356 this.errorHandler = errorHandler;
358 @Override
359 public String getFile() {
360 return "__static__/" + errorHandler.getFile();
362 @Override
363 public String getErrorCode() {
364 return errorHandler.getErrorCode();
366 @Override
367 public String getMimeType() {
368 return getMimeTypeIfStatic(getFile());
371 List<ErrorHandler> errorHandlers = new ArrayList<ErrorHandler>();
372 for (AppEngineWebXml.ErrorHandler errorHandler: appEngineWebXml.getErrorHandlers()) {
373 errorHandlers.add(new ErrorHandlerImpl(errorHandler));
375 return errorHandlers;
378 @Override
379 public String getMimeTypeIfStatic(String path) {
380 if (!path.contains("__static__/")) {
381 return null;
383 String mimeType = webXml.getMimeTypeForPath(path);
384 if (mimeType != null) {
385 return mimeType;
387 return guessContentTypeFromName(path);
391 * @param fileName path of a file with extension
392 * @return the mimetype of the file (or application/octect-stream if not recognized)
394 public static String guessContentTypeFromName(String fileName) {
395 String defaultValue = "application/octet-stream";
396 try {
397 Buffer buffer = mimeTypes.getMimeByExtension(fileName);
398 if (buffer != null) {
399 return new String(buffer.asArray());
401 String lowerName = fileName.toLowerCase();
402 if (lowerName.endsWith(".json")) {
403 return "application/json";
405 FileTypeMap typeMap = FileTypeMap.getDefaultFileTypeMap();
406 String ret = typeMap.getContentType(fileName);
407 if (ret != null) {
408 return ret;
410 ret = URLConnection.guessContentTypeFromName(fileName);
411 if (ret != null) {
412 return ret;
414 return defaultValue;
415 } catch (Throwable t) {
416 logger.log(Level.WARNING, "Error identify mimetype for " + fileName, t);
417 return defaultValue;
421 * Returns the AppEngineWebXml describing the application.
423 * @return a not {@code null} deployment descriptor
425 public AppEngineWebXml getAppEngineWebXml() {
426 return appEngineWebXml;
430 * Returns the CronXml describing the applications' cron jobs.
431 * @return a cron descriptor, possibly empty or {@code null}
433 @Override
434 public CronXml getCronXml() {
435 return cronXml;
439 * Returns the QueueXml describing the applications' task queues.
440 * @return a queue descriptor, possibly empty or {@code null}
442 @Override
443 public QueueXml getQueueXml() {
444 return queueXml;
447 @Override
448 public DispatchXml getDispatchXml() {
449 return dispatchXml;
453 * Returns the DosXml describing the applications' DoS entries.
454 * @return a dos descriptor, possibly empty or {@code null}
456 @Override
457 public DosXml getDosXml() {
458 return dosXml;
462 * Returns the pagespeed.yaml describing the applications' PageSpeed configuration.
463 * @return a pagespeed.yaml string, possibly empty or {@code null}
465 @Override
466 public String getPagespeedYaml() {
467 return pagespeedYaml;
471 * Returns the IndexesXml describing the applications' indexes.
472 * @return a index descriptor, possibly empty or {@code null}
474 @Override
475 public IndexesXml getIndexesXml() {
476 return indexesXml;
480 * Returns the WebXml describing the applications' servlets and generic web
481 * application information.
483 * @return a WebXml descriptor, possibly empty but not {@code null}
485 public WebXml getWebXml() {
486 return webXml;
489 @Override
490 public BackendsXml getBackendsXml() {
491 return backendsXml;
495 * Returns the desired API version for the current application, or
496 * {@code "none"} if no API version was used.
498 * @throws IllegalStateException if createStagingDirectory has not been called.
500 @Override
501 public String getApiVersion() {
502 if (apiVersion == null) {
503 throw new IllegalStateException("Must call createStagingDirectory first.");
505 return apiVersion;
509 * Returns a path to an exploded WAR directory for the application.
510 * This may be a temporary directory.
512 * @return a not {@code null} path pointing to a directory
514 @Override
515 public String getPath() {
516 return baseDir.getAbsolutePath();
520 * Returns the staging directory, or {@code null} if none has been created.
522 @Override
523 public File getStagingDir() {
524 return stageDir;
527 @Override
528 public void resetProgress() {
529 updateProgress = 0;
530 progressAmount = 0;
534 * Creates a new staging directory, if needed, or returns the existing one
535 * if already created.
537 * @param opts User-specified options for processing the application.
538 * @return staging directory
539 * @throws IOException
541 @Override
542 public File createStagingDirectory(ApplicationProcessingOptions opts,
543 ResourceLimits resourceLimits) throws IOException {
544 if (stageDir != null) {
545 return stageDir;
548 int i = 0;
549 while (stageDir == null && i++ < 3) {
550 try {
551 stageDir = File.createTempFile(STAGEDIR_PREFIX, null);
552 } catch (IOException ex) {
553 continue;
555 stageDir.delete();
556 if (!stageDir.mkdir()) {
557 stageDir = null;
560 if (i == 3) {
561 throw new IOException("Couldn't create a temporary directory in 3 tries.");
563 statusUpdate("Created staging directory at: '" + stageDir.getPath() + "'", 20);
565 File staticDir = new File(stageDir, "__static__");
566 staticDir.mkdir();
567 copyOrLink(baseDir, stageDir, staticDir, false, opts);
568 if (externalResourceDir != null) {
569 String previousPrefix = appEngineWebXml.getSourcePrefix();
570 String newPrefix = buildNormalizedPath(externalResourceDir);
571 try {
572 appEngineWebXml.setSourcePrefix(newPrefix);
573 copyOrLink(externalResourceDir, stageDir, staticDir, false, opts);
574 } finally {
575 appEngineWebXml.setSourcePrefix(previousPrefix);
579 apiVersion = findApiVersion(stageDir, true);
581 String runtime = getRuntime(opts);
583 if (opts.isCompileJspsSet()) {
584 compileJsps(stageDir, opts);
587 appYaml = generateAppYaml(stageDir, runtime);
589 if (GenerationDirectory.getGenerationDirectory(stageDir).mkdirs()) {
590 writePreparedYamlFile("app", appYaml);
591 writePreparedYamlFile("backends", backendsXml == null ? null : backendsXml.toYaml());
592 writePreparedYamlFile("index", indexesXml.size() == 0 ? null : indexesXml.toYaml());
593 writePreparedYamlFile("cron", cronXml == null ? null : cronXml.toYaml());
594 writePreparedYamlFile("queue", queueXml == null ? null : queueXml.toYaml());
595 writePreparedYamlFile("dos", dosXml == null ? null : dosXml.toYaml());
598 int maxJarSize = (int) resourceLimits.maxFileSize();
600 if (opts.isSplitJarsSet()) {
601 splitJars(new File(new File(stageDir, "WEB-INF"), "lib"),
602 maxJarSize, opts.getJarSplittingExcludes());
605 if (getSourceLanguage() != null) {
606 SDKRuntimePlugin runtimePlugin = SDKPluginManager.findRuntimePlugin(getSourceLanguage());
607 if (runtimePlugin != null) {
608 runtimePlugin.processStagingDirectory(stageDir);
612 return stageDir;
616 * Write yaml file to generation subdirectory within stage directory.
618 private void writePreparedYamlFile(String yamlName, String yamlString) throws IOException {
619 File f = new File(GenerationDirectory.getGenerationDirectory(stageDir), yamlName + ".yaml");
620 if (yamlString != null && f.createNewFile()) {
621 FileWriter fw = new FileWriter(f);
622 fw.write(yamlString);
623 fw.close();
627 private static String findApiVersion(File baseDir, boolean deleteApiJars) {
628 ApiVersionFinder finder = new ApiVersionFinder();
630 String foundApiVersion = null;
631 File webInf = new File(baseDir, "WEB-INF");
632 File libDir = new File(webInf, "lib");
633 for (File file : new FileIterator(libDir)) {
634 if (file.getPath().endsWith(".jar")) {
635 try {
636 String apiVersion = finder.findApiVersion(file);
637 if (apiVersion != null) {
638 if (foundApiVersion == null) {
639 foundApiVersion = apiVersion;
640 } else if (!foundApiVersion.equals(apiVersion)) {
641 logger.warning("Warning: found duplicate API version: " + foundApiVersion +
642 ", using " + apiVersion);
644 if (deleteApiJars) {
645 if (!file.delete()) {
646 logger.log(Level.SEVERE, "Could not delete API jar: " + file);
650 } catch (IOException ex) {
651 logger.log(Level.WARNING, "Could not identify API version in " + file, ex);
656 if (foundApiVersion == null) {
657 foundApiVersion = "none";
659 return foundApiVersion;
663 * Returns the runtime id to use in the generated app.yaml.
665 * This method returns {@code "java7"}, unless an explicit runtime id was specified
666 * using the {@code -r} option.
668 * Before accepting an explicit runtime id, this method validates it against the list of
669 * supported Java runtimes (currently only {@code "java7"}), unless validation was turned
670 * off using the {@code --allowAnyRuntimes} option.
672 private String getRuntime(ApplicationProcessingOptions opts) {
673 String runtime = opts.getRuntime();
674 if (runtime != null) {
675 if (!opts.isAllowAnyRuntime() && !ALLOWED_RUNTIME_IDS.contains(runtime)) {
676 throw new AppEngineConfigException("Invalid runtime id: " + runtime + ". Valid " +
677 "runtime id: java7.");
679 return runtime;
681 return JAVA_7_RUNTIME_ID;
685 * Validates a given XML document against a given schema.
687 * @param xmlFilename filename with XML document
688 * @param schema XSD schema to validate with
690 * @throws AppEngineConfigException for malformed XML, or IO errors
692 private static void validateXml(String xmlFilename, File schema) {
693 File xml = new File(xmlFilename);
694 if (!xml.exists()) {
695 return;
697 try {
698 SchemaFactory factory = SchemaFactory.newInstance(XMLConstants.W3C_XML_SCHEMA_NS_URI);
699 try {
700 factory.newSchema(schema).newValidator().validate(
701 new StreamSource(new FileInputStream(xml)));
702 } catch (SAXException ex) {
703 throw new AppEngineConfigException("XML error validating " +
704 xml.getPath() + " against " + schema.getPath(), ex);
706 } catch (IOException ex) {
707 throw new AppEngineConfigException("IO error validating " +
708 xml.getPath() + " against " + schema.getPath(), ex);
712 private static final String JSPC_MAIN = "com.google.appengine.tools.development.LocalJspC";
714 private void compileJsps(File stage, ApplicationProcessingOptions opts)
715 throws IOException {
716 statusUpdate("Scanning for jsp files.");
718 if (matchingFileExists(new File(stage.getPath()), JSP_REGEX)) {
719 statusUpdate("Compiling jsp files.");
721 File webInf = new File(stage, "WEB-INF");
723 for (File file : SdkImplInfo.getUserJspLibFiles()) {
724 copyOrLinkFile(file, new File(new File(webInf, "lib"), file.getName()));
726 for (File file : SdkImplInfo.getSharedJspLibFiles()) {
727 copyOrLinkFile(file, new File(new File(webInf, "lib"), file.getName()));
730 File classes = new File(webInf, "classes");
731 File generatedWebXml = new File(webInf, "generated_web.xml");
732 File tempDir = Files.createTempDir();
733 String classpath = getJspClasspath(classes, tempDir);
735 String javaCmd = opts.getJavaExecutable().getPath();
736 String[] args = new String[] {
737 javaCmd,
738 "-classpath", classpath,
739 JSPC_MAIN,
740 "-uriroot", stage.getPath(),
741 "-p", "org.apache.jsp",
742 "-l", "-v",
743 "-webinc", generatedWebXml.getPath(),
744 "-d", tempDir.getPath(),
745 "-javaEncoding", opts.getCompileEncoding(),
747 Process jspc = startProcess(args);
749 int status = 1;
750 try {
751 status = jspc.waitFor();
752 } catch (InterruptedException ex) { }
754 if (status != 0) {
755 detailsWriter.println("Error while executing: " + formatCommand(Arrays.asList(args)));
756 throw new JspCompilationException("Failed to compile jsp files.",
757 JspCompilationException.Source.JASPER);
760 compileJavaFiles(classpath, webInf, tempDir, opts);
762 webXml = new WebXmlReader(stage.getPath()).readWebXml();
767 private void compileJavaFiles(String classpath, File webInf, File jspClassDir,
768 ApplicationProcessingOptions opts) throws IOException {
770 JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
771 if (compiler == null) {
772 throw new RuntimeException(
773 "Cannot get the System Java Compiler. Please use a JDK, not a JRE.");
775 StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
777 ArrayList<File> files = new ArrayList<File>();
778 for (File f : new FileIterator(jspClassDir)) {
779 if (f.getPath().toLowerCase().endsWith(".java")) {
780 files.add(f);
783 if (files.isEmpty()) {
784 return;
786 List<String> optionList = new ArrayList<String>();
787 optionList.addAll(Arrays.asList("-classpath", classpath.toString()));
788 optionList.addAll(Arrays.asList("-d", jspClassDir.getPath()));
789 optionList.addAll(Arrays.asList("-encoding", opts.getCompileEncoding()));
791 Iterable<? extends JavaFileObject> compilationUnits =
792 fileManager.getJavaFileObjectsFromFiles(files);
793 boolean success = compiler.getTask(
794 null, fileManager, null, optionList, null, compilationUnits).call();
795 fileManager.close();
797 if (!success) {
798 throw new JspCompilationException("Failed to compile the generated JSP java files.",
799 JspCompilationException.Source.JSPC);
801 if (opts.isJarJSPsSet()) {
802 zipJasperGeneratedFiles(webInf, jspClassDir);
803 } else {
804 copyOrLinkDirectories(jspClassDir, new File(webInf, "classes"));
806 if (opts.isDeleteJSPs()) {
807 for (File f : new FileIterator(webInf.getParentFile())) {
808 if (f.getPath().toLowerCase().endsWith(".jsp")) {
809 f.delete();
813 if (opts.isJarClassesSet()) {
814 zipWebInfClassesFiles(webInf);
819 private void zipJasperGeneratedFiles(File webInfDir, File jspClassDir) throws IOException {
820 Set<String> fileTypesToExclude = ImmutableSet.of(".java");
821 File libDir = new File(webInfDir, "lib");
822 JarTool jarTool = new JarTool(
823 COMPILED_JSP_JAR_NAME_PREFIX, jspClassDir, libDir, MAX_COMPILED_JSP_JAR_SIZE,
824 fileTypesToExclude);
825 jarTool.run();
826 recursiveDelete(jspClassDir);
829 private void zipWebInfClassesFiles(File webInfDir) throws IOException {
830 File libDir = new File(webInfDir, "lib");
831 File classesDir = new File(webInfDir, "classes");
832 JarTool jarTool = new JarTool(
833 CLASSES_JAR_NAME_PREFIX, classesDir, libDir, MAX_CLASSES_JAR_SIZE,
834 null);
835 jarTool.run();
836 recursiveDelete(classesDir);
837 classesDir.mkdir();
840 private String getJspClasspath(File classDir, File genDir) {
841 StringBuilder classpath = new StringBuilder();
842 for (URL lib : SdkImplInfo.getImplLibs()) {
843 classpath.append(lib.getPath());
844 classpath.append(File.pathSeparatorChar);
846 for (File lib : SdkInfo.getSharedLibFiles()) {
847 classpath.append(lib.getPath());
848 classpath.append(File.pathSeparatorChar);
851 classpath.append(classDir.getPath());
852 classpath.append(File.pathSeparatorChar);
853 classpath.append(genDir.getPath());
854 classpath.append(File.pathSeparatorChar);
856 for (File f : new FileIterator(new File(classDir.getParentFile(), "lib"))) {
857 String filename = f.getPath().toLowerCase();
858 if (filename.endsWith(".jar") || filename.endsWith(".zip")) {
859 classpath.append(f.getPath());
860 classpath.append(File.pathSeparatorChar);
864 return classpath.toString();
867 private Process startProcess(String... args) throws IOException {
868 ProcessBuilder builder = new ProcessBuilder(args);
869 Process proc = builder.redirectErrorStream(true).start();
870 logger.fine(formatCommand(builder.command()));
871 new Thread(new OutputPump(proc.getInputStream(), detailsWriter)).start();
872 return proc;
875 private String formatCommand(Iterable<String> args) {
876 StringBuilder command = new StringBuilder();
877 for (String chunk : args) {
878 command.append(chunk);
879 command.append(" ");
881 return command.toString();
885 * Scans a given directory tree, testing whether any file matches a given
886 * pattern.
888 * @param dir the directory under which to scan
889 * @param regex the pattern to look for
890 * @returns Returns {@code true} on the first instance of such a file,
891 * {@code false} otherwise.
893 private static boolean matchingFileExists(File dir, Pattern regex) {
894 for (File file : dir.listFiles()) {
895 if (file.isDirectory()) {
896 if (matchingFileExists(file, regex)) {
897 return true;
899 } else {
900 if (regex.matcher(file.getName()).matches()) {
901 return true;
905 return false;
909 * Invokes the JarSplitter code on any jar files found in {@code dir}. Any
910 * jars larger than {@code max} will be split into fragments of at most that
911 * size.
912 * @param dir the directory to search, recursively
913 * @param max the maximum allowed size
914 * @param excludes a set of suffixes to exclude.
915 * @throws IOException on filesystem errors.
917 private static void splitJars(File dir, int max, Set<String> excludes) throws IOException {
918 String children[] = dir.list();
919 if (children == null) {
920 return;
922 for (String name : children) {
923 File subfile = new File(dir, name);
924 if (subfile.isDirectory()) {
925 splitJars(subfile, max, excludes);
926 } else if (name.endsWith(".jar")) {
927 if (subfile.length() > max) {
928 new JarSplitter(subfile, dir, max, false, 4, excludes).run();
929 subfile.delete();
935 private static final Pattern SKIP_FILES = Pattern.compile(
936 "^(.*/)?((#.*#)|(.*~)|(.*/RCS/.*)|)$");
939 * Copies files from the app to the upload staging directory, or makes
940 * symlinks instead if supported. Puts the files into the correct places for
941 * static vs. resource files, recursively.
943 * @param sourceDir application war dir, or on recursion a subdirectory of it
944 * @param resDir staging resource dir, or on recursion a subdirectory matching
945 * the subdirectory in {@code sourceDir}
946 * @param staticDir staging {@code __static__} dir, or an appropriate recursive
947 * subdirectory
948 * @param forceResource if all files should be considered resource files
949 * @param opts processing options, used primarily for handling of *.jsp files
950 * @throws FileNotFoundException
951 * @throws IOException
953 private void copyOrLink(File sourceDir, File resDir, File staticDir, boolean forceResource,
954 ApplicationProcessingOptions opts)
955 throws FileNotFoundException, IOException {
957 for (String name : sourceDir.list()) {
958 File file = new File(sourceDir, name);
960 String path = file.getPath();
961 if (File.separatorChar == '\\') {
962 path = path.replace('\\', '/');
965 if (file.getName().startsWith(".") ||
966 file.equals(GenerationDirectory.getGenerationDirectory(baseDir))) {
967 continue;
970 if (file.isDirectory()) {
971 if (file.getName().equals("WEB-INF")) {
972 copyOrLink(file, new File(resDir, name), new File(staticDir, name), true, opts);
973 } else {
974 copyOrLink(file, new File(resDir, name), new File(staticDir, name), forceResource,
975 opts);
977 } else {
978 if (SKIP_FILES.matcher(path).matches()) {
979 continue;
982 if (forceResource || appEngineWebXml.includesResource(path) ||
983 (opts.isCompileJspsSet() && name.toLowerCase().endsWith(".jsp"))) {
984 copyOrLinkFile(file, new File(resDir, name));
986 if (!forceResource && appEngineWebXml.includesStatic(path)) {
987 copyOrLinkFile(file, new File(staticDir, name));
994 * Attempts to symlink a single file, or copies it if symlinking is either
995 * unsupported or fails.
997 * @param source source file
998 * @param dest destination file
999 * @throws FileNotFoundException
1000 * @throws IOException
1002 private void copyOrLinkFile(File source, File dest)
1003 throws FileNotFoundException, IOException {
1004 dest.getParentFile().mkdirs();
1005 if (ln != null && !source.getName().endsWith("web.xml")) {
1007 try {
1008 dest.delete();
1009 } catch (Exception e) {
1010 System.err.println("Warning: We tried to delete " + dest.getPath());
1011 System.err.println("in order to create a symlink from " + source.getPath());
1012 System.err.println("but the delete failed with message: " + e.getMessage());
1015 Process link = startProcess(ln.getAbsolutePath(), "-s",
1016 source.getAbsolutePath(),
1017 dest.getAbsolutePath());
1018 try {
1019 int stat = link.waitFor();
1020 if (stat == 0) {
1021 return;
1023 System.err.println(ln.getAbsolutePath() + " returned status " + stat
1024 + ", copying instead...");
1025 } catch (InterruptedException ex) {
1026 System.err.println(ln.getAbsolutePath() + " was interrupted, copying instead...");
1028 if (dest.delete()) {
1029 System.err.println("ln failed but symlink was created, removed: " + dest.getAbsolutePath());
1032 byte buffer[] = new byte[1024];
1033 int readlen;
1034 FileInputStream inStream = new FileInputStream(source);
1035 FileOutputStream outStream = new FileOutputStream(dest);
1036 try {
1037 readlen = inStream.read(buffer);
1038 while (readlen > 0) {
1039 outStream.write(buffer, 0, readlen);
1040 readlen = inStream.read(buffer);
1042 } finally {
1043 try {
1044 inStream.close();
1045 } catch (IOException ex) {
1047 try {
1048 outStream.close();
1049 } catch (IOException ex) {
1053 /** Copy (or link) one directory into another one.
1055 private void copyOrLinkDirectories(File sourceDir, File destination)
1056 throws IOException {
1058 for (String name : sourceDir.list()) {
1059 File file = new File(sourceDir, name);
1060 if (file.isDirectory()) {
1061 copyOrLinkDirectories(file, new File(destination, name));
1062 } else {
1063 copyOrLinkFile(file, new File(destination, name));
1068 /** deletes the staging directory, if one was created. */
1069 @Override
1070 public void cleanStagingDirectory() {
1071 if (stageDir != null) {
1072 recursiveDelete(stageDir);
1076 /** Recursive directory deletion. */
1077 public static void recursiveDelete(File dead) {
1078 String[] files = dead.list();
1079 if (files != null) {
1080 for (String name : files) {
1081 recursiveDelete(new File(dead, name));
1084 dead.delete();
1087 @Override
1088 public void setListener(UpdateListener l) {
1089 listener = l;
1092 @Override
1093 public void setDetailsWriter(PrintWriter detailsWriter) {
1094 this.detailsWriter = detailsWriter;
1097 @Override
1098 public void statusUpdate(String message, int amount) {
1099 updateProgress += progressAmount;
1100 if (updateProgress > 99) {
1101 updateProgress = 99;
1103 progressAmount = amount;
1104 if (listener != null) {
1105 listener.onProgress(new UpdateProgressEvent(
1106 Thread.currentThread(), message, updateProgress));
1110 @Override
1111 public void statusUpdate(String message) {
1112 int amount = progressAmount / 4;
1113 updateProgress += amount;
1114 if (updateProgress > 99) {
1115 updateProgress = 99;
1117 progressAmount -= amount;
1118 if (listener != null) {
1119 listener.onProgress(new UpdateProgressEvent(
1120 Thread.currentThread(), message, updateProgress));
1124 private String generateAppYaml(File stageDir, String runtime) {
1125 Set<String> staticFiles = new HashSet<String>();
1126 for (File f : new FileIterator(new File(stageDir, "__static__"))) {
1127 staticFiles.add(Utility.calculatePath(f, stageDir));
1130 AppYamlTranslator translator =
1131 new AppYamlTranslator(getAppEngineWebXml(), getWebXml(), getBackendsXml(),
1132 getApiVersion(), staticFiles, null, runtime, getSdkVersion());
1133 String yaml = translator.getYaml();
1134 logger.fine("Generated app.yaml file:\n" + yaml);
1135 return yaml;
1139 * Returns the app.yaml string.
1141 * @throws IllegalStateException if createStagingDirectory has not been called.
1143 @Override
1144 public String getAppYaml() {
1145 if (appYaml == null) {
1146 throw new IllegalStateException("Must call createStagingDirectory first.");
1148 return appYaml;