1 files changed, 112 insertions, 0 deletions
diff --git a/src/main/java/com/google/devtools/build/lib/bazel/repository/HttpArchiveFunction.java b/src/main/java/com/google/devtools/build/lib/bazel/repository/HttpArchiveFunction.java
new file mode 100644
index 0000000000..1cd6db8d30
--- /dev/null
+++ b/src/main/java/com/google/devtools/build/lib/bazel/repository/HttpArchiveFunction.java
@@ -0,0 +1,112 @@
+// Copyright 2014 Google Inc. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//    http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.google.devtools.build.lib.bazel.repository;
+
+import com.google.devtools.build.lib.analysis.RuleDefinition;
+import com.google.devtools.build.lib.bazel.repository.DecompressorFactory.DecompressorException;
+import com.google.devtools.build.lib.bazel.rules.workspace.HttpArchiveRule;
+import com.google.devtools.build.lib.packages.AggregatingAttributeMapper;
+import com.google.devtools.build.lib.packages.PackageIdentifier.RepositoryName;
+import com.google.devtools.build.lib.packages.Rule;
+import com.google.devtools.build.lib.packages.Type;
+import com.google.devtools.build.lib.skyframe.FileValue;
+import com.google.devtools.build.lib.skyframe.RepositoryValue;
+import com.google.devtools.build.lib.syntax.EvalException;
+import com.google.devtools.build.lib.vfs.FileSystemUtils;
+import com.google.devtools.build.lib.vfs.Path;
+import com.google.devtools.build.skyframe.SkyFunctionException;
+import com.google.devtools.build.skyframe.SkyFunctionException.Transience;
+import com.google.devtools.build.skyframe.SkyFunctionName;
+import com.google.devtools.build.skyframe.SkyKey;
+import com.google.devtools.build.skyframe.SkyValue;
+
+import java.io.IOException;
+import java.net.MalformedURLException;
+import java.net.URL;
+
+/**
+ * Downloads a file over HTTP.
+ */
+public class HttpArchiveFunction extends RepositoryFunction {
+
+  @Override
+  public SkyValue compute(SkyKey skyKey, Environment env) throws SkyFunctionException {
+    RepositoryName repositoryName = (RepositoryName) skyKey.argument();
+    Rule rule = RepositoryFunction.getRule(repositoryName, HttpArchiveRule.NAME, env);
+    if (rule == null) {
+      return null;
+    }
+
+    return compute(env, rule);
+  }
+
+  protected FileValue createOutputDirectory(Environment env, String repositoryName)
+      throws RepositoryFunctionException {
+    // The output directory is always under .external-repository (to stay out of the way of
+    // artifacts from this repository) and uses the rule's name to avoid conflicts with other
+    // remote repository rules. For example, suppose you had the following WORKSPACE file:
+    //
+    // http_archive(name = "png", url = "http://example.com/downloads/png.tar.gz", sha256 = "...")
+    //
+    // This would download png.tar.gz to .external-repository/png/png.tar.gz.
+    Path outputDirectory = getExternalRepositoryDirectory().getRelative(repositoryName);
+    try {
+      FileSystemUtils.createDirectoryAndParents(outputDirectory);
+    } catch (IOException e) {
+      throw new RepositoryFunctionException(e, Transience.TRANSIENT);
+    }
+    return getRepositoryDirectory(outputDirectory, env);
+  }
+
+  protected SkyValue compute(Environment env, Rule rule)
+      throws RepositoryFunctionException {
+    FileValue directoryValue = createOutputDirectory(env, rule.getName());
+    if (directoryValue == null) {
+      return null;
+    }
+    Path outputDirectory = directoryValue.realRootedPath().asPath();
+    AggregatingAttributeMapper mapper = AggregatingAttributeMapper.of(rule);
+    URL url = null;
+    try {
+      url = new URL(mapper.get("url", Type.STRING));
+    } catch (MalformedURLException e) {
+      throw new RepositoryFunctionException(
+          new EvalException(rule.getLocation(), "Error parsing URL: " + e.getMessage()),
+              Transience.PERSISTENT);
+    }
+    String sha256 = mapper.get("sha256", Type.STRING);
+    HttpDownloader downloader = new HttpDownloader(url, sha256, outputDirectory);
+    try {
+      Path archiveFile = downloader.download();
+      outputDirectory = DecompressorFactory.create(rule, archiveFile).decompress();
+    } catch (IOException e) {
+      // Assumes all IO errors transient.
+      throw new RepositoryFunctionException(e, Transience.TRANSIENT);
+    } catch (DecompressorException e) {
+      throw new RepositoryFunctionException(new IOException(e.getMessage()), Transience.TRANSIENT);
+    }
+    return new RepositoryValue(outputDirectory, directoryValue);
+  }
+
+  @Override
+  public SkyFunctionName getSkyFunctionName() {
+    return SkyFunctionName.computed(HttpArchiveRule.NAME.toUpperCase());
+  }
+
+  @Override
+  public Class<? extends RuleDefinition> getRuleDefinition() {
+    return HttpArchiveRule.class;
+  }
+}