package fr.ifremer.quadrige3.ui.swing.table.state;
   
   /*-
    * #%L
    * Quadrige3 Core :: Quadrige3 UI Common
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2017 Ifremer
   * %%
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or
   * (at your option) any later version.
   *
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   *
   * You should have received a copy of the GNU Affero General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   * #L%
   */
  
  import fr.ifremer.quadrige3.core.dao.technical.Assert;
  import fr.ifremer.quadrige3.ui.swing.table.ColumnIdentifier;
  import fr.ifremer.quadrige3.ui.swing.table.FixedSwingTable;
  import jaxx.runtime.swing.session.State;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.jdesktop.swingx.table.TableColumnExt;
  
  import javax.swing.table.AbstractTableModel;
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  /**
   * State used for a FixedSwingTable with a AbstractTableModel
   * Only width property is handled
   * <p/>
   * Created by Ludovic on 12/03/2020.
   */
  public class FixedSwingTableSessionState implements ExtendedState {
  
      private static final Log LOG = LogFactory.getLog(FixedSwingTableSessionState.class);
  
      // session states map by context
      private Map<String, FixedSwingTableState> sessionStates;
  
      /**
       * <p>Constructor for SwingTableSessionState.</p>
       */
      public FixedSwingTableSessionState() {
          sessionStates = new HashMap<>();
      }
  
      /**
       * <p>checkComponent.</p>
       *
       * @param object a {@link Object} object.
       * @return a {@link FixedSwingTable} object.
       */
      protected FixedSwingTable checkComponent(Object object) {
          Assert.notNull(object);
          Assert.isInstanceOf(FixedSwingTable.class, object);
          Assert.isInstanceOf(AbstractTableModel.class, ((FixedSwingTable) object).getModel());
          return (FixedSwingTable) object;
      }
  
      /**
       * {@inheritDoc}
       */
      @Override
      public State getState(Object o) {
  
          return getState(o, null, null);
      }
  
      /**
       * {@inheritDoc}
       */
      @Override
      @SuppressWarnings("unchecked")
      public State getState(Object o, State previousState, String stateContextToSave) {
  
          FixedSwingTable table = checkComponent(o);
  
          // Retrieve table model and its state context
          String stateContext = stateContextToSave != null ? stateContextToSave : table.getTableModel().getStateContext();
  
          // Reuse previous state or create new one
          FixedSwingTableSessionState tableSessionState;
          if (previousState instanceof FixedSwingTableSessionState) {
              tableSessionState = (FixedSwingTableSessionState) previousState;
          } else {
              tableSessionState = new FixedSwingTableSessionState();
         }
         try {
             // Get or create the table state for the context
             FixedSwingTableState tableState = tableSessionState.getSessionStates().computeIfAbsent(stateContext, k -> new FixedSwingTableState());
 
             // get tableModel.getIdentifiers() in a new list to prevent ConcurrentModificationException
             List<ColumnIdentifier<?>> identifiers = new ArrayList<ColumnIdentifier<?>>(table.getTableModel().getIdentifiers());
 
             // Iterate over identifiers
             for (ColumnIdentifier<?> identifier : identifiers) {
                 TableColumnExt column = table.getColumnExt(identifier);
                 if (column == null) {
                     // ignore
                     continue;
                 }
 
                 //  width property
                 tableState.getWidth().put(identifier.getPropertyName(), column.getPreferredWidth());
 
             }
 
         } catch (Exception e) {
             LOG.error("Something goes wrong on getState", e);
         }
         return tableSessionState;
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     @SuppressWarnings("unchecked")
     public void setState(Object o, State state) {
         if (state == null) {
             return;
         }
         if (!(state instanceof FixedSwingTableSessionState)) {
             throw new IllegalArgumentException("invalid state");
         }
         FixedSwingTable table = checkComponent(o);
 
         // Retrieve table model
         FixedSwingTableSessionState tableSessionState = (FixedSwingTableSessionState) state;
         FixedSwingTableState tableState = tableSessionState.getSessionStates().get(table.getTableModel().getStateContext());
         if (tableState == null) {
             // Ignore if no session state for this context
             return;
         }
 
         try {
             // get tableModel.getIdentifiers() in a new list to prevent ConcurrentModificationException
             List<ColumnIdentifier<?>> identifiers = new ArrayList<ColumnIdentifier<?>>(table.getTableModel().getIdentifiers());
 
             for (ColumnIdentifier<?> identifier : identifiers) {
                 TableColumnExt column = table.getColumnExt(identifier);
                 if (column == null) {
                     // ignore
                     continue;
                 }
 
                 // Get the width property
                 if (column.getResizable()) {
                     Integer width = tableState.getWidth().get(identifier.getPropertyName());
                     if (width != null) {
                         column.setPreferredWidth(width);
                     }
                 }
 
             }
 
         } catch (Exception e) {
             LOG.error("Something goes wrong on setState", e);
         }
     }
 
     // <Dont' delete> these getters ans setters used by SwingSession
 
     public Map<String, FixedSwingTableState> getSessionStates() {
         return sessionStates;
     }
 
     @SuppressWarnings("unused")
     public void setSessionStates(Map<String, FixedSwingTableState> sessionStates) {
         this.sessionStates = sessionStates;
     }
 
 }