47dc04ed9a746cde670f712740b5934503f31de6
[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     return executeNuGet(params.getNuGetParameters(), argz, packagesConfig.getParentFile(), factory);\r
47   }\r
48 \r
49   @NotNull\r
50   public <T> T createUpdate(@NotNull PackagesUpdateParameters params, @NotNull File packagesConfig, @NotNull File targetFolder, @NotNull Callback<T> factory) throws RunBuildException {\r
51     final List<String> argz = new ArrayList<String>();\r
52     argz.add("update");\r
53     argz.add(FileUtil.getCanonicalFile(packagesConfig).getPath()); //path to package\r
54     if (params.getUseSafeUpdate()) {\r
55       argz.add("-Safe");\r
56     }\r
57     argz.add("-Verbose");\r
58     argz.add("-RepositoryPath");\r
59     argz.add(FileUtil.getCanonicalFile(targetFolder).getPath());\r
60 \r
61     for (String id : params.getPackagesToUpdate()) {\r
62       argz.add("-Id");\r
63       argz.add(id);\r
64     }\r
65 \r
66     return executeNuGet(params.getNuGetParameters(), argz, packagesConfig.getParentFile(), factory);\r
67   }\r
68 \r
69   public <T> T createPack(@NotNull final NuGetPackParameters params,\r
70                           @NotNull final File workdir,\r
71                           @NotNull final Callback<T> factory) throws RunBuildException {\r
72     final List<String> arguments = new ArrayList<String>();\r
73     arguments.add("pack");\r
74     arguments.add(params.getSpecFile().getPath());\r
75 \r
76     arguments.add("-OutputDirectory");\r
77     arguments.add(params.getOutputDirectory().getPath());\r
78 \r
79     arguments.add("-BasePath");\r
80     arguments.add(params.getBaseDirectory().getPath());\r
81 \r
82     arguments.add("-Verbose");\r
83 \r
84     arguments.add("-Version");\r
85     arguments.add(params.getVersion());\r
86 \r
87     for (String exclude : params.getExclude()) {\r
88       arguments.add("-Exclude");\r
89       arguments.add(exclude);\r
90     }\r
91 \r
92     if (params.packSymbols()) {\r
93       arguments.add("-Symbols");\r
94     }\r
95 \r
96     if (params.packTool()) {\r
97       arguments.add("-Tool");\r
98     }\r
99 \r
100     for (String prop : params.getProperties()) {\r
101       arguments.add("-Properties");\r
102       arguments.add(prop);\r
103     }\r
104 \r
105     for (String cmd : params.getCustomCommandline()) {\r
106       //TODO: check if -Build was added\r
107       arguments.add(cmd);\r
108     }\r
109 \r
110     return executeNuGet(params, Collections.<String>emptyList(), arguments, workdir, factory);\r
111   }\r
112 \r
113   @NotNull\r
114   public <T> T createPush(@NotNull final NuGetPublishParameters params,\r
115                           @NotNull final File packagePath,\r
116                           @NotNull final Callback<T> factory) throws RunBuildException {\r
117     final List<String> arguments = new ArrayList<String>();\r
118     arguments.add("push");\r
119     arguments.add(packagePath.getPath());\r
120     arguments.add(params.getApiKey());\r
121     if (params.getCreateOnly()) {\r
122       arguments.add("-CreateOnly");\r
123     }\r
124 \r
125     final String source = params.getPublishSource();\r
126     return executeNuGet(\r
127             params,\r
128             StringUtil.isEmptyOrSpaces(source)\r
129                     ? Collections.<String>emptyList()\r
130                     : Arrays.asList(source),\r
131             arguments,\r
132             packagePath.getParentFile(),\r
133             factory);\r
134   }\r
135 \r
136   @NotNull\r
137   private <T> T executeNuGet(@NotNull final NuGetFetchParameters nuget,\r
138                              @NotNull final Collection<String> arguments,\r
139                              @NotNull final File workingDir,\r
140                              @NotNull final Callback<T> factory) throws RunBuildException {\r
141     return executeNuGet(nuget, nuget.getNuGetPackageSources(), arguments, workingDir, factory);\r
142   }\r
143 \r
144   private <T> T executeNuGet(NuGetParameters nuget, Collection<String> sources, Collection<String> arguments, File workingDir, Callback<T> factory) throws RunBuildException {\r
145     final List<String> argz = new ArrayList<String>(arguments);\r
146     for (String source : sources) {\r
147       argz.add("-Source");\r
148       argz.add(source);\r
149     }\r
150 \r
151     return factory.createCommand(\r
152             nuget.getNuGetExeFile(),\r
153             workingDir,\r
154             argz\r
155     );\r
156   }\r
157 \r
158 }\r