module HammerCLICsv
  module Utils
    module Subscriptions
      SUBSCRIPTIONS = 'Subscriptions'
      SUBS_NAME = 'Subscription Name'
      SUBS_TYPE = 'Subscription Type'
      SUBS_QUANTITY = 'Subscription Quantity'
      SUBS_SKU = 'Subscription SKU'
      SUBS_CONTRACT = 'Subscription Contract'
      SUBS_ACCOUNT = 'Subscription Account'
      SUBS_START = 'Subscription Start'
      SUBS_END = 'Subscription End'

      def get_all_subscriptions(organization)
        @api.resource(:subscriptions).call(:index, {
            :per_page => 999999,
            'organization_id' => foreman_organization(:name => organization)
        })['results']
      end

      def get_subscription(organization, options = {})
        @subscriptions ||= {}
        @subscriptions[organization] ||= {}

        if options[:name]
          return nil if options[:name].nil? || options[:name].empty?
          options[:id] = @subscriptions[organization][options[:name]]
          if !options[:id]
            results = @api.resource(:subscriptions).call(:index, {
                :per_page => 999999,
                'organization_id' => foreman_organization(:name => organization),
                'search' => "name = \"#{options[:name]}\""
            })
            raise "No subscriptions match '#{options[:name]}'" if results['subtotal'] == 0
            raise "Too many subscriptions match '#{options[:name]}'" if results['subtotal'] > 1
            subscription = results['results'][0]
            @subscriptions[organization][options[:name]] = subscription['id']
            options[:id] = @subscriptions[organization][options[:name]]
            raise "Subscription '#{options[:name]}' not found" if !options[:id]
          end
          result = options[:id]
        else
          return nil if options[:id].nil?
          options[:name] = @subscriptions.key(options[:id])
          if !options[:name]
            subscription = @api.resource(:subscriptions).call(:show, {'id' => options[:id]})
            raise "Subscription '#{options[:name]}' not found" if !subscription || subscription.empty?
            options[:name] = subscription['name']
            @subscriptions[options[:name]] = options[:id]
          end
          result = options[:name]
        end

        result
      end

      def matches_by_sku_and_name(matches, line, subscriptions)
        if line[SUBS_SKU]
          matches = subscriptions.select do |subscription|
            line[SUBS_SKU] == subscription['product_id']
          end
          raise _("No subscriptions match SKU '%{sku}'") % {:sku => line[SUBS_SKU]} if matches.empty?
        elsif line[SUBS_NAME]
          matches = subscriptions.select do |subscription|
            line[SUBS_NAME] == subscription['name']
          end
          raise _("No subscriptions match name '%{name}'") % {:name => line[SUBS_NAME]} if matches.empty?
        end
        matches
      end

      def matches_by_type(matches, line)
        if line[SUBS_TYPE] == 'Red Hat' || line[SUBS_TYPE] == 'Custom'
          matches = matches.select do |subscription|
            subscription['type'] == 'NORMAL'
          end
        elsif line[SUBS_TYPE] == 'Red Hat Guest'
          matches = matches.select do |subscription|
            subscription['type'] == 'STACK_DERIVED'
          end
        elsif line[SUBS_TYPE] == 'Red Hat Temporary'
          matches = matches.select do |subscription|
            subscription['type'] == 'UNMAPPED_GUEST'
          end
        end
        raise _("No subscriptions match type '%{type}'") % {:type => line[SUBS_TYPE]} if matches.empty?
        matches
      end

      def matches_by_account(matches, line)
        if matches.length > 1 && line[SUBS_ACCOUNT]
          refined = matches.select do |subscription|
            line[SUBS_ACCOUNT] == subscription['account_number']
          end
          matches = refined unless refined.empty?
        end
        matches
      end

      def matches_by_contract(matches, line)
        if matches.length > 1 && line[SUBS_CONTRACT]
          refined = matches.select do |subscription|
            line[SUBS_CONTRACT] == subscription['contract_number']
          end
          matches = refined unless refined.empty?
        end
        matches
      end

      def matches_by_quantity(matches, line)
        if line[SUBS_QUANTITY] && line[SUBS_QUANTITY] != 'Automatic'
          refined = matches.select do |subscription|
            subscription['available'] == -1 || line[SUBS_QUANTITY].to_i <= subscription['available']
          end
          raise _("No '%{name}' subscription with quantity %{quantity} or more available") %
            {:name => matches[0]['name'], :quantity => line[SUBS_QUANTITY]} if refined.empty?
          matches = refined
        end
        matches
      end

      def match_with_quantity_to_attach(match, line)
        if line[SUBS_QUANTITY] && line[SUBS_QUANTITY] != 'Automatic' && !line[SUBS_QUANTITY].empty?
          match['quantity'] = line[SUBS_QUANTITY]
        else
          match['quantity'] = -1
        end
        match
      end

      # Subscription amount, SKU, name, contract number, and account number separated by '|'
      # or simply the subscription name.
      def split_subscription_details(details)
        details = details.split('|')
        details.length == 1 ? ['Automatic', nil, details[0], nil, nil] : details
      end
    end
  end
end