check NuGet tool is valid
[teamcity/dotNetPackagesSupport.git] / nuget-server / src / jetbrains / buildServer / nuget / server / toolRegistry / impl / ToolsRegistry.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 com.intellij.openapi.diagnostic.Logger;\r
20 import jetbrains.buildServer.nuget.server.toolRegistry.NuGetInstalledTool;\r
21 import jetbrains.buildServer.util.FileUtil;\r
22 import org.jetbrains.annotations.NotNull;\r
23 \r
24 import java.io.File;\r
25 import java.util.ArrayList;\r
26 import java.util.Collection;\r
27 import java.util.Collections;\r
28 \r
29 /**\r
30  * Created by Eugene Petrenko (eugene.petrenko@gmail.com)\r
31  * Date: 16.08.11 0:25\r
32  */\r
33 public class ToolsRegistry {\r
34   private static final Logger LOG = Logger.getInstance(ToolsRegistry.class.getName());\r
35   private final ToolPaths myPaths;\r
36   private final PluginNaming myNaming;\r
37 \r
38   public ToolsRegistry(@NotNull final ToolPaths paths,\r
39                        @NotNull final PluginNaming naming) {\r
40     myPaths = paths;\r
41     myNaming = naming;\r
42   }\r
43 \r
44   @NotNull\r
45   public Collection<? extends NuGetInstalledTool> getTools() {\r
46     return getToolsInternal();\r
47   }\r
48 \r
49   private Collection<InstalledTool> getToolsInternal() {\r
50     final File[] tools = myPaths.getTools().listFiles();\r
51     if (tools == null) return Collections.emptyList();\r
52     final Collection<InstalledTool> result = new ArrayList<InstalledTool>();\r
53     for (final File path : tools) {\r
54       final InstalledTool e = new InstalledTool(path);\r
55       if (!e.getPath().isFile()) {\r
56         LOG.warn("NuGet.exe is not found at " + e);\r
57         continue;\r
58       }\r
59 \r
60       if (myNaming.getAgentToolFilePath(e).isFile()) {\r
61         LOG.warn("NuGet tool is not packed for agent. " + e);\r
62         continue;\r
63       }\r
64       result.add(e);\r
65     }\r
66     return result;\r
67   }\r
68 \r
69   public void removeTool(@NotNull final String toolId) {\r
70     for (InstalledTool tool : getToolsInternal()) {\r
71       if (tool.getId().equals(toolId)) {\r
72         LOG.info("Removing NuGet plugin: " + tool);\r
73 \r
74         final File agentPlugin = myNaming.getAgentToolFilePath(tool);\r
75         LOG.info("Removing NuGet plugin agent tool : " + agentPlugin);\r
76         FileUtil.delete(agentPlugin);\r
77 \r
78         final File toolHome = tool.getRootPath();\r
79         LOG.info("Removing NuGet files from: " + toolHome);\r
80         FileUtil.delete(toolHome);\r
81         return;\r
82       }\r
83     }\r
84   }\r
85 \r
86   private static class InstalledTool implements NuGetInstalledTool {\r
87     private final File myPath;\r
88 \r
89     public InstalledTool(@NotNull final File path) {\r
90       myPath = path;\r
91     }\r
92 \r
93     @NotNull\r
94     public File getRootPath() {\r
95       return myPath;\r
96     }\r
97 \r
98     @NotNull\r
99     public File getPath() {\r
100       return new File(myPath, "tools/NuGet.exe");\r
101     }\r
102 \r
103     @NotNull\r
104     public String getId() {\r
105       return myPath.getName();\r
106     }\r
107 \r
108     @NotNull\r
109     public String getVersion() {\r
110       return myPath.getName();\r
111     }\r
112 \r
113     @Override\r
114     public String toString() {\r
115       return "InstalledTool{version=" + getVersion() +\r
116               ", myPath=" + myPath +\r
117               '}';\r
118     }\r
119   }\r
120 }\r