package prison;

import org.json.JSONArray;
import org.json.JSONObject;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class Prison {

    private List<Prisoner> prisoners = new ArrayList<>();

    public Prison() {

    }

    public List<Prisoner> getPrisoners() {
        return prisoners;
    }

    public void addPrisoner(Prisoner prisoner) {
        prisoners.add(prisoner);
    }

    public void removePrisoner(Scanner scanner) {
        System.out.println("Enter the name of the prisoner you want to remove: ");
        String name = scanner.nextLine();

        for (Prisoner p : prisoners) {
            if (p.getName().equalsIgnoreCase(name)) {
                prisoners.remove(p);
                return;
            }
        }
    }

    public void loadFile(Scanner scanner) {
        System.out.println("Enter path:");
        String path = scanner.nextLine();
        try {

            String jsonString = new String(Files.readAllBytes(Paths.get(path)));
            JSONArray jsonArray = new JSONArray(jsonString);
            for (int i = 0; i < jsonArray.length(); i++) {
                JSONObject jsonObject = jsonArray.getJSONObject(i);
                Prisoner prisoner = new Prisoner(jsonObject);
                prisoners.add(prisoner);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void printPrisoners() {
        for (Prisoner p : prisoners) {
            System.out.println(p.toString());
        }
    }
}