ScreenEntityTablesBinder.java

/*******************************************************************************
 * Copyright (c) 2012 OpenLegacy Inc.
 * All rights reserved. This program and the accompanying materials 
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     OpenLegacy Inc. - initial API and implementation
 *******************************************************************************/
package org.openlegacy.terminal.support.binders;

import org.apache.commons.lang.ArrayUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.openlegacy.FieldFormatter;
import org.openlegacy.terminal.ScreenEntityBinder;
import org.openlegacy.terminal.ScreenPojoFieldAccessor;
import org.openlegacy.terminal.TerminalField;
import org.openlegacy.terminal.TerminalPosition;
import org.openlegacy.terminal.TerminalSendAction;
import org.openlegacy.terminal.TerminalSnapshot;
import org.openlegacy.terminal.definitions.ScreenTableDefinition;
import org.openlegacy.terminal.definitions.ScreenTableDefinition.ScreenColumnDefinition;
import org.openlegacy.terminal.providers.TablesDefinitionProvider;
import org.openlegacy.terminal.support.SimpleTerminalPosition;
import org.openlegacy.terminal.utils.SimpleScreenPojoFieldAccessor;
import org.openlegacy.utils.ReflectionUtil;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.inject.Inject;

@Component
public class ScreenEntityTablesBinder implements ScreenEntityBinder {

	@Inject
	private TablesDefinitionProvider tablesDefinitionProvider;

	@Inject
	private FieldFormatter fieldFormatter;

	private final static Log logger = LogFactory.getLog(ScreenEntityTablesBinder.class);

	public void populateEntity(Object screenEntity, TerminalSnapshot terminalSnapshot) {

		ScreenPojoFieldAccessor fieldAccessor = new SimpleScreenPojoFieldAccessor(screenEntity);

		Map<String, ScreenTableDefinition> tableDefinitions = tablesDefinitionProvider.getTableDefinitions(screenEntity.getClass());

		Set<String> tableFieldNames = tableDefinitions.keySet();

		if (logger.isDebugEnabled()) {
			logger.debug(MessageFormat.format("Found {0} tables to bind to entity {1}: {2}", tableFieldNames.size(),
					screenEntity.getClass().getName(), ArrayUtils.toString(tableFieldNames.toArray())));
		}

		for (String tableFieldName : tableFieldNames) {

			ScreenTableDefinition tableDefinition = tableDefinitions.get(tableFieldName);
			List<Object> rows = new ArrayList<Object>();

			List<ScreenColumnDefinition> columnDefinitions = tableDefinition.getColumnDefinitions();
			int startRow = tableDefinition.getStartRow();
			int endRow = tableDefinition.getEndRow();
			for (int currentRow = startRow; currentRow <= endRow; currentRow += tableDefinition.getRowGaps()) {

				Object row = ReflectionUtil.newInstance(tableDefinition.getTableClass());
				ScreenPojoFieldAccessor rowAccessor = new SimpleScreenPojoFieldAccessor(row);

				boolean keyIsEmpty = false;

				for (ScreenColumnDefinition columnDefinition : columnDefinitions) {
					TerminalPosition position = SimpleTerminalPosition.newInstance(currentRow + columnDefinition.getRowsOffset(),
							columnDefinition.getStartColumn());
					String cellText = getCellContent(terminalSnapshot, position, columnDefinition);
					if (columnDefinition.isKey() && cellText.length() == 0) {
						if (logger.isDebugEnabled()) {
							logger.debug(MessageFormat.format(
									"Key field {0} is empty in row {1}. Aborting table rows collecting",
									columnDefinition.getName(), position.getRow()));
						}
						keyIsEmpty = true;
					}
					rowAccessor.setFieldValue(columnDefinition.getName(), cellText);

					TerminalField terminalField = terminalSnapshot.getField(position);
					rowAccessor.setTerminalField(columnDefinition.getName(), terminalField);
				}
				if (!keyIsEmpty) {
					rows.add(row);
				}
			}
			fieldAccessor.setFieldValue(tableFieldName, rows);
		}
	}

	private String getCellContent(TerminalSnapshot terminalSnapshot, TerminalPosition position,
			ScreenColumnDefinition columnDefinition) {
		int length = columnDefinition.getEndColumn() - columnDefinition.getStartColumn() + 1;
		String columnText = terminalSnapshot.getLogicalText(position, length);
		columnText = fieldFormatter.format(columnText);
		return columnText;
	}

	public void setFieldFormatter(FieldFormatter fieldFormatter) {
		this.fieldFormatter = fieldFormatter;
	}

	public void populateSendAction(TerminalSendAction sendAction, TerminalSnapshot terminalScreen, Object entity) {
		ScreenPojoFieldAccessor fieldAccessor = new SimpleScreenPojoFieldAccessor(entity);

		Map<String, ScreenTableDefinition> tableDefinitions = tablesDefinitionProvider.getTableDefinitions(entity.getClass());

		Set<String> tableFieldNames = tableDefinitions.keySet();

		for (String tableFieldName : tableFieldNames) {
			ScreenTableDefinition tableDefinition = tableDefinitions.get(tableFieldName);
			List<?> rows = (List<?>)fieldAccessor.getFieldValue(tableFieldName);
			// TODO send tables values
			if (rows == null) {
				continue;
			}
			int rowCount = 0;
			for (Object row : rows) {
				List<ScreenColumnDefinition> columnDefinitions = tableDefinition.getColumnDefinitions();
				fieldAccessor = new SimpleScreenPojoFieldAccessor(row);
				for (ScreenColumnDefinition columnDefinition : columnDefinitions) {
					if (columnDefinition.isEditable()) {
						Object value = fieldAccessor.getFieldValue(columnDefinition.getName());
						if (value == null) {
							continue;
						}
						String valueString = value.toString();
						if (StringUtils.hasLength(valueString)) {
							int screenRow = tableDefinition.getStartRow() + rowCount;
							TerminalField terminalField = terminalScreen.getField(SimpleTerminalPosition.newInstance(screenRow,
									columnDefinition.getStartColumn()));
							terminalField.setValue(valueString);
							sendAction.getModifiedFields().add(terminalField);
						}
					}

				}
				rowCount++;
			}

		}

	}
}