package io.guixer.tools.monkey;
   
   import static com.google.common.base.Preconditions.checkNotNull;
   import static com.google.common.collect.Lists.newArrayList;
   import static io.guixer.tools.Locators.inferLocator;
   import static org.apache.commons.lang3.StringUtils.isBlank;
   import static org.apache.commons.lang3.StringUtils.substringAfterLast;
   
   import java.io.IOException;
  import java.util.List;
  import java.util.Locale;
  
  import javax.annotation.Nullable;
  
  import org.apache.commons.lang3.ArrayUtils;
  import org.apache.commons.lang3.NotImplementedException;
  import org.apache.commons.lang3.RandomUtils;
  import org.openqa.selenium.By;
  import org.openqa.selenium.WebDriver;
  import org.openqa.selenium.WebElement;
  
  import io.guixer.tools.GuixerConsumer;
  import io.guixer.tools.RunnerContext;
  import io.guixer.types.AttributeScope;
  
  public class MonkeySteps implements GuixerConsumer {
  
  	private final int stepCount;
  
  	public MonkeySteps(
  		final int stepCount
  	) {
  
  		this.stepCount = stepCount;
  	}
  
  	@Override
  	public void accept(
  		final RunnerContext context
  	) throws IOException {
  
  		checkNotNull(context, "context");
  
  		for (int stepIndex = 1; stepIndex <= stepCount; ++stepIndex) {
  
  			final WebDriver driver = context.getDriver();
  
  			final List<WebElement> anchors = driver.findElements(By.tagName("a"));
  
  			final List<WebElement> buttons = driver.findElements(By.tagName("button"));
  
  			final List<WebElement> inputs = driver.findElements(By.tagName("input"));
  
  			final List<Interactable> interactables = newArrayList();
  
  			for (final WebElement anchor : anchors) {
  
  				@Nullable
  				final Interactable interactable = Interactable.fromAnchor(driver, anchor);
  
  				if (interactable != null) {
  
  					interactables.add(interactable);
  				}
  			}
  
  			for (final WebElement button : buttons) {
  
  				@Nullable
  				final Interactable interactable = Interactable.fromButton(driver, button);
  
  				if (interactable != null) {
  
  					interactables.add(interactable);
  				}
  			}
  
  			for (final WebElement input : inputs) {
  
  				@Nullable
  				final Interactable interactable = Interactable.fromInput(driver, input);
  
  				if (interactable != null) {
  
  					interactables.add(interactable);
  				}
  			}
  
  			if (interactables.isEmpty()) {
  
  				// context.failure("No more interactable elements");
  
  				// context.message("No more interactable elements");
  
  				// break;
  
  				context.intent("Monkey " + stepIndex + "/" + stepCount + ": No more interactable elements: We go back");
  
  				context.tag(AttributeScope.STEP, "no_more_interactables");
 
 				context.executeScript("history.back()");
 
 				context.sleep(2);
 
 				context.takeScreenshot();
 
 			} else {
 
 				final Interactable interactable = interactables.get(RandomUtils.nextInt(0, interactables.size()));
 
 				context.intent("Monkey " + stepIndex + "/" + stepCount + ": We click at random: " + interactable.label);
 
 				interactable.interact(context);
 
 				context.sleep(2);
 
 				context.takeScreenshot();
 			}
 		}
 	}
 
 	private static abstract class Interactable {
 
 		public final String label;
 
 		private Interactable(
 			final String label
 		) {
 
 			this.label = checkNotNull(label, "label");
 		}
 
 		public abstract void interact(
 			RunnerContext context
 		) throws IOException;
 
 		@Nullable
 		public static Interactable fromAnchor(
 			final WebDriver driver,
 			final WebElement anchor
 		) {
 
 			if (!anchor.isDisplayed() || !anchor.isEnabled()) {
 
 				return null;
 			}
 
 			@Nullable
 			final String href = anchor.getAttribute("href");
 
 			if (href == null) {
 
 				return null;
 			}
 
 			final String label;
 
 			@Nullable
 			final String text = anchor.getText();
 
 			if (!isBlank(text)) {
 
 				label = text;
 
 			} else {
 
 				final List<WebElement> imgs = anchor.findElements(By.tagName("img"));
 
 				if (!imgs.isEmpty()) {
 
 					label = substringAfterLast(imgs.get(0).getAttribute("src"), "/");
 
 				} else {
 
 					throw new NotImplementedException("<a> without text or <img>, href: " + href + ": " + anchor);
 				}
 			}
 
 			// final String locator = inferLocator(driver, anchor);
 
 			return new Interactable(label) {
 
 				@Override
 				public void interact(
 					final RunnerContext context
 				) throws IOException {
 
 					final String locator = inferLocator(driver, anchor);
 
 					context.click(locator);
 				}
 			};
 		}
 
 		@Nullable
 		public static Interactable fromButton(
 			final WebDriver driver,
 			final WebElement button
 		) {
 
 			if (!button.isDisplayed() || !button.isEnabled()) {
 
 				return null;
 			}
 
 			final String label;
 
 			@Nullable
 			final String text = button.getText();
 
 			if (!isBlank(text)) {
 
 				label = text;
 
 			} else {
 
 				throw new NotImplementedException("<button> without text: " + button);
 			}
 
 			// final String locator = inferLocator(driver, button);
 
 			return new Interactable(label) {
 
 				@Override
 				public void interact(
 					final RunnerContext context
 				) throws IOException {
 
 					final String locator = inferLocator(driver, button);
 
 					context.click(locator);
 				}
 			};
 		}
 
 		@Nullable
 		public static Interactable fromInput(
 			final WebDriver driver,
 			final WebElement input
 		) {
 
 			if (!input.isDisplayed() || !input.isEnabled()) {
 
 				return null;
 			}
 
 			@Nullable
 			final String type = input.getAttribute("type");
 
 			if (type == null || !ArrayUtils.contains(new String[] { //
 					"button", //
 					"checkbox", "radio", //
 					"submit" //
 			}, type.toLowerCase(Locale.ENGLISH))) {
 
 				return null;
 			}
 
 			final String label;
 
 			@Nullable
 			final String value = input.getAttribute("value");
 
 			if (!isBlank(value)) {
 
 				label = value;
 
 			} else {
 
 				throw new NotImplementedException("<input> without value: " + input);
 			}
 
 			// final String locator = inferLocator(driver, input);
 
 			return new Interactable(label) {
 
 				@Override
 				public void interact(
 					final RunnerContext context
 				) throws IOException {
 
 					final String locator = inferLocator(driver, input);
 
 					context.click(locator);
 				}
 			};
 		}
 	}
 }