module CloudstackCli
  module OptionResolver

    def vm_options_to_params
      resolve_zone
      resolve_project
      resolve_compute_offering
      resolve_template
      resolve_disk_offering
      resolve_iso
      unless options[:template_id]
        say "Error: Template or ISO is required.", :red
        exit 1
      end
      resolve_networks
    end

    def resolve_zone
      if options[:zone]
        zones = client.list_zones
        zone = zones.find {|z| z['name'] == options[:zone] }
        if !zone
          msg = options[:zone] ? "Zone '#{options[:zone]}' is invalid." : "No zone found."
          say "Error: #{msg}", :red
          exit 1
        end
        options[:zone_id] = zone['id']
      end
      options
    end

    def resolve_domain
      if options[:domain]
        if domain = client.list_domains(name: options[:domain]).first
          options[:domain_id] = domain['id']
        else
          say "Error: Domain #{options[:domain]} not found.", :red
          exit 1
        end
      end
      options
    end

    def resolve_project
      if options[:project]
        if %w(ALL -1).include? options[:project]
          options[:project_id] = "-1"
        elsif project = client.list_projects(name: options[:project], listall: true).first
          options[:project_id] = project['id']
        else
          say "Error: Project #{options[:project]} not found.", :red
          exit 1
        end
      end
      options
    end

    def resolve_account
      if options[:account]
        if account = client.list_accounts(name: options[:account], listall: true).first
          options[:account_id] = account['id']
          options[:domain_id] = account['domainid']
        else
          say "Error: Account #{options[:account]} not found.", :red
          exit 1
        end
      end
      options
    end

    def resolve_networks
      networks = []
      available_networks = network = client.list_networks(
        zone_id: options[:zone_id],
        project_id: options[:project_id]
      )
      if options[:networks]
        options[:networks].each do |name|
          unless network = available_networks.find { |n| n['name'] == name }
            say "Error: Network '#{name}' not found.", :red
            exit 1
          end
          networks << network['id'] rescue nil
        end
      end
      networks.compact!
      if networks.empty?
        #unless default_network = client.list_networks(project_id: options[:project_id]).find {
        #  |n| n['isdefault'] == true }
        unless default_network = client.list_networks(project_id: options[:project_id]).first
          say "Error: No default network found.", :red
          exit 1
        end
        networks << available_networks.first['id'] rescue nil
      end
      options[:network_ids] = networks.join(',')
      options
    end

    def resolve_iso
      if options[:iso]
        unless iso = client.list_isos(
            name: options[:iso],
            project_id: options[:project_id]
          ).first
          say "Error: Iso '#{options[:iso]}' is invalid.", :red
          exit 1
        end
        unless options[:diskoffering_id]
          say "Error: a disk offering is required when using iso.", :red
          exit 1
        end
        options[:template_id] = iso['id']
        options['hypervisor'] = (options[:hypervisor] || 'vmware')
      end
      options
    end

    def resolve_template
      if options[:template]
        if template = client.list_templates(
            name: options[:template],
            template_filter: "executable",
            project_id: options[:project_id]
          ).first
          options[:template_id] = template['id']
        else
          say "Error: Template #{options[:template]} not found.", :red
          exit 1
        end
      end
      options
    end

    def resolve_compute_offering
      if offering = client.list_service_offerings(name: options[:offering]).first
        options[:service_offering_id] = offering['id']
      else
        say "Error: Offering #{options[:offering]} not found.", :red
        exit 1
      end
      options
    end

    def resolve_disk_offering
      if options[:disk_offering]
        unless disk_offering = client.list_disk_offerings(name: options[:disk_offering]).first
          say "Error: Disk offering '#{options[:disk_offering]}' not found.", :red
          exit 1
        end
        options[:disk_offering_id] = disk_offering['id']
      end
      options
    end

    def resolve_virtual_machine
      if options[:virtual_machine]
        args = { name: options[:virtual_machine], listall: true }
        args[:project_id] = options[:project_id]
        unless vm = client.list_virtual_machines(args).first
          say "Error: VM '#{options[:virtual_machine]}' not found.", :red
          exit 1
        end
        options[:virtual_machine_id] = vm['id']
      end
      options
    end

    def resolve_snapshot
      if options[:snapshot]
        args = { name: options[:snapshot], listall: true }
        args[:project_id] = options[:project_id]
        unless snapshot = client.list_snapshots(args).first
          say "Error: Snapshot '#{options[:snapshot]}' not found.", :red
          exit 1
        end
        options[:snapshot_id] = snapshot['id']
      end
      options
    end

    def resolve_host(type = "routing")
      if options[:host]
        args = { name: options[:host], type: type, listall: true }
        unless host = client.list_hosts(args).first
          say "Error: Host '#{options[:host]}' not found.", :red
          exit 1
        end
        options[:host_id] = host['id']
      end
      options
    end

  end
end