wrap nuget call with cmd.exe calls, do not show api key
[teamcity/dotNetPackagesSupport.git] / nuget-agent / src / jetbrains / buildServer / nuget / agent / commands / impl / CommandFactoryImpl.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.agent.commands.impl;\r
18 \r
19 import jetbrains.buildServer.RunBuildException;\r
20 import jetbrains.buildServer.nuget.agent.commands.CommandFactory;\r
21 import jetbrains.buildServer.nuget.agent.parameters.*;\r
22 import jetbrains.buildServer.util.FileUtil;\r
23 import jetbrains.buildServer.util.StringUtil;\r
24 import org.jetbrains.annotations.NotNull;\r
25 \r
26 import java.io.File;\r
27 import java.util.*;\r
28 \r
29 /**\r
30  * Created by Eugene Petrenko (eugene.petrenko@gmail.com)\r
31  * Date: 21.07.11 16:10\r
32  */\r
33 public class CommandFactoryImpl implements CommandFactory {\r
34 \r
35   @NotNull\r
36   public <T> T createInstall(@NotNull PackagesInstallParameters params, @NotNull File packagesConfig, @NotNull File targetFolder, @NotNull Callback<T> factory) throws RunBuildException {\r
37     final List<String> argz = new ArrayList<String>();\r
38     argz.add("install");\r
39     argz.add(FileUtil.getCanonicalFile(packagesConfig).getPath()); //path to package\r
40     if (params.getExcludeVersion()) {\r
41       argz.add("-ExcludeVersion");\r
42     }\r
43     argz.add("-OutputDirectory");\r
44     argz.add(FileUtil.getCanonicalFile(targetFolder).getPath());\r
45 \r
46     final NuGetFetchParameters nuget = params.getNuGetParameters();\r
47     return executeNuGet(nuget, nuget.getNuGetPackageSources(), argz, packagesConfig.getParentFile(), factory);\r
48   }\r
49 \r
50   @NotNull\r
51   public <T> T createUpdate(@NotNull PackagesUpdateParameters params, @NotNull File packagesConfig, @NotNull File targetFolder, @NotNull Callback<T> factory) throws RunBuildException {\r
52     final List<String> argz = new ArrayList<String>();\r
53     argz.add("update");\r
54     argz.add(FileUtil.getCanonicalFile(packagesConfig).getPath()); //path to package\r
55     if (params.getUseSafeUpdate()) {\r
56       argz.add("-Safe");\r
57     }\r
58     argz.add("-Verbose");\r
59     argz.add("-RepositoryPath");\r
60     argz.add(FileUtil.getCanonicalFile(targetFolder).getPath());\r
61 \r
62     for (String id : params.getPackagesToUpdate()) {\r
63       argz.add("-Id");\r
64       argz.add(id);\r
65     }\r
66 \r
67     final NuGetFetchParameters nuget = params.getNuGetParameters();\r
68     return executeNuGet(nuget, nuget.getNuGetPackageSources(), argz, packagesConfig.getParentFile(), factory);\r
69   }\r
70 \r
71   public <T> T createPack(@NotNull final NuGetPackParameters params,\r
72                           @NotNull final File workdir,\r
73                           @NotNull final Callback<T> factory) throws RunBuildException {\r
74     final List<String> arguments = new ArrayList<String>();\r
75     arguments.add("pack");\r
76     arguments.add(params.getSpecFile().getPath());\r
77 \r
78     arguments.add("-OutputDirectory");\r
79     arguments.add(params.getOutputDirectory().getPath());\r
80 \r
81     arguments.add("-BasePath");\r
82     arguments.add(params.getBaseDirectory().getPath());\r
83 \r
84     arguments.add("-Verbose");\r
85 \r
86     arguments.add("-Version");\r
87     arguments.add(params.getVersion());\r
88 \r
89     for (String exclude : params.getExclude()) {\r
90       arguments.add("-Exclude");\r
91       arguments.add(exclude);\r
92     }\r
93 \r
94     if (params.packSymbols()) {\r
95       arguments.add("-Symbols");\r
96     }\r
97 \r
98     if (params.packTool()) {\r
99       arguments.add("-Tool");\r
100     }\r
101 \r
102     for (String prop : params.getProperties()) {\r
103       arguments.add("-Properties");\r
104       arguments.add(prop);\r
105     }\r
106 \r
107     for (String cmd : params.getCustomCommandline()) {\r
108       //TODO: check if -Build was added\r
109       arguments.add(cmd);\r
110     }\r
111 \r
112     return executeNuGet(params, Collections.<String>emptyList(), arguments, workdir, factory);\r
113   }\r
114 \r
115   @NotNull\r
116   public <T> T createPush(@NotNull final NuGetPublishParameters params,\r
117                           @NotNull final File packagePath,\r
118                           @NotNull final Callback<T> factory) throws RunBuildException {\r
119     final String apiKey = "teamcity_nuget_api_key_" + System.currentTimeMillis();\r
120 \r
121     final List<String> arguments = new ArrayList<String>();\r
122     arguments.add("push");\r
123     arguments.add(packagePath.getPath());\r
124     arguments.add("%%" + apiKey + "%%");\r
125     if (params.getCreateOnly()) {\r
126       arguments.add("-CreateOnly");\r
127     }\r
128 \r
129     final String source = params.getPublishSource();\r
130     return executeNuGet(\r
131             params,\r
132             StringUtil.isEmptyOrSpaces(source)\r
133                     ? Collections.<String>emptyList()\r
134                     : Arrays.asList(source),\r
135             arguments,\r
136             packagePath.getParentFile(),\r
137             Collections.singletonMap(apiKey, params.getApiKey()),\r
138             factory);\r
139   }\r
140 \r
141   private <T> T executeNuGet(@NotNull final NuGetParameters nuget,\r
142                              @NotNull final Collection<String> sources,\r
143                              @NotNull final Collection<String> arguments,\r
144                              @NotNull final File workingDir,\r
145                              @NotNull final Callback<T> factory) throws RunBuildException {\r
146     return executeNuGet(nuget, sources, arguments, workingDir, Collections.<String,String>emptyMap(), factory);\r
147   }\r
148 \r
149   private <T> T executeNuGet(@NotNull final NuGetParameters nuget,\r
150                              @NotNull final Collection<String> sources,\r
151                              @NotNull final Collection<String> arguments,\r
152                              @NotNull final File workingDir,\r
153                              @NotNull final Map<String,String> additionalEnvironment,\r
154                              @NotNull final Callback<T> factory) throws RunBuildException {\r
155     final List<String> argz = new ArrayList<String>(arguments);\r
156     for (String source : sources) {\r
157       argz.add("-Source");\r
158       argz.add(source);\r
159     }\r
160 \r
161     return factory.createCommand(\r
162             nuget.getNuGetExeFile(),\r
163             workingDir,\r
164             argz,\r
165             additionalEnvironment\r
166     );\r
167   }\r
168 \r
169 }\r