RangeProviderForm.java

/**
 * Copyright (C) 2012 Red Hat, Inc. and/or its affiliates.
 *
 * 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.jbpm.formModeler.core.model;

import java.net.URI;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;
import javax.inject.Inject;
import javax.inject.Named;

import org.guvnor.common.services.project.model.Project;
import org.jbpm.formModeler.api.client.FormEditorContext;
import org.jbpm.formModeler.api.client.FormEditorContextManager;
import org.jbpm.formModeler.api.model.RangeProvider;
import org.jbpm.formModeler.editor.service.FormModelerService;
import org.kie.workbench.common.services.datamodeller.util.FileUtils;
import org.kie.workbench.common.services.shared.project.KieProjectService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.uberfire.backend.server.util.Paths;
import org.uberfire.backend.vfs.Path;
import org.uberfire.io.IOService;
import org.uberfire.util.URIUtil;

public class RangeProviderForm implements RangeProvider {

    private Logger log = LoggerFactory.getLogger( RangeProviderForm.class );

    @Inject
    @Named("ioStrategy")
    private IOService ioService;

    @Inject
    private KieProjectService projectService;

    @Inject
    protected FormModelerService formModelerService;

    @Inject
    protected FormEditorContextManager formEditorContextManager;

    @Override
    public String getType() {
        return "{$range_provider_form}";
    }

    @Override
    public Map<String, String> getRangesMap( String namespace ) {
        TreeMap treeMap = new TreeMap<String, String>();

        FormEditorContext context = formEditorContextManager.getRootEditorContext( namespace );

        if ( context == null ) {
            return treeMap;
        }

        Path currentForm = null;
        try {
            currentForm = Paths.convert( ioService.get( new URI( context.getPath() ) ) );
        } catch ( Exception e ) {
            log.warn( "Unable to load asset on '" + context.getPath() + "': ", e );
            return treeMap;
        }
        String currentFormDirUri = getFormDirUri( currentForm );
        String currentFormName = currentForm.getFileName();

        Project project = projectService.resolveProject( currentForm );

        FileUtils utils = FileUtils.getInstance();

        List<org.uberfire.java.nio.file.Path> nioPaths = new ArrayList<org.uberfire.java.nio.file.Path>();
        nioPaths.add( Paths.convert( project.getRootPath() ) );

        Collection<FileUtils.ScanResult> forms = utils.scan( ioService, nioPaths, "form", true );

        Path formPath;
        String formDirUri;
        String formName;
        for ( FileUtils.ScanResult form : forms ) {
            formPath = Paths.convert( form.getFile() );
            formDirUri = getFormDirUri( formPath );
            formName = formPath.getFileName();

            if ( currentFormDirUri.equals( formDirUri ) && !formName.startsWith( "." ) && !currentFormName.equals( formName ) ) {
                treeMap.put( formPath.getFileName(), formPath.getFileName() );
            }
        }
        return treeMap;
    }

    private String getFormDirUri( Path formPath ) {
        String fileName = formPath.getFileName();
        try {
            fileName = URIUtil.encode(fileName);
        } catch ( Exception e ) {

        }
        return formPath.toURI().substring( 0, formPath.toURI().lastIndexOf( fileName ) - 1 );
    }

}