package sx.register.services.remoteservices.pfr.smev3.benefitscommon.beans;

import sx.datastore.SXObj;
import sx.datastore.db.utils.SqlEnum;
import sx.datastore.params.SXObjListParams;
import sx.register.services.remoteservices.pfr.benefits.bean.Payment;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

public final class MonthlyPensionsList {

  private List<MonthlyPension> pensions;

  private MonthlyPensionsList() {
  }

  public static MonthlyPensionsList of(BenefitsResponse benefitsResponse) {
    List<sx.register.services.remoteservices.pfr.benefits.bean.MonthlyPayment> monthlyPensions = benefitsResponse.getListOfMonthlyPayments();
    List<SXObj> sxObjList = new ArrayList<>();

    List<SXObj> list = monthlyPensions.stream()
        .flatMap(o -> o.getListOfPayments().stream()
            .map(p -> {
              String type = p.getType();
              SXObjListParams params = new SXObjListParams(WsPFRCorrPension.CLASS_NAME)
                  .addSelectedAttr(WsPFRCorrPension.Attribute.ATTR_SIZE)
                  .addSelectedAttr(WsPFRCorrPension.Attribute.GROUP_NUMBER)
                  .addSelectedAttr(WsPFRCorrPension.Attribute.CODE_FULL)
                  .addSelectedAttr(WsPFRCorrPension.Attribute.PENSION)
                  .addCondition(WsPFRCorrPension.Attribute.CODE, type, SqlEnum.EQUAL);

              return params.getObj();
            }))
        .collect(Collectors.toList());


  }

  public List<MonthlyPension> getPensions() {
    return pensions;
  }


}