package model.collections;

import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

import model.data.Korisnik;
import exceptions.ElementNePostoji;
import exceptions.ElementPostoji;

public class Korisnici implements ITableModel {

private Map<String, Korisnik> mapaKorisnika = new HashMap<String, Korisnik>();

    public void addKorisnik(Korisnik novi) throws ElementPostoji {
        if (mapaKorisnika.containsKey(novi.getKey()))
            throw new ElementPostoji("Korisnik već postoji u kolekciji!");
        mapaKorisnika.put(novi.getKey(), novi);
    }

    public void removeKorisnik(Korisnik stari) throws ElementNePostoji {
        if (!mapaKorisnika.containsKey(stari.getKey()))
            throw new ElementNePostoji("Traženi korisnik ne postoji u kolekciji!");
        mapaKorisnika.remove(stari.getKey());
    }

    public Korisnik getKorisnik(String key) throws ElementNePostoji {
        if (!mapaKorisnika.containsKey(key))
            throw new ElementNePostoji("Traženi korisnik ne postoji u kolekciji!");
        return mapaKorisnika.get(key);
    }

    public Iterator<String> getKorisniciIterator() {
        Set<String> brojeviIndeksa = mapaKorisnika.keySet();
        return brojeviIndeksa.iterator();
    }

    @Override
    public Object getValueAt(int rowIndex, int columnIndex) {
        ArrayList<Korisnik> nizKorisnika = new ArrayList<Korisnik>(mapaKorisnika.values());
        switch(columnIndex){
            case 0: return nizKorisnika.get(rowIndex).getKorisnickoIme();
            case 1: return nizKorisnika.get(rowIndex).getLozinka();
            case 2: return nizKorisnika.get(rowIndex).getPrezime();
            case 3: return nizKorisnika.get(rowIndex).getIme();
            case 4: return nizKorisnika.get(rowIndex).getStatus();
            case 5: return nizKorisnika.get(rowIndex).getPol();
            case 6: return nizKorisnika.get(rowIndex).isAdministrator();
            case 7: return nizKorisnika.get(rowIndex).getNapomena();
        }
        return "";
    }

    @Override
    public int getRowCount() {
        return mapaKorisnika.size();
    }
    public void save(ObjectOutputStream out) {
        try {
            out.writeObject(mapaKorisnika);
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    @SuppressWarnings("unchecked")
    public void load(ObjectInputStream in) {
        try {
            mapaKorisnika = (Map<String, Korisnik>) in.readObject();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

    }
}