package edu.pdx.cs.joy.grader;

import edu.pdx.cs.joy.ParserException;
import edu.pdx.cs.joy.grader.gradebook.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;

public class ProjectGradesImporter {

  private final GradeBook gradeBook;
  private final Assignment assignment;
  private final Logger logger;

  public ProjectGradesImporter(GradeBook gradeBook, Assignment assignment, Logger logger) {
    this.gradeBook = gradeBook;
    this.assignment = assignment;
    this.logger = logger;
  }

  public static TestedProjectSubmissionOutputParser.ProjectScore getScoreFrom(Reader reader) throws TestedProjectSubmissionOutputParser.TestedProjectSubmissionOutputParsingException, IOException {
    return TestedProjectSubmissionOutputParser.parseTestedSubmissionOutput(reader);
  }

  public void recordScoreFromProjectReport(String studentId, Reader report) throws TestedProjectSubmissionOutputParser.TestedProjectSubmissionOutputParsingException, IOException {
    TestedProjectSubmissionOutputParser.ProjectScore score = getScoreFrom(report);

    if (score.getTotalPoints() != this.assignment.getPoints()) {
      String message = "Assignment " + this.assignment.getName() + " should be worth " + this.assignment.getPoints() +
        " points, but the report for " + studentId + " was out of " + score.getTotalPoints();
      throw new IllegalStateException(message);
    }

    Optional<Student> maybeStudent = gradeBook.getStudent(studentId);
    if (maybeStudent.isEmpty()) {
      warn("Student \"" + studentId + "\" not found in gradebook");
      return;
    }

    Student student = maybeStudent.get();
    Grade grade = student.getGrade(this.assignment);
    if (grade == null) {
      grade = new Grade(assignment, score.getScore());
      student.setGrade(assignment.getName(), grade);

    } else {
      grade.setScore(score.getScore());
    }

    info("Recorded grade of " + score.getScore() + " for " + studentId);
  }

  private void info(String message) {
    this.logger.info(message);
  }

  private void warn(String message) {
    logger.warn(message);
  }

  public static void main(String[] args) {
    String gradeBookFileName = null;
    String assignmentName = null;
    List<String> projectFileNames = new ArrayList<>();

    for (String arg : args) {
      if (gradeBookFileName == null) {
        gradeBookFileName = arg;

      } else if (assignmentName == null) {
        assignmentName = arg;

      } else {
        projectFileNames.add(arg);
      }
    }

    usageIfNull(gradeBookFileName, "Missing grade book file");
    usageIfNull(assignmentName, "Missing assignment name");
    usageIfEmpty(projectFileNames, "No project file names provided");

    GradeBook gradeBook = getGradeBook(gradeBookFileName);
    Assignment assignment = getAssignment(assignmentName, gradeBook);
    Logger logger = LoggerFactory.getLogger(ProjectGradesImporter.class.getPackage().getName());

    ProjectGradesImporter importer = new ProjectGradesImporter(gradeBook, assignment, logger);

    for (String projectFileName : projectFileNames) {
      File projectFile = getProjectFile(projectFileName);
      String studentId = getStudentIdFromFileName(projectFile);
      try {
        importer.recordScoreFromProjectReport(studentId, new FileReader(projectFile));

      } catch (FileNotFoundException ex) {
        throw new IllegalStateException("Could not find file \"" + projectFile + "\"");

      } catch (IllegalStateException | IOException ex) {
        throw new IllegalStateException("While recording score from " + projectFileName, ex);

      } catch (TestedProjectSubmissionOutputParser.TestedProjectSubmissionOutputParsingException e) {
        logger.warn("Could not find score in " + projectFileName);
      }
    }

    saveGradeBookIfModified(gradeBook, gradeBookFileName);
  }

  private static void saveGradeBookIfModified(GradeBook gradeBook, String gradeBookFileName) {
    if (gradeBook.isDirty()) {
      File file = new File(gradeBookFileName);
      try {
        XmlDumper dumper = new XmlDumper(file);
        dumper.dump(gradeBook);

      } catch (IOException e) {
        usage("Can't write grade book in \"" + gradeBookFileName + "\"");
      }
    }
  }

  private static String getStudentIdFromFileName(File projectFile) {
    String fileName = projectFile.getName();
    int index = fileName.lastIndexOf('.');
    if (index < 0) {
      return usage("Project file \"" + fileName + "\" does not have a file extension");
    }

    return fileName.substring(0, index);
  }

  private static File getProjectFile(String projectFileName) {
    File projectFile = new File(projectFileName);
    if (!projectFile.exists()) {
      return usage("Project file \"" + projectFileName + "\" does not exist");

    } else if (!projectFile.isFile()) {
      return usage("Project file \"" + projectFileName + "\" is not a file");
    }

    return projectFile;
  }

  private static Assignment getAssignment(String assignmentName, GradeBook gradeBook) {
    Assignment assignment = gradeBook.getAssignment(assignmentName);
    if (assignment == null) {
      return usage("Could not find assignment \"" + assignmentName + "\" in grade book");
    }
    return assignment;
  }

  private static GradeBook getGradeBook(String gradeBookFileName) {
    try {
      XmlGradeBookParser parser = new XmlGradeBookParser(gradeBookFileName);
      return parser.parse();

    } catch (IOException ex) {
      return usage("Couldn't read grade book file \"" + gradeBookFileName + "\"");

    } catch (ParserException ex) {
      return usage("Couldn't parse grade book file \"" + gradeBookFileName + "\"");
    }
  }

  private static void usageIfEmpty(List<String> list, String message) {
    if (list.isEmpty()) {
      usage(message);
    }
  }

  private static void usageIfNull(String argument, String message) {
    if (argument == null) {
      usage(message);
    }
  }

  private static <T> T usage(String message) {
    PrintStream err = System.err;

    err.println("** " + message);

    err.println("java ProjectGradesImporter gradeBookFileName assignmentName projectFileName+");
    err.println();
    err.println("Imports grades (of the form \"5.8 out of 6.0\") from project reports into a grade book.");
    err.println("The name of the project report file begins with the student's id.");
    err.println();
    err.println("  gradeBookFileName     Name of file containing grade book");
    err.println("  assignmentName        Assignment/project whose score is to be recorded");
    err.println("  projectFileName       Name of file containing graded project");
    err.println();

    System.exit(1);

    return null;
  }

}