
package de.ivu.elect.business.gruppe.boundary;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.ejb.Stateless;
import javax.inject.Inject;

import de.ivu.elect.business.gruppe.control.PolitischeGruppeCreate;
import de.ivu.elect.business.gruppe.entity.GruppeUebernehmenParameter;
import de.ivu.elect.business.gruppe.entity.GruppeUebernehmenResult;
import de.ivu.elect.business.gruppe.entity.PolitischeGruppe;
import de.ivu.elect.business.gruppe.entity.PolitischeGruppeTK;
import de.ivu.elect.business.wahl.boundary.WahlDAO;
import de.ivu.elect.business.wahl.entity.Wahl;


import static java.util.function.Function.identity;
import static java.util.stream.Collectors.toMap;

@Stateless
public class GruppeUebernehmen {

  @Inject
  private transient WahlDAO wahlDAO;

  @Inject
  private transient PolitischeGruppeDAO politischeGruppeDAO;

  @Inject
  private transient PolitischeGruppeFactory politischeGruppeFactory;

  @Inject
  private transient PolitischeGruppeCreate politischeGruppeCreate;

  @Inject
  private transient GruppeEdit gruppeEdit;

  public GruppeUebernehmen() {
    super();
  }

  public GruppeUebernehmenResult save(GruppeUebernehmenParameter gruppeUebernehmenParameter) {
    if (gruppeUebernehmenParameter.getWahlQuelle() == null || gruppeUebernehmenParameter.getWahlZiel() == null) {
      return GruppeUebernehmenResult.NICHTS_UEBRNOMMEN;
    }

    Wahl quelle = wahlDAO.attach(gruppeUebernehmenParameter.getWahlQuelle());
    Wahl ziel = wahlDAO.attach(gruppeUebernehmenParameter.getWahlZiel());

    List<PolitischeGruppe> gruppenQuelle = politischeGruppeDAO.findByWahl(quelle);
    List<PolitischeGruppe> gruppenZiel = politischeGruppeDAO.findByWahl(ziel);

    return uebernehmeGruppen(ziel, gruppenQuelle, gruppenZiel,
        gruppeUebernehmenParameter.isGruppenAktualisierenWennSchonVorhanden(), gruppeUebernehmenParameter.isUeberzaehligeGruppenLoeschen());
  }

  private GruppeUebernehmenResult uebernehmeGruppen(Wahl zielWahl, List<PolitischeGruppe> gruppenQuelle, List<PolitischeGruppe> gruppenZiel, boolean aktualisieren, boolean ueberzaehligeLoeschen) {
    Map<PolitischeGruppeTK, PolitischeGruppe> gruppenNachTKZiel = gruppenZiel.stream().collect(toMap(PolitischeGruppe::getPolitischeGruppeTK, identity()));

    Map<PolitischeGruppeTK, PolitischeGruppe> zuLoeschendeGruppen = new HashMap<>();
    if (ueberzaehligeLoeschen) {
      for (PolitischeGruppe zielGruppe : gruppenZiel) {
        zuLoeschendeGruppen.put(zielGruppe.getPolitischeGruppeTK(), zielGruppe);
      }
    }

    long anzahlUebernommenerGruppem = 0L;
    long anzahlBereitsVorhandenerGruppen = 0L;
    long anzahlAktualisierterGruppen = 0L;
    long anzahlGeloeschterGruppen = 0L;

    for (PolitischeGruppe quelle : gruppenQuelle) {
      zuLoeschendeGruppen.remove(quelle.getPolitischeGruppeTK());

      PolitischeGruppeTK tk = quelle.getPolitischeGruppeTK();

      if (gruppenNachTKZiel.containsKey(tk)) {
        if (aktualisieren) {
          PolitischeGruppe bisherigeGruppe = gruppenNachTKZiel.get(tk);
          aktualisiere(bisherigeGruppe, quelle);
          anzahlAktualisierterGruppen++;
        } else {
          anzahlBereitsVorhandenerGruppen++;
        }
      } else {
        PolitischeGruppe neueZielGruppe = politischeGruppeFactory.create();
        kopiereGruppeFuerWahl(quelle, neueZielGruppe, zielWahl);
        persistiereNeueGruppe(neueZielGruppe, gruppenNachTKZiel);

        anzahlUebernommenerGruppem++;
      }
    }

    if (!zuLoeschendeGruppen.isEmpty()) {
      politischeGruppeDAO.flush();
      politischeGruppeDAO.clear();
      for (PolitischeGruppe ueberzaehligeGruppeNotAttached : zuLoeschendeGruppen.values()) {
        PolitischeGruppe ueberzaehligeGruppe = politischeGruppeDAO.attach(ueberzaehligeGruppeNotAttached);
        anzahlGeloeschterGruppen++;
        gruppeEdit.delete(ueberzaehligeGruppe);
      }
    }

    return new GruppeUebernehmenResult(anzahlUebernommenerGruppem, anzahlBereitsVorhandenerGruppen,
        anzahlAktualisierterGruppen, anzahlGeloeschterGruppen);
  }

  private void aktualisiere(PolitischeGruppe ziel, PolitischeGruppe quelle) {
    quelle.copyEntityAttributesUsingSetter(ziel);
  }

  protected Map<String, Object> createPostCreateContext(Wahl zielWahl) {
    return new HashMap<>();
  }

  private void persistiereNeueGruppe(PolitischeGruppe neueGruppe, Map<PolitischeGruppeTK, PolitischeGruppe> politischeGruppeTKPolitischeGruppeMap) {
    politischeGruppeCreate.save(neueGruppe);
    politischeGruppeTKPolitischeGruppeMap.put(neueGruppe.getPolitischeGruppeTK(), neueGruppe);
  }

  private void kopiereGruppeFuerWahl(PolitischeGruppe quelle, PolitischeGruppe ziel, Wahl zielWahl) {
    ziel.setWahl(zielWahl);
    aktualisiere(ziel, quelle);
  }

}
