helen

defmodule Dutycycle.State do
  @moduledoc false

  require Logger
  use Timex
  use Ecto.Schema

  import Repo, only: [update!: 1]
  import Ecto.Changeset, only: [cast: 3, validate_required: 2]

  alias Dutycycle.Profile
  alias Dutycycle.State
  alias TimeSupport

  schema "dutycycle_state" do
    field(:state, :binary, default: "stopped")
    field(:dev_state, :boolean)
    field(:run_at, :utc_datetime_usec)
    field(:run_end_at, :utc_datetime_usec)
    field(:idle_at, :utc_datetime_usec)
    field(:idle_end_at, :utc_datetime_usec)
    field(:started_at, :utc_datetime_usec)
    field(:state_at, :utc_datetime_usec)
    belongs_to(:dutycycle, Dutycycle)

    timestamps(usec: true)
  end

  def state(%Dutycycle{state: state}),
    do: Map.get(state, :state, %State{state: "stopped"})

  def mode(%State{state: state}), do: state

  def next_phase(
        mode,
        %Dutycycle{name: name, state: %State{state: prev_mode}} = dc,
        opts
      )
      when is_atom(mode) and is_list(opts) do
    log = Keyword.get(opts, :log_transition, false)

    log &&
      Logger.info([
        inspect(name, pretty: true),
        " transitioning from ",
        inspect(prev_mode, pretty: true),
        " to ",
        inspect(Atom.to_string(mode), pretty: true)
      ])

    next_phase(mode, dc) |> started_at(prev_mode, :reload)
  end

  def next_phase(:run, %Dutycycle{} = dc), do: run(dc)
  def next_phase(:idle, %Dutycycle{} = dc), do: idle(dc)
  def next_phase(:offline, %Dutycycle{} = dc), do: offline(dc)
  def next_phase(:stop, %Dutycycle{} = dc), do: stop(dc)

  def idle(%Dutycycle{state: st} = dc) do
    {dc, update_states_only(st, state: "idling", dev_state: false)}
  end

  def idling?(%Dutycycle{state: %State{state: "idling"}}), do: true
  def idling?(%Dutycycle{state: %State{state: _}}), do: false

  def offline(%Dutycycle{} = dc) do
    stop(dc, "offline")
  end

  def offline?(%Dutycycle{state: %State{state: "offline"}}), do: true
  def offline?(%Dutycycle{state: %State{state: _}}), do: false

  def reload(%State{id: id}), do: Repo.get!(__MODULE__, id)

  def run(%Dutycycle{name: name, state: st, log: log} = dc) do
    log &&
      Logger.debug([
        "dutycycle ",
        inspect(name, pretty: true),
        " setting state to run"
      ])

    {dc, update_states_only(st, state: "running", dev_state: true)}
  end

  def running?(%Dutycycle{state: %State{state: "running"}}), do: true
  def running?(%Dutycycle{state: %State{state: _}}), do: false

  def stop(%Dutycycle{state: st} = dc, state \\ "stopped")
      when is_binary(state) do
    {dc, update_states_only(st, state: state, dev_state: false)}
  end

  def stopped?(%Dutycycle{state: %State{state: "stopped"}}), do: true
  def stopped?(%Dutycycle{state: %State{state: _}}), do: false

  def persist_phase_timer(
        %State{state: state} = st,
        %Profile{idle_ms: idle_ms, run_ms: run_ms},
        timer
      ) do
    cond do
      state === "running" ->
        Logger.debug(["persist_phase_timer() handling state 'running')"])

        update(
          st,
          calculate_at(timer, run_ms, at_key: :run_at, at_end_key: :run_end_at)
        )

      state === "idling" ->
        update(
          st,
          calculate_at(timer, idle_ms,
            at_key: :idle_at,
            at_end_key: :idle_end_at
          )
        )

      true ->
        Logger.warn([
          "persist_phase_timer(): unhandled state",
          inspect(state, pretty: true),
          "setting at times to nil"
        ])

        update(st, nil_phase_at_times())
    end
  end

  #####################
  # Private Functions #
  #####################

  defp add_state_at(opts), do: opts ++ [state_at: TimeSupport.utc_now()]

  defp calculate_at(timer, total_ms, opts)
       when is_reference(timer) and is_list(opts) do
    {:at_key, at_key} = Keyword.take(opts, [:at_key]) |> hd()
    {:at_end_key, at_end_key} = Keyword.take(opts, [:at_end_key]) |> hd()

    # state start at:
    #  current time minus total phase_ms minus the remaining timer
    #
    # state end at:
    #  current time plus the remaining timer

    remaining_ms = Process.read_timer(timer)

    []
    |> Keyword.put(at_key, shift_ms((total_ms - remaining_ms) * -1))
    |> Keyword.put(at_end_key, shift_ms(remaining_ms))
  end

  defp changeset(st, params),
    do:
      st
      |> cast(params, possible_changes())
      |> validate_required(required_changes())

  defp nil_phase_at_times,
    do: [idle_at: nil, idle_end_at: nil, run_at: nil, run_end_at: nil]

  defp possible_changes,
    do: [
      :state,
      :dev_state,
      :run_at,
      :run_end_at,
      :idle_at,
      :idle_end_at,
      :started_at,
      :state_at
    ]

  defp required_changes, do: [:state, :dev_state]

  defp started_at({dc, {:ok, %State{} = st}}, prev_mode, :reload),
    do: started_at({dc, {:ok, reload(st)}}, prev_mode)

  # if we just transitioned from offline or stopped to running then
  # record the current time as started at
  defp started_at({dc, {:ok, %State{state: mode} = st}}, prev_mode)
       when mode in ["running", "idling"] and
              prev_mode in ["stopped", "offline"] do
    {dc, update(st, started_at: TimeSupport.utc_now())}
  end

  # if we just transitioned to stopped or offline from any state
  # clear the started at time
  defp started_at({dc, {:ok, %State{state: state} = st}}, _prev_mode)
       when state in ["stopped", "offline"] do
    {dc, update(st, started_at: nil)}
  end

  # all other cases, don't change started at
  defp started_at(passthrough, _prev_mode), do: passthrough

  defp shift_ms(ms),
    do:
      TimeSupport.utc_now()
      |> Timex.shift(milliseconds: ms)

  defp update(%State{} = st, opts) when is_list(opts) do
    opts = add_state_at(opts)
    set = Keyword.take(opts, possible_changes()) |> Enum.into(%{})

    cs = changeset(st, set)

    if cs.valid? do
      Logger.debug(["state update\n", inspect(cs, pretty: true)])
      st = update!(cs)
      {:ok, st}
    else
      {:invalid_changes, cs}
    end
  end

  defp update_states_only(%State{} = st, opts)
       when is_list(opts) do
    opts = add_state_at(opts)

    set =
      (Keyword.take(opts, [:state, :dev_state, :state_at]) ++
         nil_phase_at_times())
      |> Enum.into(%{})

    cs = changeset(st, set)

    if cs.valid? do
      Logger.debug([
        "state update_states_only cs ",
        inspect(cs, pretty: true)
      ])

      st = update!(cs)
      {:ok, st}
    else
      {:invalid_changes, cs}
    end
  end
end