content: /*****************************************************************
   *   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
   *
   *    https://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.cayenne.dbsync.merge.token.db;
  
  import org.apache.cayenne.access.DataNode;
  import org.apache.cayenne.dba.DbAdapter;
  import org.apache.cayenne.dbsync.merge.context.MergerContext;
  import org.apache.cayenne.dbsync.merge.factory.MergerTokenFactory;
  import org.apache.cayenne.dbsync.merge.token.MergerToken;
  import org.apache.cayenne.map.DbAttribute;
  import org.apache.cayenne.map.DbEntity;
  import org.apache.cayenne.validation.SimpleValidationFailure;
  
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.List;
  
  public class CreateTableToDb extends AbstractToDbToken.Entity {
  
      public CreateTableToDb(DbEntity entity) {
          super("Create Table", 40, entity);
      }
  
      @Override
      public List<String> createSql(DbAdapter adapter) {
          List<String> sqls = new ArrayList<>();
          if(needAutoPkSupport()) {
              sqls.addAll(adapter.getPkGenerator().createAutoPkStatements(
                      Collections.singletonList(getEntity())));
          }
          sqls.add(adapter.createTable(getEntity()));
          return sqls;
      }
  
      @Override
      public void execute(MergerContext mergerContext) {
          try {
              DataNode node = mergerContext.getDataNode();
              DbAdapter adapter = node.getAdapter();
              if(needAutoPkSupport()) {
                  adapter.getPkGenerator().createAutoPk(
                          node,
                          Collections.singletonList(getEntity()));
              }
              executeSql(mergerContext, adapter.createTable(getEntity()));
          }
          catch (Exception e) {
              mergerContext.getValidationResult().addFailure(
                      new SimpleValidationFailure(this, e.getMessage()));
          }
      }
  
      private boolean needAutoPkSupport() {
          DbEntity entity = getEntity();
          if(entity.getPrimaryKeyGenerator() != null) {
              return false;
          }
  
          for(DbAttribute attribute : entity.getPrimaryKeys()) {
              if(attribute.isGenerated()) {
                  return false;
              }
          }
  
          return true;
      }
  
      public MergerToken createReverse(MergerTokenFactory factory) {
          return factory.createDropTableToModel(getEntity());
      }
  
  }
  