From 593b86227aa306c9b8da9d9bf681b2f17c3b87be Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Ko=C5=82odziejski?= Date: Thu, 6 Nov 2014 11:29:19 +0100 Subject: [PATCH 1/2] Integration with Vertabelo - native support for Vertabelo XML file --- jOOQ-meta/pom.xml | 50 +- .../util/vertabelo/VertabeloXMLDatabase.java | 449 ++++++++++++++++++ .../VertabeloXMLTableDefinition.java | 144 ++++++ .../main/resources/xjb/vertabelo/binding.xjb | 27 ++ .../src/main/resources/xsd/vertabelo-2.1.xsd | 439 +++++++++++++++++ 5 files changed, 1108 insertions(+), 1 deletion(-) create mode 100644 jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloXMLDatabase.java create mode 100644 jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloXMLTableDefinition.java create mode 100644 jOOQ-meta/src/main/resources/xjb/vertabelo/binding.xjb create mode 100644 jOOQ-meta/src/main/resources/xsd/vertabelo-2.1.xsd diff --git a/jOOQ-meta/pom.xml b/jOOQ-meta/pom.xml index 73257f4a81..ec2b4bd807 100644 --- a/jOOQ-meta/pom.xml +++ b/jOOQ-meta/pom.xml @@ -131,6 +131,54 @@ + + + vertabelo-xml + + generate + + + true + true + false + src/main/resources/xsd + src/main/resources/xjb/vertabelo + + vertabelo-2.1.xsd + + org.jooq.util.vertabelo.jaxb + + -Xxew + -Xxew:instantiate lazy + -Xxew:delete + -Xfluent-api + -Xdefault-value + -Xannotate + + + + com.github.jaxb-xew-plugin + jaxb-xew-plugin + 1.0 + + + org.jvnet.jaxb2_commons + jaxb2-fluent-api + 3.0 + + + org.jvnet.jaxb2_commons + jaxb2-default-value + 1.1 + + + org.jvnet.jaxb2_commons + jaxb2-basics-annotate + 0.6.2 + + + + @@ -165,4 +213,4 @@ jooq - \ No newline at end of file + diff --git a/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloXMLDatabase.java b/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloXMLDatabase.java new file mode 100644 index 0000000000..18fceb76e3 --- /dev/null +++ b/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloXMLDatabase.java @@ -0,0 +1,449 @@ +package org.jooq.util.vertabelo; + +import java.io.File; +import java.sql.SQLException; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; + +import javax.xml.bind.JAXB; + +import org.jooq.DSLContext; +import org.jooq.SQLDialect; +import org.jooq.impl.DSL; +import org.jooq.tools.JooqLogger; +import org.jooq.tools.StringUtils; +import org.jooq.util.AbstractDatabase; +import org.jooq.util.ArrayDefinition; +import org.jooq.util.CheckConstraintDefinition; +import org.jooq.util.ColumnDefinition; +import org.jooq.util.DataTypeDefinition; +import org.jooq.util.DefaultCheckConstraintDefinition; +import org.jooq.util.DefaultDataTypeDefinition; +import org.jooq.util.DefaultRelations; +import org.jooq.util.DefaultSequenceDefinition; +import org.jooq.util.EnumDefinition; +import org.jooq.util.PackageDefinition; +import org.jooq.util.RoutineDefinition; +import org.jooq.util.SchemaDefinition; +import org.jooq.util.SequenceDefinition; +import org.jooq.util.TableDefinition; +import org.jooq.util.UDTDefinition; +import org.jooq.util.vertabelo.jaxb.AlternateKey; +import org.jooq.util.vertabelo.jaxb.AlternateKeyColumn; +import org.jooq.util.vertabelo.jaxb.Column; +import org.jooq.util.vertabelo.jaxb.DatabaseModel; +import org.jooq.util.vertabelo.jaxb.Property; +import org.jooq.util.vertabelo.jaxb.Reference; +import org.jooq.util.vertabelo.jaxb.ReferenceColumn; +import org.jooq.util.vertabelo.jaxb.Sequence; +import org.jooq.util.vertabelo.jaxb.Table; +import org.jooq.util.vertabelo.jaxb.TableCheck; +import org.jooq.util.vertabelo.jaxb.View; + +/** + * The Vertabelo XML Database + * + * @author Michał Kołodziejski + */ +public class VertabeloXMLDatabase extends AbstractDatabase { + + interface TableOperation { + void invoke(Table table, String schemaName); + } + + interface ViewOperation { + void invoke(View view, String schemaName); + } + + + private static final JooqLogger log = JooqLogger.getLogger(VertabeloXMLDatabase.class); + + // codegen properties + private static final String XML_FILE_PARAM = "xml-file"; + private static final String DIALECT_PARAM = "dialect"; + + // XML additional properties + private static final String SCHEMA_ADDITIONAL_PROPERTY_NAME = "Schema"; + private static final String PK_ADDITIONAL_PROPERTY_NAME = "Primary key name"; + + + protected DatabaseModel databaseModel; + + protected DatabaseModel databaseModel() { + if(databaseModel == null) { + databaseModel = JAXB.unmarshal(new File(getProperties().getProperty(XML_FILE_PARAM)), DatabaseModel.class); + } + + return databaseModel; + } + + + @Override + protected DSLContext create0() { + @SuppressWarnings("deprecation") + SQLDialect dialect = SQLDialect.SQL99; + + try { + dialect = SQLDialect.valueOf(getProperties().getProperty(DIALECT_PARAM)); + } + catch (Exception ignore) {} + + return DSL.using(dialect); + } + + + @Override + protected void loadPrimaryKeys(final DefaultRelations relations) throws SQLException { + + filterTablesBySchema(databaseModel().getTables(), new TableOperation() { + @Override + public void invoke(Table table, String schemaName) { + + SchemaDefinition schema = getSchema(schemaName); + TableDefinition tableDefinition = getTable(schema, table.getName()); + + if (tableDefinition != null) { + String pkName = getTablePkName(table); + + // iterate through all columns and find PK columns + for (Column column : table.getColumns()) { + if (column.isPK()) { + relations.addPrimaryKey(pkName, tableDefinition.getColumn(column.getName())); + } + } + } + } + }); + } + + + private String getTablePkName(Table table) { + Property pkAdditionalProperty = VertabeloXMLDatabase.findAdditionalProperty(PK_ADDITIONAL_PROPERTY_NAME, + table.getProperties()); + String pkName = VertabeloXMLDatabase.getAdditionalPropertyValueOrEmpty(pkAdditionalProperty); + if (StringUtils.isEmpty(pkName)) { + pkName = table.getName().toUpperCase() + "_PK"; + } + return pkName; + } + + + @Override + protected void loadUniqueKeys(final DefaultRelations relations) throws SQLException { + + filterTablesBySchema(databaseModel().getTables(), new TableOperation() { + @Override + public void invoke(Table table, String schemaName) { + SchemaDefinition schema = getSchema(schemaName); + TableDefinition tableDefinition = getTable(schema, table.getName()); + + if (tableDefinition != null) { + // iterate through all UNIQUE keys for this table + for (AlternateKey alternateKey : table.getAlternateKeys()) { + + // iterate through all columns of this key + for (AlternateKeyColumn alternateKeyColumn : alternateKey.getColumns()) { + Column column = (Column) alternateKeyColumn.getColumn(); + relations.addUniqueKey(alternateKey.getName(), tableDefinition.getColumn(column.getName())); + } + + } + } + } + }); + + } + + + @Override + protected void loadForeignKeys(final DefaultRelations relations) throws SQLException { + + for (final Reference reference : databaseModel().getReferences()) { + final Table pkTable = (Table) reference.getPKTable(); + final Table fkTable = (Table) reference.getFKTable(); + + filterTablesBySchema(Arrays.asList(pkTable), new TableOperation() { + @Override + public void invoke(Table table, String schemaName) { + SchemaDefinition schema = getSchema(schemaName); + TableDefinition pkTableDefinition = getTable(schema, pkTable.getName()); + TableDefinition fkTableDefinition = getTable(schema, fkTable.getName()); + + // we need to find unique key among PK and all alternate keys... + String uniqueKeyName = findUniqueConstraintNameForReference(reference); + if(uniqueKeyName == null) { + // no matching key - ignore this foreign key + return; + } + + for (ReferenceColumn referenceColumn : reference.getReferenceColumns()) { + Column fkColumn = (Column) referenceColumn.getFKColumn(); + ColumnDefinition fkColumnDefinition = fkTableDefinition.getColumn(fkColumn.getName()); + + relations.addForeignKey( + reference.getName(), + uniqueKeyName, + fkColumnDefinition, + pkTableDefinition.getSchema()); + } + } + }); + + } + } + + + private String findUniqueConstraintNameForReference(Reference reference) { + // list of referenced columns + List uniqueKeyColumns = new ArrayList(); + for (ReferenceColumn referenceColumn : reference.getReferenceColumns()) { + uniqueKeyColumns.add((Column) referenceColumn.getPKColumn()); + } + + + // list of PK columns + Table pkTable = (Table) reference.getPKTable(); + List pkColumns = new ArrayList(); + for (Column column : pkTable.getColumns()) { + if (column.isPK()) { + pkColumns.add(column); + } + } + + if (uniqueKeyColumns.equals(pkColumns)) { + // PK matches FK + log.info("Primary key constraint matches foreign key: " + reference.getName()); + return getTablePkName((Table) reference.getPKTable()); + } + + // need to check alternate keys + for (AlternateKey alternateKey : pkTable.getAlternateKeys()) { + List akColumns = new ArrayList(); + for (AlternateKeyColumn column : alternateKey.getColumns()) { + akColumns.add((Column) column.getColumn()); + } + + if (uniqueKeyColumns.equals(akColumns)) { + // AK matches FK + log.info("Alternate key constraint matches foreign key: " + reference.getName()); + return alternateKey.getName(); + } + } + + // no match + log.info("No matching unique constraint for foreign key: " + reference.getName()); + return null; + } + + + + @Override + protected void loadCheckConstraints(final DefaultRelations relations) throws SQLException { + + filterTablesBySchema(databaseModel().getTables(), new TableOperation() { + @Override + public void invoke(Table table, String schemaName) { + SchemaDefinition schema = getSchema(schemaName); + TableDefinition tableDefinition = getTable(schema, table.getName()); + + if (tableDefinition != null) { + + // iterate through all table checks + for (TableCheck tableCheck : table.getTableChecks()) { + CheckConstraintDefinition checkConstraintDefinition = new DefaultCheckConstraintDefinition( + schema, + tableDefinition, + tableCheck.getName(), + tableCheck.getCheckExpression()); + + relations.addCheckConstraint(tableDefinition, checkConstraintDefinition); + } + + // iterate through all columns and find columns with checks + for (Column column : table.getColumns()) { + if (! StringUtils.isBlank(column.getCheckExpression())) { + CheckConstraintDefinition checkConstraintDefinition = new DefaultCheckConstraintDefinition( + schema, + tableDefinition, + table.getName() + "_" + column.getName() + "_check", + column.getCheckExpression()); + + relations.addCheckConstraint(tableDefinition, checkConstraintDefinition); + } + } + } + } + }); + } + + + @Override + protected List getSchemata0() throws SQLException { + List result = new ArrayList(); + List schemaNames = new ArrayList(); + + // search in tables + for (Table table : databaseModel().getTables()) { + Property additionalProperty = findAdditionalProperty(SCHEMA_ADDITIONAL_PROPERTY_NAME, table.getProperties()); + addUniqueSchemaName(additionalProperty, schemaNames); + } + + // search in views + for (View view : databaseModel().getViews()) { + Property additionalProperty = findAdditionalProperty(SCHEMA_ADDITIONAL_PROPERTY_NAME, view.getProperties()); + addUniqueSchemaName(additionalProperty, schemaNames); + } + + + // transform + for (String schemaName : schemaNames) { + result.add(new SchemaDefinition(this, schemaName, null)); + } + + return result; + } + + + private void addUniqueSchemaName(Property additionalProperty, List schemaNames) { + String schemaName = ""; // default to empty string + if (additionalProperty != null) { + // additional property is set + schemaName = additionalProperty.getValue(); + } + + if (!schemaNames.contains(schemaName)) { + schemaNames.add(schemaName); + } + } + + + @Override + protected List getSequences0() throws SQLException { + List result = new ArrayList(); + + for (Sequence sequence : databaseModel().getSequences()) { + Property additionalProperty = VertabeloXMLDatabase.findAdditionalProperty(SCHEMA_ADDITIONAL_PROPERTY_NAME, + sequence.getProperties()); + String schemaName = VertabeloXMLDatabase.getAdditionalPropertyValueOrEmpty(additionalProperty); + + if (getInputSchemata().contains(schemaName)) { + SchemaDefinition schema = getSchema(schemaName); + + DataTypeDefinition type = new DefaultDataTypeDefinition( + this, + schema, + "BIGINT" + ); + + result.add(new DefaultSequenceDefinition(schema, sequence.getName(), type)); + } + } + + return result; + } + + + @Override + protected List getTables0() throws SQLException { + final List result = new ArrayList(); + + // tables + filterTablesBySchema(databaseModel().getTables(), new TableOperation() { + @Override + public void invoke(Table table, String schemaName) { + SchemaDefinition schema = getSchema(schemaName); + result.add(new VertabeloXMLTableDefinition(schema, table)); + } + }); + + // views + filterViewsBySchema(databaseModel().getViews(), new ViewOperation() { + @Override + public void invoke(View view, String schemaName) { + SchemaDefinition schema = getSchema(schemaName); + result.add(new VertabeloXMLTableDefinition(schema, view)); + } + }); + + return result; + } + + + @Override + protected List getEnums0() { + List result = new ArrayList(); + return result; + } + + @Override + protected List getUDTs0() { + List result = new ArrayList(); + return result; + } + + @Override + protected List getArrays0() { + List result = new ArrayList(); + return result; + } + + @Override + protected List getRoutines0() { + List result = new ArrayList(); + return result; + } + + @Override + protected List getPackages0() { + List result = new ArrayList(); + return result; + } + + + protected void filterTablesBySchema(List tables, TableOperation operation) { + for (Table table : tables) { + Property schemaAdditionalProperty = VertabeloXMLDatabase.findAdditionalProperty(SCHEMA_ADDITIONAL_PROPERTY_NAME, + table.getProperties()); + String schemaName = VertabeloXMLDatabase.getAdditionalPropertyValueOrEmpty(schemaAdditionalProperty); + + if (getInputSchemata().contains(schemaName)) { + + operation.invoke(table, schemaName); + + } + } + } + + + protected void filterViewsBySchema(List views, ViewOperation operation) { + for (View view : views) { + Property schemaAdditionalProperty = VertabeloXMLDatabase.findAdditionalProperty(SCHEMA_ADDITIONAL_PROPERTY_NAME, + view.getProperties()); + String schemaName = VertabeloXMLDatabase.getAdditionalPropertyValueOrEmpty(schemaAdditionalProperty); + + if (getInputSchemata().contains(schemaName)) { + + operation.invoke(view, schemaName); + + } + } + } + + + public static Property findAdditionalProperty(String name, List properties) { + for (Property property : properties) { + if (property.getName().equalsIgnoreCase(name)) { + return property; + } + } + return null; + } + + public static String getAdditionalPropertyValueOrEmpty(Property additionalProperty) { + if (additionalProperty != null) { + return additionalProperty.getValue(); + } + return ""; + } +} diff --git a/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloXMLTableDefinition.java b/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloXMLTableDefinition.java new file mode 100644 index 0000000000..88677b3185 --- /dev/null +++ b/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloXMLTableDefinition.java @@ -0,0 +1,144 @@ +package org.jooq.util.vertabelo; + +import java.sql.SQLException; +import java.util.ArrayList; +import java.util.List; + +import org.jooq.DataType; +import org.jooq.impl.DefaultDataType; +import org.jooq.tools.StringUtils; +import org.jooq.util.AbstractTableDefinition; +import org.jooq.util.ColumnDefinition; +import org.jooq.util.DataTypeDefinition; +import org.jooq.util.DefaultColumnDefinition; +import org.jooq.util.DefaultDataTypeDefinition; +import org.jooq.util.SchemaDefinition; +import org.jooq.util.vertabelo.jaxb.Column; +import org.jooq.util.vertabelo.jaxb.Property; +import org.jooq.util.vertabelo.jaxb.Table; +import org.jooq.util.vertabelo.jaxb.View; +import org.jooq.util.vertabelo.jaxb.ViewColumn; + +/** + * Definition of the Vertabelo XML Table + * + * @author Michał Kołodziejski + */ +public class VertabeloXMLTableDefinition extends AbstractTableDefinition { + + protected Table table; + protected View view; + + public VertabeloXMLTableDefinition(SchemaDefinition schema, Table table) { + super(schema, table.getName(), ""); + + this.table = table; + } + + public VertabeloXMLTableDefinition(SchemaDefinition schema, View view) { + super(schema, view.getName(), ""); + + this.view = view; + } + + + @Override + protected List getElements0() throws SQLException { + if(table != null) { + // table + return getTableElements(); + + } else { + // view + return getViewElements(); + } + } + + + protected List getTableElements() { + List result = new ArrayList(); + + String schemaName = getSchemaName(); + SchemaDefinition schema = getDatabase().getSchema(schemaName); + + int position = 0; + for(Column column : table.getColumns()) { + ++position; + + // convert data type + DataType dataType = DefaultDataType.getDataType(getDialect(), column.getType()); + + DataTypeDefinition type = new DefaultDataTypeDefinition( + getDatabase(), + schema, + dataType.getTypeName(), + dataType.hasLength() ? dataType.length() : null, + dataType.hasPrecision() ? dataType.precision() : null, + dataType.hasScale() ? dataType.scale() : null, + column.isNullable(), + !StringUtils.isEmpty(column.getDefaultValue())); + + ColumnDefinition columnDefinition = new DefaultColumnDefinition( + this, + column.getName(), + position, + type, + false, + column.getDescription()); + + result.add(columnDefinition); + } + + return result; + } + + + protected List getViewElements() { + List result = new ArrayList(); + + String schemaName = getSchemaName(); + SchemaDefinition schema = getDatabase().getSchema(schemaName); + + int position = 0; + for(ViewColumn column : view.getViewColumns()) { + ++position; + + // convert data type + DataType dataType = DefaultDataType.getDataType(getDialect(), column.getType()); + + DataTypeDefinition type = new DefaultDataTypeDefinition( + getDatabase(), + schema, + dataType.getTypeName(), + dataType.hasLength() ? dataType.length() : null, + dataType.hasPrecision() ? dataType.precision() : null, + dataType.hasScale() ? dataType.scale() : null, + true, + false); + + ColumnDefinition columnDefinition = new DefaultColumnDefinition( + this, + column.getName(), + position, + type, + false, + column.getDescription()); + + result.add(columnDefinition); + } + + return result; + } + + protected String getSchemaName() { + Property additionalProperty; + + if(table != null) { + additionalProperty = VertabeloXMLDatabase.findAdditionalProperty("Schema", table.getProperties()); + } else { + additionalProperty = VertabeloXMLDatabase.findAdditionalProperty("Schema", view.getProperties()); + } + + return VertabeloXMLDatabase.getAdditionalPropertyValueOrEmpty(additionalProperty); + } +} diff --git a/jOOQ-meta/src/main/resources/xjb/vertabelo/binding.xjb b/jOOQ-meta/src/main/resources/xjb/vertabelo/binding.xjb new file mode 100644 index 0000000000..03fefad849 --- /dev/null +++ b/jOOQ-meta/src/main/resources/xjb/vertabelo/binding.xjb @@ -0,0 +1,27 @@ + + + + + + + + + + + + + + + + + + + + \ No newline at end of file diff --git a/jOOQ-meta/src/main/resources/xsd/vertabelo-2.1.xsd b/jOOQ-meta/src/main/resources/xsd/vertabelo-2.1.xsd new file mode 100644 index 0000000000..e38cf7de9d --- /dev/null +++ b/jOOQ-meta/src/main/resources/xsd/vertabelo-2.1.xsd @@ -0,0 +1,439 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + From e65f62d833286ea0689748e3be47dabe94a9f511 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Ko=C5=82odziejski?= Date: Fri, 7 Nov 2014 16:32:33 +0100 Subject: [PATCH 2/2] Integration with Vertabelo - possibility to download Vertabelo XML model directly from the API --- .../util/vertabelo/VertabeloAPIDatabase.java | 130 ++++++++ .../jooq/util/vertabelo/util/Base64Coder.java | 283 ++++++++++++++++++ 2 files changed, 413 insertions(+) create mode 100644 jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloAPIDatabase.java create mode 100644 jOOQ-meta/src/main/java/org/jooq/util/vertabelo/util/Base64Coder.java diff --git a/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloAPIDatabase.java b/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloAPIDatabase.java new file mode 100644 index 0000000000..1585266190 --- /dev/null +++ b/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/VertabeloAPIDatabase.java @@ -0,0 +1,130 @@ +package org.jooq.util.vertabelo; + +import java.io.BufferedReader; +import java.io.ByteArrayInputStream; +import java.io.IOException; +import java.io.InputStreamReader; +import java.io.UnsupportedEncodingException; +import java.net.MalformedURLException; +import java.net.URL; + +import javax.net.ssl.HttpsURLConnection; +import javax.xml.bind.JAXB; + +import org.jooq.tools.JooqLogger; +import org.jooq.tools.StringUtils; +import org.jooq.util.vertabelo.jaxb.DatabaseModel; +import org.jooq.util.vertabelo.util.Base64Coder; + +/** + * The Vertabelo API Database + * + * @author Michał Kołodziejski + */ +public class VertabeloAPIDatabase extends VertabeloXMLDatabase { + + private static final JooqLogger log = JooqLogger.getLogger(VertabeloAPIDatabase.class); + + protected static final String API_URL_PREFIX = "https://my.vertabelo.com/api/xml/"; + + // codegen properties + protected static final String API_TOKEN_PARAM = "api-token"; + protected static final String MODEL_ID_PARAM = "model-id"; + protected static final String TAG_NAME_PARAM = "tag-name"; + + protected String apiToken; + protected String modelId; + protected String tagName; + + + @Override + protected DatabaseModel databaseModel() { + if(databaseModel == null) { + readSettings(); + String xml = getXMLFromAPI(); + + ByteArrayInputStream stream; + try { + stream = new ByteArrayInputStream(xml.getBytes("UTF-8")); + } + catch (UnsupportedEncodingException e) { + throw new RuntimeException(e); + } + databaseModel = JAXB.unmarshal(stream, DatabaseModel.class); + } + + return databaseModel; + } + + + protected void readSettings() { + apiToken = getProperties().getProperty(API_TOKEN_PARAM); + if (StringUtils.isEmpty(apiToken)) { + throw new IllegalStateException("Lack of \"" + API_TOKEN_PARAM + "\" parameter."); + } + + + modelId = getProperties().getProperty(MODEL_ID_PARAM); + if (StringUtils.isEmpty(modelId)) { + throw new IllegalStateException("Lack of \"" + MODEL_ID_PARAM + "\" parameter."); + } + + tagName = getProperties().getProperty(TAG_NAME_PARAM); + } + + + protected String getXMLFromAPI() { + String xml = null; + + String apiUrl = getApiUrl(); + + try { + log.info("Creating connection to Vertabelo server..."); + + URL url = new URL(apiUrl); + HttpsURLConnection connection = (HttpsURLConnection) url.openConnection(); + connection.setConnectTimeout(5000); + connection.setReadTimeout(5000); + + // authorization data + String encodedAuthData = Base64Coder.encodeString(apiToken + ":"); + connection.addRequestProperty("Authorization", "Basic " + encodedAuthData); + + + // do request + int responseCode=connection.getResponseCode(); + log.info("Response code: " + responseCode); + + if (responseCode != HttpsURLConnection.HTTP_OK) { + throw new RuntimeException("Request failed with status code: " + responseCode); + } + + // read response + String response = ""; + String line; + BufferedReader br = new BufferedReader(new InputStreamReader(connection.getInputStream())); + while ((line = br.readLine()) != null) { + response += line; + } + + xml = response; + + } catch (MalformedURLException e) { + throw new RuntimeException(e); + } catch (IOException e) { + throw new RuntimeException(e); + } + + return xml; + } + + + protected String getApiUrl() { + String apiUrl = API_URL_PREFIX + modelId; + if(!StringUtils.isEmpty(tagName)) { + apiUrl += "/" + tagName; + } + + return apiUrl; + } +} diff --git a/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/util/Base64Coder.java b/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/util/Base64Coder.java new file mode 100644 index 0000000000..7f9b980de5 --- /dev/null +++ b/jOOQ-meta/src/main/java/org/jooq/util/vertabelo/util/Base64Coder.java @@ -0,0 +1,283 @@ +//Copyright 2003-2010 Christian d'Heureuse, Inventec Informatik AG, Zurich, Switzerland +//www.source-code.biz, www.inventec.ch/chdh +// +//This module is multi-licensed and may be used under the terms +//of any of the following licenses: +// +//EPL, Eclipse Public License, V1.0 or later, http://www.eclipse.org/legal +//LGPL, GNU Lesser General Public License, V2.1 or later, http://www.gnu.org/licenses/lgpl.html +//GPL, GNU General Public License, V2 or later, http://www.gnu.org/licenses/gpl.html +//AGPL, GNU Affero General Public License V3 or later, http://www.gnu.org/licenses/agpl.html +//AL, Apache License, V2.0 or later, http://www.apache.org/licenses +//BSD, BSD License, http://www.opensource.org/licenses/bsd-license.php +//MIT, MIT License, http://www.opensource.org/licenses/MIT +// +//Please contact the author if you need another license. +//This module is provided "as is", without warranties of any kind. +// +//Project home page: www.source-code.biz/base64coder/java + +package org.jooq.util.vertabelo.util; + +/** + * A Base64 encoder/decoder. + *

+ * This class is used to encode and decode data in Base64 format as described in + * RFC 1521. + * + * @author Christian d'Heureuse, Inventec Informatik AG, Zurich, Switzerland, + * www.source-code.biz + */ +public class Base64Coder { + + // The line separator string of the operating system. + private static final String systemLineSeparator = System.getProperty("line.separator"); + + // Mapping table from 6-bit nibbles to Base64 characters. + private static final char[] map1 = new char[64]; + static { + int i = 0; + for (char c = 'A'; c <= 'Z'; c++) + map1[i++] = c; + for (char c = 'a'; c <= 'z'; c++) + map1[i++] = c; + for (char c = '0'; c <= '9'; c++) + map1[i++] = c; + map1[i++] = '+'; + map1[i++] = '/'; + } + + // Mapping table from Base64 characters to 6-bit nibbles. + private static final byte[] map2 = new byte[128]; + static { + for (int i = 0; i < map2.length; i++) + map2[i] = -1; + for (int i = 0; i < 64; i++) + map2[map1[i]] = (byte) i; + } + + /** + * Encodes a string into Base64 format. No blanks or line breaks are + * inserted. + * + * @param s A String to be encoded. + * @return A String containing the Base64 encoded data. + */ + public static String encodeString(String s) { + return new String(encode(s.getBytes())); + } + + /** + * Encodes a byte array into Base 64 format and breaks the output into lines + * of 76 characters. This method is compatible with + * sun.misc.BASE64Encoder.encodeBuffer(byte[]). + * + * @param in An array containing the data bytes to be encoded. + * @return A String containing the Base64 encoded data, broken into lines. + */ + public static String encodeLines(byte[] in) { + return encodeLines(in, 0, in.length, 76, systemLineSeparator); + } + + /** + * Encodes a byte array into Base 64 format and breaks the output into + * lines. + * + * @param in An array containing the data bytes to be encoded. + * @param iOff Offset of the first byte in in to be processed. + * @param iLen Number of bytes to be processed in in, starting + * at iOff. + * @param lineLen Line length for the output data. Should be a multiple of + * 4. + * @param lineSeparator The line separator to be used to separate the output + * lines. + * @return A String containing the Base64 encoded data, broken into lines. + */ + public static String encodeLines(byte[] in, int iOff, int iLen, int lineLen, String lineSeparator) { + int blockLen = (lineLen * 3) / 4; + if (blockLen <= 0) + throw new IllegalArgumentException(); + int lines = (iLen + blockLen - 1) / blockLen; + int bufLen = ((iLen + 2) / 3) * 4 + lines * lineSeparator.length(); + StringBuilder buf = new StringBuilder(bufLen); + int ip = 0; + while (ip < iLen) { + int l = Math.min(iLen - ip, blockLen); + buf.append(encode(in, iOff + ip, l)); + buf.append(lineSeparator); + ip += l; + } + return buf.toString(); + } + + /** + * Encodes a byte array into Base64 format. No blanks or line breaks are + * inserted in the output. + * + * @param in An array containing the data bytes to be encoded. + * @return A character array containing the Base64 encoded data. + */ + public static char[] encode(byte[] in) { + return encode(in, 0, in.length); + } + + /** + * Encodes a byte array into Base64 format. No blanks or line breaks are + * inserted in the output. + * + * @param in An array containing the data bytes to be encoded. + * @param iLen Number of bytes to process in in. + * @return A character array containing the Base64 encoded data. + */ + public static char[] encode(byte[] in, int iLen) { + return encode(in, 0, iLen); + } + + /** + * Encodes a byte array into Base64 format. No blanks or line breaks are + * inserted in the output. + * + * @param in An array containing the data bytes to be encoded. + * @param iOff Offset of the first byte in in to be processed. + * @param iLen Number of bytes to process in in, starting at + * iOff. + * @return A character array containing the Base64 encoded data. + */ + public static char[] encode(byte[] in, int iOff, int iLen) { + int oDataLen = (iLen * 4 + 2) / 3; // output length without padding + int oLen = ((iLen + 2) / 3) * 4; // output length including padding + char[] out = new char[oLen]; + int ip = iOff; + int iEnd = iOff + iLen; + int op = 0; + while (ip < iEnd) { + int i0 = in[ip++] & 0xff; + int i1 = ip < iEnd ? in[ip++] & 0xff : 0; + int i2 = ip < iEnd ? in[ip++] & 0xff : 0; + int o0 = i0 >>> 2; + int o1 = ((i0 & 3) << 4) | (i1 >>> 4); + int o2 = ((i1 & 0xf) << 2) | (i2 >>> 6); + int o3 = i2 & 0x3F; + out[op++] = map1[o0]; + out[op++] = map1[o1]; + out[op] = op < oDataLen ? map1[o2] : '='; + op++; + out[op] = op < oDataLen ? map1[o3] : '='; + op++; + } + return out; + } + + /** + * Decodes a string from Base64 format. No blanks or line breaks are allowed + * within the Base64 encoded input data. + * + * @param s A Base64 String to be decoded. + * @return A String containing the decoded data. + * @throws IllegalArgumentException If the input is not valid Base64 encoded + * data. + */ + public static String decodeString(String s) { + return new String(decode(s)); + } + + /** + * Decodes a byte array from Base64 format and ignores line separators, tabs + * and blanks. CR, LF, Tab and Space characters are ignored in the input + * data. This method is compatible with + * sun.misc.BASE64Decoder.decodeBuffer(String). + * + * @param s A Base64 String to be decoded. + * @return An array containing the decoded data bytes. + * @throws IllegalArgumentException If the input is not valid Base64 encoded + * data. + */ + public static byte[] decodeLines(String s) { + char[] buf = new char[s.length()]; + int p = 0; + for (int ip = 0; ip < s.length(); ip++) { + char c = s.charAt(ip); + if (c != ' ' && c != '\r' && c != '\n' && c != '\t') + buf[p++] = c; + } + return decode(buf, 0, p); + } + + /** + * Decodes a byte array from Base64 format. No blanks or line breaks are + * allowed within the Base64 encoded input data. + * + * @param s A Base64 String to be decoded. + * @return An array containing the decoded data bytes. + * @throws IllegalArgumentException If the input is not valid Base64 encoded + * data. + */ + public static byte[] decode(String s) { + return decode(s.toCharArray()); + } + + /** + * Decodes a byte array from Base64 format. No blanks or line breaks are + * allowed within the Base64 encoded input data. + * + * @param in A character array containing the Base64 encoded data. + * @return An array containing the decoded data bytes. + * @throws IllegalArgumentException If the input is not valid Base64 encoded + * data. + */ + public static byte[] decode(char[] in) { + return decode(in, 0, in.length); + } + + /** + * Decodes a byte array from Base64 format. No blanks or line breaks are + * allowed within the Base64 encoded input data. + * + * @param in A character array containing the Base64 encoded data. + * @param iOff Offset of the first character in in to be + * processed. + * @param iLen Number of characters to process in in, starting + * at iOff. + * @return An array containing the decoded data bytes. + * @throws IllegalArgumentException If the input is not valid Base64 encoded + * data. + */ + public static byte[] decode(char[] in, int iOff, int iLen) { + if (iLen % 4 != 0) + throw new IllegalArgumentException("Length of Base64 encoded input string is not a multiple of 4."); + while (iLen > 0 && in[iOff + iLen - 1] == '=') + iLen--; + int oLen = (iLen * 3) / 4; + byte[] out = new byte[oLen]; + int ip = iOff; + int iEnd = iOff + iLen; + int op = 0; + while (ip < iEnd) { + int i0 = in[ip++]; + int i1 = in[ip++]; + int i2 = ip < iEnd ? in[ip++] : 'A'; + int i3 = ip < iEnd ? in[ip++] : 'A'; + if (i0 > 127 || i1 > 127 || i2 > 127 || i3 > 127) + throw new IllegalArgumentException("Illegal character in Base64 encoded data."); + int b0 = map2[i0]; + int b1 = map2[i1]; + int b2 = map2[i2]; + int b3 = map2[i3]; + if (b0 < 0 || b1 < 0 || b2 < 0 || b3 < 0) + throw new IllegalArgumentException("Illegal character in Base64 encoded data."); + int o0 = (b0 << 2) | (b1 >>> 4); + int o1 = ((b1 & 0xf) << 4) | (b2 >>> 2); + int o2 = ((b2 & 3) << 6) | b3; + out[op++] = (byte) o0; + if (op < oLen) + out[op++] = (byte) o1; + if (op < oLen) + out[op++] = (byte) o2; + } + return out; + } + + // Dummy constructor. + private Base64Coder() {} + +} // end class Base64Coder