Codion and Complexity

package is.codion.demos.store;

import is.codion.common.db.database.Database;
import is.codion.common.user.User;
import is.codion.framework.db.EntityConnectionProvider;
import is.codion.framework.db.local.LocalEntityConnectionProvider;
import is.codion.framework.domain.DomainModel;
import is.codion.framework.domain.DomainType;
import is.codion.framework.domain.entity.EntityType;
import is.codion.framework.domain.entity.attribute.Column;
import is.codion.framework.domain.entity.attribute.Column.Generator;
import is.codion.framework.domain.entity.attribute.ForeignKey;
import is.codion.swing.framework.model.SwingEntityApplicationModel;
import is.codion.swing.framework.model.SwingEntityEditModel;
import is.codion.swing.framework.model.SwingEntityModel;
import is.codion.swing.framework.ui.EntityApplication;
import is.codion.swing.framework.ui.EntityApplicationPanel;
import is.codion.swing.framework.ui.EntityEditPanel;
import is.codion.swing.framework.ui.EntityPanel;

import java.util.List;

import static is.codion.framework.domain.DomainType.domainType;

public final class StoreAppPanel extends EntityApplicationPanel<SwingEntityApplicationModel> {

  public static final DomainType DOMAIN = domainType("Store");

  public interface Customer {
    EntityType TYPE = DOMAIN.entityType("store.customer");

    Column<Long> ID = TYPE.longColumn("id");
    Column<String> FIRST_NAME = TYPE.stringColumn("first_name");
  }

  public interface Address {
    EntityType TYPE = DOMAIN.entityType("store.address");

    Column<Long> ID = TYPE.longColumn("id");
    Column<String> STREET = TYPE.stringColumn("street");
    Column<Long> CUSTOMER_ID = TYPE.longColumn("customer_id");
    ForeignKey CUSTOMER_FK = TYPE.foreignKey("customer_fk", CUSTOMER_ID, Customer.ID);
  }

  private static class Store extends DomainModel {

    private Store() {
      super(DOMAIN);
      add(Customer.TYPE.as()
              .attributes(
                      Customer.ID.as()
                              .primaryKey()
                              .generator(Generator.identity()),
                      Customer.FIRST_NAME.as()
                              .column()
                              .caption("First name")
                              .nullable(false)
                              .maximumLength(40))
              .caption("Customer")
              .build());
      
      add(Address.TYPE.as()
              .attributes(
                      Address.ID.as()
                              .primaryKey()
                              .generator(Generator.identity()),
                      Address.STREET.as()
                              .column()
                              .nullable(false)
                              .maximumLength(120)
                              .caption("Street"),
                      Address.CUSTOMER_ID.as()
                              .column(),
                      Address.CUSTOMER_FK.as()
                              .foreignKey()
                              .caption("Customer"))
              .caption("Address")
              .build());
    }
  }

  public StoreAppPanel(SwingEntityApplicationModel applicationModel) {
    super(applicationModel, createEntityPanels(applicationModel), List.of());
  }

  private static List<EntityPanel> createEntityPanels(SwingEntityApplicationModel applicationModel) {
    SwingEntityModel customerModel = applicationModel.models().get(Customer.TYPE);
    SwingEntityModel addressModel = customerModel.detail().get(Address.TYPE);

    EntityPanel customerPanel = new EntityPanel(customerModel, new CustomerEditPanel(customerModel.editModel()));
    EntityPanel addressPanel = new EntityPanel(addressModel, new AddressEditPanel(addressModel.editModel()));
    customerPanel.detail().add(addressPanel);

    return List.of(customerPanel);
  }

  private static SwingEntityApplicationModel createApplicationModel(EntityConnectionProvider connectionProvider) {
    SwingEntityModel customerModel = new SwingEntityModel(Customer.TYPE, connectionProvider);
    SwingEntityModel addressModel = new SwingEntityModel(Address.TYPE, connectionProvider);
    customerModel.detail().add(addressModel);

    return new SwingEntityApplicationModel(connectionProvider, List.of(customerModel));
  }

  private static class CustomerEditPanel extends EntityEditPanel {

    public CustomerEditPanel(SwingEntityEditModel editModel) {
      super(editModel);
    }

    @Override
    protected void initializeUI() {
      create().textField(Customer.FIRST_NAME);
      addInputPanel(Customer.FIRST_NAME);
    }
  }

  private static class AddressEditPanel extends EntityEditPanel {

    public AddressEditPanel(SwingEntityEditModel editModel) {
      super(editModel);
    }

    @Override
    protected void initializeUI() {
      create().comboBox(Address.CUSTOMER_FK);
      create().textField(Address.STREET);
      addInputPanel(Address.CUSTOMER_FK);
      addInputPanel(Address.STREET);
    }
  }

  public static void main(String[] args) {
    Database.URL.set("jdbc:h2:mem:h2db");
    Database.INIT_SCRIPTS.set("src/main/sql/create_schema.sql");
    EntityApplication.builder(SwingEntityApplicationModel.class, StoreAppPanel.class)
            .connectionProvider(LocalEntityConnectionProvider.builder()
                    .domain(new Store())
                    .user(User.parse("scott:tiger"))
                    .build())
            .model(StoreAppPanel::createApplicationModel)
            .start();
  }
}