gchan/wits

require 'wits/nodes'

module Wits
  module FinalInterimPrices
    module ConvenienceMethods

      def self.extended(_base)
        Wits::Nodes.nodes.each do |node, name|
          node_methods(node, name)
          node_prices_methods(node, name)
          node_interim_prices_methods(node, name)
          node_forecast_prices_methods(node, name)
        end
      end

      private

      def self.node_methods(node, name)
        # def ben2201
        create_prices_method(node.downcase, node)

        # def hay
        create_prices_method(node.downcase[0..2], node)

        # def halfway_bush
        create_prices_method(name.downcase.tr(' ', '_'), node)
      end

      def self.node_prices_methods(node, name)
        # def ben2201_prices
        create_prices_method("#{node.downcase}_prices", node)

        # def hay_prices
        create_prices_method("#{node.downcase[0..2]}_prices", node)

        # def halfway_bush_prices
        create_prices_method("#{name.downcase.tr(' ', '_')}_prices", node)
      end

      def self.node_interim_prices_methods(node, name)
        # def ben2201_interim_prices
        create_interim_prices_method(node.downcase, node)

        # def hay_interim_prices
        create_interim_prices_method(node.downcase[0..2], node)

        # def halfway_bush_interim_prices
        create_interim_prices_method(name.downcase.tr(' ', '_'), node)
      end

      def self.node_forecast_prices_methods(node, name)
        # def ben2201_forecast_prices
        create_forecast_method(node.downcase, node)

        # def hay_forecast_prices
        create_forecast_method(node.downcase[0..2], node)

        # def halfway_bush_forecast_prices
        create_forecast_method(name.downcase.tr(' ', '_'), node)
      end

      def self.create_prices_method(method_name, node)
        define_method method_name do |*args|
          prices(node, *args)
        end
      end

      def self.create_interim_prices_method(method_prefix, node)
        define_method "#{method_prefix}_interim_prices" do |*args|
          interim_prices(node, *args)
        end
      end

      def self.create_forecast_method(method_prefix, node)
        define_method "#{method_prefix}_forecast_prices" do |*args|
          forecast_prices(node, *args)
        end
      end
    end
  end
end