CommitteeGhantChartManagerImpl.java

/*
 * Copyright 2014 James Pether Sörling
 *
 * Licensed 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.
 *
 *	$Id$
 *  $HeadURL$
*/
package com.hack23.cia.web.impl.ui.application.views.common.chartfactory.impl;

import java.util.Comparator;
import java.util.Date;
import java.util.Locale;
import java.util.function.Function;

import org.springframework.stereotype.Service;

import com.hack23.cia.model.internal.application.data.committee.impl.ViewRiksdagenCommitteeRoleMember;
import com.hack23.cia.web.impl.ui.application.views.common.chartfactory.api.CommitteeGhantChartManager;

/**
 * The Class CommitteeGhantChartManagerImpl.
 */
@Service
public final class CommitteeGhantChartManagerImpl
		extends AbstractGhantChartManagerImpl<ViewRiksdagenCommitteeRoleMember> implements CommitteeGhantChartManager {

	/**
	 * Instantiates a new committee ghant chart manager impl.
	 */
	public CommitteeGhantChartManagerImpl() {
		super();
	}

	@Override
	protected Comparator<ViewRiksdagenCommitteeRoleMember> getComparator() {
		return (o1, o2) -> o1.getFromDate().compareTo(o2.getFromDate());
	}

	@Override
	protected Function<ViewRiksdagenCommitteeRoleMember, String> getRoleMapping() {
		return new RoleMapping();
	}

	@Override
	protected StepMapping<ViewRiksdagenCommitteeRoleMember> getStepMapping() {
		return new StepMapping<ViewRiksdagenCommitteeRoleMember>() {

			@Override
			public Date getFromDate(final ViewRiksdagenCommitteeRoleMember t) {
				return t.getFromDate();
			}

			@Override
			public Date getToDate(final ViewRiksdagenCommitteeRoleMember t) {
				return t.getToDate();
			}

			@Override
			public String getRoleCode(final ViewRiksdagenCommitteeRoleMember t) {
				return t.getRoleCode();
			}

			@Override
			public String getOrg(final ViewRiksdagenCommitteeRoleMember t) {
				return t.getDetail();
			}

			@Override
			public String getParty(final ViewRiksdagenCommitteeRoleMember t) {
				return t.getParty();
			}

			@Override
			public String getBackgroundColor(final ViewRiksdagenCommitteeRoleMember t) {
				String color;

				if (t.getRoleCode().toLowerCase(Locale.ENGLISH).contains("suppleant")) {
					color = "ded858";
				} else if (t.getRoleCode().toLowerCase(Locale.ENGLISH).contains("ordförande")) {
					color = "3271c8";
				} else {
					color = "0eab76";
				}

				return color;
			}

			@Override
			public Object getFirstName(final ViewRiksdagenCommitteeRoleMember t) {
				return t.getFirstName();
			}

			@Override
			public Object getLastName(final ViewRiksdagenCommitteeRoleMember t) {
				return t.getLastName();
			}

		};
	}

	/**
	 * The Class RoleMapping.
	 */
	private static final class RoleMapping implements Function<ViewRiksdagenCommitteeRoleMember, String> {

		@Override
		public String apply(final ViewRiksdagenCommitteeRoleMember t) {
			return t.getRoleCode();
		}
	}

}