1 | |
package liquibase.changelog; |
2 | |
|
3 | |
import liquibase.util.StringUtils; |
4 | |
import liquibase.database.Database; |
5 | |
|
6 | |
import java.util.List; |
7 | |
import java.util.ArrayList; |
8 | |
import java.util.Map; |
9 | |
import java.util.Collection; |
10 | |
import java.util.regex.Pattern; |
11 | |
import java.util.regex.Matcher; |
12 | |
|
13 | 9 | public class ChangeLogParameters { |
14 | |
|
15 | 25 | private List<ChangeLogParameter> changeLogParameters = new ArrayList<ChangeLogParameter>(); |
16 | |
private ExpressionExpander expressionExpander; |
17 | |
private Database currentDatabase; |
18 | |
private List<String> currentContexts; |
19 | |
|
20 | |
public ChangeLogParameters() { |
21 | 20 | this(null); |
22 | 20 | } |
23 | |
|
24 | 25 | public ChangeLogParameters(Database currentDatabase) { |
25 | 25 | for (Map.Entry entry : System.getProperties().entrySet()) { |
26 | 1400 | changeLogParameters.add(new ChangeLogParameter(entry.getKey().toString(), entry.getValue())); |
27 | |
} |
28 | 25 | this.expressionExpander = new ExpressionExpander(this); |
29 | 25 | this.currentDatabase = currentDatabase; |
30 | 25 | this.currentContexts = new ArrayList<String>(); |
31 | 25 | } |
32 | |
|
33 | |
public void addContext(String context) { |
34 | 0 | this.currentContexts.add(context); |
35 | 0 | } |
36 | |
|
37 | |
public void setContexts(Collection<String> contexts) { |
38 | 2 | this.currentContexts = new ArrayList<String>(); |
39 | 2 | if (contexts != null) { |
40 | 2 | this.currentContexts.addAll(contexts); |
41 | |
} |
42 | 2 | } |
43 | |
|
44 | |
public void set(String paramter, Object value) { |
45 | 6 | changeLogParameters.add(new ChangeLogParameter(paramter, value)); |
46 | 6 | } |
47 | |
|
48 | |
public void set(String key, String value, String contexts, String databases) { |
49 | 4 | changeLogParameters.add(new ChangeLogParameter(key, value, contexts, databases)); |
50 | 4 | } |
51 | |
|
52 | |
|
53 | |
|
54 | |
|
55 | |
|
56 | |
|
57 | |
|
58 | |
|
59 | |
|
60 | |
public Object getValue(String key) { |
61 | 11 | ChangeLogParameter parameter = findParameter(key); |
62 | 11 | return parameter != null ? parameter.getValue() : null; |
63 | |
} |
64 | |
|
65 | |
private ChangeLogParameter findParameter(String key) { |
66 | 11 | for (ChangeLogParameter param : changeLogParameters) { |
67 | 608 | if (param.getKey().equalsIgnoreCase(key) && param.isValid()) { |
68 | 8 | return param; |
69 | |
} |
70 | |
} |
71 | 3 | return null; |
72 | |
} |
73 | |
|
74 | |
public boolean hasValue(String key) { |
75 | 0 | return findParameter(key) != null; |
76 | |
} |
77 | |
|
78 | |
public String expandExpressions(String string) { |
79 | 558 | return expressionExpander.expandExpressions(string); |
80 | |
} |
81 | |
|
82 | |
private class ChangeLogParameter { |
83 | |
private String key; |
84 | |
private Object value; |
85 | |
private List<String> validContexts; |
86 | |
private List<String> validDatabases; |
87 | |
|
88 | 1406 | public ChangeLogParameter(String key, Object value) { |
89 | 1406 | this.key = key; |
90 | 1406 | this.value = value; |
91 | 1406 | } |
92 | |
|
93 | |
public ChangeLogParameter(String key, Object value, String validContexts, String validDatabases) { |
94 | 4 | this(key, value, StringUtils.splitAndTrim(validContexts, ","), StringUtils |
95 | |
.splitAndTrim(validDatabases, ",")); |
96 | 4 | } |
97 | |
|
98 | 4 | public ChangeLogParameter(String key, Object value, List<String> validContexts, List<String> validDatabases) { |
99 | 4 | this.key = key; |
100 | 4 | this.value = value; |
101 | 4 | this.validContexts = validContexts; |
102 | 4 | this.validDatabases = validDatabases; |
103 | 4 | } |
104 | |
|
105 | |
public String getKey() { |
106 | 608 | return key; |
107 | |
} |
108 | |
|
109 | |
public Object getValue() { |
110 | 8 | return value; |
111 | |
} |
112 | |
|
113 | |
public List<String> getValidDatabases() { |
114 | 0 | return validDatabases; |
115 | |
} |
116 | |
|
117 | |
public List<String> getValidContexts() { |
118 | 0 | return validContexts; |
119 | |
} |
120 | |
|
121 | |
@Override |
122 | |
public String toString() { |
123 | 0 | return getValue().toString(); |
124 | |
} |
125 | |
|
126 | |
public boolean isValid() { |
127 | 10 | boolean isValid = true; |
128 | 10 | if (validContexts != null && validContexts.size() > 0) { |
129 | 2 | if (ChangeLogParameters.this.currentContexts != null |
130 | |
&& ChangeLogParameters.this.currentContexts.size() > 0) { |
131 | 2 | isValid = false; |
132 | 2 | for (String currentContext : ChangeLogParameters.this.currentContexts) { |
133 | 2 | if (validContexts.contains(currentContext)) { |
134 | 1 | isValid = true; |
135 | |
} |
136 | |
} |
137 | |
} |
138 | |
} |
139 | |
|
140 | 10 | if (isValid && validDatabases != null && validDatabases.size() > 0) { |
141 | 3 | isValid = validDatabases.contains(currentDatabase.getTypeName()); |
142 | |
} |
143 | |
|
144 | 10 | return isValid; |
145 | |
} |
146 | |
} |
147 | |
|
148 | |
protected static class ExpressionExpander { |
149 | |
private ChangeLogParameters changeLogParameters; |
150 | |
|
151 | 31 | public ExpressionExpander(ChangeLogParameters changeLogParameters) { |
152 | 31 | this.changeLogParameters = changeLogParameters; |
153 | 31 | } |
154 | |
|
155 | |
public String expandExpressions(String text) { |
156 | 565 | if (text == null) { |
157 | 287 | return null; |
158 | |
} |
159 | 278 | Pattern expressionPattern = Pattern.compile("(\\$\\{[^\\}]+\\})"); |
160 | 278 | Matcher matcher = expressionPattern.matcher(text); |
161 | 278 | String originalText = text; |
162 | 283 | while (matcher.find()) { |
163 | 5 | String expressionString = originalText.substring(matcher.start(), matcher.end()); |
164 | 5 | String valueTolookup = expressionString.replaceFirst("\\$\\{", "").replaceFirst("\\}$", ""); |
165 | |
|
166 | 5 | int dotIndex = valueTolookup.indexOf('.'); |
167 | 5 | Object value = changeLogParameters.getValue(valueTolookup); |
168 | |
|
169 | 5 | if (value != null) { |
170 | 3 | text = text.replace(expressionString, value.toString()); |
171 | |
} |
172 | 5 | } |
173 | 278 | return text; |
174 | |
} |
175 | |
} |
176 | |
} |