View Javadoc
1   /**
2    * This file Copyright (c) 2014-2018 Magnolia International
3    * Ltd.  (http://www.magnolia-cms.com). All rights reserved.
4    *
5    *
6    * This file is dual-licensed under both the Magnolia
7    * Network Agreement and the GNU General Public License.
8    * You may elect to use one or the other of these licenses.
9    *
10   * This file is distributed in the hope that it will be
11   * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the
12   * implied warranty of MERCHANTABILITY or FITNESS FOR A
13   * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT.
14   * Redistribution, except as permitted by whichever of the GPL
15   * or MNA you select, is prohibited.
16   *
17   * 1. For the GPL license (GPL), you can redistribute and/or
18   * modify this file under the terms of the GNU General
19   * Public License, Version 3, as published by the Free Software
20   * Foundation.  You should have received a copy of the GNU
21   * General Public License, Version 3 along with this program;
22   * if not, write to the Free Software Foundation, Inc., 51
23   * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
24   *
25   * 2. For the Magnolia Network Agreement (MNA), this file
26   * and the accompanying materials are made available under the
27   * terms of the MNA which accompanies this distribution, and
28   * is available at http://www.magnolia-cms.com/mna.html
29   *
30   * Any modifications to this file must keep this entire header
31   * intact.
32   *
33   */
34  package info.magnolia.config.source.yaml;
35  
36  import info.magnolia.config.registry.Registry;
37  import info.magnolia.config.source.AbstractConfigurationSourceBuilder;
38  import info.magnolia.init.MagnoliaConfigurationProperties;
39  import info.magnolia.map2bean.Map2BeanTransformer;
40  import info.magnolia.module.ModuleRegistry;
41  import info.magnolia.objectfactory.Components;
42  import info.magnolia.resourceloader.ResourceOrigin;
43  
44  import java.util.regex.Pattern;
45  import java.util.regex.PatternSyntaxException;
46  
47  import org.slf4j.Logger;
48  import org.slf4j.LoggerFactory;
49  import org.yaml.snakeyaml.constructor.Construct;
50  
51  /**
52   * Builder for {@link info.magnolia.config.source.yaml.YamlConfigurationSource} instances.
53   */
54  public class YamlConfigurationSourceBuilder extends AbstractConfigurationSourceBuilder {
55  
56      private static final Logger log = LoggerFactory.getLogger(YamlConfigurationSourceBuilder.class);
57  
58      /**
59       * Regex format for config files according to Magnolia module conventions.
60       * Mind the %s part of the regex, which should be replaced with the expected definition type.
61       * @see #moduleConventionsFormatFor(Registry)
62       */
63      private static final String MODULE_CONVENTIONS_FORMAT = "^/(?<module>[a-zA-Z0-9-_]+)/(?<deftype>%s)/(?<relPath>.*/)?(?<name>[a-zA-Z0-9-_\\.]+)\\.yaml$";
64  
65      private final ResourceOrigin origin;
66      private final YamlReader yamlReader;
67      private final MagnoliaConfigurationProperties magnoliaConfigurationProperties;
68      private final Map2BeanTransformer map2BeanTransformer;
69      private final ModuleRegistry moduleRegistry;
70      private String pathPattern;
71  
72      /**
73       * @deprecated since 5.5 - use {@link #YamlConfigurationSourceBuilder(ResourceOrigin, Map2BeanTransformer, YamlReader, MagnoliaConfigurationProperties, ModuleRegistry)} instead.
74       */
75      @Deprecated
76      public YamlConfigurationSourceBuilder(ResourceOrigin origin, info.magnolia.config.map2bean.Map2BeanTransformer map2BeanTransformer, YamlReader yamlReader) {
77          this(origin, map2BeanTransformer, yamlReader, Components.getComponent(MagnoliaConfigurationProperties.class), Components.getComponent(ModuleRegistry.class));
78      }
79  
80      /**
81       * @deprecated since 5.5.6 - use {@link #YamlConfigurationSourceBuilder(ResourceOrigin, Map2BeanTransformer, YamlReader, MagnoliaConfigurationProperties, ModuleRegistry)} instead.
82       */
83      @Deprecated
84      public YamlConfigurationSourceBuilder(ResourceOrigin origin, Map2BeanTransformer map2BeanTransformer, YamlReader yamlReader, MagnoliaConfigurationProperties magnoliaConfigurationProperties) {
85          this(origin, map2BeanTransformer, yamlReader, magnoliaConfigurationProperties, Components.getComponent(ModuleRegistry.class));
86      }
87  
88      public YamlConfigurationSourceBuilder(ResourceOrigin origin, Map2BeanTransformer map2BeanTransformer, YamlReader yamlReader, MagnoliaConfigurationProperties magnoliaConfigurationProperties, ModuleRegistry moduleRegistry) {
89          this.origin = origin;
90          this.map2BeanTransformer = map2BeanTransformer;
91          this.yamlReader = yamlReader;
92          this.magnoliaConfigurationProperties = magnoliaConfigurationProperties;
93          this.moduleRegistry = moduleRegistry;
94      }
95  
96      public YamlConfigurationSourceBuilder withPattern(String pathPattern) {
97          this.pathPattern = pathPattern;
98          return this;
99      }
100 
101     public YamlConfigurationSourceBuilder withCustomMultiConstruct(String tagPrefix, Construct construct) {
102         this.yamlReader.registerCustomMultiConstruct(tagPrefix, construct);
103         return this;
104     }
105 
106     public YamlConfigurationSourceBuilder withCustomConstruct(String tag, Construct construct) {
107         this.yamlReader.registerCustomConstruct(tag, construct);
108         return this;
109     }
110 
111     @Override
112     public void bindTo(Registry<?> registry) {
113         final Pattern pathPattern = validatePathPattern(this.pathPattern);
114         if (registry == null) {
115             throw new NullPointerException("Must pass a registry instance");
116         }
117 
118         new YamlConfigurationSource<>(origin, map2BeanTransformer, registry, pathPattern, yamlReader, magnoliaConfigurationProperties, moduleRegistry).start();
119     }
120 
121     public void bindWithDefaults(Registry registry) {
122         if (pathPattern != null) {
123             throw new IllegalStateException(this + " is already partially configured, don't use bindWithDefaults().");
124         }
125         withPattern(moduleConventionsFormatFor(registry)).bindTo(registry);
126     }
127 
128     protected Pattern validatePathPattern(String regex) {
129         if (pathPattern == null) {
130             throw new IllegalStateException(this + " is incomplete.");
131         }
132         try {
133             // Ideally, we'd validate the pattern contains 1 group.
134             return Pattern.compile(regex);
135         } catch (PatternSyntaxException e) {
136             log.warn("Pattern {} is invalid: {}", regex, e.getMessage());
137             return Pattern.compile("$^");
138         }
139     }
140 
141     // TODO Move this (and the regex pattern) to a component that exposes this and more conventions-wrapping objects (e.g Predicate<Path>)
142     protected String moduleConventionsFormatFor(Registry registry) {
143         return String.format(MODULE_CONVENTIONS_FORMAT, registry.type().getPluralName());
144     }
145 
146 }