module GradeSchool (..) where

import Dict exposing (..)
import Result exposing (toMaybe)


-- TODO: implement these classes
-- type Grade = Int
-- type Student = String
-- type School = Dict Int List String


schoolFromList : List ( Int, String ) -> Dict Int (List String)
schoolFromList schoolList =
  List.foldl
    (\tuple -> \dict -> addStudent (fst tuple) (snd tuple) dict)
    newSchool
    schoolList


schoolToList : Dict Int (List String) -> List ( Int, List String )
schoolToList school =
  Dict.toList school


newSchool : Dict Int (List String)
newSchool =
  Dict.empty


addStudent : Int -> String -> Dict Int (List String) -> Dict Int (List String)
addStudent grade student school =
  Dict.insert grade (List.sort (student :: (studentsInGrade grade school))) school


gradeWithStudents : Int -> List String -> Dict Int (List String)
gradeWithStudents grade students =
  Dict.singleton grade (List.sort students)


studentsInGrade : Int -> Dict Int (List String) -> List String
studentsInGrade grade school =
  case (Dict.get grade school) of
    Just list ->
      list

    Nothing ->
      []