/*
 * eGov suite of products aim to improve the internal efficiency,transparency,
 *    accountability and the service delivery of the government  organizations.
 *
 *     Copyright (C) <2015>  eGovernments Foundation
 *
 *     The updated version of eGov suite of products as by eGovernments Foundation
 *     is available at http://www.egovernments.org
 *
 *     This program is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     any later version.
 *
 *     This program is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU General Public License for more details.
 *
 *     You should have received a copy of the GNU General Public License
 *     along with this program. If not, see http://www.gnu.org/licenses/ or
 *     http://www.gnu.org/licenses/gpl.html .
 *
 *     In addition to the terms of the GPL license to be adhered to in using this
 *     program, the following additional terms are to be complied with:
 *
 *         1) All versions of this program, verbatim or modified must carry this
 *            Legal Notice.
 *
 *         2) Any misrepresentation of the origin of the material is prohibited. It
 *            is required that all modified versions of this material be marked in
 *            reasonable ways as different from the original version.
 *
 *         3) This license does not grant any rights to any user of the program
 *            with regards to rights under trademark law for use of the trade names
 *            or trademarks of eGovernments Foundation.
 *
 *   In case of any queries, you can reach eGovernments Foundation at contact@egovernments.org.
 */
package org.egov.wtms.masters.service;

import org.egov.wtms.masters.entity.WaterSource;
import org.egov.wtms.masters.repository.WaterSourceRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.List;

@Service
@Transactional(readOnly = true)
public class WaterSourceService {

    private final WaterSourceRepository waterSourceRepository;

    @Autowired
    public WaterSourceService(final WaterSourceRepository waterSourceRepository) {
        this.waterSourceRepository = waterSourceRepository;
    }

    public WaterSource findOne(final Long waterSourceId) {
        return waterSourceRepository.findOne(waterSourceId);
    }

    @Transactional
    public WaterSource createWaterSource(final WaterSource waterSource) {
        waterSource.setActive(true);
        return waterSourceRepository.save(waterSource);
    }

    @Transactional
    public void updateWaterSource(final WaterSource waterSource) {
        waterSourceRepository.save(waterSource);
    }

    public List<WaterSource> findAll() {
        return waterSourceRepository.findAll(new Sort(Sort.Direction.DESC, "waterSourceType"));
    }

    public WaterSource findByWaterSourceType(final String waterSourceType) {
        return waterSourceRepository.findByWaterSourceType(waterSourceType);
    }

    public WaterSource load(final Long id) {
        return waterSourceRepository.getOne(id);
    }

    public Page<WaterSource> getListOfWaterSources(final Integer pageNumber, final Integer pageSize) {
        final Pageable pageable = new PageRequest(pageNumber - 1, pageSize, Sort.Direction.ASC, "waterSourceType");
        return waterSourceRepository.findAll(pageable);
    }

    public WaterSource findByCode(final String code) {
        return waterSourceRepository.findByCode(code);
    }

    public WaterSource findByCodeIgnoreCase(final String code) {
        return waterSourceRepository.findByCodeIgnoreCase(code);
    }

    public WaterSource findByWaterSourceTypeIgnoreCase(final String waterSourceType) {
        return waterSourceRepository.findByWaterSourceTypeIgnoreCase(waterSourceType);
    }

    public List<WaterSource> getAllActiveWaterSourceTypes() {
        return waterSourceRepository.findByActiveTrueOrderByWaterSourceTypeAsc();
    }

    public WaterSource findByCodeAndWaterSourceType(final String code, final String waterSourceType) {
        return waterSourceRepository.findByCodeAndWaterSourceType(code, waterSourceType);
    }

    public List<WaterSource> getWaterSourceListForRest() {
        final List<WaterSource> waterSourceList = waterSourceRepository.findByActiveTrueOrderByWaterSourceTypeAsc();
        final List<WaterSource> prepareListForRest = new ArrayList<WaterSource>(0);

        for (final WaterSource waterSource : waterSourceList) {
            final WaterSource waterSourceRest = new WaterSource();
            waterSourceRest.setCode(waterSource.getCode());
            waterSourceRest.setWaterSourceType(waterSource.getWaterSourceType());
            waterSourceRest.setActive(waterSource.isActive());
            prepareListForRest.add(waterSourceRest);
        }
        return prepareListForRest;
    }

}