View Javadoc
1   /*
2    * Copyright (c) 2002-2021, City of Paris
3    * All rights reserved.
4    *
5    * Redistribution and use in source and binary forms, with or without
6    * modification, are permitted provided that the following conditions
7    * are met:
8    *
9    *  1. Redistributions of source code must retain the above copyright notice
10   *     and the following disclaimer.
11   *
12   *  2. Redistributions in binary form must reproduce the above copyright notice
13   *     and the following disclaimer in the documentation and/or other materials
14   *     provided with the distribution.
15   *
16   *  3. Neither the name of 'Mairie de Paris' nor 'Lutece' nor the names of its
17   *     contributors may be used to endorse or promote products derived from
18   *     this software without specific prior written permission.
19   *
20   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
21   * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23   * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
24   * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25   * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26   * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27   * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28   * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29   * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30   * POSSIBILITY OF SUCH DAMAGE.
31   *
32   * License 1.0
33   */
34  package fr.paris.lutece.plugins.ctv.util;
35  
36  import java.util.ArrayList;
37  import java.util.Arrays;
38  import java.util.List;
39  
40  import com.google.gson.Gson;
41  
42  import fr.paris.lutece.plugins.ctv.view.object.JsonResultViewObject;
43  import fr.paris.lutece.plugins.unittree.business.unit.Unit;
44  import fr.paris.lutece.plugins.unittree.business.unit.Unit.TypeUnit;
45  
46  public class UnitTreeUtils
47  {
48  
49      private static final String ROOT_LABEL = "DVD";
50  
51      public static String getUnitTreeJson( List<Unit> units )
52      {
53          Gson json = new Gson( );
54          List<JsonResultViewObject> jsonList = filterResult( units, ROOT_LABEL );
55          return json.toJson( jsonList );
56      }
57  
58      public static String getUnitTreeJson( List<Unit> units, Integer... selectedUnitId )
59      {
60          return getUnitTreeJson( units, false, selectedUnitId );
61      }
62  
63      public static String getUnitTreeJson( List<Unit> units, boolean disabled, Integer... selectedUnitId )
64      {
65          List<Integer> selectedUnitIds = selectedUnitId != null ? Arrays.asList( selectedUnitId ) : new ArrayList<Integer>( );
66          Gson json = new Gson( );
67          List<JsonResultViewObject> jsonList = filterResult( units, ROOT_LABEL );
68          for ( JsonResultViewObject jsonResultViewObject : jsonList )
69          {
70              if ( selectedUnitIds.contains( Integer.parseInt( jsonResultViewObject.getId( ) ) ) )
71              {
72                  jsonResultViewObject.getState( ).put( "selected", true );
73              }
74              else
75                  if ( disabled )
76                  {
77                      jsonResultViewObject.getState( ).put( "disabled", true );
78                  }
79          }
80          return json.toJson( jsonList );
81      }
82  
83      private static List<JsonResultViewObject> filterResult( List<Unit> allUnitWithInfo, String root )
84      {
85  
86          List<JsonResultViewObject> inputList = new ArrayList<>( );
87          List<Unit> allUnitWithInfoFiltered = filterList( allUnitWithInfo, root );
88  
89          for ( Unit unit : allUnitWithInfoFiltered )
90          {
91              if ( TypeUnit.VILLE.equals( unit.getTypeunit( ) ) )
92              {
93                  String id = String.valueOf( unit.getIdUnit( ) );
94                  String parentId = String.valueOf( unit.getIdParent( ) );
95  
96                  if ( unit.getLabel( ).equals( root ) )
97                  {
98                      parentId = "#";
99                  }
100                 inputList.add( new JsonResultViewObject( id, parentId, unit.getLabel( ) ) );
101             }
102         }
103 
104         return inputList;
105     }
106 
107     private static List<Unit> filterList( List<Unit> allUnitWithInfo, String root )
108     {
109         List<String> elementsToRemove = new ArrayList<>( );
110         List<Unit> tmp = new ArrayList<>( );
111         int idParentToRemove = 0;
112         elementsToRemove.add( String.valueOf( idParentToRemove ) );
113 
114         for ( Unit unit : allUnitWithInfo )
115         {
116             if ( TypeUnit.VILLE.equals( unit.getTypeunit( ) ) )
117             {
118                 idParentToRemove = unit.getIdUnit( );
119                 if ( elementsToRemove.contains( String.valueOf( unit.getIdParent( ) ) ) && !unit.getLabel( ).equals( root ) )
120                 {
121                     elementsToRemove.add( String.valueOf( idParentToRemove ) );
122                 }
123             }
124         }
125 
126         for ( Unit child : allUnitWithInfo )
127         {
128             if ( TypeUnit.VILLE.equals( child.getTypeunit( ) ) )
129             {
130                 if ( !elementsToRemove.contains( String.valueOf( ( child.getIdUnit( ) ) ) ) )
131                 {
132                     tmp.add( child );
133                 }
134             }
135         }
136 
137         return tmp;
138     }
139 
140 }