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