All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.netbeans.modules.analysis.ui.Nodes Maven / Gradle / Ivy

The newest version!
/*
 * 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.
 */

package org.netbeans.modules.analysis.ui;

import java.awt.Image;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.HashSet;
import java.util.IdentityHashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.swing.Action;
import org.netbeans.api.project.FileOwnerQuery;
import org.netbeans.api.project.Project;
import org.netbeans.modules.analysis.AnalysisResult;
import org.netbeans.modules.analysis.DescriptionReader;
import org.netbeans.modules.analysis.SPIAccessor;
import org.netbeans.modules.analysis.spi.Analyzer.AnalyzerFactory;
import org.netbeans.modules.analysis.spi.Analyzer.WarningDescription;
import org.netbeans.spi.editor.hints.ErrorDescription;
import org.netbeans.spi.project.ui.LogicalViewProvider;
import org.openide.actions.OpenAction;
import org.openide.cookies.LineCookie;
import org.openide.cookies.OpenCookie;
import org.openide.filesystems.FileObject;
import org.openide.filesystems.FileUtil;
import org.openide.loaders.DataObject;
import org.openide.loaders.DataObjectNotFoundException;
import org.openide.nodes.AbstractNode;
import org.openide.nodes.ChildFactory;
import org.openide.nodes.Children;
import org.openide.nodes.FilterNode;
import org.openide.nodes.Node;
import org.openide.nodes.NodeOp;
import org.openide.text.Line;
import org.openide.text.Line.ShowOpenType;
import org.openide.text.Line.ShowVisibilityType;
import org.openide.text.PositionBounds;
import org.openide.util.Exceptions;
import org.openide.util.Lookup;
import org.openide.util.NbBundle.Messages;
import org.openide.util.UserQuestionException;
import org.openide.util.lookup.Lookups;

/**
 *
 * @author lahvac
 */
public class Nodes {

    private static final Logger LOG = Logger.getLogger(Nodes.class.getName());
    
    public static Node constructSemiLogicalView(final AnalysisResult errors, final boolean byCategory) {
        return new AbstractNode(Children.create(new ChildFactory() {
            @Override protected boolean createKeys(List toPopulate) {
                constructSemiLogicalView(errors, byCategory, toPopulate);
                return true;
            }
            @Override protected Node createNodeForKey(Node key) {
                return key;
            }
        }, true));
    }
    
