/*
 * Copyright (c) 2012 - 2019 the original author or authors.
 *
 * 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 com.github.ferstl.maven.pomenforcers.model;

import java.util.Collection;
import java.util.Map;

import com.github.ferstl.maven.pomenforcers.priority.PriorityOrdering;
import com.github.ferstl.maven.pomenforcers.priority.PriorityOrderingFactory;
import com.google.common.base.Function;
import com.google.common.collect.Maps;

import static com.github.ferstl.maven.pomenforcers.model.functions.StringStartsWithEquivalence.stringStartsWith;
import static java.util.Objects.requireNonNull;

public enum PluginElement implements PriorityOrderingFactory<String, PluginModel>, Function<PluginModel, String> {

  GROUP_ID("groupId") {
    @Override
    public PriorityOrdering<String, PluginModel> createPriorityOrdering(Collection<String> priorityCollection) {
      return new PriorityOrdering<>(priorityCollection, this, stringStartsWith());
    }

    @Override
    public String apply(PluginModel input) {
      return input.getGroupId();
    }
  },

  ARTIFACT_ID("artifactId") {
    @Override
    public PriorityOrdering<String, PluginModel> createPriorityOrdering(Collection<String> priorityCollection) {
      return new PriorityOrdering<>(priorityCollection, this, stringStartsWith());
    }

    @Override
    public String apply(PluginModel input) {
      return input.getArtifactId();
    }
  };

  private static final Function<String, PluginElement> STRING_TO_PLUGIN_ELEMENT =
      new StringToPluginElementTransformer();

  private static final Map<String, PluginElement> elementMap;

  static {
    elementMap = Maps.newLinkedHashMap();
    for (PluginElement element : values()) {
      elementMap.put(element.getElementName(), element);
    }
  }

  private static PluginElement getByElementName(String elementName) {
    requireNonNull(elementName, "Element name is null");

    PluginElement result = elementMap.get(elementName);
    if (result == null) {
      throw new IllegalArgumentException("No plugin element with name " + elementName);
    }

    return result;
  }

  public static Function<String, PluginElement> stringToPluginElement() {
    return STRING_TO_PLUGIN_ELEMENT;
  }

  private final String elementName;

  private PluginElement(String elementName) {
    this.elementName = elementName;
  }

  public String getElementName() {
    return this.elementName;
  }

  private static class StringToPluginElementTransformer implements Function<String, PluginElement> {
    @Override
    public PluginElement apply(String input) {
      return getByElementName(input);
    }
  }
}

@@ -723,17 +723,16 @@
 il.Map;%0A
-%0A
 import c
@@ -951,17 +951,16 @@
 t.Maps;%0A
-%0A
 import s
@@ -2290,337 +2290,8 @@
  %7D%0A%0A
-  private static PluginElement getByElementName(String elementName) %7B%0A    requireNonNull(elementName, %22Element name is null%22);%0A%0A    PluginElement result = elementMap.get(elementName);%0A    if (result == null) %7B%0A      throw new IllegalArgumentException(%22No plugin element with name %22 + elementName);%0A    %7D%0A%0A    return result;%0A  %7D%0A%0A
   pu
@@ -2356,24 +2356,24 @@
 Element() %7B%0A
+
     return S
@@ -2696,16 +2696,17 @@
 ment%3E %7B%0A
+%0A
     @Ove
@@ -2750,32 +2750,32 @@
 String input) %7B%0A
-
       return get
@@ -2797,19 +2797,364 @@
 nput);%0A    %7D
+%0A%0A    private static PluginElement getByElementName(String elementName) %7B%0A      requireNonNull(elementName, %22Element name is null%22);%0A%0A      PluginElement result = elementMap.get(elementName);%0A      if (result == null) %7B%0A        throw new IllegalArgumentException(%22No plugin element with name %22 + elementName);%0A      %7D%0A%0A      return result;%0A    %7D
 %0A  %7D%0A%7D%0A

Move getElementByName() to StringToPluginElementTransformer