runtime/org.argeo.slc.core/src/main/java/org/argeo/slc/core/deploy/DefaultResourceSet.java

   1 package org.argeo.slc.core.deploy;
   2
   3 import java.io.IOException;
   4 import java.util.ArrayList;
   5 import java.util.List;
   6 import java.util.Map;
   7 import java.util.StringTokenizer;
   8 import java.util.TreeMap;
   9
  10 import org.apache.commons.logging.Log;
  11 import org.apache.commons.logging.LogFactory;
  12 import org.argeo.slc.SlcException;
  13 import org.springframework.beans.factory.InitializingBean;
  14 import org.springframework.context.ResourceLoaderAware;
  15 import org.springframework.core.io.Resource;
  16 import org.springframework.core.io.ResourceLoader;
  17 import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
  18 import org.springframework.core.io.support.ResourcePatternResolver;
  19 import org.springframework.util.AntPathMatcher;
  20 import org.springframework.util.PathMatcher;
  21
  22 public class DefaultResourceSet implements ResourceLoaderAware,
  23                 InitializingBean, ResourceSet {
  24         private final static Log log = LogFactory.getLog(DefaultResourceSet.class);
  25         public final static String DEFAULT_EXCLUDES = "**/.svn/**";
  26
  27         private String base;
  28         private String include;
  29         private List<String> includes = new ArrayList<String>();
  30         private String exclude;
  31         private List<String> excludes = new ArrayList<String>();
  32         private Boolean useDefaultExcludes = true;
  33         private ResourcePatternResolver resourcePatternResolver;
  34         private PathMatcher excludePathMatcher = new AntPathMatcher();
  35
  36         private ResourceLoader resourceLoader;
  37
  38         /** List the resources, identified by their relative path. */
  39         public Map<String, Resource> listResources() {
  40                 try {
  41                         String baseResUrl = getResourceLoaderToUse().getResource(base)
  42                                         .getURL().toString();
  43                         Map<String, Resource> res = new TreeMap<String, Resource>();
  44                         for (String includePattern : includes)
  45                                 processInclude(res, includePattern, baseResUrl);
  46                         return res;
  47                 } catch (IOException e) {
  48                         throw new SlcException("Cannot list resource from " + base, e);
  49                 }
  50         }
  51
  52         protected void processInclude(Map<String, Resource> res, String include,
  53                         String baseResUrl) throws IOException {
  54                 String pattern = base + "/" + include;
  55                 if (log.isDebugEnabled())
  56                         log.debug("Look for resources with pattern '" + pattern
  57                                         + "' in base url " + baseResUrl);
  58                 Resource[] resources = resourcePatternResolver.getResources(pattern);
  59                 resources: for (Resource resource : resources) {
  60                         String url = resource.getURL().toString();
  61                         String relPath = url.substring(baseResUrl.length());
  62
  63                         // skip dir
  64                         if (relPath.charAt(relPath.length() - 1) == '/') {
  65                                 if (log.isTraceEnabled())
  66                                         log.debug("Skip directory " + relPath + "=" + resource);
  67                                 continue resources;
  68                         }
  69
  70                         // make sure there is not starting '/'
  71                         if (relPath.charAt(0) == '/')
  72                                 relPath = relPath.substring(1);
  73
  74                         // skip excludes
  75                         for (String exclude : excludes)
  76                                 if (excludePathMatcher.match(exclude, relPath)) {
  77                                         if (log.isTraceEnabled())
  78                                                 log.debug("Exclude " + relPath + "=" + resource);
  79                                         continue resources;
  80                                 }
  81
  82                         // check if already exists
  83                         if (res.containsKey(relPath))
  84                                 log.warn(relPath + " already matched by " + res.get(relPath)
  85                                                 + ", " + resource + " will override it.");
  86
  87                         // store the marched resource
  88                         res.put(relPath, resource);
  89                         if (log.isDebugEnabled())
  90                                 log.debug(relPath + "=" + resource);
  91                 }
  92
  93         }
  94
  95         public void afterPropertiesSet() throws Exception {
  96                 if (resourcePatternResolver == null)
  97                         resourcePatternResolver = new PathMatchingResourcePatternResolver(
  98                                         getResourceLoaderToUse());
  99                 if (include != null)
 100                         addCommaSeparatedToList(include, includes);
 101                 if (exclude != null)
 102                         addCommaSeparatedToList(exclude, excludes);
 103
 104                 if (includes.size() == 0)
 105                         includes.add("**");
 106
 107                 if (useDefaultExcludes)
 108                         addCommaSeparatedToList(DEFAULT_EXCLUDES, excludes);
 109         }
 110
 111         private void addCommaSeparatedToList(String str, List<String> lst) {
 112                 StringTokenizer st = new StringTokenizer(str, ",");
 113                 while (st.hasMoreTokens()) {
 114                         String token = st.nextToken();
 115                         if (!lst.contains(token))
 116                                 lst.add(token);
 117                 }
 118         }
 119
 120         public void setResourceLoader(ResourceLoader resourceLoader) {
 121                 this.resourceLoader = resourceLoader;
 122         }
 123
 124         /**
 125          * Can be overridden in order to provide the proper resource loader used to
 126          * resolve resources.
 127          */
 128         public ResourceLoader getResourceLoaderToUse() {
 129                 return resourceLoader;
 130         }
 131
 132         public void setBase(String base) {
 133                 this.base = base;
 134         }
 135
 136         public void setInclude(String include) {
 137                 this.include = include;
 138         }
 139
 140         public void setIncludes(List<String> includes) {
 141                 this.includes = includes;
 142         }
 143
 144         public void setExclude(String exclude) {
 145                 this.exclude = exclude;
 146         }
 147
 148         public void setExcludes(List<String> excludes) {
 149                 this.excludes = excludes;
 150         }
 151
 152         public void setUseDefaultExcludes(Boolean useDefaultExcludes) {
 153                 this.useDefaultExcludes = useDefaultExcludes;
 154         }
 155
 156         public void setExcludePathMatcher(PathMatcher excludePathMatcher) {
 157                 this.excludePathMatcher = excludePathMatcher;
 158         }
 159
 160         public void setResourcePatternResolver(
 161                         ResourcePatternResolver resourcePatternResolver) {
 162                 this.resourcePatternResolver = resourcePatternResolver;
 163         }
 164
 165         public ResourcePatternResolver getResourcePatternResolver() {
 166                 return resourcePatternResolver;
 167         }
 168
 169 }