org.uberfire.ext.apps.impl.AppsPersistenceImpl Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 JBoss, by Red Hat, Inc
 *
 * 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.uberfire.ext.apps.impl;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.annotation.PostConstruct;
import javax.enterprise.context.ApplicationScoped;
import javax.inject.Inject;
import javax.inject.Named;

import org.jboss.errai.bus.server.annotations.Service;
import org.uberfire.ext.apps.api.AppsPersistenceAPI;
import org.uberfire.ext.apps.api.Directory;
import org.uberfire.ext.layout.editor.api.PerspectiveServices;
import org.uberfire.ext.layout.editor.api.editor.LayoutTemplate;
import org.uberfire.ext.plugin.type.TagsConverterUtil;
import org.uberfire.io.IOService;
import org.uberfire.java.nio.file.DirectoryStream;
import org.uberfire.java.nio.file.FileSystem;
import org.uberfire.java.nio.file.Files;
import org.uberfire.java.nio.file.Path;
import org.uberfire.java.nio.file.StandardDeleteOption;

@Service
@ApplicationScoped
public class AppsPersistenceImpl implements AppsPersistenceAPI {

    public static final String HOME_DIR = ".app_dir/home";
    @Inject
    PerspectiveServices perspectiveServices;
    @Inject
    @Named("ioStrategy")
    private IOService ioService;

    @Inject
    @Named("pluginsFS")
    private FileSystem fileSystem;

    private Path root;

    @PostConstruct
    public void setup() {
        this.root = fileSystem.getRootDirectories().iterator().next();
    }

    @Override
    public Directory getRootDirectory() {

        final Map> tagMap = generateTagMap();

        final Directory root = buildDirectories(tagMap);

        return root;
    }

    private Map> generateTagMap() {
        Map> tagsMap = new HashMap<>();
        final Collection layoutTemplates = perspectiveServices.listLayoutTemplates();
        for (LayoutTemplate layoutTemplate : layoutTemplates) {
            List tags = TagsConverterUtil.extractTags(layoutTemplate.getLayoutProperties());
            for (String tag : tags) {
                List perspectives = tagsMap.get(tag.toUpperCase());
                if (perspectives == null) {
                    perspectives = new ArrayList<>();
                }
                perspectives.add(layoutTemplate.getName());
                tagsMap.put(tag.toUpperCase(),
                            perspectives);
            }
        }

        return tagsMap;
    }

    private Directory buildDirectories(Map> tagMap) {
        Path homeDir = getHomeDir();

        Directory root = new Directory(homeDir.getFileName().toString(),
                                       homeDir.toString(),
                                       homeDir.toUri().toString(),
                                       tagMap);

        root.addChildDirectories(extractAllChildDirectories(root,
                                                            homeDir));

        return root;
    }

    private List extractAllChildDirectories(Directory parent,
                                                       Path dir) {

        List childs = new ArrayList();

        if (ioService.exists(dir) && Files.isDirectory(dir)) {
            final DirectoryStream paths = ioService.newDirectoryStream(dir);
            for (Path childPath : paths) {
                if (Files.isDirectory(childPath)) {
                    final Directory child = getDirectory(childPath.getFileName().toString(),
                                                         childPath.toString(),
                                                         childPath.toUri().toString(),
                                                         parent);
                    final List childsOfChilds = extractAllChildDirectories(child,
                                                                                      childPath);
                    child.addChildDirectories(childsOfChilds);
                    childs.add(child);
                }
            }
        }
        return childs;
    }

    private Directory getDirectory(String name,
                                   String fullpath,
                                   String uri,
                                   Directory parent) {
        return new Directory(name,
                             fullpath,
                             uri,
                             parent);
    }

    @Override
    public Directory createDirectory(Directory parentDirectory,
                                     String name) {
        final Path parentDir = recursiveSearchForDir(getHomeDir(),
                                                     parentDirectory);
        Path newDir = parentDir.resolve(name);
        if (!ioService.exists(newDir)) {
            createDir(newDir);
        }
        newDir = ioService.get(newDir.toUri());
        return getDirectory(name,
                            newDir.toString(),
                            newDir.toUri().toString(),
                            parentDirectory);
    }

    @Override
    public Boolean deleteDirectory(String uri) {
        Path dir = ioService.get(uri);
        return ioService.deleteIfExists(dir,
                                        StandardDeleteOption.NON_EMPTY_DIRECTORIES);
    }

    private Path recursiveSearchForDir(Path dir,
                                       Directory parentDirectory) {
        if (ioService.exists(dir) && Files.isDirectory(dir)) {
            if (isThisPathRelativeToThisDir(dir,
                                            parentDirectory)) {
                return dir;
            } else {
                Path desiredPath = null;
                final DirectoryStream paths = ioService.newDirectoryStream(dir);
                for (Path path : paths) {
                    if (Files.isDirectory(path)) {
                        desiredPath = recursiveSearchForDir(path,
                                                            parentDirectory);
                    }
                    if (desiredPath != null) {
                        break;
                    }
                }
                return desiredPath;
            }
        }
        return null;
    }

    private boolean isThisPathRelativeToThisDir(Path dir,
                                                Directory parentDirectory) {
        return dir.getFileName().toString().equals(parentDirectory.getName());
    }

    private Path getHomeDir() {
        final Path homeDir = root.resolve(HOME_DIR);

        if (!ioService.exists(homeDir)) {
            createDir(homeDir);
        }
        return homeDir;
    }

    private void createDir(Path dir) {
        final Path dummy_file = dir.resolve("dummy_file");
        ioService.write(dummy_file,
                        ".");
    }
}