{-# LANGUAGE TemplateHaskell #-}

module Data.Aviation.Casr.Logbook.Reports.FlightTimeReport(
  FlightTimeReport(..)
, HasFlightTimeReport(..)
, singleFlightTimeReport
, getFlightTimeReport
) where

import Control.Category((.))
import Control.Lens ( preview, (^.), view, isn't, makeClassy )
import Data.Aviation.Casr.Logbook.Types
    ( TimeAmount,
      AsEngine(_Multi, _Single),
      Aviator,
      HasAircraft(aircraftRegistration, aircraftType, aircraftEngine),
      HasDayNight(nightDayNight, dayDayNight),
      totalDayNight,
      AsCommand(_InCommandInstructing, _InCommand, _Dual, _ICUS),
      HasAircraftFlight(instrumentflightTime, daynight, flightaircraft,
                        command),
      Entry(AircraftFlightEntry),
      Entries(Entries),
      Logbook(Logbook),
      HasInstructionRating(instructionRating),
      AsInstructionRating(_GA3InstructionRating, _GA2InstructionRating,
                          _GA1InstructionRating, _RASIInstructionRating,
                          _RAInstructionRating),
      getUnderInstructionPic,
      isRAInstruction,
      isGAInstruction,
      aeronauticalHours )
import Data.Eq(Eq)
import Data.Foldable(foldl')
import Data.Int(Int)
import qualified Data.Map as Map(unionWith, singleton, empty)
import Data.Map(Map)
import Data.Maybe(Maybe(Just, Nothing))
import Data.Monoid(Monoid(mappend, mempty))
import Data.Ord(Ord)
import Data.Semigroup(Semigroup((<>)))
import Data.String(String)
import Prelude(Show, (+))

data FlightTimeReport =
  FlightTimeReport {
    _flightsTotal ::
      Int
  , _hoursTotal ::
      TimeAmount
  , _hoursTotalICUS ::
      TimeAmount
  , _hoursTotalDual ::
      TimeAmount
  , _hoursTotalInCommand ::
      TimeAmount
  , _hoursInAircraftType ::
      Map String (TimeAmount, TimeAmount, TimeAmount, TimeAmount)
  , _hoursInAircraftRegistration ::
      Map String (TimeAmount, TimeAmount, TimeAmount, TimeAmount)
  , _hoursSingleEngine ::
      TimeAmount
  , _hoursSingleEngineICUS ::
      TimeAmount
  , _hoursSingleEngineDual ::
      TimeAmount
  , _hoursSingleEngineInCommand ::
      TimeAmount
  , _hoursMultiEngine ::
      TimeAmount
  , _hoursMultiEngineICUS ::
      TimeAmount
  , _hoursMultiEngineDual ::
      TimeAmount
  , _hoursMultiEngineInCommand ::
      TimeAmount
  , _hoursDay ::
      TimeAmount
  , _hoursDayICUS ::
      TimeAmount
  , _hoursDayDual ::
      TimeAmount
  , _hoursDayInCommand ::
      TimeAmount
  , _hoursInstructing ::
      TimeAmount
  , _hoursGAInstructing ::
      TimeAmount
  , _hoursGA1Instructing ::
      TimeAmount
  , _hoursGA2Instructing ::
      TimeAmount
  , _hoursGA3Instructing ::
      TimeAmount
  , _hoursRAInstructing ::
      TimeAmount
  , _hoursRAJuniorInstructing ::
      TimeAmount
  , _hoursRASeniorInstructing ::
      TimeAmount
  , _hoursNight ::
      TimeAmount
  , _hoursNightICUS ::
      TimeAmount
  , _hoursNightDual ::
      TimeAmount
  , _hoursNightInCommand ::
      TimeAmount
  , _hoursWithPiC ::
      Map Aviator TimeAmount
  , _hoursInstrument ::
      TimeAmount
  } deriving (Eq, Ord, Show)

makeClassy ''FlightTimeReport

instance Semigroup FlightTimeReport where
  FlightTimeReport ft1 tl1 tli1 tld1 tlc1 tp1 rg1 se1 sei1 sed1 sec1 me1 mei1 med1 mec1 dy1 dyi1 dyd1 dyc1 ins1 insg1 insg1_1 insg1_2 insg1_3 insr1  insr1_j insr1_s nt1 nti1 ntd1 ntc1 wpc1 is1 <> FlightTimeReport ft2 tl2 tli2 tld2 tlc2 tp2 rg2 se2 sei2 sed2 sec2 me2 mei2 med2 mec2 dy2 dyi2 dyd2 dyc2 ins2 insg2 insg2_1 insg2_2 insg2_3 insr2 insr2_j insr2_s nt2 nti2 ntd2 ntc2 wpc2 is2 =
    FlightTimeReport
      (ft1 + ft2)
      (tl1 <> tl2)
      (tli1 <> tli2)
      (tld1 <> tld2)
      (tlc1 <> tlc2)
      (Map.unionWith mappend tp1 tp2)
      (Map.unionWith mappend rg1 rg2)
      (se1 <> se2)
      (sei1 <> sei2)
      (sed1 <> sed2)
      (sec1 <> sec2)
      (me1 <> me2)
      (mei1 <> mei2)
      (med1 <> med2)
      (mec1 <> mec2)
      (dy1 <> dy2)
      (dyi1 <> dyi2)
      (dyd1 <> dyd2)
      (dyc1 <> dyc2)
      (ins1 <> ins2)
      (insg1 <> insg2)
      (insg1_1 <> insg2_1)
      (insg1_2 <> insg2_2)
      (insg1_3 <> insg2_3)
      (insr1 <> insr2)
      (insr1_j <> insr2_j)
      (insr1_s <> insr2_s)
      (nt1 <> nt2)
      (nti1 <> nti2)
      (ntd1 <> ntd2)
      (ntc1 <> ntc2)
      (Map.unionWith mappend wpc1 wpc2)
      (is1 <> is2)

instance Monoid FlightTimeReport where
  mempty =
    FlightTimeReport
      0
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
      mempty
  mappend =
    (<>)

singleFlightTimeReport ::
  Entry a b c d
  -> FlightTimeReport
singleFlightTimeReport (AircraftFlightEntry fl _) =
  let hoursdaynight = totalDayNight (aeronauticalHours fl)
      icus x =
        case preview (command . _ICUS) fl of
          Just _ ->
            x
          Nothing ->
            mempty
      dual x =
        case preview (command . _Dual) fl of
          Just _ ->
            x
          Nothing ->
            mempty
      comd x =
        case preview (command . _InCommand) fl of
          Just _ ->
            x
          Nothing ->
            mempty
      instr x =
        case preview (command . _InCommandInstructing) fl of
          Just _ ->
            x
          Nothing ->
            mempty
      instrRA x =
        case preview (command . _InCommandInstructing) fl of
          Just a ->
            if isRAInstruction (view instructionRating a) then x else mempty
          Nothing ->
            mempty
      instrGA x =
        case preview (command . _InCommandInstructing) fl of
          Just a ->
            if isGAInstruction (view instructionRating a) then x else mempty
          Nothing ->
            mempty
      instrRAJunior x =
        case preview (command . _InCommandInstructing) fl of
          Just a ->
            if isn't _RAInstructionRating (view instructionRating a) then mempty else x
          Nothing ->
            mempty
      instrRASenior x =
        case preview (command . _InCommandInstructing) fl of
          Just a ->
            if isn't _RASIInstructionRating (view instructionRating a) then mempty else x
          Nothing ->
            mempty
      instrGA1 x =
        case preview (command . _InCommandInstructing) fl of
          Just a ->
            if isn't _GA1InstructionRating (view instructionRating a) then mempty else x
          Nothing ->
            mempty
      instrGA2 x =
        case preview (command . _InCommandInstructing) fl of
          Just a ->
            if isn't _GA2InstructionRating (view instructionRating a) then mempty else x
          Nothing ->
            mempty
      instrGA3 x =
        case preview (command . _InCommandInstructing) fl of
          Just a ->
            if isn't _GA3InstructionRating (view instructionRating a) then mempty else x
          Nothing ->
            mempty
      hoursmap k =
        Map.singleton k (hoursdaynight, icus hoursdaynight, dual hoursdaynight, comd hoursdaynight)
      singleengine x =
        case preview (flightaircraft . aircraftEngine . _Single) fl of
          Just _ ->
            x
          Nothing ->
            mempty
      multiengine x =
        case preview (flightaircraft . aircraftEngine . _Multi) fl of
          Just _ ->
            x
          Nothing ->
            mempty
      totalhoursday =
        fl ^. daynight . dayDayNight
      totalhoursnight =
        fl ^. daynight . nightDayNight
      pic x =
        case getUnderInstructionPic (fl ^. command) of
          Just a ->
            Map.singleton a x
          Nothing ->
            Map.empty
  in  FlightTimeReport
        1
        hoursdaynight
        (icus hoursdaynight)
        (dual hoursdaynight)
        (comd hoursdaynight)
        (hoursmap (fl ^. flightaircraft . aircraftType))
        (hoursmap (fl ^. flightaircraft . aircraftRegistration))
        (singleengine hoursdaynight)
        (singleengine (icus hoursdaynight))
        (singleengine (dual hoursdaynight))
        (singleengine (comd hoursdaynight))
        (multiengine hoursdaynight)
        (multiengine (icus hoursdaynight))
        (multiengine (dual hoursdaynight))
        (multiengine (comd hoursdaynight))
        totalhoursday
        (icus hoursdaynight)
        (dual hoursdaynight)
        (comd hoursdaynight)
        (instr hoursdaynight)
        (instrGA hoursdaynight)
        (instrGA1 hoursdaynight)
        (instrGA2 hoursdaynight)
        (instrGA3 hoursdaynight)
        (instrRA hoursdaynight)
        (instrRAJunior hoursdaynight)
        (instrRASenior hoursdaynight)
        totalhoursnight
        (icus totalhoursnight)
        (dual totalhoursnight)
        (comd totalhoursnight)
        (pic hoursdaynight)
        (fl ^. instrumentflightTime)
singleFlightTimeReport _ =
  mempty

getFlightTimeReport ::
  Logbook a b c d
  -> FlightTimeReport
getFlightTimeReport (Logbook _ (Entries es)) =
  foldl' (\a -> mappend a . singleFlightTimeReport) mempty es