archiva-modules/archiva-web/archiva-rest/archiva-rest-services/src/main/java/org/apache/archiva/rest/services/DefaultPluginsServices.java

package org.apache.archiva.rest.services;
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

import org.apache.archiva.rest.api.services.ArchivaRestServiceException;
import org.apache.archiva.rest.api.services.PluginsService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.core.io.Resource;
import org.springframework.stereotype.Service;

import javax.inject.Inject;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * @author Eric Barboni
 * @since 1.4.0
 */
@Service( "pluginsService#rest" )
public class DefaultPluginsServices
    implements PluginsService
{

    private List<String> repositoryType = new ArrayList<>();

    private List<String> adminFeatures = new ArrayList<>();

    private ApplicationContext applicationContext;

    private Logger log = LoggerFactory.getLogger( getClass() );

    private String adminPlugins;

    @Inject
    public DefaultPluginsServices( ApplicationContext applicationContext )
        throws IOException
    {
        log.debug( "init DefaultPluginsServices" );
        this.applicationContext = applicationContext;

        // rebuild
        repositoryType = feed( "repository" );
        log.debug( "feed {}:{}", "repository" , repositoryType);
        adminFeatures = feed( "features" );
        log.debug( "feed {}:{}", "features", adminFeatures );
        StringBuilder sb = new StringBuilder();
        for ( String repoType : repositoryType )
        {
            sb.append( repoType ).append( "|" );
        }
        for ( String repoType : adminFeatures )
        {
            sb.append( repoType ).append( "|" );
        }
        log.debug( "getAdminPlugins: {}", sb );
        if ( sb.length() > 1 )
        {
            adminPlugins = sb.substring( 0, sb.length() - 1 );
        }
        else
        {
            adminPlugins = sb.toString();
        }
    }

    private List<String> feed( String key )
        throws IOException
    {
        log.info( "Feeding: {}", key );
        Resource[] xmlResources = applicationContext.getResources( "/**/" + key + "/**/main.js" );
        if (xmlResources == null)
        {
            return Collections.emptyList();
        }
        List<String> repository = new ArrayList<>( xmlResources.length );
        for ( Resource rc : xmlResources )
        {
            String tmp = rc.getURL().toString();
            tmp = tmp.substring( tmp.lastIndexOf( key ) + key.length() + 1, tmp.length() - 8 );
            repository.add( "archiva/admin/" + key + "/" + tmp + "/main" );
        }
        return repository;
    }

    @Override
    public String getAdminPlugins()
        throws ArchivaRestServiceException
    {
        return adminPlugins;
    }
}