cloudfoundry/cf

require "cf/cli/space/base"

module CF
  module Space
    class Create < Base
      desc "Create a space in an organization"
      group :spaces
      input :name, :desc => "Space name", :argument => :optional
      input :organization, :desc => "Parent organization",
        :argument => :optional, :aliases => ["--org", "-o"],
        :from_given => by_name(:organization),
        :default => proc { client.current_organization }
      input :target, :desc => "Switch to the space after creation",
        :alias => "-t", :default => false
      input :manager, :desc => "Add yourself as manager", :default => true
      input :developer, :desc => "Add yourself as developer", :default => true
      input :auditor, :desc => "Add yourself as auditor", :default => false
      input :find_if_exists, :desc => "Use an existing space if one already exists with the given name", :default => false
      def create_space
        space = client.space
        space.organization = org
        space.name = input[:name]

        begin
          with_progress("Creating space #{c(space.name, :name)}") { space.create! }
        rescue CFoundry::SpaceNameTaken
          raise unless input[:find_if_exists]
          space = client.space_by_name(input[:name])
          already_exists = true
        end

        if input[:manager]
          with_progress("Adding you as a manager") { space.add_manager client.current_user }
        end

        if input[:developer]
          with_progress("Adding you as a developer") { space.add_developer client.current_user }
        end

        if input[:auditor]
          with_progress("Adding you as an auditor") { space.add_auditor client.current_user }
        end

        if input[:target]
          invoke :target, :organization => org, :space => space
        else
          if already_exists
            line c("Space already exists!", :good)
            line
            line "#{b("cf switch-space #{space.name}")}    # targets existing space"
          else    
            line c("Space created!", :good)
            line
            line "#{b("cf switch-space #{space.name}")}    # targets new space"
          end
        end
      end

      private

      def ask_name
        ask("Name")
      end
    end
  end
end