download nuget package test
[teamcity/dotNetPackagesSupport.git] / nuget-server / src / jetbrains / buildServer / nuget / server / feed / reader / impl / NuGetFeedReaderImpl.java
1 /*\r
2  * Copyright 2000-2011 JetBrains s.r.o.\r
3  *\r
4  * Licensed under the Apache License, Version 2.0 (the "License");\r
5  * you may not use this file except in compliance with the License.\r
6  * You may obtain a copy of the License at\r
7  *\r
8  * http://www.apache.org/licenses/LICENSE-2.0\r
9  *\r
10  * Unless required by applicable law or agreed to in writing, software\r
11  * distributed under the License is distributed on an "AS IS" BASIS,\r
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
13  * See the License for the specific language governing permissions and\r
14  * limitations under the License.\r
15  */\r
16 \r
17 package jetbrains.buildServer.nuget.server.feed.reader.impl;\r
18 \r
19 import com.intellij.openapi.diagnostic.Logger;\r
20 import com.intellij.openapi.util.Pair;\r
21 import jetbrains.buildServer.nuget.server.feed.reader.*;\r
22 import jetbrains.buildServer.util.FileUtil;\r
23 import jetbrains.buildServer.util.XmlUtil;\r
24 import org.apache.http.HttpHeaders;\r
25 import org.apache.http.HttpResponse;\r
26 import org.apache.http.HttpStatus;\r
27 import org.apache.http.StatusLine;\r
28 import org.apache.http.client.methods.HttpGet;\r
29 import org.jdom.Element;\r
30 import org.jdom.JDOMException;\r
31 import org.jetbrains.annotations.NotNull;\r
32 \r
33 import java.io.*;\r
34 import java.util.Collection;\r
35 \r
36 /**\r
37  * Created by Eugene Petrenko (eugene.petrenko@gmail.com)\r
38  * Date: 11.08.11 15:42\r
39  */\r
40 public class NuGetFeedReaderImpl implements NuGetFeedReader {\r
41   private static final Logger LOG = Logger.getInstance(NuGetFeedReader.class.getName());\r
42   private final FeedClient myClient;\r
43   private final UrlResolver myResolver;\r
44   private final FeedGetMethodFactory myMethodFactory;\r
45   private final PackagesFeedParser myParser;\r
46 \r
47   public NuGetFeedReaderImpl(FeedClient client, UrlResolver resolver, FeedGetMethodFactory methodFactory, PackagesFeedParser parser) {\r
48     myClient = client;\r
49     myResolver = resolver;\r
50     myMethodFactory = methodFactory;\r
51     myParser = parser;\r
52   }\r
53 \r
54   @NotNull\r
55   public Collection<FeedPackage> queryPackageVersions(@NotNull String feedUrl,\r
56                                                       @NotNull String packageId) throws IOException {\r
57     LOG.debug("Connecting to NuGet feed url: " + feedUrl);\r
58     final Pair<String, HttpResponse> pair = myResolver.resolvePath(feedUrl);\r
59     feedUrl = pair.first;\r
60     LOG.debug("Resolved NuGet feed URL to " + feedUrl);\r
61     final Element element = toDocument(pair.second);\r
62     LOG.debug("Recieved xml: " + XmlUtil.to_s(element));\r
63 \r
64     final HttpGet get = myMethodFactory.createGet(feedUrl + "/Packages()",\r
65             new Param("$filter", "Id eq '" + packageId + "'")\r
66     );\r
67     get.setHeader(HttpHeaders.ACCEPT_ENCODING, "application/atom+xml");\r
68 \r
69     LOG.debug("Query for packages: " + get.getURI());\r
70 \r
71     final HttpResponse execute = myClient.getClient().execute(get);\r
72     return myParser.readPackages(toDocument(execute));\r
73   }\r
74 \r
75   public void downloadPackage(@NotNull FeedPackage pkg, @NotNull File file) throws IOException {\r
76     FileUtil.createParentDirs(file);\r
77     final String url = pkg.getDownloadUrl();\r
78 \r
79     final HttpGet get = myMethodFactory.createGet(url);\r
80     final HttpResponse resp = myClient.getClient().execute(get);\r
81     final StatusLine statusLine = resp.getStatusLine();\r
82     if (statusLine.getStatusCode() != HttpStatus.SC_OK) {\r
83       throw new IOException("Failed to download package " + pkg + ". Server returned " + statusLine);\r
84     }\r
85 \r
86     OutputStream os = null;\r
87     try {\r
88       os = new BufferedOutputStream(new FileOutputStream(file));\r
89       resp.getEntity().writeTo(os);\r
90     } catch (final IOException e) {\r
91       throw new IOException("Failed to download package " + pkg + ". " + e.getMessage()) {{ initCause(e); }};\r
92     } finally {\r
93       FileUtil.close(os);\r
94     }\r
95   }\r
96 \r
97   private Element toDocument(HttpResponse response) throws IOException {\r
98     try {\r
99       return FileUtil.parseDocument(response.getEntity().getContent(), false);\r
100     } catch (final JDOMException e) {\r
101       throw new IOException("Failed to parse xml document. " + e.getMessage()) {{\r
102         initCause(e);\r
103       }};\r
104     }\r
105   }\r
106 }\r