@@ -101,25 +101,38 @@ File getOutputFolder() {
101
101
}
102
102
103
103
RevengStrategy setupReverseEngineeringStrategy () {
104
+ File [] revengFiles = getRevengFiles ();
104
105
RevengStrategy result = RevengStrategyFactory
105
- .createReverseEngineeringStrategy (getExtension ().revengStrategy );
106
+ .createReverseEngineeringStrategy (getExtension ().revengStrategy , revengFiles );
106
107
RevengSettings settings = new RevengSettings (result );
107
108
settings .setDefaultPackageName (getExtension ().packageName );
108
109
result .setSettings (settings );
109
110
return result ;
110
111
}
111
-
112
- private File getPropertyFile () {
113
- String hibernatePropertiesFile = getExtension ().hibernateProperties ;
112
+
113
+ private File getFile (String filename ) {
114
114
SourceSetContainer ssc = getProject ().getExtensions ().getByType (SourceSetContainer .class );
115
115
SourceSet ss = ssc .getByName (SourceSet .MAIN_SOURCE_SET_NAME );
116
116
SourceDirectorySet sds = ss .getResources ();
117
117
for (File f : sds .getFiles ()) {
118
- if (hibernatePropertiesFile .equals (f .getName ())) {
118
+ if (filename .equals (f .getName ())) {
119
119
return f ;
120
120
}
121
121
}
122
- throw new BuildException ("File '" + hibernatePropertiesFile + "' could not be found" );
122
+ throw new BuildException ("File '" + filename + "' could not be found" );
123
+ }
124
+
125
+ private File getPropertyFile () {
126
+ return getFile (getExtension ().hibernateProperties );
127
+ }
128
+
129
+ private File [] getRevengFiles () {
130
+ String revengFile = getExtension ().revengFile ;
131
+ if (revengFile == null ) {
132
+ return null ;
133
+ }
134
+
135
+ return new File [] { getFile (revengFile ) };
123
136
}
124
137
125
138
private void loadPropertiesFile (File propertyFile ) {
0 commit comments