    private static void constructSemiLogicalView(AnalysisResult errors, boolean byCategory, List toPopulate) {
        for (Node n : errors.extraNodes) {
            toPopulate.add(n.cloneNode());
        }
        
        if (!byCategory) {
            toPopulate.addAll(constructSemiLogicalViewNodes(new LogicalViewCache(), sortErrors(errors.provider2Hints, BY_FILE), errors.errorsToProjects));
        } else {
//            Map> analyzerId2Description = new HashMap>();
            Map>> byCategoryId = sortErrors(errors.provider2Hints, new ByCategoryRetriever(errors.analyzerId2Description.get()));
            List categoryNodes = new ArrayList(byCategoryId.size());
            LogicalViewCache lvc = new LogicalViewCache();

            for (Entry>> categoryEntry : byCategoryId.entrySet()) {
                Map>> byId = sortErrors(categoryEntry.getValue(), BY_ID);
                List warningTypNodes = new ArrayList(byId.size());
                long categoryWarnings = 0;

                for (Entry>> typeEntry : byId.entrySet()) {
                    AnalyzerFactory analyzer = typeEntry.getValue().keySet().iterator().next();
                    final Image icon = SPIAccessor.ACCESSOR.getAnalyzerIcon(analyzer);

                    WarningDescription wd = typeEntry.getKey() != null ? findWarningDescription(errors.analyzerId2Description.get(), analyzer, typeEntry.getKey()) : null;
                    String typeDisplayName = wd != null ? SPIAccessor.ACCESSOR.getWarningDisplayName(wd) : null;
                    long typeWarnings = 0;

                    for (List v1 : typeEntry.getValue().values()) {
                        typeWarnings += v1.size();
                    }

                    final String typeHtmlDisplayName = (typeDisplayName != null ? translate(typeDisplayName) : "Unknown") + " (" + typeWarnings + ")";
                    AbstractNode typeNode = new AbstractNode(constructSemiLogicalViewChildren(lvc, sortErrors(typeEntry.getValue(), BY_FILE), errors.errorsToProjects)) {
                        @Override public Image getIcon(int type) {
                            return icon;
                        }
                        @Override public Image getOpenedIcon(int type) {
                            return icon;
                        }
                        @Override public String getHtmlDisplayName() {
                            return typeHtmlDisplayName;
                        }
                        @Override public Action[] getActions(boolean context) {
                            return new Action[0];
                        }
                    };

                    warningTypNodes.add(typeNode);
                    categoryWarnings += typeWarnings;
                }
                
                warningTypNodes.sort(new Comparator() {
                    @Override public int compare(Node o1, Node o2) {
                        return o1.getDisplayName().compareTo(o2.getDisplayName());
                    }
                });

                AnalyzerFactory analyzer = categoryEntry.getValue().keySet().iterator().next();//TODO: multiple Analyzers for this category
                final Image icon = SPIAccessor.ACCESSOR.getAnalyzerIcon(analyzer);
                final String categoryHtmlDisplayName = translate(categoryEntry.getKey()) + " (" + categoryWarnings + ")";
                AbstractNode categoryNode = new AbstractNode(new DirectChildren(warningTypNodes)) {
                    @Override public Image getIcon(int type) {
                        return icon;
                    }
                    @Override public Image getOpenedIcon(int type) {
                        return icon;
                    }
                    @Override public String getHtmlDisplayName() {
                        return categoryHtmlDisplayName;
                    }
                    @Override public Action[] getActions(boolean context) {
                        return new Action[0];
                    }
                };

                categoryNodes.add(categoryNode);
            }

            categoryNodes.sort(new Comparator() {
                @Override public int compare(Node o1, Node o2) {
                    return o1.getDisplayName().compareTo(o2.getDisplayName());
                }
            });

            toPopulate.addAll(categoryNodes);
        }
    }

    private static  Map>> sortErrors(Map> errs, AttributeRetriever attributeRetriever) {
        Map>> sorted = new HashMap>>();

        for (Entry> e : errs.entrySet()) {
            for (ErrorDescription ed : e.getValue()) {
                if (ed == null) {
                    //XXX:
                    LOG.log(Level.FINE, "null ErrorDescription produced by {0} ({1})", new Object[] {SPIAccessor.ACCESSOR.getAnalyzerDisplayName(e.getKey()), e.getKey().getClass()});
                    continue;
                }
                
                A attribute = attributeRetriever.getAttribute(e.getKey(), ed);
                Map> errorsPerAttributeValue = sorted.get(attribute);

                if (errorsPerAttributeValue == null) {
                    sorted.put(attribute, errorsPerAttributeValue = new HashMap>());
                }

                List errors = errorsPerAttributeValue.get(e.getKey());

                if (errors == null) {
                    errorsPerAttributeValue.put(e.getKey(), errors = new ArrayList());
                }

                errors.add(ed);
            }
        }

        return sorted;
    }

    private static interface AttributeRetriever {
        public A getAttribute(AnalyzerFactory a, ErrorDescription ed);
    }

    private static final AttributeRetriever BY_FILE = new AttributeRetriever() {
        @Override public FileObject getAttribute(AnalyzerFactory a, ErrorDescription ed) {
            return ed.getFile();
        }
    };

    private static final AttributeRetriever BY_ID = new AttributeRetriever() {
        @Override public String getAttribute(AnalyzerFactory a, ErrorDescription ed) {
            return ed.getId();
        }
    };

