/builder/extra/tools/ruby/configure/packages/configure_service_package.rb

TEMP_DEPLOYMENT_SERVICE = "temp"

class ConfigureServicePackage < ConfigurePackage
  SERVICE_CREATE = "create_service"
  SERVICE_DELETE = "delete_service"
  SERVICE_UPDATE = "update_service"
  
  @deploy_service_key = nil
  
  def parsed_options?(arguments)
    if Configurator.instance.display_help? && !Configurator.instance.display_preview?()
      return true
    end
    
    arguments = load_target_config(arguments)
    
    unless Configurator.instance.display_preview?
      unless @config.props.size > 0
        error("Unable to run configure-service because this directory has not been setup")
        error("Try running this command from the directory where Tungsten is installed or adding --host, --user and --release-directory for the server and directory you are trying to modify. These values should match what you provided when calling 'tools/configure'.")
        return false
      end
    end
    
    @config.setProperty(DEPLOYMENT_TYPE, nil)
    @config.setProperty(DEPLOY_CURRENT_PACKAGE, nil)
    @config.setProperty(DEPLOY_PACKAGE_URI, nil)
    
    service_config = Properties.new
    
    datasource_options = Properties.new()
    datasource_options.setProperty([DATASOURCES, "master"], {})
    datasource_options.setProperty([DATASOURCES, "ds"], {})
    
    opts=OptionParser.new
    
    opts.on("-C", "--create")   { @config.setProperty(DEPLOYMENT_TYPE, SERVICE_CREATE) }
    opts.on("-D", "--delete")   { @config.setProperty(DEPLOYMENT_TYPE, SERVICE_DELETE) }
    opts.on("-U", "--update")   { @config.setProperty(DEPLOYMENT_TYPE, SERVICE_UPDATE) }
    
    each_service_prompt{
      |prompt|
      if (av = prompt.get_command_line_argument_value()) != nil
        opts.on("--#{prompt.get_command_line_argument()}") {
          service_config.setProperty(prompt.name, av)
        }
      else
        opts.on("--#{prompt.get_command_line_argument()} String") {
          |val|
          service_config.setProperty(prompt.name, val)
        }
      end
      
      if Configurator.instance.advanced_mode?()
        case prompt.class().name()
        when "ReplicationServiceDatasource"
          each_datasource_prompt{
            |dsp|
            if (av = dsp.get_command_line_argument_value()) != nil
              opts.on("--#{dsp.get_command_line_argument()}") {
                datasource_options.setProperty([DATASOURCES, "ds", dsp.name], av)
              }
            else
              opts.on("--#{dsp.get_command_line_argument()} String") {
                |val|
                datasource_options.setProperty([DATASOURCES, "ds", dsp.name], val)
              }
            end
          }
        when "ReplicationServiceMasterDatasource"
          each_datasource_prompt{
            |dsp|
            if (av = dsp.get_command_line_argument_value()) != nil
              opts.on("--master-#{dsp.get_command_line_argument()}") {
                datasource_options.setProperty([DATASOURCES, "master", dsp.name], av)
              }
            else
              opts.on("--master-#{dsp.get_command_line_argument()} String") {
                |val|
                datasource_options.setProperty([DATASOURCES, "master", dsp.name], val)
              }
            end
          }
          opts.on("--disable-relay-logs") {
            datasource_options.setProperty([DATASOURCES, "master", REPL_DISABLE_RELAY_LOGS], "true")
            datasource_options.setProperty([DATASOURCES, "ds", REPL_DISABLE_RELAY_LOGS], "true")
          }
        end
      end
    }
    opts.on("--master-host String") {
      |val|
      service_config.setProperty(REPL_MASTERHOST, val)
      warning("--master-host is deprecated, use --master-thl-host instead.")
    }
    
    begin
      remainder = Configurator.instance.run_option_parser(opts, arguments, false, "Invalid option for configure-service")
      
      unless @config.getNestedProperty([DEPLOYMENT_TYPE])
        error("You must specify -C, -D or -U")
      end
      
      begin
        ds_alias = service_config.getProperty(REPL_DATASOURCE)
        if ds_alias && @config.getPropertyOr(DATASOURCES, {}).keys().include?(ds_alias)
          if datasource_options.getPropertyOr([DATASOURCES, "ds"], {}).size > 0
            raise "The '#{ds_alias}' datasource already exists, but you provided configuration settings for it."
          else
            # The datasource is properly configured
          end
        else
          if datasource_options.getPropertyOr([DATASOURCES, "ds"], {}).size > 0
            unless ds_alias
              ds_alias = datasource_options.getProperty([DATASOURCES, "ds", REPL_DBHOST]).gsub(".", "_") + "_" + datasource_options.getProperty([DATASOURCES, "ds", REPL_DBPORT])
            end
            if @config.getPropertyOr(DATASOURCES, {}).keys().include?(ds_alias)
              raise "The '#{ds_alias}' datasource already exists, but you provided configuration settings for it."
            end
            service_config.setProperty(REPL_DATASOURCE, ds_alias)
            @config.setProperty([DATASOURCES, ds_alias], datasource_options.getProperty([DATASOURCES, "ds"]))
          elsif @config.getNestedProperty([DEPLOYMENT_TYPE]) == SERVICE_CREATE
            raise "You must specify a datasource alias or datasource configuration information"
          end
        end
      rescue => e
        error(e.message)
      end
      
      begin
        master_alias = service_config.getProperty(REPL_MASTER_DATASOURCE)
        if master_alias && @config.getPropertyOr(DATASOURCES, {}).keys().include?(master_alias)
          if datasource_options.getPropertyOr([DATASOURCES, "master"], {}).size > 0
            raise "The '#{master_alias}' master datasource already exists, but you provided configuration settings for it."
          else
            # The master datasource is properly configured
          end
        else
          if service_config.getProperty(REPL_ROLE) == REPL_ROLE_DI && datasource_options.getPropertyOr([DATASOURCES, "master"], {}).size > 0
            unless master_alias
              master_alias = datasource_options.getProperty([DATASOURCES, "master", REPL_DBHOST]).gsub(".", "_") + "_" + datasource_options.getProperty([DATASOURCES, "master", REPL_DBPORT])
            end
            if @config.getPropertyOr(DATASOURCES, {}).keys().include?(master_alias)
              raise "The '#{master_alias}' master datasource already exists, but you provided configuration settings for it."
            end
            service_config.setProperty(REPL_MASTER_DATASOURCE, master_alias)
            @config.setProperty([DATASOURCES, master_alias], datasource_options.getProperty([DATASOURCES, "master"]))
          end
        end
      rescue => e
        error(e.message)
      end
      
      case remainder.size()
      when 0
        raise "No service_name specified"
      when 1
        @deploy_service_key = false
        @config.getPropertyOr(REPL_SERVICES, {}).each_key{
          |s_key|
          if @config.getProperty([REPL_SERVICES, s_key, DEPLOYMENT_SERVICE]) == remainder[0]
            @deploy_service_key = s_key
          end
        }
        
        case @config.getProperty(DEPLOYMENT_TYPE)
        when SERVICE_CREATE
          if @deploy_service_key != false
            raise "A service named '#{remainder[0]}' already exists"
          else
            @deploy_service_key = remainder[0]
            service_config.setProperty(DEPLOYMENT_SERVICE, @deploy_service_key)
            @config.setProperty([REPL_SERVICES, remainder[0]], service_config.props)
          end
          
          @config.setProperty([REPL_SERVICES, remainder[0], FIXED_PROPERTY_STRINGS], 
            Configurator.instance.fixed_properties)
        when SERVICE_UPDATE
          if @deploy_service_key == false
            raise "Unable to find an existing service config for '#{remainder[0]}'"
          else
            service_config.props.each{
              |sc_key, sc_val|
              @config.setProperty([REPL_SERVICES, @deploy_service_key, sc_key], sc_val)
            }
            
            @config.setProperty([REPL_SERVICES, @deploy_service_key, FIXED_PROPERTY_STRINGS], 
              @config.getPropertyOr([REPL_SERVICES, @deploy_service_key, FIXED_PROPERTY_STRINGS], []) + 
              Configurator.instance.fixed_properties)
          end
        when SERVICE_DELETE
          if @deploy_service_key == false
            raise "Unable to find an existing service config for '#{remainder[0]}'"
          end
        end
      else
        raise "Multiple service names specified: #{remainder.join(', ')}"
      end
    rescue => e
      error("Argument parsing failed: #{e.to_s()}")
    end
    
    if Configurator.instance.display_preview?()
      if @deploy_service_key == nil
        @deploy_service_key = 'service-name'
      end
      
      service_config.props.each{
        |sc_key, sc_val|
        @config.setProperty([REPL_SERVICES, @deploy_service_key, sc_key], sc_val)
      }
    end
    
    @config.setProperty(DEPLOYMENT_SERVICE, @deploy_service_key)
    
    is_valid?()
  end
  
  def output_usage
    if Configurator.instance.display_preview?
      svc = @deploy_service_key
      applier = @config.getProperty([REPL_SERVICES, svc, REPL_DATASOURCE])
      extractor = @config.getProperty([REPL_SERVICES, svc, REPL_MASTER_DATASOURCE])
    end
    
    puts "Usage: configure-service [general-options] {-C|-D|-U} [target-options] [service-options] service-name"
    output_general_usage()
    
    Configurator.instance.write_divider()
    puts "Target options:"
    output_usage_line("--host", "Host to connect to configure the service", @target_host)
    output_usage_line("--user", "User to connect to the host as", @target_user)
    output_usage_line("--release-directory", "The release directory that holds the Tungsten runtime files", @target_home_directory)
    
    Configurator.instance.write_divider()
    puts "Service options:"
    output_usage_line("-C", "Create a replication service")
    output_usage_line("-D", "Delete a replication service")
    output_usage_line("-U", "Update a replication service")
    
    each_service_prompt{
      |prompt|
      if Configurator.instance.display_preview?
        prompt.set_member(svc)
      end
      prompt.output_usage()
      
      if Configurator.instance.advanced_mode?()
        case prompt.class().name()
        when "ReplicationServiceDatasource"
          output_usage_line("--disable-relay-logs", "Disable the use of relay-logs?")
          
          each_datasource_prompt{
            |prompt|
            if Configurator.instance.display_preview?
              prompt.set_member(applier)
            end
            
            prompt.output_usage()
          }
        when "ReplicationServiceMasterDatasource"
          each_datasource_prompt{
            |prompt|
            if Configurator.instance.display_preview?
              prompt.set_member(extractor)
            end
            
            output_usage_line("--master-#{prompt.get_command_line_argument()}", prompt.get_prompt(), prompt.get_value(true, true), nil, prompt.get_prompt_description())
          }
        end
      end
    }
  end
  
  def get_prompts
    [
      Datasources.new(),
      ReplicationServices.new(),
      DeploymentServicePrompt.new()
    ]
  end
  
  def get_non_interactive_prompts
    cluster_prompts = ConfigurePackageCluster.new(@config).get_prompts()
    cluster_prompts.delete_if{ |prompt| 
      (prompt.is_a?(ReplicationServices) || prompt.is_a?(Datasources))
    }
    
    cluster_prompts
  end
  
  def get_validation_checks
    [
      ReplicationServiceChecks.new()
    ]
  end
  
  def allow_interactive?
    false
  end
  
  def allow_batch?
    false
  end
end

module ConfigureServicePrompt
  def enabled_for_command_line?
    if Configurator.instance.package.is_a?(ConfigureServicePackage)
      super() && true
    else
      false
    end
  end
end