# frozen_string_literal: true

require "dry-transformer"
require "json"

module DjiMqttConnect
  module Sys::Product
    class StatusMarshal < MessageMarshal
      include Utils::MessageParsing

      # Rename pesky `method` argument to `_method` and makes a copy of the raw data
      class AttributeTransformer < Dry::Transformer::Pipe
        import Dry::Transformer::HashTransformations

        define! do
          copy_keys "data" => "_data"
          rename_keys "method" => "_method"
        end
      end

      # Attempts to look a the method attribute, and builds a specific Message class for the message
      def load(raw_message)
        # Parse the message from JSON
        parsed_message = parse_json(raw_message)

        # Transform the message
        transformed_message = attribute_transformer.call(parsed_message)
        message_method = transformed_message["_method"]

        # Build an instance of the class, or a generic message from the current class
        message_class = message_class_from_message_method(message_method, StatusMessage)
        build_message(message_class, transformed_message)
      end

      private

      def attribute_transformer
        @attribute_transformer ||= AttributeTransformer.new
      end
    end
  end
end