#include <cstddef>
#include <iostream>
#include <memory>
#include <string>
#include <vector>

class Adres {
  std::string ulica;
  int nr_mieszkania;

public:
  Adres(std::string ulica, int nr_mieszkania)
      : ulica(ulica), nr_mieszkania(nr_mieszkania) {}

  friend std::ostream &operator<<(std::ostream &os, const Adres &adres) {
    os << adres.ulica << " mieszkania " << adres.nr_mieszkania;
    return os;
  }
};

class Pracownik {
protected:
  std::string imie;
  float pensja;
  std::unique_ptr<Adres> adres;

public:
  virtual float obliczWyplate() const = 0;

  Pracownik(std::string imie, float pensja, std::unique_ptr<Adres> adres)
      : imie(imie), pensja(pensja), adres(std::move(adres)) {}

  virtual ~Pracownik() = default;

  friend std::ostream &operator<<(std::ostream &os, const Pracownik &p) {
    os << p.imie << " zarabia " << p.obliczWyplate() << " i mieszka na "
       << *(p.adres);
    return os;
  }
};

class PracownikEtatowy : public Pracownik {
public:
  PracownikEtatowy(std::string imie, float pensja, std::unique_ptr<Adres> adres)
      : Pracownik(imie, pensja, std::move(adres)) {}

  float obliczWyplate() const override { return pensja; }
};

class Zleceniobiorca : public Pracownik {
  float godziny;

public:
  Zleceniobiorca(std::string imie, float pensja, std::unique_ptr<Adres> adres,
                 float godziny)
      : Pracownik(imie, pensja, std::move(adres)), godziny(godziny) {}

  float obliczWyplate() const override { return pensja * godziny; }
};

class Dzial {
  std::vector<std::shared_ptr<Pracownik>> pracownicy;

public:
  Dzial() = default;

  void dodajPracownika(std::shared_ptr<Pracownik> pracownik) {
    pracownicy.push_back(pracownik);
  }

  float sumaWyplat() const {
    float sum = 0;
    for (const auto &pracownik : pracownicy) {
      sum += pracownik->obliczWyplate();
    }
    return sum;
  }

  void wypiszPracownikow() const {
    std::cout << "=== Wszyscy pracownicy ===\n";
    for (const auto &pracownik : pracownicy) {
      std::cout << *pracownik << std::endl;
    }
  }
};

class Kierownik : public PracownikEtatowy {
  std::vector<std::shared_ptr<Pracownik>> podwladni;

public:
  Kierownik(std::string imie, float pensja, std::unique_ptr<Adres> adres)
      : PracownikEtatowy(imie, pensja, std::move(adres)) {}

  float obliczWyplate() const override { return pensja; }

  float sumaWyplatZespolu() const {
    float sum = 0;
    for (const auto &pracownik : podwladni) {
      sum += pracownik->obliczWyplate();
    }
    return sum;
  }

  void dodajPracownika(std::shared_ptr<Pracownik> p) { podwladni.push_back(p); }
};

int main() {
  Kierownik kierownik("Stasiek", 238943,
                      std::make_unique<Adres>("Wojska polskiego", 23));

  kierownik.dodajPracownika(std::make_shared<PracownikEtatowy>(
      "Bożena", 2500, std::make_unique<Adres>("Żołnierska", 18)));

  kierownik.dodajPracownika(std::make_shared<Zleceniobiorca>(
      "Adam", 6700, std::make_unique<Adres>("Wiejska", 1), 30));

  std::cout << "Pracownicy sumarycznie zarabiają: "
            << kierownik.sumaWyplatZespolu() << std::endl;

  return 0;
}