    private static final class ByCategoryRetriever implements AttributeRetriever {
        private final Map> analyzerId2Description;
        public ByCategoryRetriever(Map> analyzerId2Description) {
            this.analyzerId2Description = analyzerId2Description;
        }
        @Override public String getAttribute(AnalyzerFactory a, ErrorDescription ed) {
            String id = ed.getId();

            if (id == null) {
                Logger.getLogger(Nodes.class.getName()).log(Level.FINE, "No ID for: {0}", ed.toString());
                return "Unknown";
            }

            WarningDescription wd = findWarningDescription(analyzerId2Description, a, id);

            if (wd == null) return "Unknown";
            else return SPIAccessor.ACCESSOR.getWarningCategoryDisplayName(wd);
        }
    }

    private static WarningDescription findWarningDescription(Map> analyzerId2Description, AnalyzerFactory a, String id) {
        Map warnings = analyzerId2Description.get(a);
        
        return warnings != null ? warnings.get(id) : null;
    }

    private static Children constructSemiLogicalViewChildren(final LogicalViewCache lvc, final Map>> errors, final Map errorsToProjects) {
        return Children.create(new ChildFactory() {
            @Override protected boolean createKeys(List toPopulate) {
                toPopulate.addAll(constructSemiLogicalViewNodes(lvc, errors, errorsToProjects));
                return true;
            }
            @Override protected Node createNodeForKey(Node key) {
                return key;
            }
        }, true);
    }

    private static List constructSemiLogicalViewNodes(LogicalViewCache lvc, Map>> errors, Map errorsToProjects) {
        Map>>> projects = new IdentityHashMap>>>();
        Map> file2Projects = new HashMap<>();
        for (Map.Entry>> fileEntry : errors.entrySet()) {
            if (!errorsToProjects.isEmpty()) {
                for (Iterator>> analyzerEntryIt = fileEntry.getValue().entrySet().iterator(); analyzerEntryIt.hasNext();) {
                    final Map.Entry> analyzerEntry = analyzerEntryIt.next();
                    for (Iterator errorIt = analyzerEntry.getValue().iterator(); errorIt.hasNext();) {
                        final ErrorDescription error = errorIt.next();
                        Project project = errorsToProjects.get(error);
                        if (project != null) {
                            Set inProjects = file2Projects.get(fileEntry.getKey());
                            if (inProjects == null) {
                                inProjects = new HashSet<>();
                                file2Projects.put(fileEntry.getKey(), inProjects);
                            }
                            inProjects.add(project);
                            Map>> projectErrors = projects.get(project);
                            if (projectErrors == null) {
                                projects.put(project, projectErrors = new HashMap<>());
                            }
                            Map> analyzerErrors = projectErrors.get(fileEntry.getKey());
                            if (analyzerErrors == null) {
                                analyzerErrors = new HashMap<>();
                                projectErrors.put(fileEntry.getKey(), analyzerErrors);
                            }
                            List errorList = analyzerErrors.get(analyzerEntry.getKey());
                            if (errorList == null) {
                                errorList = new ArrayList<>();
                                analyzerErrors.put(analyzerEntry.getKey(), errorList);
                            }
                            errorList.add(error);
                            errorIt.remove();
                        }
                    }
                    if (analyzerEntry.getValue().isEmpty()) {
                        analyzerEntryIt.remove();
                    }
                }
            }
            if (fileEntry.getValue().isEmpty()) {
                continue;
            }
            final FileObject file = fileEntry.getKey();
            Project project = FileOwnerQuery.getOwner(file);
            if (project == null) {
                Logger.getLogger(Nodes.class.getName()).log(Level.WARNING, "Cannot find project for: {0}", FileUtil.getFileDisplayName(file));
            }

            Map>> projectErrors = projects.get(project);

            if (projectErrors == null) {
                projects.put(project, projectErrors = new HashMap>>());
            }

            projectErrors.put(file, errors.get(file));
        }
        
        projects.remove(null);
        Map> needsCacheClean = new HashMap<>();
        for (Map.Entry> f2p : file2Projects.entrySet()) {
            if (f2p.getValue().size() > 1) {
                for (Project p : f2p.getValue()) {
                    Set filesToClear = needsCacheClean.get(p);
                    if (filesToClear == null) {
                        filesToClear = new HashSet<>();
                        needsCacheClean.put(p, filesToClear);
                    }
                    filesToClear.add(f2p.getKey());
                }
            }
        }
        List nodes = new ArrayList(projects.size());
        
        for (Project p : projects.keySet()) {
            final Set filesToClear = needsCacheClean.get(p);
            if (filesToClear != null) {
                lvc.file2FileNode.keySet().removeAll(filesToClear);
            }
            nodes.add(constructSemiLogicalView(p, lvc, projects.get(p)));
        }

        nodes.sort(new Comparator() {
            @Override public int compare(Node o1, Node o2) {
                return o1.getDisplayName().compareToIgnoreCase(o2.getDisplayName());
            }
        });
        
//        Children.Array subNodes = new Children.Array();
//        
//        subNodes.add(nodes.toArray(new Node[0]));
        
        return nodes;
    }
    
