net.mindengine.oculus.frontend.web.Session.java Source code

Introduction

Here is the source code for net.mindengine.oculus.frontend.web.Session.java

Source

/*******************************************************************************
* 2012 Ivan Shubin http://mindengine.net
* 
* This file is part of MindEngine.net Oculus Frontend.
* 
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU 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 General Public License
* along with Oculus Frontend.  If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/
package net.mindengine.oculus.frontend.web;

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

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;

import net.mindengine.oculus.frontend.db.search.SearchColumn;
import net.mindengine.oculus.frontend.domain.report.TestRunSearchData;
import net.mindengine.oculus.frontend.domain.user.PermissionList;
import net.mindengine.oculus.frontend.domain.user.User;

import org.jfree.chart.JFreeChart;

public class Session {

    private HttpSession session;

    public Session(HttpSession session) {
        this.session = session;
    }

    public void saveChart(String chartId, JFreeChart chart) {
        session.setAttribute("jfreechart_" + chartId, chart);
    }

    public JFreeChart getChart(String chartId) {
        return (JFreeChart) session.getAttribute("jfreechart_" + chartId);
    }

    public void setIssueColumnList(Collection<SearchColumn> columns) {
        session.setAttribute("issueColumnList", columns);
    }

    @SuppressWarnings("unchecked")
    public Collection<SearchColumn> getIssueColumnList() {
        return (Collection<SearchColumn>) session.getAttribute("issueColumnList");
    }

    @SuppressWarnings("unchecked")
    public List<TestRunSearchData> getCollectedTestRuns() {
        List<TestRunSearchData> list = (List<TestRunSearchData>) session.getAttribute("collectedTestRuns");
        if (list == null) {
            list = new ArrayList<TestRunSearchData>();
            setCollectedTestRuns(list);
        }
        return list;
    }

    public void setCollectedTestRuns(List<TestRunSearchData> testRuns) {
        // Redefining runs ids
        if (testRuns != null) {
            int id = 0;
            for (TestRunSearchData data : testRuns) {
                data.setId(id);
                id++;
            }
        }
        session.setAttribute("collectedTestRuns", testRuns);
    }

    /**
     * Returns the temporary message text and deletes it from the session. This
     * method couldn't be run twice as in the second time it will return null
     * value
     * 
     * @return temporary message text
     */
    public String getTemporaryMessage() {
        String message = (String) session.getAttribute("temporary_message");
        if (message != null)
            session.removeAttribute("temporary_message");
        return message;
    }

    /**
     * Set the temporary message text to session variable. This message lives
     * only between two methods: set and get.
     * 
     * @param text
     */
    public void setTemporaryMessage(String message) {
        session.setAttribute("temporary_message", message);
    }

    public HttpSession getSession() {
        return session;
    }

    public void setSession(HttpSession session) {

        this.session = session;
    }

    public static Session create(HttpSession session) {
        return new Session(session);
    }

    public static Session create(HttpServletRequest request) {
        return new Session(request.getSession(true));
    }
}