minor
[teamcity/dotNetPackagesSupport.git] / nuget-agent / src / jetbrains / buildServer / nuget / agent / install / impl / NuGetActionFactoryImpl.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.install.impl;\r
18 \r
19 import com.intellij.openapi.diagnostic.Logger;\r
20 import jetbrains.buildServer.RunBuildException;\r
21 import jetbrains.buildServer.agent.BuildFinishedStatus;\r
22 import jetbrains.buildServer.agent.BuildProcess;\r
23 import jetbrains.buildServer.agent.BuildRunnerContext;\r
24 import jetbrains.buildServer.nuget.agent.install.NuGetActionFactory;\r
25 import jetbrains.buildServer.nuget.agent.install.PackageUsages;\r
26 import jetbrains.buildServer.nuget.agent.parameters.NuGetFetchParameters;\r
27 import jetbrains.buildServer.nuget.agent.parameters.PackagesInstallParameters;\r
28 import jetbrains.buildServer.nuget.agent.parameters.PackagesUpdateParameters;\r
29 import jetbrains.buildServer.nuget.agent.util.BuildProcessBase;\r
30 import jetbrains.buildServer.nuget.agent.util.CommandlineBuildProcessFactory;\r
31 import jetbrains.buildServer.util.FileUtil;\r
32 import org.jetbrains.annotations.NotNull;\r
33 \r
34 import java.io.File;\r
35 import java.util.ArrayList;\r
36 import java.util.Collection;\r
37 import java.util.List;\r
38 \r
39 /**\r
40  * Created by Eugene Petrenko (eugene.petrenko@gmail.com)\r
41  * Date: 07.07.11 17:49\r
42  */\r
43 public class NuGetActionFactoryImpl implements NuGetActionFactory {\r
44   private static final Logger LOG = Logger.getInstance(NuGetActionFactoryImpl.class.getName());\r
45 \r
46   private final CommandlineBuildProcessFactory myFactory;\r
47   private final PackageUsages myPackageUsages;\r
48 \r
49   public NuGetActionFactoryImpl(@NotNull final CommandlineBuildProcessFactory factory,\r
50                                 @NotNull final PackageUsages packageUsages) {\r
51     myFactory = factory;\r
52     myPackageUsages = packageUsages;\r
53   }\r
54 \r
55   @NotNull\r
56   public BuildProcess createInstall(@NotNull final BuildRunnerContext context,\r
57                                     @NotNull final PackagesInstallParameters params,\r
58                                     @NotNull final File packagesConfig,\r
59                                     @NotNull final File targetFolder) throws RunBuildException {\r
60     final List<String> argz = new ArrayList<String>();\r
61     argz.add("install");\r
62     argz.add(FileUtil.getCanonicalFile(packagesConfig).getPath()); //path to package\r
63     if (params.getExcludeVersion()) {\r
64       argz.add("-ExcludeVersion");\r
65     }\r
66     argz.add("-OutputDirectory");\r
67     argz.add(FileUtil.getCanonicalFile(targetFolder).getPath());\r
68 \r
69     return executeNuGet(context, params.getNuGetParameters(), argz, packagesConfig.getParentFile());\r
70   }\r
71 \r
72 \r
73   @NotNull\r
74   public BuildProcess createUpdate(@NotNull final BuildRunnerContext context,\r
75                                    @NotNull final PackagesUpdateParameters params,\r
76                                    @NotNull final File packagesConfig,\r
77                                    @NotNull final File targetFolder) throws RunBuildException {\r
78     final List<String> argz = new ArrayList<String>();\r
79     argz.add("update");\r
80     argz.add(FileUtil.getCanonicalFile(packagesConfig).getPath()); //path to package\r
81     if (params.getUseSafeUpdate()) {\r
82       argz.add("-Safe");\r
83     }\r
84     argz.add("-Verbose");\r
85     argz.add("-RepositoryPath");\r
86     argz.add(FileUtil.getCanonicalFile(targetFolder).getPath());\r
87 \r
88     for (String id : params.getPackagesToUpdate()) {\r
89       argz.add("-Id");\r
90       argz.add(id);\r
91     }\r
92 \r
93     return executeNuGet(context, params.getNuGetParameters(), argz, packagesConfig.getParentFile());\r
94   }\r
95 \r
96   @NotNull\r
97   public BuildProcess createUsageReport(@NotNull final BuildRunnerContext context,\r
98                                         @NotNull final NuGetFetchParameters params,\r
99                                         @NotNull final File packagesConfig,\r
100                                         @NotNull final File targetFolder) throws RunBuildException {\r
101     return new BuildProcessBase() {\r
102       @NotNull\r
103       @Override\r
104       protected BuildFinishedStatus waitForImpl() throws RunBuildException {\r
105         myPackageUsages.createReport(packagesConfig);\r
106         return BuildFinishedStatus.FINISHED_SUCCESS;\r
107       }\r
108     };\r
109   }\r
110 \r
111   @NotNull\r
112   private BuildProcess executeNuGet(@NotNull final BuildRunnerContext context,\r
113                                     @NotNull final NuGetFetchParameters nuget,\r
114                                     @NotNull final Collection<String> arguments,\r
115                                     @NotNull final File workingDirectory) throws RunBuildException {\r
116     final List<String> argz = new ArrayList<String>(arguments);\r
117     for (String source : nuget.getNuGetPackageSources()) {\r
118       argz.add("-Source");\r
119       argz.add(source);\r
120     }\r
121 \r
122     return myFactory.executeCommandLine(\r
123             context,\r
124             nuget.getNuGetExeFile(),\r
125             argz,\r
126             workingDirectory\r
127     );\r
128   }\r
129 }\r