/**
    * This file Copyright (c) 2019 Magnolia International
    * Ltd.  (http://www.magnolia-cms.com). All rights reserved.
    *
    *
    * This file is dual-licensed under both the Magnolia
    * Network Agreement and the GNU General Public License.
    * You may elect to use one or the other of these licenses.
    *
   * This file is distributed in the hope that it will be
   * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the
   * implied warranty of MERCHANTABILITY or FITNESS FOR A
   * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT.
   * Redistribution, except as permitted by whichever of the GPL
   * or MNA you select, is prohibited.
   *
   * 1. For the GPL license (GPL), you can redistribute and/or
   * modify this file under the terms of the GNU General
   * Public License, Version 3, as published by the Free Software
   * Foundation.  You should have received a copy of the GNU
   * General Public License, Version 3 along with this program;
   * if not, write to the Free Software Foundation, Inc., 51
   * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
   *
   * 2. For the Magnolia Network Agreement (MNA), this file
   * and the accompanying materials are made available under the
   * terms of the MNA which accompanies this distribution, and
   * is available at http://www.magnolia-cms.com/mna.html
   *
   * Any modifications to this file must keep this entire header
   * intact.
   *
   */
  package info.magnolia.ui.form.field.definition.migration;
  
  import info.magnolia.ui.contentapp.browser.BrowserSubAppDescriptor;
  import info.magnolia.ui.contentapp.configuration.ContentViewDefinition;
  import info.magnolia.ui.contentapp.configuration.TreeViewDefinition;
  import info.magnolia.ui.contentapp.configuration.WorkbenchDefinition;
  import info.magnolia.ui.contentapp.configuration.column.ColumnDefinition;
  import info.magnolia.ui.contentapp.configuration.column.ConfiguredColumnDefinition;
  import info.magnolia.ui.datasource.jcr.JcrDatasourceDefinition;
  import info.magnolia.ui.chooser.definition.AppAwareWorkbenchChooserDefinition;
  import info.magnolia.ui.datasource.DatasourceDefinition;
  import info.magnolia.ui.vaadin.integration.contentconnector.ConfiguredJcrContentConnectorDefinition;
  import info.magnolia.ui.vaadin.integration.contentconnector.NodeTypeDefinition;
  import info.magnolia.ui.workbench.tree.TreePresenterDefinition;
  
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.List;
  import java.util.Optional;
  import java.util.stream.Collectors;
  
  import org.apache.commons.lang3.StringUtils;
  
  
  /**
   * Compatibility chooser definition which can auto-resolve datasource from old apps.
   *
   * @param <T> type of items to be chosen
   */
  public class CompatibilityAppAwareWorkbenchChooserDefinition<T> extends AppAwareWorkbenchChooserDefinition<T> {
  
      @Override
      public DatasourceDefinition getDatasource() {
          try {
              return super.getDatasource();
          } catch (IllegalStateException e) {
              // App is most likely from old framework
              return resolveDatasourceFromAppName();
          }
      }
  
      @Override
      public WorkbenchDefinition<T> getWorkbench() {
          try {
              return super.getWorkbench();
          } catch (IllegalStateException e) {
              // App is most likely from old framework
              return resolveWorkbenchDefinitionFromAppName();
          }
      }
  
      private WorkbenchDefinition<T> resolveWorkbenchDefinitionFromAppName() {
          Optional<List<ColumnDefinition<T>>> columns = resolveBrowserDescriptorFromAppName()
                  .map(BrowserSubAppDescriptor::getWorkbench)
                  .map(workbench -> workbench.getContentViews()
                          .stream()
                          .filter(TreePresenterDefinition.class::isInstance)
                          .map(TreePresenterDefinition.class::cast)
                          .findFirst())
                  .map(treePresenterDefinition -> treePresenterDefinition
                          .map(TreePresenterDefinition::getColumns)
                          .get().stream()
                          .filter(info.magnolia.ui.workbench.column.definition.ColumnDefinition::isDisplayInChooseDialog)
                          .filter(columnDefinition -> !"path".equals(columnDefinition.getName()))
                          .map(oldColumnDefinition -> {
                              ConfiguredColumnDefinition<T> column = new ConfiguredColumnDefinition<>();
                             column.setName(oldColumnDefinition.getPropertyName());
                             column.setLabel(StringUtils.defaultString(oldColumnDefinition.getLabel(), oldColumnDefinition.getName()));
                             return column;
                         })
                         .collect(Collectors.toList()));
 
         TreeViewDefinition<T> treeDefinition = new TreeViewDefinition<>();
         treeDefinition.setColumns(columns.orElse(Collections.emptyList()));
 
         List<ContentViewDefinition<T>> contentViews = new ArrayList<>();
         contentViews.add(treeDefinition);
 
         WorkbenchDefinition<T> definition = new WorkbenchDefinition<>();
         definition.setContentViews(contentViews);
         return definition;
     }
 
     private JcrDatasourceDefinition resolveDatasourceFromAppName() {
         return resolveBrowserDescriptorFromAppName()
                 .map(BrowserSubAppDescriptor::getContentConnector)
                 .filter(ConfiguredJcrContentConnectorDefinition.class::isInstance)
                 .map(ConfiguredJcrContentConnectorDefinition.class::cast)
                 .map(contentConnectorDefinition -> {
                     JcrDatasourceDefinition jcrDatasourceDefinition = new JcrDatasourceDefinition();
                     jcrDatasourceDefinition.setWorkspace(contentConnectorDefinition.getWorkspace());
                     jcrDatasourceDefinition.setRootPath(contentConnectorDefinition.getRootPath());
                     jcrDatasourceDefinition.setAllowedNodeTypes(contentConnectorDefinition.getNodeTypes().stream()
                             .map(NodeTypeDefinition::getName)
                             .collect(Collectors.toSet()));
                     jcrDatasourceDefinition.setIncludeProperties(contentConnectorDefinition.isIncludeProperties());
                     return jcrDatasourceDefinition;
                 })
                 .orElse(null);
     }
 
     private Optional<BrowserSubAppDescriptor> resolveBrowserDescriptorFromAppName() {
         return Optional.ofNullable(getAppName())
                 .map(getAppDescriptorRegistry()::getProvider)
                 .map(provider -> provider.get().getSubApps().values()
                         .stream()
                         .filter(BrowserSubAppDescriptor .class::isInstance)
                         .map(BrowserSubAppDescriptor .class::cast)
                         .findFirst()
                         .orElseThrow(() -> new IllegalStateException("Could not resolve browser descriptor from app " + getAppName())));
     }
 }