ro.fortsoft.pf4j.spring.boot.Pf4jUpdateAutoConfiguration Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2018, hiwepy (https://github.com/hiwepy).
 *
 * 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 ro.fortsoft.pf4j.spring.boot;

import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

import org.springframework.beans.factory.ObjectProvider;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.client.RestTemplate;

import ro.fortsoft.pf4j.PluginManager;
import ro.fortsoft.pf4j.spring.boot.ext.property.Pf4jPluginRepoProperties;
import ro.fortsoft.pf4j.spring.boot.ext.update.RestTemplateUpdateRepository;
import ro.fortsoft.pf4j.update.DefaultUpdateRepository;
import ro.fortsoft.pf4j.update.UpdateManager;
import ro.fortsoft.pf4j.update.UpdateRepository;

/**
 * TODO
 * @author 		： hiwepy
 */
@Configuration
@AutoConfigureAfter({ Pf4jMavenAutoConfiguration.class })
@ConditionalOnClass({ UpdateManager.class })
@ConditionalOnProperty(prefix = Pf4jUpdateProperties.PREFIX, value = "enabled", havingValue = "true")
@EnableConfigurationProperties({Pf4jUpdateProperties.class})
public class Pf4jUpdateAutoConfiguration {
	
	@Bean
	public UpdateManager updateManager(
			PluginManager pluginManager,
			@Autowired(required = false) ObjectProvider repoProvider,
			@Autowired(required = false) ObjectProvider restTemplateProvider,
			Pf4jUpdateProperties properties) {
		
		List repositories = new ArrayList();
		if(!CollectionUtils.isEmpty(properties.getRepos())) {
			for (Pf4jPluginRepoProperties repo : properties.getRepos()) {
				repositories.add(new DefaultUpdateRepository(repo.getId(), repo.getUrl(), repo.getPluginsJsonFileName()));
			}
		}
		List repos = repoProvider.orderedStream().collect(Collectors.toList());
		if(!CollectionUtils.isEmpty(repos)) {
			for (UpdateRepository newRepo : repos) {
				if (newRepo != null) {
					repositories.add(newRepo);
				}
			}
		}
		if(StringUtils.hasText(properties.getReposRestPath())) {
			restTemplateProvider.ifAvailable(restTemplate -> {
				repositories.add(new RestTemplateUpdateRepository(properties.getReposRestPath(), restTemplate));
			});
		}
		UpdateManager updateManager = null;
		if (StringUtils.hasText(properties.getReposJsonPath())) {
			updateManager = new UpdateManager(pluginManager, Paths.get(properties.getReposJsonPath()));
			if(!CollectionUtils.isEmpty(repositories)) {
				updateManager.setRepositories(repositories);
			}
		} else if(!CollectionUtils.isEmpty(repositories)) {
			updateManager = new UpdateManager(pluginManager, repositories);
		} else {
			updateManager = new UpdateManager(pluginManager);
		}
		
		return updateManager;
	}
	
}