package second_term.sixth_chapter;

import java.io.*;
import java.util.Collections;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;

public class ProgramOfStudy implements Iterable<Course>,Serializable {

    private List<Course> list;

    public ProgramOfStudy()
    {
        list = new LinkedList<Course>();
    }

    public void addCourse(Course course)
    {
        if (course != null)
            list.add(course);
    }

    public Course find(String prefix, int number)
    {
        for (Course course : list)
            if (prefix.equals(course.getPrefix()) &&
                    number == course.getNumber())
                return course;

        return null;
    }

    public void addCourseAfter(Course target, Course newCourse)
    {
        if (target == null || newCourse == null)
            return;

        int targetIndex = list.indexOf(target);
        if (targetIndex != -1)
            list.add(targetIndex + 1, newCourse);
    }

    public void replace(Course target, Course newCourse)
    {
        if (target == null || newCourse == null)
            return;

        int targetIndex = list.indexOf(target);
        if (targetIndex != -1)
            list.set(targetIndex, newCourse);
    }

    public void sort(){
        Collections.sort(list);
    }

    public String toString()
    {
        String result = "";
        for (Course course : list)
            result += course + "\n";
        return result;
    }

    public Iterator<Course> iterator()
    {
        return list.iterator();
    }


    public void save(String fileName) throws IOException
    {
        FileOutputStream fos = new FileOutputStream(fileName);
        ObjectOutputStream oos = new ObjectOutputStream(fos);
        oos.writeObject(this);
        oos.flush();
        oos.close();
    }


    public static ProgramOfStudy load(String fileName) throws IOException, ClassNotFoundException
    {
        FileInputStream fis = new FileInputStream(fileName);
        ObjectInputStream ois = new ObjectInputStream(fis);
        ProgramOfStudy pos = (ProgramOfStudy) ois.readObject();
        ois.close();

        return pos;
    }
}