CustomNavigationTransformer.java

/*
 * Copyright 2016 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed under the Eclipse Public License version 1.0, available at
 * http://www.eclipse.org/legal/epl-v10.html
 */

package org.jboss.forge.addon.swarm.ui;

import java.util.stream.StreamSupport;

import org.jboss.forge.addon.ui.context.UINavigationContext;
import org.jboss.forge.addon.ui.result.NavigationResult;
import org.jboss.forge.addon.ui.result.navigation.NavigationResultBuilder;
import org.jboss.forge.addon.ui.result.navigation.NavigationResultTransformer;

/**
 *
 * @author <a href="mailto:[email protected]">George Gastaldi</a>
 */
public class CustomNavigationTransformer implements NavigationResultTransformer
{

   @Override
   public boolean handles(UINavigationContext context)
   {
      if (context.getCurrentCommand() instanceof AddFractionCommand)
      {
         AddFractionCommand cmd = (AddFractionCommand) context.getCurrentCommand();
         return StreamSupport.stream(cmd.getFractionElements().getValue().spliterator(), false)
                  .anyMatch((f) -> f.artifactId().equals("jaxrs"));
      }
      return false;
   }

   @Override
   public NavigationResult transform(UINavigationContext context, NavigationResult original)
   {
      return NavigationResultBuilder.create(original).add(CreateMainClassCommand.class).build();
   }

   @Override
   public int priority()
   {
      return 0;
   }

}