    private static Node constructSemiLogicalView(final Project p, LogicalViewCache lvc, final Map>> errors) {
        Node view = lvc.project2LogicalViewRootNode.get(p);
        final LogicalViewProvider lvp = p.getLookup().lookup(LogicalViewProvider.class);
        
        if (view == null) {
            if (lvp != null) {
                view = lvp.createLogicalView();
            } else {
                try {
                    view = DataObject.find(p.getProjectDirectory()).getNodeDelegate();
                } catch (DataObjectNotFoundException ex) {
                    Exceptions.printStackTrace(ex);
                    return new AbstractNode(Children.LEAF);
                }
            }
            
            lvc.project2LogicalViewRootNode.put(p, view);
        }

        int warnings = 0;

        for (Map> v1 : errors.values()) {
            for (List v2 : v1.values()) {
                warnings += v2.size();
            }
        }
        
        Map>> foundNodes = resolveFileNodes(lvp, lvc, view, errors);

        return new Wrapper(view, warnings, foundNodes, true);
    }

    private static Map>> resolveFileNodes(LogicalViewProvider lvp, LogicalViewCache lvc, final Node view, Map>> errors) {
        Map>> fileNodes = new HashMap>>();

        for (Entry>> entry : errors.entrySet()) {
            FileObject file = entry.getKey();
            Map> eds = entry.getValue();
            Node foundChild = lvc.file2FileNode.get(file);
            
            if (foundChild == null) {
                foundChild = locateChild(view, lvp, file);
                if (foundChild == null) {
                    try {
                        foundChild = DataObject.find(file).getNodeDelegate();
                    } catch (DataObjectNotFoundException ex) {
                        LOG.log(Level.FINE, null, ex);
                        continue;
                    }
                }
                lvc.file2FileNode.put(file, foundChild);
            }

            if (foundChild == null) {
                return null;
            }

            fileNodes.put(foundChild, eds);
        }

        return fileNodes;
    }
    
    private static Node locateChild(Node parent, LogicalViewProvider lvp, FileObject file) {
        if (lvp != null) {
            return lvp.findPath(parent, file);
        }

        throw new UnsupportedOperationException("Not done yet");
    }

    private static class Wrapper extends FilterNode {

        private final int warningsCount;

        public Wrapper(Node orig, int warningsCount, Map>> fileNodes, boolean topLevel) {
            super(orig, new WrapperChildren(orig, fileNodes, topLevel), Lookup.EMPTY);
            this.warningsCount = warningsCount;
        }
        
