0d26570d8354c61a987a359bcfda7e6f8c10acef
[teamcity/dotNetPackagesSupport.git] / nuget-server / src / jetbrains / buildServer / nuget / server / toolRegistry / impl / ToolPacker.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.toolRegistry.impl;\r
18 \r
19 import jetbrains.buildServer.util.ArchiveUtil;\r
20 import jetbrains.buildServer.util.FileUtil;\r
21 import org.jetbrains.annotations.NotNull;\r
22 \r
23 import java.io.BufferedOutputStream;\r
24 import java.io.File;\r
25 import java.io.FileOutputStream;\r
26 import java.io.IOException;\r
27 import java.util.Collection;\r
28 import java.util.zip.ZipEntry;\r
29 import java.util.zip.ZipOutputStream;\r
30 \r
31 /**\r
32  * Created by Eugene Petrenko (eugene.petrenko@gmail.com)\r
33  * Date: 15.08.11 20:51\r
34  */\r
35 public class ToolPacker {\r
36 \r
37   public File packTool(@NotNull final String toolId,\r
38                        @NotNull final File rootDir) throws IOException {\r
39     final File tool = new File(FileUtil.getTempDirectory(), toolId + ".zip");\r
40     final ZipOutputStream fos = new ZipOutputStream(new BufferedOutputStream(new FileOutputStream(tool)));\r
41     try {\r
42       fos.putNextEntry(new ZipEntry("teamcity-plugin.xml"));\r
43       fos.write(TOOL_DESCRIPTOR.getBytes("utf-8"));\r
44       fos.closeEntry();\r
45       if (!ArchiveUtil.packZip(rootDir, fos)) {\r
46         throw new IOException("Failed to create agent tool plugin");\r
47       }\r
48     } finally {\r
49       FileUtil.close(fos);\r
50     }\r
51 \r
52     return tool;\r
53   }\r
54 \r
55   private static final String TOOL_DESCRIPTOR = "" +\r
56           "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" +\r
57           "<teamcity-agent-plugin xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\n" +\r
58           "                       xsi:noNamespaceSchemaLocation=\"urn:shemas-jetbrains-com:teamcity-agent-plugin-v1-xml\">\n" +\r
59           "  <tool-deployment />\n" +\r
60           "</teamcity-agent-plugin>";\r
61 }\r