org.apache.openmeetings.web.common.GroupChoiceProvider.java Source code

Introduction

Here is the source code for org.apache.openmeetings.web.common.GroupChoiceProvider.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.openmeetings.web.common;

import static org.apache.openmeetings.web.app.WebSession.getUserId;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Locale;

import org.apache.openmeetings.db.dao.user.GroupDao;
import org.apache.openmeetings.db.dao.user.UserDao;
import org.apache.openmeetings.db.entity.user.Group;
import org.apache.openmeetings.db.entity.user.GroupUser;
import org.apache.openmeetings.db.entity.user.User;
import org.apache.openmeetings.web.app.WebSession;
import org.apache.wicket.injection.Injector;
import org.apache.wicket.spring.injection.annot.SpringBean;
import org.apache.wicket.util.string.Strings;
import org.wicketstuff.select2.ChoiceProvider;
import org.wicketstuff.select2.Response;

public class GroupChoiceProvider extends ChoiceProvider<Group> {
    private static final long serialVersionUID = 1L;
    @SpringBean
    private GroupDao groupDao;
    @SpringBean
    private UserDao userDao;

    public GroupChoiceProvider() {
        Injector.get().inject(this);
    }

    @Override
    public void query(String term, int page, Response<Group> response) {
        if (WebSession.getRights().contains(User.Right.Admin)) {
            List<Group> groups = groupDao.get(0, Integer.MAX_VALUE);
            for (Group g : groups) {
                if (Strings.isEmpty(term)
                        || g.getName().toLowerCase(Locale.ROOT).contains(term.toLowerCase(Locale.ROOT))) {
                    response.add(g);
                }
            }
        } else {
            User u = userDao.get(getUserId());
            for (GroupUser ou : u.getGroupUsers()) {
                if (Strings.isEmpty(term) || ou.getGroup().getName().toLowerCase(Locale.ROOT)
                        .contains(term.toLowerCase(Locale.ROOT))) {
                    response.add(ou.getGroup());
                }
            }
        }
    }

    @Override
    public Collection<Group> toChoices(Collection<String> ids) {
        Collection<Group> c = new ArrayList<>();
        for (String id : ids) {
            c.add(groupDao.get(Long.valueOf(id)));
        }
        return c;
    }

    @Override
    public String getDisplayValue(Group choice) {
        return choice.getName();
    }

    @Override
    public String getIdValue(Group choice) {
        Long id = choice.getId();
        return id == null ? null : "" + id;
    }
}