/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill 庄骞 (smallchill@163.com)
 */
package org.springblade.community.unit.wrapper;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import net.logstash.logback.encoder.org.apache.commons.lang3.StringUtils;
import org.springblade.community.agency.entity.Agency;
import org.springblade.community.agency.service.IAgencyService;
import org.springblade.community.agency.vo.AgencyVO;
import org.springblade.community.residential.entity.Residential;
import org.springblade.community.residential.service.IResidentialService;
import org.springblade.community.unit.entity.Unit;
import org.springblade.community.unit.vo.UnitVO;
import org.springblade.core.mp.support.BaseEntityWrapper;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.core.tool.utils.StringUtil;
import org.springblade.sysarea.entity.SysArea;
import org.springblade.sysarea.service.ISysAreaService;
import org.springblade.system.street.entity.Street;
import org.springblade.system.street.service.IStreetService;

import java.util.Objects;

/**
 * 包装类,返回视图层所需的字段
 *
 * @author BladeX
 * @since 2020-11-06
 */
public class UnitWrapper extends BaseEntityWrapper<Unit, UnitVO> {

	private IAgencyService agencyService;
	private IResidentialService residentialService;

	private UnitWrapper(IAgencyService agencyService, IResidentialService residentialService){
		this.agencyService = agencyService;
		this.residentialService = residentialService;
	}

	public static UnitWrapper build(IAgencyService agencyService, IResidentialService residentialService) {
		return new UnitWrapper(agencyService,residentialService);
 	}

	@Override
	public UnitVO entityVO(Unit unit) {
		UnitVO unitVO = Objects.requireNonNull(BeanUtil.copy(unit, UnitVO.class));

		Residential residential = residentialService.getById(unit.getResidentialId());
		if(residential != null){
			unitVO.setResidentialName(residential.getName());
			unitVO.setAgencyId(residential.getAgencyId());
			Agency agency = agencyService.getById(residential.getAgencyId());
			if(agency != null){
				unitVO.setAgencyName(agency.getAgencyName());
			}
		}

		return unitVO;
	}

}