        public Wrapper(Node orig, Map> errors, boolean file) {
            super(orig, new ErrorDescriptionChildren(errors), lookupForFileNode(orig, errors));
            int warnings = 0;
            for (List e : errors.values()) {
                warnings += e.size();
            }
            this.warningsCount = warnings;
        }

        @Override
        public Action[] getActions(boolean context) {
            return new Action[0];
        }

        private String displayName;

        @Override
        public String getDisplayName() {
            if (displayName == null) {
                return getOriginal().getDisplayName();
            }
            return displayName;
        }

        private String htmlDisplayName;
        
        @Override
        public String getHtmlDisplayName() {
            if (htmlDisplayName == null) {
                if (getOriginal().getHtmlDisplayName() == null) {
                    return translate(getOriginal().getDisplayName()) + " (" + warningsCount + ")";
                }
                return getOriginal().getHtmlDisplayName() + " (" + warningsCount + ")";
            }
            return htmlDisplayName;
        }

        private void fireDisplayNameChange() {
            fireDisplayNameChange(null, getDisplayName());
        }

    }

    private static Lookup lookupForFileNode(Node n, Map> errors) {
        return Lookups.fixed();
    }
    
    private static boolean isParent(Node parent, Node child) {
        if (NodeOp.isSon(parent, child)) {
            return true;
        }

        Node p = child.getParentNode();

        if (p == null) {
            return false;
        }

        return isParent(parent, p);
    }
        
    private static class WrapperChildren extends Children.Keys {

        private final Node orig;
        private final java.util.Map>> fileNodes;
        private final List extraNodes;

        public WrapperChildren(Node orig, java.util.Map>> fileNodes, boolean topLevel) {
            super();
            this.orig = orig;
            this.fileNodes = fileNodes;
            if (topLevel) {
                extraNodes = new ArrayList();
                for (Node n : fileNodes.keySet()) {
                    if (!isParent(orig, n)) {
                        extraNodes.add(n);
                    }
                }
            } else {
                extraNodes = Collections.emptyList();
            }
        }

        @Override
        protected void addNotify() {
            if (extraNodes.isEmpty()) {
                setKeys(orig.getChildren().getNodes(true));
            } else {
                List children = new ArrayList();
                children.addAll(Arrays.asList(orig.getChildren().getNodes(true)));
                children.addAll(extraNodes);
                setKeys(children);
            }
        }

        @Override
        protected void removeNotify() {
            setKeys(Collections.emptyList());
        }
        
        @Override
        protected synchronized Node[] createNodes(Node key) {
            if (fileNodes.containsKey(key)) {
                return new Node[] {new Wrapper(key, fileNodes.get(key), true)};
            }
            final java.util.Map>> fileNodesInside = new HashMap>>();
            int warnings = 0;
            for (Entry>> e : fileNodes.entrySet()) {
                if (isParent(key, e.getKey())) {
                    fileNodesInside.put(e.getKey(), e.getValue());
                    for (List w : e.getValue().values()) {
                        warnings += w.size();
                    }
                }
            }
            if (fileNodesInside.isEmpty()) {
                return new Node[0];
            }
            return new Node[] {new Wrapper(key, warnings, fileNodesInside, false)};
        }
        
    }
    
    private static final class DirectChildren extends Children.Keys {

        public DirectChildren(Collection nodes) {
            setKeys(nodes);
        }
        
        @Override
        protected Node[] createNodes(Node key) {
            return new Node[] {key};
        }
    }

    private static final class ErrorDescriptionChildren extends Children.Keys {

