@@ -118,25 +118,38 @@ File getOutputFolder() {
118
118
}
119
119
120
120
RevengStrategy setupReverseEngineeringStrategy () {
121
+ File [] revengFiles = getRevengFiles ();
121
122
RevengStrategy result = RevengStrategyFactory
122
- .createReverseEngineeringStrategy (getExtension ().revengStrategy );
123
+ .createReverseEngineeringStrategy (getExtension ().revengStrategy , revengFiles );
123
124
RevengSettings settings = new RevengSettings (result );
124
125
settings .setDefaultPackageName (getExtension ().packageName );
125
126
result .setSettings (settings );
126
127
return result ;
127
128
}
128
-
129
- private File getPropertyFile () {
130
- String hibernatePropertiesFile = getExtension ().hibernateProperties ;
129
+
130
+ private File getFile (String filename ) {
131
131
SourceSetContainer ssc = getProject ().getExtensions ().getByType (SourceSetContainer .class );
132
132
SourceSet ss = ssc .getByName (SourceSet .MAIN_SOURCE_SET_NAME );
133
133
SourceDirectorySet sds = ss .getResources ();
134
134
for (File f : sds .getFiles ()) {
135
- if (hibernatePropertiesFile .equals (f .getName ())) {
135
+ if (filename .equals (f .getName ())) {
136
136
return f ;
137
137
}
138
138
}
139
- throw new BuildException ("File '" + hibernatePropertiesFile + "' could not be found" );
139
+ throw new BuildException ("File '" + filename + "' could not be found" );
140
+ }
141
+
142
+ private File getPropertyFile () {
143
+ return getFile (getExtension ().hibernateProperties );
144
+ }
145
+
146
+ private File [] getRevengFiles () {
147
+ String revengFile = getExtension ().revengFile ;
148
+ if (revengFile == null ) {
149
+ return null ;
150
+ }
151
+
152
+ return new File [] { getFile (revengFile ) };
140
153
}
141
154
142
155
private void loadPropertiesFile (File propertyFile ) {
0 commit comments