/*
 * Copyright 2015 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.cloud.dataflow.admin.config;

import org.springframework.boot.autoconfigure.AutoConfigureBefore;
import org.springframework.boot.autoconfigure.redis.RedisAutoConfiguration;
import org.springframework.cloud.Cloud;
import org.springframework.cloud.CloudFactory;
import org.springframework.cloud.dataflow.module.deployer.ModuleDeployer;
import org.springframework.cloud.dataflow.module.deployer.lattice.LrpModuleDeployer;
import org.springframework.cloud.dataflow.module.deployer.lattice.TaskModuleDeployer;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;
import org.springframework.data.redis.connection.RedisConnectionFactory;

/**
 * Configuration used when running <i>in a cloud</i>, triggered by cloud profiles (cloud, lattice).
 *
 * @author Mark Fisher
 * @author Thomas Risberg
 * @author Ilayaperumal Gopinathan
 */
@Configuration
@Profile("cloud")
public class CloudConfiguration {

	@AutoConfigureBefore(RedisAutoConfiguration.class)
	protected static class RedisConfig {

		@Bean
		public Cloud cloud() {
			return new CloudFactory().getCloud();
		}

		@Bean
		RedisConnectionFactory redisConnectionFactory(Cloud cloud) {
			return cloud.getSingletonServiceConnector(RedisConnectionFactory.class, null);
		}
	}

	@Profile("lattice")
	protected static class LatticeConfig {

		@Bean
		public ModuleDeployer processModuleDeployer() {
			return new LrpModuleDeployer();
		}

		@Bean
		public ModuleDeployer taskModuleDeployer() {
			return new TaskModuleDeployer();
		}
	}
}

@@ -735,16 +735,98 @@
 Before;%0A
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;%0A
 import o
@@ -933,16 +933,65 @@
 .Cloud;%0A
+import org.springframework.cloud.CloudConnector;%0A
 import o
@@ -1266,32 +1266,101 @@
 ModuleDeployer;%0A
+import org.springframework.cloud.lattice.connector.LatticeConnector;%0A
 import org.sprin
@@ -1968,16 +1968,176 @@
 d cloud(
+CloudFactory cloudFactory) %7B%0A%09%09%09return cloudFactory.getCloud();%0A%09%09%7D%0A%0A%09%09@Bean%0A%09%09@ConditionalOnMissingBean(CloudFactory.class)%0A%09%09public CloudFactory cloudFactory(
 ) %7B%0A%09%09%09r
@@ -2160,27 +2160,16 @@
 actory()
-.getCloud()
 ;%0A%09%09%7D%0A%0A%09
@@ -2578,18 +2578,668 @@
 loyer();%0A%09%09%7D
+%0A%0A%09%09@Bean%0A%09%09public CloudFactory cloudFactory() %7B%0A%09%09%09return new LatticeCloudFactory();%0A%09%09%7D%0A%09%7D%0A%0A%09private static class LatticeCloudFactory extends CloudFactory %7B%0A%0A%09%09@Override%0A%09%09public void registerCloudConnector(CloudConnector cloudConnector) %7B%0A%09%09%09// Make sure we only register Lattice connector in case of Lattice environment%0A%09%09%09// Since lattice environment also has %60VCAP_APPLICATION%60 as environment property, the CloudFoundry connector%0A%09%09%09// can also be matched as the possible cloud connectors. Hence, we exclude CF connector on lattice.%0A%09%09%09if (cloudConnector instanceof LatticeConnector) %7B%0A%09%09%09%09super.registerCloudConnector(cloudConnector);%0A%09%09%09%7D%0A%09%09%7D
 %0A%09%7D%0A%7D%0A

Fix CloudConnector for lattice