require 'set'

module VagrantPlugins
  module Vocker
    class Config < Vagrant.plugin("2", :config)
      attr_reader :images, :containers

      def initialize
        @images     = Set.new
        @containers = Hash.new
      end

      def pull_images(*images)
        @images += images.map(&:to_s)
      end

      def run(*args)
        container_name = args.shift
        params         = {}

        if args.empty?
          params[:image] = container_name
        elsif args.first.is_a?(String)
          params[:image] = args.shift
          params[:cmd]   = container_name
        else
          params = args.shift
          params[:cmd] ||= container_name
        end

        @containers[container_name.to_s] = params
      end

      def merge(other)
        super.tap do |result|
          result.pull_images *(other.images + self.images)
        end
      end
    end
  end
end