1 files changed, 88 insertions, 0 deletions
diff --git a/das-java/trunk/tools/src/main/java/org/apache/tuscany/das/rdb/util/DBToXSDGenerator.java b/das-java/trunk/tools/src/main/java/org/apache/tuscany/das/rdb/util/DBToXSDGenerator.java
new file mode 100644
index 0000000000..a5e510dedf
--- /dev/null
+++ b/das-java/trunk/tools/src/main/java/org/apache/tuscany/das/rdb/util/DBToXSDGenerator.java
@@ -0,0 +1,88 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ * 
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.    
+ */
+package org.apache.tuscany.das.rdb.util;
+
+import java.io.InputStream;
+
+public class DBToXSDGenerator {
+	/**
+	 * Output is dbSchemaFile as specified in dbInfoFileName.schemaFile
+	 * @param dbInfoFileName  name of file which contains db connection info and output files names for dbschema and model 
+	 * @throws Exception
+	 */	
+	public static void getSchemaFileFromDB(String dbInfoFileName) throws Exception {
+		DBToSchemaFile.read(dbInfoFileName);
+		DBToSchemaFile.schemaFileFromDB();
+	}
+	
+	/**
+	 * If user supplies schemaFileName and modelFileName will use it instead of the one from DBToSchemaFile.
+	 * schemaFile should exist. If modelFileName is null, will use STDOUT.
+	 * @param schemaFileName the result of Torque output
+	 * @throws Exception
+	 */
+	public static void getModelFileFromSchemaFile(String schemaFileName, String modelFileName) throws Exception {
+		if(schemaFileName == null || schemaFileName.trim().equals("")) {
+			throw new RuntimeException("Null or empty schemaFileName");
+		}
+		
+		SchemaFileToXSD.convert(schemaFileName, modelFileName);
+	}
+	
+	/**
+	 * If user supplies schemaFileName and modelFileName through ModelXSDGenOption, will use it instead of 
+	 * the one from DBToSchemaFile. schemaFile should exist. If modelFileName is null, will use STDOUT.
+	 * @param ModelXSDGenOption 
+	 * @throws Exception
+	 */
+	public static void getModelFileFromSchemaFile(ModelXSDGenOption mo) throws Exception {
+		getModelFileFromSchemaFile(mo.getSchemaFile(), mo.getModelFile());
+	}
+	
+	/**
+	 * All in one
+	 * @param dbInfoFileName e.g. DBConnectionConfig.xml
+	 * @throws Exception
+	 */
+	public static void getModelFileFromDB(String dbInfoFileName) throws Exception {
+		DBToSchemaFile.read(dbInfoFileName);
+		DBToSchemaFile.schemaFileFromDB();
+		getModelFileFromSchemaFile(DBToSchemaFile.getMo());
+	}
+	
+	/**
+	 * Useful for plugin
+	 */
+	public static void getModelFileFromDB(ModelXSDGenOption mo) throws Exception {
+		if(mo.getSchemaFile() == null || mo.getSchemaFile().trim().equals("") || 
+			mo.getDriverClass() == null || mo.getDriverClass().trim().equals("") || 
+			mo.getDatabaseURL() == null || mo.getDatabaseURL().trim().equals("") ||
+			mo.getSchemaName() == null || mo.getSchemaName().trim().equals("")) {
+			throw new RuntimeException("Required inputs missing - check driverClass, url, schemaName, schemaFile!");
+		}
+		
+		DBToSchemaFile.setMo(mo);	
+		DBToSchemaFile.schemaFileFromDB();
+		getModelFileFromSchemaFile(DBToSchemaFile.getMo().getSchemaFile(), DBToSchemaFile.getMo().getModelFile());
+	}
+	
+    protected static InputStream getStream(String fileName) {
+        return Thread.currentThread().getContextClassLoader().getResourceAsStream(fileName);
+    }    
+}