ru.codeinside.gses.webui.data.ControlledTasksQuery.java Source code

Introduction

Here is the source code for ru.codeinside.gses.webui.data.ControlledTasksQuery.java

Source

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 * Copyright (c) 2013, MPL CodeInside http://codeinside.ru
 */

package ru.codeinside.gses.webui.data;

import com.google.common.base.Objects;
import com.google.common.collect.Lists;
import org.activiti.engine.task.Task;
import org.activiti.engine.task.TaskQuery;
import org.apache.commons.lang.StringUtils;
import org.apache.commons.lang.time.DateUtils;
import ru.codeinside.adm.AdminServiceProvider;
import ru.codeinside.adm.database.*;
import ru.codeinside.gses.webui.Flash;
import ru.codeinside.gses.webui.components.TasksQueryFilter;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

import static ru.codeinside.gses.service.impl.DeclarantServiceImpl.*;

public class ControlledTasksQuery extends AbstractLazyLoadingQuery<Task> implements TasksQueryFilter {
    private static final long serialVersionUID = 1L;
    private List<String> controlledGroups;
    private String processInstanceId;
    private String serviceId;
    private ProcedureType type;
    private String taskKey;
    private String procedureId;
    private String declarantTypeName;
    private String declarantTypeValue;
    private String requester;
    private Date fromDate;
    private Date toDate;
    private boolean overdue;
    private List<String> orgGroups;
    private List<String> empGroups;
    private boolean superSupervisor;

    public ControlledTasksQuery(ItemBuilder<Task> itemBuilder, Employee emp) {
        super(itemBuilder);
        superSupervisor = emp.getRoles().contains(Role.SuperSupervisor);
        controlledGroups = new ArrayList<String>(
                emp.getOrganizationGroups().size() + emp.getEmployeeGroups().size());
        for (Group group : emp.getOrganizationGroups()) {
            controlledGroups.add(group.getName());
        }
        for (Group group : emp.getEmployeeGroups()) {
            controlledGroups.add(group.getName());
        }
    }

    List<Task> items(int start, int count) {
        TaskQuery query = createTaskQuery();
        query.orderByTaskPriority().desc();
        if ("name".equals(orderBy)) {
            query.orderByTaskName();
        } else if ("id".equals(orderBy)) {
            query.orderByTaskId();
        } else {
            query.orderByDueDate();
        }
        return listPage(query, start, count);
    }

    private TaskQuery createTaskQuery() {
        TaskQuery query = Flash.flash().getProcessEngine().getTaskService().createTaskQuery();
        ((TaskQueryImpl2) query).setIgnoreAssignee(false);
        List<String> resultGroups;
        if (superSupervisor) {
            if (orgGroups == null && empGroups != null) {
                resultGroups = empGroups;
            } else if (empGroups == null && orgGroups != null) {
                resultGroups = orgGroups;
            } else if (empGroups != null && orgGroups != null) {
                orgGroups.addAll(empGroups);
                resultGroups = orgGroups;
            } else {
                resultGroups = Lists.newArrayListWithExpectedSize(0);
            }
        } else {
            resultGroups = Lists.newArrayListWithExpectedSize(controlledGroups.size());
            resultGroups.addAll(controlledGroups);
            if (orgGroups == null && empGroups != null) {
                resultGroups.retainAll(empGroups);
            } else if (empGroups == null && orgGroups != null) {
                resultGroups.retainAll(orgGroups);
            } else if (empGroups != null && orgGroups != null) {
                orgGroups.addAll(empGroups);
                resultGroups.retainAll(orgGroups);
            }
        }
        if (!resultGroups.isEmpty()) {
            query.taskCandidateGroupIn(resultGroups);
        }
        if (processInstanceId != null) {
            query.processInstanceId(processInstanceId);
        }
        if (type != null) {
            query.processVariableValueEquals(VAR_PROCEDURE_TYPE_NAME, Integer.toString(type.ordinal()));
        }
        if (!StringUtils.isEmpty(serviceId)) {
            query.processVariableValueEquals(VAR_SERVICE_ID, serviceId);
        }
        if (taskKey != null && !taskKey.isEmpty()) {
            query.taskDefinitionKey(taskKey);
        }
        if (procedureId != null && !procedureId.isEmpty()) {
            query.processVariableValueEquals(VAR_PROCEDURE_ID, procedureId);
        }
        if (declarantTypeName != null && declarantTypeValue != null) {
            query.processVariableValueEquals(declarantTypeName, declarantTypeValue);
        }
        if (requester != null && !requester.isEmpty()) {
            query.processVariableValueEquals(VAR_REQUESTER_LOGIN, requester);
        }
        if (fromDate != null) {
            query.taskCreatedAfter(DateUtils.addSeconds(fromDate, -1));
        }
        if (toDate != null) {
            query.taskCreatedBefore(DateUtils.addSeconds(toDate, 1));
        }
        if (overdue) {
            ((TaskQueryImpl2) query).setOverdue(true);
        }
        return query;
    }

    @Override
    Task singleResult(String id) {
        return createTaskQuery().taskDefinitionKey(id).singleResult();
    }

    @Override
    public int size() {
        return (int) createTaskQuery().count();
    }

    @Override
    public void setFromDate(Date fromDate) {
        this.fromDate = fromDate;
    }

    @Override
    public void setToDate(Date toDate) {
        this.toDate = toDate;
    }

    @Override
    public void setRequester(String requester) {
        this.requester = requester;
    }

    @Override
    public void setBidId(String bidId) {
        Bid bid = AdminServiceProvider.get().getBid(bidId);
        processInstanceId = bid != null ? bid.getProcessInstanceId() : null;
    }

    @Override
    public void setProcedureType(String procedureType) {
        if (Objects.equal("??? ", procedureType)) {
            type = ProcedureType.Administrative;
        } else if (Objects.equal("?? ", procedureType)) {
            type = ProcedureType.Interdepartmental;
        } else {
            type = null;
        }
    }

    @Override
    public void setServiceId(String serviceId) {
        this.serviceId = serviceId;
    }

    @Override
    public void setProcedureId(String procedureId) {
        this.procedureId = procedureId;
    }

    @Override
    public void setTaskKey(String taskKey) {
        this.taskKey = taskKey;
    }

    @Override
    public void setDeclarantTypeName(String declarantTypeName) {
        this.declarantTypeName = declarantTypeName;
    }

    @Override
    public void setDeclarantTypeValue(String declarantTypeValue) {
        this.declarantTypeValue = declarantTypeValue;
    }

    @Override
    public void setOverdue(boolean value) {
        this.overdue = value;
    }

    @Override
    public void setControlledOrgGroups(List groups) {
        orgGroups = groups;
    }

    @Override
    public void setControlledEmpGroups(List groups) {
        empGroups = groups;
    }

}