Electricity Transmission

transmit.fs

type Transmit = Station -> Station -> Supply option -> Station * Station

let transmit (fStation: Station) (tStation: Station) (supply: Supply option) =
  match fStation, tStation, supply with
  | PowerStation fStation, TransmissionSubstation tStation, Some (Supply supply) ->
    match fStation.Supply, tStation.Supply with
    | Some (Supply fStationSupply), Some (Supply tStationSupply) ->
      if fStationSupply > supply then
        PowerStation { fStation with Supply = Supply (fStationSupply - supply) },
        TransmissionSubstation { tStation with Supply = Supply (tStationSupply + supply) }
      else
        PowerStation { fStation with Supply = None },
        TransmissionSubstation { tStation with Supply = Supply (tStationSupply + fStationSupply)}    
    | None, _ | _, _ -> PowerStation fStation, TransmissionSubstation tStation
  | TransmissionSubstation fStation, TransmissionSubstation tStation, Some (Supply supply) ->
    match fStation.Supply, tStation.Supply with
    | Some (Supply fStationSupply), Some (Supply tStationSupply) ->
      if fStationSupply > supply then
        TransmissionSubstation { fStation with Supply = Supply (fStationSupply - supply) },
        TransmissionSubstation { tStation with Supply = Supply (tStationSupply + supply) }
      else
        TransmissionSubstation { fStation with Supply = None },
        TransmissionSubstation { tStation with Supply = Supply (tStationSupply + fStationSupply)}
    | None, _ | _, _ -> TransmissionSubstation fStation, TransmissionSubstation tStation
  | _, _, None | _, _, _ -> fStation, tStation