SCP-66

package JavaBasicTech.L16_Lists_Exercise;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Scanner;
import java.util.stream.Collectors;

public class Pr10SoftUniCoursePlanning {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        List<String> result = new ArrayList<>();
        List<String> lessons = Arrays.stream(scanner.nextLine()
                .split(", "))
                .collect(Collectors.toList());
        String initialScheduleLessons = scanner.nextLine();
        while (!"course start".equals(initialScheduleLessons)) {
            String[] commandArr = initialScheduleLessons.split(":");
            String command = commandArr[0];
            switch (command) {
                case "Add":
                    result = addLessons(lessons, commandArr[1]);
                    break;
                case "Insert":
                    result = insertLessons(lessons, commandArr[1], Integer.parseInt(commandArr[2]));
                    break;
                case "Remove":
                    result = removeLessons(lessons, commandArr[1]);
                    break;
                case "Swap":
                    result = swapLessons(lessons, commandArr[1], commandArr[2]);
                    break;
                case "Exercise":
                    result = exerciseLessons(lessons, commandArr[1]);
                    break;
            }
            initialScheduleLessons = scanner.nextLine();
        }
        for (int i = 0; i < result.size(); i++) {
            System.out.println(i + 1 + "." + result.get(i));
        }
    }

    private static List<String> addLessons(List<String> lessons, String lessonTitle) {
        lessons.add(lessonTitle);
        return lessons;
    }

    private static List<String> insertLessons(List<String> lessons, String lessonTitle, int index) {
        if (lessons.contains(lessonTitle)) return lessons;
        lessons.add(index, lessonTitle);
        return lessons;
    }

    private static List<String> removeLessons(List<String> lessons, String lessonTitle) {
        String nameOfLesson = lessonTitle;
        String nameOfLessonExercise = lessonTitle + "-Exercise";
        for (int i = 0; i < lessons.size(); i++) {
            String searchLesseon = lessons.get(i);
            if (searchLesseon.equals(nameOfLesson)) {
                lessons.remove(i);
            }
        }
        for (int i = 0; i < lessons.size(); i++) {
            String searchLesseon = lessons.get(i);
            if (searchLesseon.equals(nameOfLessonExercise)) {
                lessons.remove(i);
            }
        }
        return lessons;
    }

    private static List<String> swapLessons(List<String> lessons, String lessonFirst, String lessonSecond) {
        int indexLessonFirst = lessons.indexOf(lessonFirst);
        int indexLessonSecond = lessons.indexOf(lessonSecond);
        String lessonFirstExercise = lessonFirst + "-Exercise";
        String lessonSecondExercise = lessonSecond + "-Exercise";

        if (!lessons.contains(lessonFirst)) {
            return lessons;
        }
        if (!lessons.contains(lessonSecond)) {
            return lessons;
        }

        if (lessons.contains(lessonFirst)) {
            lessons.add(indexLessonSecond, lessonFirst);
            if (lessons.contains(lessonFirstExercise)) {
                lessons.add(indexLessonSecond + 1, lessonFirstExercise);
            }
            lessons.remove(indexLessonFirst);
        }

        boolean isContainsEx = false;
        if (lessons.contains(lessonSecond)) {
            lessons.remove(lessonSecond);
            if (lessons.contains(lessonSecondExercise)) {
                isContainsEx = true;
                lessons.remove(lessonSecondExercise);
            }
            lessons.add(indexLessonFirst, lessonSecond);
            if (isContainsEx){
                lessons.add(indexLessonFirst + 1, lessonSecondExercise);
            }
        }
        return lessons;
    }

    private static List<String> exerciseLessons(List<String> lessons, String lessonTitle) {

        String lessonTitleExer = lessonTitle + "-Exercise";
        if (!(lessons.contains(lessonTitle))) {
            lessons.add(lessonTitle);
            lessons.add(lessonTitleExer);
        } else if (lessons.contains(lessonTitle)) {
                if (!lessons.contains(lessonTitleExer)) {
                    int position = lessons.indexOf(lessonTitle);
                    lessons.add(position + 1, lessonTitleExer);
                }
        }
        return lessons;
    }
}