        private final java.util.Map error2Analyzer = new HashMap();
        public ErrorDescriptionChildren(java.util.Map> errors) {
            List eds = new ArrayList();

            for (Entry> e : errors.entrySet()) {
                for (ErrorDescription ed : e.getValue()) {
                    error2Analyzer.put(ed, e.getKey());
                    eds.add(ed);
                }
            }

            eds.sort(new Comparator() {
                @Override public int compare(ErrorDescription o1, ErrorDescription o2) {
                    try {
                        try {
                            final PositionBounds r1 = o1.getRange();
                            final PositionBounds r2 = o2.getRange();
                            if (r1 != null) {
                                return r2 != null
                                        ? r1.getBegin().getLine() - r2.getBegin().getLine()
                                        : -1;
                            } else {
                                return r2 != null
                                        ? 1
                                        : o1.getDescription().compareTo(o2.getDescription());
                            }
                        } catch (UserQuestionException uqe) {
                            uqe.confirmed();
                            return compare(o1, o2);
                        }
                    } catch (IOException ex) {
                        throw new IllegalStateException(ex); //XXX
                    }
                }
            });

            setKeys(eds);
        }

        @Override
        protected Node[] createNodes(ErrorDescription key) {
            AnalyzerFactory analyzerFactory = error2Analyzer.get(key);
            if (analyzerFactory==null) {
                return null;
            } else return new Node[] {new ErrorDescriptionNode(analyzerFactory, key)};
        }
    }

    private static final class ErrorDescriptionNode extends AbstractNode {

        private final Image icon;

        public ErrorDescriptionNode(AnalyzerFactory provider, final ErrorDescription ed) {
            super(Children.LEAF, Lookups.fixed(ed, new OpenErrorDescription(provider, ed), new DescriptionReader() {
                @Override public CharSequence getDescription() {
                    return ed.getDetails();
                }
            }));
            int line = -1;
            try {
                final PositionBounds range = ed.getRange();
                if (range != null) {
                    line = range.getBegin().getLine();
                }
            } catch (IOException ex) {
                Exceptions.printStackTrace(ex);
            }
            setDisplayName((line != (-1) ? (line + 1 + ":") : "") + ed.getDescription());
            icon = SPIAccessor.ACCESSOR.getAnalyzerIcon(provider);
        }

        @Override
        public Action getPreferredAction() {
            return OpenAction.get(OpenAction.class);
        }

        @Override
        public Image getIcon(int type) {
            return icon;
        }

        @Override
        public Action[] getActions(boolean context) {
            return new Action[0];
        }
    }

    private static final class OpenErrorDescription implements OpenCookie {

        private final AnalyzerFactory analyzer;
        private final ErrorDescription ed;

        public OpenErrorDescription(AnalyzerFactory analyzer, ErrorDescription ed) {
            this.analyzer = analyzer;
            this.ed = ed;
        }

        @Override
        public void open() {
            openErrorDescription(analyzer, ed);
        }
        
    }

    @Messages("ERR_CannotOpen=Cannot open target file")
    static void openErrorDescription(AnalyzerFactory analyzer, ErrorDescription ed) throws IndexOutOfBoundsException {
        try {
            DataObject od = DataObject.find(ed.getFile());
            LineCookie lc = od.getLookup().lookup(LineCookie.class);

            if (lc != null) {
                Line line = lc.getLineSet().getCurrent(ed.getRange().getBegin().getLine());

                line.show(ShowOpenType.OPEN, ShowVisibilityType.FOCUS);
                analyzer.warningOpened(ed);
            }
        } catch (IOException ex) {
            Exceptions.printStackTrace(Exceptions.attachLocalizedMessage(Exceptions.attachSeverity(ex, Level.WARNING), Bundle.ERR_CannotOpen()));
        }
    }

    private static String[] c = new String[] {"&", "<", ">", "\n", "\""}; // NOI18N
    private static String[] tags = new String[] {"&", "<", ">", "
", """}; // NOI18N private static String translate(String input) { for (int cntr = 0; cntr < c.length; cntr++) { input = input.replaceAll(c[cntr], tags[cntr]); } return input; } interface FutureValue { T get(); } public static final class LogicalViewCache { private final Map project2LogicalViewRootNode = new IdentityHashMap(); private final Map file2FileNode = new HashMap(); } }




© 2015 - 2025 Weber Informatics LLC | Privacy Policy