UserMenus.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2019 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * 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 Affero 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/>.
 */
 
package org.turro.elephant.user.menu;
 
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
import org.turro.string.Strings;
import org.turro.action.UserSummaries;
import org.turro.elephant.context.Application;
import org.turro.elephant.context.IConstructor;
import org.turro.path.Path;
import org.turro.reflection.ClassNames;
import org.turro.reflection.Reflections;
 
public class UserMenus {
  
  public static Set<IUserMenu> getUserMenu(IConstructor constructor) {
    if(constructor.getUser() != null) {
      setActive();
      UserSummaries.initSummaries(constructor);
      return userMenus();
    } else {
      return Collections.EMPTY_SET;
    }
  }
  
  public static Object executeMenu(String idMenu) {
    return doExecuteMenu(userMenus(), idMenu);
  }
  
  public static boolean isInRole(String path) {
    if(!Strings.isBlank(path) && path.startsWith("/user/")) {
      IUserMenu userMenu = getUserMenuByPath(userMenus(), Path.reduce(path, 2));
      return userMenu != null ? userMenu.isInRole() : true;
    }
    return true;
  }
 
  private static IUserMenu getUserMenuByPath(Set<IUserMenu> menus, String path) {
    IUserMenu result = null; 
    for(IUserMenu userMenu : menus) {
      if(path.equals(userMenu.getLink())) {
        return userMenu;
      }
      if(userMenu.getChildren() != null) {
        result = getUserMenuByPath(userMenu.getChildren(), path);
      }
      if(result != null) break;
    }
    return result;
  }
 
  private static Object doExecuteMenu(Set<IUserMenu> menus, String idMenu) {
    for(IUserMenu userMenu : menus) {
      if(idMenu.equals(userMenu.getId())) {
        return userMenu.execute();
      }
      if(userMenu.getChildren() != null) doExecuteMenu(userMenu.getChildren(), idMenu);
    }
    return null;
  }
  
  public static IUserMenu getSessionMenu() {
    for(IUserMenu um : userMenus()) {
      if(um instanceof SessionUserMenu) {
        return um;
      }
    }
    return null;
  }
  
  public static IUserMenu getSettingsMenu() {
    for(IUserMenu um : userMenus()) {
      if(um instanceof SettingsUserMenu) {
        return um;
      }
    }
    return null;
  }
  
  public static void setActive() {
    Application.getApplication().getConstructor().setSessionAttribute("currPath_usrmx", 
            Application.getApplication().getConstructor().getCurrentContext().getPath());
  }
 
  public static String getActive() {
    return (String) Application.getApplication().getConstructor().getSessionAttribute("currPath_usrmx");
  }
  
  /* Attributes */
  
  public static Object getAttribute(String attribute) {
    return Application.getApplication().getConstructor().findAttribute("um_" + attribute);
  }
  
  public static void setAttribute(String attribute, Object value) {
    Application.getApplication().getConstructor().setSessionAttribute("um_" + attribute, value);
  }
  
  public static Map<String, Object> getAttributes() {
    return Application.getApplication().getConstructor().getSessionAttributeMap("um_");
  }
  
  public static void removeAttributes() {
    for(String key : getAttributes().keySet()) {
      Application.getApplication().getConstructor().removeSessionAttribute("um_" + key);
    }
  }
 
  /* User menus */
  
  private static TreeSet<IUserMenu> _userMenus;
  private static final Lock lock = new ReentrantLock();
  
  private static Set<IUserMenu> userMenus() {
    if(_userMenus == null) {
      lock.lock();
      try {
        if(_userMenus == null) {
          _userMenus = new TreeSet<>();
          List<String> menus = ClassNames.cached().byAnnotation(UserMenu.class.getName());
          if(menus != null) {
            for(String menu : menus) {
              Object obj = Reflections.of(menu).instance();
              if(obj instanceof IUserMenu) {
                if(((IUserMenu) obj).isValid()) {
                  _userMenus.add((IUserMenu) obj);
                }
              }
            }
          }
          for(IUserMenu um : _userMenus) {
            um.initMenu();
          }
        }
      } finally {
        lock.unlock();
      }
    }
    return _userMenus;
  }
 
  private UserMenus() {
  }
  
}