Examples with UniverseManager org.jboss.galleon.cli.UniverseManager used on opensource projects

Example 1 with UniverseManager

use of org.jboss.galleon.cli.UniverseManager in project galleon by wildfly.

the class FPLocationCompleter method doComplete.

private void doComplete(PmCompleterInvocation completerInvocation) throws ProvisioningException {
    // Legacy completer first
    PmSession pmSession = completerInvocation.getPmSession();
    UniverseManager resolver = pmSession.getUniverse();
    installation = completerInvocation.getCommand() instanceof CommandWithInstallationDirectory ? ((CommandWithInstallationDirectory) completerInvocation.getCommand()).getInstallationDirectory(completerInvocation.getAeshContext()) : null;
    UniverseSpec defaultUniverse = pmSession.getUniverse().getDefaultUniverseSpec(installation);
    Set<String> aliases = pmSession.getUniverse().getUniverseNames(installation);
    // producer[@universe]:channel/frequency#build
    // producer[@factory-id/location]:channel/frequency#build
    String buffer = completerInvocation.getGivenCompleteValue();
    List<String> candidates = new ArrayList<>();
    FPLocationParser.ParsedFPLocation loc = null;
    try {
        if (buffer.isEmpty()) {
            if (defaultUniverse != null) {
                getAllProducers(null, defaultUniverse, resolver.getUniverse(defaultUniverse), candidates);
            }
            for (String name : aliases) {
                UniverseSpec u = pmSession.getUniverse().getUniverseSpec(installation, name);
                if (u.getFactory().equals(LegacyGalleon1UniverseFactory.ID)) {
                    continue;
                }
                if (!u.equals(defaultUniverse)) {
                    getAllProducers(u.toString(), u, resolver.getUniverse(u), candidates);
                }
            }
        } else {
            loc = FPLocationParser.parse(buffer, new FPLocationParser.FPLocationCompletionConsumer() {

                @Override
                public void completeProducer(String producer) throws FPLocationParserException, ProvisioningException {
                    // Lookup in all universes for a producer, we don't know the universe yet
                    if (defaultUniverse != null) {
                        getProducers(producer, null, resolver.getUniverse(defaultUniverse), candidates);
                    }
                    for (String name : aliases) {
                        UniverseSpec u = pmSession.getUniverse().getUniverseSpec(installation, name);
                        if (!u.equals(defaultUniverse)) {
                            getProducers(producer, name, resolver.getUniverse(u), candidates);
                        }
                    }
                }

                @Override
                public void completeUniverse(FPLocationParser.ParsedFPLocation parsedLocation, String universe) throws FPLocationParserException, ProvisioningException {
                    for (String name : aliases) {
                        UniverseSpec spec = pmSession.getUniverse().getUniverseSpec(installation, name);
                        if (spec != null && resolver.getUniverse(spec).hasProducer(parsedLocation.getProducer())) {
                            if (name.equals(universe)) {
                                candidates.add(name + FeaturePackLocation.CHANNEL_START);
                            } else if (name.startsWith(universe)) {
                                candidates.add(name);
                            }
                        }
                    }
                }

                @Override
                public void completeUniverseLocation(FPLocationParser.ParsedFPLocation parsedLocation, String universeLocation) throws FPLocationParserException, ProvisioningException {
                    for (String name : aliases) {
                        UniverseSpec spec = pmSession.getUniverse().getUniverseSpec(installation, name);
                        if (spec == null || spec.getFactory().equals(LegacyGalleon1UniverseFactory.ID)) {
                            continue;
                        }
                        if (!candidates.contains(spec.getLocation())) {
                            if (spec.getFactory().equals(parsedLocation.getUniverseFactory()) && resolver.getUniverse(spec).hasProducer(parsedLocation.getProducer())) {
                                if (spec.getLocation().equals(universeLocation)) {
                                    candidates.add(spec.getLocation() + FeaturePackLocation.UNIVERSE_LOCATION_END);
                                } else if (spec.getLocation().startsWith(universeLocation)) {
                                    candidates.add(spec.getLocation());
                                }
                                break;
                            }
                        }
                    }
                }

                @Override
                public void completeChannel(FPLocationParser.ParsedFPLocation parsedLocation, String channel) throws FPLocationParserException, ProvisioningException {
                    Producer<?> p = getProducer(parsedLocation, pmSession);
                    if (p == null) {
                        return;
                    }
                    for (Channel c : p.getChannels()) {
                        if (c.getName().equals(channel)) {
                            // Do nothing, do not inline separators. Separators are to be added explicitly
                            // this could be revisited.
                            candidates.add(channel);
                        } else if (c.getName().startsWith(channel)) {
                            candidates.add(c.getName());
                        }
                    }
                }

                @Override
                public void completeFrequency(FPLocationParser.ParsedFPLocation parsedLocation, String frequency) throws FPLocationParserException, ProvisioningException {
                    Producer<?> p = getProducer(parsedLocation, pmSession);
                    if (p == null) {
                        return;
                    }
                    for (String freq : p.getFrequencies()) {
                        if (freq.equals(frequency)) {
                            // Do not inline the build separator, separator is to be added explicitly
                            // this could be revisited.
                            candidates.add(freq);
                        } else if (freq.startsWith(frequency)) {
                            candidates.add(freq);
                        }
                    }
                }

                @Override
                public void completeChannelSeparator(FPLocationParser.ParsedFPLocation parsedLocation) throws FPLocationParserException, ProvisioningException {
                    candidates.add("" + FeaturePackLocation.CHANNEL_START);
                }

                @Override
                public void completeBuild(FPLocationParser.ParsedFPLocation parsedLocation, String build) throws FPLocationParserException, ProvisioningException {
                    UniverseSpec spec = null;
                    if (parsedLocation.getUniverseName() != null) {
                        spec = pmSession.getUniverse().getUniverseSpec(installation, parsedLocation.getUniverseName());
                    } else if (parsedLocation.getUniverseFactory() == null) {
                        spec = pmSession.getUniverse().getDefaultUniverseSpec(installation);
                    } else {
                        spec = new UniverseSpec(parsedLocation.getUniverseFactory(), parsedLocation.getUniverseLocation());
                    }
                    if (spec != null) {
                        String latestBuild = null;
                        // FPID
                        if (parsedLocation.getFrequency() == null) {
                            FeaturePackLocation.FPID id = new FeaturePackLocation(spec, parsedLocation.getProducer(), parsedLocation.getChannel(), null, null).getFPID();
                            latestBuild = pmSession.getUniverse().getUniverse(spec).getProducer(parsedLocation.getProducer()).getChannel(parsedLocation.getChannel()).getLatestBuild(id);
                        } else {
                            FeaturePackLocation loc = new FeaturePackLocation(spec, parsedLocation.getProducer(), parsedLocation.getChannel(), parsedLocation.getFrequency(), null);
                            latestBuild = pmSession.getUniverse().getUniverse(spec).getProducer(parsedLocation.getProducer()).getChannel(parsedLocation.getChannel()).getLatestBuild(loc);
                        }
                        if (latestBuild != null) {
                            if (latestBuild.startsWith(build)) {
                                candidates.add(latestBuild);
                            }
                        }
                    }
                }
            });
        }
    } catch (Exception ex) {
        CliLogging.completionException(ex);
        return;
    }
    completerInvocation.addAllCompleterValues(candidates);
    if (candidates.size() == 1) {
        if (completerInvocation.getGivenCompleteValue().endsWith(candidates.get(0))) {
            completerInvocation.setAppendSpace(true);
        } else {
            completerInvocation.setAppendSpace(false);
        }
        completerInvocation.setOffset(completerInvocation.getGivenCompleteValue().length() - (loc == null ? 0 : (loc.getMarker() + 1)));
    }
}